From 0ddf56a52aa729f3e806b31ab44bea2ece2bf6b2 Mon Sep 17 00:00:00 2001
From: paituo <330435863@qq.com>
Date: Tue, 8 Apr 2025 15:16:15 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=8E=AF=E5=A2=83=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE=E6=94=AF=E6=8C=81=E5=85=AC=E5=8F=B8=E5=A4=A7=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env.template  | 12 +++++-------
 agentic_rag.py |  3 +++
 2 files changed, 8 insertions(+), 7 deletions(-)

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