'commit'

dsLightRag/Config/Config.py

@@ -11,23 +11,23 @@ LLM_MODEL_NAME = "deepseek-chat"
 #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"
 #LLM_MODEL_NAME = "deepseek-r1"  # 使用更牛B的r1模型
 
-# 免费的嵌入模型
+# 嵌入向量模型
 EMBED_MODEL_NAME = "BAAI/bge-m3"
 EMBED_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"
 EMBED_BASE_URL = "https://api.siliconflow.cn/v1"
 EMBED_DIM = 1024
 EMBED_MAX_TOKEN_SIZE = 8192
 
+# 重排模型
+RERANK_MODEL='BAAI/bge-reranker-v2-m3'
+RERANK_BASE_URL='https://api.siliconflow.cn/v1/rerank'
+RERANK_BINDING_API_KEY='sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl'
+
 # 图数据库
 NEO4J_URI = "bolt://localhost:7687"
 NEO4J_USERNAME = "neo4j"
 NEO4J_PASSWORD = "DsideaL147258369"
 NEO4J_AUTH = (NEO4J_USERNAME, NEO4J_PASSWORD)
-
-# 免费的重排模型
-RERANK_MODEL='BAAI/bge-reranker-v2-m3'
-RERANK_BASE_URL='https://api.siliconflow.cn/v1/rerank'
-RERANK_BINDING_API_KEY='sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl'

dsLightRag/Start.py

@@ -58,7 +58,6 @@ async def rag(request: fastapi.Request):
     async def generate_response_stream(query: str):
         try:
             rag = await initialize_rag(WORKING_PATH)
-
             await rag.initialize_storages()
             await initialize_pipeline_status()
             resp = await rag.aquery(

dsLightRag/Util/LightRagUtil.py

@@ -97,8 +97,6 @@ async def embedding_func(texts: list[str]) -> np.ndarray:
         base_url=EMBED_BASE_URL
     )
 
-
-
 async def rerank_func(query: str, documents: list, top_k: int = None, **kwargs):
     return await custom_rerank(
         query=query,
@@ -124,10 +122,8 @@ async def initialize_rag(working_dir, graph_storage=None):
         ),
         rerank_model_func=rerank_func,
     )
-
     await rag.initialize_storages()
     await initialize_pipeline_status()
-
     return rag
 
 

dsLightRag/static/ai.html

@@ -177,12 +177,12 @@
             <a href="SuShi.html" class="btn" target="_blank">进入 <i class="fas fa-arrow-right"></i></a>
         </div>
 
-        <!--        <div class="card">-->
+        <div class="card">
-        <!--            <i class="fas fa-project-diagram"></i>-->
+            <i class="fas fa-project-diagram"></i>
-        <!--            <h3>少年读史记</h3>-->
+            <h3>少年读史记</h3>
-        <!--            <p>关系图谱应用示例</p>-->
+            <p>关系图谱应用示例</p>
-        <!--            <a href="ShiJi.html" class="btn" target="_blank">进入 <i class="fas fa-arrow-right"></i></a>-->
+            <a href="ShiJi.html" class="btn" target="_blank">进入 <i class="fas fa-arrow-right"></i></a>
-        <!--        </div>-->
+        </div>
 
         <div class="card">
             <i class="fas fa-globe"></i>