diff --git a/.env.template b/.env.template index 88a50c1..96c7fea 100644 --- a/.env.template +++ b/.env.template @@ -2,13 +2,11 @@ # DB_URL=postgresql+psycopg://ai:ai@localhost:5532/ai # OpenAI API配置 -API_KEY=sk-oftybjqntjpxlhkcxkieluljb -EMBEDDING_MODEL=BAAI/bge-large-zh-v1.5 -EMBEDDING_BASE_URL=https://api.siliconflow.cn/v1 -MODEL_LIST=Qwen2.5-72B=openai:Qwen/Qwen2.5-72B-Instruct&gpt-4o=openai:gpt-4o - -# 模型配置 -MODEL_BASE_URL=https://api.siliconflow.cn/v1 +API_KEY=sk-oftybjqntjpxlhkc +EMBEDDING_MODEL=bge-m3 +EMBEDDING_BASE_URL=http://10.1.16.39:9995/v1 +MODEL_LIST=Qwen2.5-72B=openai:Qwen2.5-72B-Instruct-GPTQ-Int8 +MODEL_BASE_URL=http://172.20.0.145:9995/v1 # 文件路径配置 KNOWLEDGE_SOURCE_DIR=data diff --git a/agentic_rag.py b/agentic_rag.py index 89f511a..38bc10d 100644 --- a/agentic_rag.py +++ b/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}")