修改环境配置支持公司大模型配置

This commit is contained in:
2025-04-08 15:16:15 +08:00
parent 01ed406dfa
commit 0ddf56a52a
2 changed files with 8 additions and 7 deletions
+3
View File
@@ -31,6 +31,7 @@
from pathlib import Path
from agno.document.chunking.document import DocumentChunking
from agno.models.deepseek import DeepSeek
from dotenv import load_dotenv
# 加载.env文件
@@ -107,6 +108,8 @@ def get_model_by_provider(provider: str, model_name: str):
# return Claude(id=model_name)
# elif provider == "groq":
# return Groq(id=model_name)
elif provider == "deepseek":
return DeepSeek(id=model_name, base_url=model_baseUrl, api_key=api_key)
else:
raise ValueError(f"Unsupported model provider: {provider}")