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

.env.template

@@ -2,13 +2,11 @@
 # DB_URL=postgresql+psycopg://ai:ai@localhost:5532/ai
 
 # OpenAI API配置
-API_KEY=sk-oftybjqntjpxlhkcxkieluljb
+API_KEY=sk-oftybjqntjpxlhkc
-EMBEDDING_MODEL=BAAI/bge-large-zh-v1.5
+EMBEDDING_MODEL=bge-m3
-EMBEDDING_BASE_URL=https://api.siliconflow.cn/v1
+EMBEDDING_BASE_URL=http://10.1.16.39:9995/v1
-MODEL_LIST=Qwen2.5-72B=openai:Qwen/Qwen2.5-72B-Instruct&gpt-4o=openai:gpt-4o
+MODEL_LIST=Qwen2.5-72B=openai:Qwen2.5-72B-Instruct-GPTQ-Int8
-
+MODEL_BASE_URL=http://172.20.0.145:9995/v1
-# 模型配置
-MODEL_BASE_URL=https://api.siliconflow.cn/v1
 
 # 文件路径配置
 KNOWLEDGE_SOURCE_DIR=data

agentic_rag.py

@@ -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}")