fix model property key

langgenius · Jan 2, 2024 · 9344091 · 9344091
1 parent 7777c34
commit 9344091
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/core/model_runtime/model_providers/openai_api_compatible/_common.py b/api/core/model_runtime/model_providers/openai_api_compatible/_common.py
@@ -117,9 +117,9 @@ def get_customizable_model_schema(self, model: str, credentials: dict) -> AIMode
 
         if model_type == ModelType.LLM:
             if credentials['mode'] == 'chat':
-                entity.model_properties['mode'] = LLMMode.CHAT
+                entity.model_properties[ModelPropertyKey.MODE] = LLMMode.CHAT
             elif credentials['mode'] == 'completion':
-                entity.model_properties['mode'] = LLMMode.COMPLETION
+                entity.model_properties[ModelPropertyKey.MODE] = LLMMode.COMPLETION
             else:
                 raise ValueError(f"Unknown completion type {credentials['completion_type']}")