mirror of
https://github.com/Zippland/Snap-Solver.git
synced 2026-02-14 08:16:46 +08:00
修复gemini接口,添加豆包接口
This commit is contained in:
@@ -114,6 +114,25 @@ class ModelFactory:
|
||||
)
|
||||
# 对于阿里巴巴模型,也需要传递正确的模型名称
|
||||
elif 'qwen' in model_name.lower() or 'qvq' in model_name.lower() or 'alibaba' in model_name.lower():
|
||||
return model_class(
|
||||
api_key=api_key,
|
||||
temperature=temperature,
|
||||
system_prompt=system_prompt,
|
||||
language=language,
|
||||
model_name=model_name
|
||||
)
|
||||
# 对于Google模型,也需要传递正确的模型名称
|
||||
elif 'gemini' in model_name.lower() or 'google' in model_name.lower():
|
||||
return model_class(
|
||||
api_key=api_key,
|
||||
temperature=temperature,
|
||||
system_prompt=system_prompt,
|
||||
language=language,
|
||||
model_name=model_name,
|
||||
api_base_url=api_base_url
|
||||
)
|
||||
# 对于豆包模型,也需要传递正确的模型名称
|
||||
elif 'doubao' in model_name.lower():
|
||||
return model_class(
|
||||
api_key=api_key,
|
||||
temperature=temperature,
|
||||
|
||||
Reference in New Issue
Block a user