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

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

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:
azure_chat_open_ai = ChatOpenAI(
model=model_name,
openai_api_base=model_credential.get('api_base'),
openai_api_key=model_credential.get('api_key')
)