'commit'

dsRagAnything/Config/Config.py

@@ -1,33 +1,33 @@
-# 嵌入模型
+# 嵌入向量模型
 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
 
-# 大模型 【DeepSeek深度求索官方】
-#LLM_API_KEY = "sk-44ae895eeb614aa1a9c6460579e322f1"
-#LLM_BASE_URL = "https://api.deepseek.com"
-#LLM_MODEL_NAME = "deepseek-chat"
+# 重排模型
+RERANK_MODEL = 'BAAI/bge-reranker-v2-m3'
+RERANK_BASE_URL = 'https://api.siliconflow.cn/v1/rerank'
+RERANK_BINDING_API_KEY = 'sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl'
 
-# 阿里云提供的大模型服务
-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"
+# GLM
+GLM_BASE_URL = "https://api.siliconflow.cn/v1/chat/completions"
+GLM_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"
+GLM_MODEL_NAME = "THUDM/GLM-4.1V-9B-Thinking"
 
-# 视觉模型
-VISION_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"
-VISION_BASE_URL = "https://api.siliconflow.cn/v1/chat/completions"
-VISION_MODEL_NAME = "GLM-4.1V-9B-Thinking"
+# 阿里云API信息【YLT】
+#ALY_LLM_API_KEY = "sk-f6da0c787eff4b0389e4ad03a35a911f"
+# 阿里云API信息【绘智科技】
+ALY_LLM_API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
+ALY_LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
+ALY_LLM_MODEL_NAME = "qwen-plus"
 
-# Neo4j
-NEO4J_URI = "bolt://localhost:7687"
-NEO4J_USERNAME = "neo4j"
-NEO4J_PASSWORD = "DsideaL147258369"
-NEO4J_AUTH = (NEO4J_USERNAME, NEO4J_PASSWORD)
 
-# 写入Neo4j的批次大小
-BATCH_SIZE_NODES = 100
-# 写入Neo4j的批次大小
-BATCH_SIZE_EDGES = 100
+
+# 智谱的API KEY【吴缤申请个人版免费】
+ZHIPU_API_KEY = "78dc1dfe37e04f29bd4ca9a49858a969.gn7TIZTfzpY35nx9"
+
+# GPTNB的API KEY
+GPTNB_API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662"
+GPTNB_BASE_URL="https://goapi.gptnb.ai"
+