修改配置

backend/app/config.py

@@ -1,50 +1,89 @@
 """
 环境变量集中管理模块
 所有配置项统一定义，避免散落在各个文件中
+配置分组：相关配置放在一起，URL 和 API Key 配对
+所有配置直接从环境变量读取，无默认值，避免配置混乱
+需要类型转换的配置在此处理
 """
 
 import os
 
 
-# ========== Graph 执行追踪配置 ==========
-# 是否启用 Graph 流转追踪（通过环境变量控制）
-ENABLE_GRAPH_TRACE = os.getenv("ENABLE_GRAPH_TRACE", "true").lower() == "true"
+# ========== 辅助函数：类型转换 ==========
+def _get_str(key: str) -> str | None:
+    """获取字符串配置"""
+    return os.getenv(key)
 
-# ========== 记忆提取配置 ==========
-# 记忆提取间隔：每 N 轮对话生成一次摘要
-MEMORY_SUMMARIZE_INTERVAL = int(os.getenv("MEMORY_SUMMARIZE_INTERVAL", "10"))
 
-# ========== Mem0 记忆层配置 ==========
-# Qdrant 向量数据库地址
-QDRANT_URL = os.getenv("QDRANT_URL", "http://127.0.0.1:6333")
-QDRANT_COLLECTION_NAME = os.getenv("QDRANT_COLLECTION_NAME", "mem0_user_memories")
-QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "your-qdrant-api-key")
+def _get_int(key: str) -> int | None:
+    """获取整数配置，自动转换"""
+    value = os.getenv(key)
+    if value is not None:
+        try:
+            return int(value)
+        except (ValueError, TypeError):
+            pass
+    return None
 
-# ==========    llm 配置 ==========
-# LLM 模型配置
-VLLM_BASE_URL = os.getenv("VLLM_BASE_URL", "http://127.0.0.1:8081/v1")
-LLM_API_KEY = os.getenv("LLM_API_KEY", "your-ai-api-key")
 
-# llama.cpp Embedding 服务地址 (用于 Mem0 的向量化)
-LLAMACPP_EMBEDDING_URL = os.getenv("LLAMACPP_EMBEDDING_URL", "http://127.0.0.1:8082/v1")
-LLAMACPP_API_KEY = os.getenv("LLAMACPP_API_KEY", "your-llamacpp-api-key")
+def _get_bool(key: str) -> bool | None:
+    """获取布尔配置，自动转换"""
+    value = os.getenv(key)
+    if value is not None:
+        return value.lower() in ("true", "1", "yes", "on")
+    return None
 
-# ========== 后端服务配置 ==========
-# 数据库连接字符串
-DB_URI = os.getenv(
-    "DB_URI",
-    "postgresql://postgres:huang1998@ai-postgres:5432/langgraph_db?sslmode=disable"
-)
-# 后端服务端口
-BACKEND_PORT = int(os.getenv("BACKEND_PORT", "8079"))
-
-# ========== 日志配置 ==========
-LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()
-DEBUG = os.getenv("DEBUG", "false").lower() == "true"
-
-# ========== Reranker 服务配置 ==========
-LLAMACPP_RERANKER_URL = os.getenv("LLAMACPP_RERANKER_URL", "http://127.0.0.1:8083")
 
 # ========== 第三方 API 密钥 ==========
