zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
shaohuzhang1 2024-04-17 17:14:01 +08:00 committed by GitHub
parent 8e55a2c9eb
commit df850894c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py
View File

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