更新 config.py

2026-07-07 10:22:30 +08:00 · 2026-07-07 10:22:30 +08:00 · ec302b7539
commit ec302b7539
parent a79b29fb30
2 changed files with 224 additions and 0 deletions
--- a/0
+++ b/0
--- a/config.py
+++ b/config.py
@ -0,0 +1,224 @@
+"""
+项目配置文件
+从.env文件读取配置
+"""
+import os
+from dotenv import load_dotenv
+from contextvars import ContextVar
+from typing import Optional, Dict, Any
+
+# 加载.env文件
+load_dotenv()
+
+# ==================== 大模型配置 ====================
+LLM_CONFIG = {
+#    "model": "Qwen3.5-35B-A3B",
+    "model": "Qwen3-32B",
+#    "model": "Qwen3-14B",
+    "base_url": "http://192.168.0.164:9800/v1",
+    "api_key": "gpustack_dee9ca823290886c_5edfc86aeeeceb1e9ee5162941cb2cb5",
+    "temperature": float("0.1"),
+    "max_tokens": int("20096"),
+    "timeout": int("30")
+}
+
+# ==================== 嵌入模型配置 ====================
+EMBEDDING_CONFIG = {
+    "model": "bge-m3",
+    "base_url": "http://embedding:9700/v1/embeddings",
+    "api_key": "gpustack_dee9ca823290886c_5edfc86aeeeceb1e9ee5162941cb2cb5",
+    "base_url_v2" : "http://embedding:9700/v1/embeddings",
+}
+
+
+# ==================== 线程安全的请求上下文配置 ====================
+# 使用 contextvars 存储每个请求的用户配置，确保多用户并发时不会相互干扰
+_request_user_config: ContextVar[Optional[Dict[str, str]]] = ContextVar('request_user_config', default=None)
+
+# RAG配置的默认值
+_RAG_CONFIG_DEFAULT = {
+    "search_endpoint": "http://htknow:8080/api/v1/knowledge/search/",
+    "x-user-id": "1",
+    "x-user-name": "testuser",
+    "x-role": "admin",
+    "kb-id": None,
+    "file-id": None
+}
+
+class DynamicRAGConfig(dict):
+    """
+    动态RAG配置类，支持线程安全的请求级配置
+    当访问用户相关配置时，自动从当前请求上下文读取
+    其他配置项直接返回默认值
+    兼容字典的所有操作方式（如 get(), [], in 等）
+    """
+    def __getitem__(self, key: str):
+        # 如果是用户相关配置或知识库参数，从当前请求上下文读取
+        if key in ["x-user-id", "x-user-name", "x-role", "kb-id", "file-id"]:
+            request_config = _request_user_config.get()
+            if request_config and key in request_config:
+                return request_config[key]
+            # 如果没有设置，返回默认值
+            return _RAG_CONFIG_DEFAULT[key]
+        # 其他配置项直接返回默认值
+        return _RAG_CONFIG_DEFAULT.get(key)
+    
+    def get(self, key: str, default=None):
+        """支持 get() 方法"""
+        try:
+            return self[key]
+        except KeyError:
+            return default
+    
+    def __contains__(self, key: str):
+        """支持 in 操作符"""
+        return key in _RAG_CONFIG_DEFAULT
+    
+    def keys(self):
+        """返回所有键"""
+        return _RAG_CONFIG_DEFAULT.keys()
+    
+    def values(self):
+        """返回所有值（动态读取用户配置）"""
+        return [self[key] for key in _RAG_CONFIG_DEFAULT.keys()]
+    
+    def items(self):
+        """返回所有键值对（动态读取用户配置）"""
+        return [(key, self[key]) for key in _RAG_CONFIG_DEFAULT.keys()]
+
+# ==================== RAG服务配置 ====================
+# 使用动态配置类，支持多用户并发时自动读取各自的配置
+RAG_CONFIG = DynamicRAGConfig()
+
+def set_request_user_config(x_user_id: Optional[str] = None, 
+                            x_user_name: Optional[str] = None, 
+                            x_role: Optional[str] = None,
+                            text_kb_id: Optional[str] = None,
+                            text_file_id: Optional[str] = None):
+    """
+    设置当前请求的用户配置和知识库参数（线程安全）
+    
+    Args:
+        x_user_id: 用户ID
+        x_user_name: 用户名称
+        x_role: 角色
+        text_kb_id: 文本检索知识库ID
+        text_file_id: 文本检索文件ID
+    """
+    config = {}
+    if x_user_id is not None:
+        config["x-user-id"] = x_user_id
+    if x_user_name is not None:
+        config["x-user-name"] = x_user_name
+    if x_role is not None:
+        config["x-role"] = x_role
+    # 仅当非空字符串时才设置，空字符串时不传入 RAG 接口
+    if text_kb_id is not None and str(text_kb_id).strip():
+        config["kb-id"] = str(text_kb_id).strip()
+    if text_file_id is not None and str(text_file_id).strip():
+        config["file-id"] = str(text_file_id).strip()
+    _request_user_config.set(config if config else None)
+
+# ==================== 图谱检索服务配置 ====================
+GRAPH_SEARCH_CONFIG = {
+    "search_endpoint": "http://kgrag:9085/search_graph",
+    "graph_timeout": float("30.0"),
+    "default_top_k": int("10"),
+    "operation_search_endpoint": "http://kgrag:9085/operation_graph_search"
+}
+
+# ==================== 设备到系统检索服务配置 ====================
+DEVICE_SYSTEM_CONFIG = {
+    "endpoint": "http://kgrag:9085/device_to_system",
+    "timeout": float("30.0"),
+    "default_min_hops": int("2"),
+    "default_max_hops": int("8"),
+    "default_top_k": int("10")
+}
+
+# ==================== 图册检索服务配置 ====================
+ATLAS_CONFIG = {
+    "endpoint": "http://kgrag:9085/atlas_retrieval",
+    "timeout": float("30.0"),
+    "default_top_k": int("10")
+}
+
+# ==================== 工作流配置 ====================
+WORKFLOW_CONFIG = {
+    "max_retry_count": int("2"),  # 最大重试次数
+    "image_clarity_threshold": float("0.7"),  # 图片清晰度阈值
+    "default_timeout": int("30")  # 默认超时时间（秒）
+}
+
+# ==================== 服务器配置 ====================
+SERVER_CONFIG = {
+    "host": "agent",
+    "port": "9088",
+    "base_url": "http://agent:9088",
+    "report" : "http://192.168.0.164:9088"
+}
+
+# ==================== ASR服务配置 ====================
+ASR_CONFIG = {
+    "api_base": "http://asr:9900/v1",
+    "model": "base",
+    "language": "zh",
+    "timeout": int("30")
+}
+
+# ==================== VLM服务配置 ====================
+VLM_CONFIG = {
+    "api_base" : "http://kgrag:9085/v1",
+    "model" :  "Qwen3-VL-8B"
+}
+
+# ==================== Neo4J配置 ====================
+NEO4J_CONFIG = {
+    "uri": "neo4j://neo4j:7687",
+    "user": "neo4j",
+    "password": "zdht123"
+}
+
+# ==================== 知识库匹配配置 ====================
+KB_TREE_CONFIG = {
+    "url": "http://htknow:8080/api/v1/knowledge/knowledge_base/tree"
+}
+
+# ==================== 维修反馈统计配置 ====================
+REPAIR_FEEDBACK_CONFIG = {
+    "url": "http://webui:22000",
+    "email": "15088888888@163.com",
+    "password": "11223344"
+}
+
+# ==================== 数据库配置 ====================
+POSTGRES_CONNECTION_STRING = "postgresql://postgres:zdht123@postgres:5432/hj_webui"
+#POSTGRES_CONNECTION_STRING = "postgresql+psycopg://postgres:zdht123@postgres:5432/hj_webui"
+POSTGRES_SQLALCHEMY_URL = "postgresql+psycopg://postgres:zdht123@postgres:5432/hj_webui"
+
+
+# ==================== 索引配置 ====================
+INDEX_CONFIG = {
+    "NAME_PROPERTY" : "名称",
+    "FULLTEXT_PROPERTY" :"fulltext",
+    "EMBEDDING_PROPERTY" : "embedding",
+    "SEARCH_LABEL" : "Searchable",
+    "VECTOR_INDEX_NAME" : "global_searchable_embedding",
+    "FULLTEXT_INDEX_NAME" : "global_searchable_content_search"
+}
+
+# ==================== sql表格字段与neo4j节点映射 ====================
+MAP_INS={
+    "故障模式" : "fault",
+    "系统" : "system_name",
+    "设备" : "device_name",
+    "舷号" : "ship_number",
+    "消耗备件" : "spare_parts"
+}
+
+# ==================== neo4j配置 ====================
+NEO4J_CONFIG = {
+    "uri": "neo4j://neo4j:7687",
+    "user": "neo4j",
+    "password": "zdht123@"
+}