-ZHIPUAI_API_KEY = os.getenv("ZHIPUAI_API_KEY", "")
-DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY", "")
+ZHIPUAI_API_KEY = _get_str("ZHIPUAI_API_KEY")
+DEEPSEEK_API_KEY = _get_str("DEEPSEEK_API_KEY")
+
+
+# ========== llama.cpp 服务配置（URL + API密钥 配对） ==========
+# 主 LLM 服务
+VLLM_BASE_URL = _get_str("VLLM_BASE_URL")
+LLM_API_KEY = _get_str("LLAMACPP_API_KEY")
+
+# Embedding 服务 (用于 Mem0 的向量化)
+LLAMACPP_EMBEDDING_URL = _get_str("LLAMACPP_EMBEDDING_URL")
+LLAMACPP_API_KEY = _get_str("LLAMACPP_API_KEY")
+
+# Reranker 服务
+LLAMACPP_RERANKER_URL = _get_str("LLAMACPP_RERANKER_URL")
+
+
+# ========== Qdrant 向量数据库配置（URL + API密钥 配对） ==========
+QDRANT_URL = _get_str("QDRANT_URL")
+QDRANT_API_KEY = _get_str("QDRANT_API_KEY")
+QDRANT_COLLECTION_NAME = _get_str("QDRANT_COLLECTION_NAME")
+
+
+# ========== PostgreSQL 数据库配置（分离配置 + 完整URI） ==========
+# 分离配置（优先使用）
+DB_HOST = _get_str("DB_HOST")
+DB_PORT = _get_int("DB_PORT")
+DB_USER = _get_str("DB_USER")
+DB_PASSWORD = _get_str("DB_PASSWORD")
+DB_NAME = _get_str("DB_NAME")
+
+# 完整连接字符串（直接从环境变量读取）
+DB_URI = _get_str("DB_URI")
+
+
+# ========== 后端服务配置 ==========
+BACKEND_PORT = _get_int("BACKEND_PORT")
+
+
+# ========== Mem0 记忆层配置 ==========
+# 记忆提取间隔：每 N 轮对话生成一次摘要
+MEMORY_SUMMARIZE_INTERVAL = _get_int("MEMORY_SUMMARIZE_INTERVAL")
+
+
+# ========== Graph 执行追踪配置 ==========
+# 是否启用 Graph 流转追踪（通过环境变量控制）
+ENABLE_GRAPH_TRACE = _get_bool("ENABLE_GRAPH_TRACE")
+
+
+# ========== 日志配置 ==========
+LOG_LEVEL = _get_str("LOG_LEVEL")
+DEBUG = _get_bool("DEBUG")

backend/rag_core/config.py

@@ -1,24 +1,54 @@
 """
 RAG Core 配置管理模块
 集中管理所有环境变量配置项，避免散落在各个文件中
+所有配置直接从环境变量读取，无默认值，避免配置混乱
+需要类型转换的配置在此处理
 """
 
 import os
 
-# ========== 向量数据库配置 ==========
-QDRANT_URL = os.getenv("QDRANT_URL", "http://127.0.0.1:6333")
-QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "")
 
-# ========== 嵌入服务配置 ==========
-LLAMACPP_EMBEDDING_URL = os.getenv("LLAMACPP_EMBEDDING_URL", "http://127.0.0.1:8082")
-LLAMACPP_API_KEY = os.getenv("LLAMACPP_API_KEY", "")
+# ========== 辅助函数：类型转换 ==========
+def _get_str(key: str) -> str | None:
+    """获取字符串配置"""
+    return os.getenv(key)
+
+
+def _get_int(key: str) -> int | None:
+    """获取整数配置，自动转换"""
+    value = os.getenv(key)
+    if value is not None:
+        try:
+            return int(value)
+        except (ValueError, TypeError):
+            pass
+    return None
+
+
+# ========== 向量数据库配置（URL + API密钥 配对） ==========
+QDRANT_URL = _get_str("QDRANT_URL")
+QDRANT_API_KEY = _get_str("QDRANT_API_KEY")
+
+
+# ========== 嵌入服务配置（URL + API密钥 配对） ==========
+LLAMACPP_EMBEDDING_URL = _get_str("LLAMACPP_EMBEDDING_URL")
+LLAMACPP_API_KEY = _get_str("LLAMACPP_API_KEY")
+
+
+# ========== 文档存储配置（分离配置 + 完整URI） ==========
+# 分离配置（优先使用）
+DB_HOST = _get_str("DB_HOST")
+DB_PORT = _get_int("DB_PORT")
+DB_USER = _get_str("DB_USER")
+DB_PASSWORD = _get_str("DB_PASSWORD")
+DB_NAME = _get_str("DB_NAME")
+
+# 完整连接字符串（直接从环境变量读取）
+DB_URI = _get_str("DB_URI")
+
+# 文档存储 URI（直接从环境变量读取，默认同 DB_URI）
+DOCSTORE_URI = _get_str("DOCSTORE_URI") or DB_URI
 
-# ========== 文档存储配置 ==========
-DB_URI = os.getenv(
-    "DB_URI",
-    "postgresql://postgres:***@ai-postgres:5432/langgraph_db?sslmode=disable"
-)
-DOCSTORE_URI = os.getenv("DOCSTORE_URI", DB_URI)
 
 # ========== 其他配置 ==========
-# 可以在此添加其他 RAG Core 专用的配置项
+# 可以在此添加其他 RAG Core 专用的配置项