'commit'

dsLightRag/Config/Config.py

@@ -3,15 +3,15 @@ ALY_AK = 'LTAI5tE4tgpGcKWhbZg6C4bh'
 ALY_SK = 'oizcTOZ8izbGUouboC00RcmGE8vBQ1'
 
 # 大模型 【DeepSeek深度求索官方】
-#LLM_API_KEY = "sk-44ae895eeb614aa1a9c6460579e322f1"
-#LLM_BASE_URL = "https://api.deepseek.com"
-#LLM_MODEL_NAME = "deepseek-chat"
+LLM_API_KEY = "sk-44ae895eeb614aa1a9c6460579e322f1"
+LLM_BASE_URL = "https://api.deepseek.com"
+LLM_MODEL_NAME = "deepseek-chat"
 
-# 阿里云提供的大模型服务
-LLM_API_KEY = "sk-f6da0c787eff4b0389e4ad03a35a911f"
-LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
+# 阿里云提供的大模型服务 【阿里云在处理文字材料时，容易引发绿网拦截，导致数据上报异常】
+#LLM_API_KEY = "sk-f6da0c787eff4b0389e4ad03a35a911f"
+#LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
 #LLM_MODEL_NAME = "qwen-plus" # 不要使用通义千问，会导致化学方程式不正确！
-LLM_MODEL_NAME = "deepseek-v3"
+#LLM_MODEL_NAME = "deepseek-v3"
 
 EMBED_MODEL_NAME = "BAAI/bge-m3"
 EMBED_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"

dsLightRag/Start.py

@@ -49,7 +49,7 @@ app.mount("/static", StaticFiles(directory="Static"), name="static")
 @app.post("/api/rag")
 async def rag(request: fastapi.Request):
     data = await request.json()
-    workspace = data.get("topic")  # Chinese, Math
+    workspace = data.get("topic","ShiJi")  # Chinese, Math ,ShiJi 默认是少年读史记
     mode = data.get("mode", "hybrid")  # 默认为hybrid模式
     # 查询的问题
     query = data.get("query")

dsLightRag/T1_Train.py

@@ -25,17 +25,20 @@ WORKING_DIR = f"./output"
 # 1、工作空间【知识库名称】
 # 2、文档名称【不允许出现重复，因为后面需要以此为条件查询】
 tasks = [
-    {  # 苏轼
-        "workspace": "SuShi", "docx_name": "苏轼.docx",
-    },
-    {  # 化学
-        "workspace": "Chemistry", "docx_name": "Chemistry.docx",
-    },
-    {  # 几何
-        "workspace": "JiHe", "docx_name": "JiHe.docx",
-    },
-    {  # 数学
-        "workspace": "Math", "docx_name": "Math.docx",
+    # {  # 苏轼
+    #     "workspace": "SuShi", "docx_name": "苏轼.docx",
+    # },
+    # {  # 化学
+    #     "workspace": "Chemistry", "docx_name": "Chemistry.docx",
+    # },
+    # {  # 几何
+    #     "workspace": "JiHe", "docx_name": "JiHe.docx",
+    # },
+    # {  # 数学
+    #     "workspace": "Math", "docx_name": "Math.docx",
+    # },
+    {  # 史记
+        "workspace": "ShiJi", "docx_name": "少年读史记张嘉骅.docx",
     },
 ]
 for task in tasks: