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

shaohuzhang1 · web-flow · commit df850894c689 · 2024-04-17T17:14:01.000+08:00
diff --git 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 @@ def get_dialogue_number(self):
 
     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')
         )

Original file line number	Diff line number	Diff line change
`@@ -73,6 +73,7 @@ def get_dialogue_number(self):`
`73`	`73`
`74`	`74`	`def get_model(self, model_type, model_name, model_credential: Dict[str, object], **model_kwargs) -> ChatOpenAI:`
`75`	`75`	`azure_chat_open_ai = ChatOpenAI(`
	`76`	`+ model=model_name,`
`76`	`77`	`openai_api_base=model_credential.get('api_base'),`
`77`	`78`	`openai_api_key=model_credential.get('api_key')`
`78`	`79`	`)`