diff --git a/config b/config deleted file mode 100644 index e69de29..0000000 diff --git a/config.py b/config.py new file mode 100644 index 0000000..18df755 --- /dev/null +++ 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@" +}