修改环境配置支持公司大模型配置
This commit is contained in:
@@ -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}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user