feat: 添加更多LLM模型

This commit is contained in:
RockChinQ
2024-01-30 16:29:54 +08:00
parent e9e458c877
commit 28bd232dda
3 changed files with 178 additions and 6 deletions
+1 -1
View File
@@ -47,7 +47,7 @@ class OpenAIChatCompletion(api.LLMAPIRequester):
self.client.api_key = conversation.use_model.token_mgr.get_token()
args = self.ap.cfg_mgr.data["completion_api_params"].copy()
args["model"] = conversation.use_model.name
args["model"] = conversation.use_model.name if conversation.use_model.model_name is None else conversation.use_model.model_name
if conversation.use_model.tool_call_supported:
tools = await self.ap.tool_mgr.generate_tools_for_openai(conversation)