fix: 支持 OpenAI 兼容模型 #142 (#146)

2024-04-17 17:14:01 +08:00 · 2024-04-17 17:14:01 +08:00 · df850894c6
commit df850894c6
parent 8e55a2c9eb
1 changed files with 1 additions and 0 deletions
--- a/apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py
+++ b/apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py
@ -73,6 +73,7 @@ class OpenAIModelProvider(IModelProvider):

    def get_model(self, model_type, model_name, model_credential: Dict[str, object], **model_kwargs) -> ChatOpenAI:
        azure_chat_open_ai = ChatOpenAI(
+            model=model_name,
            openai_api_base=model_credential.get('api_base'),
            openai_api_key=model_credential.get('api_key')
        )