agent功能开发增加MCP后端

2025-11-07 17:42:06 +08:00
parent 3574f5391f
commit a1c76a257c
5 changed files with 2329 additions and 0 deletions
--- a/mcp_config.py
+++ b/mcp_config.py
@@ -0,0 +1,108 @@
+"""
+MCP服务器配置文件
+集中管理所有配置项
+"""
+
+from typing import Dict
+from pydantic import BaseSettings
+
+class Settings(BaseSettings):
+    """应用配置"""
+
+    # 服务器配置
+    SERVER_HOST: str = "0.0.0.0"
+    SERVER_PORT: int = 8900
+    DEBUG: bool = True
+
+    # 后端API服务端点
+    NEWS_API_URL: str = "http://222.128.1.157:21891"
+    ROADSHOW_API_URL: str = "http://222.128.1.157:19800"
+    CONCEPT_API_URL: str = "http://222.128.1.157:16801"
+    STOCK_ANALYSIS_API_URL: str = "http://222.128.1.157:8811"
+
+    # HTTP客户端配置
+    HTTP_TIMEOUT: float = 60.0
+    HTTP_MAX_RETRIES: int = 3
+
+    # 日志配置
+    LOG_LEVEL: str = "INFO"
+    LOG_FORMAT: str = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+
+    # CORS配置
+    CORS_ORIGINS: list = ["*"]
+    CORS_CREDENTIALS: bool = True
+    CORS_METHODS: list = ["*"]
+    CORS_HEADERS: list = ["*"]
+
+    # LLM配置（如果需要集成）
+    LLM_PROVIDER: str = "openai"  # openai, anthropic, etc.
+    LLM_API_KEY: str = ""
+    LLM_MODEL: str = "gpt-4"
+    LLM_BASE_URL: str = ""
+
+    # 速率限制
+    RATE_LIMIT_ENABLED: bool = False
+    RATE_LIMIT_PER_MINUTE: int = 60
+
+    # 缓存配置
+    CACHE_ENABLED: bool = True
+    CACHE_TTL: int = 300  # 秒
+
+    class Config:
+        env_file = ".env"
+        case_sensitive = True
+
+
+# 全局设置实例
+settings = Settings()
+
+
+# 工具类别映射（用于组织和展示）
+TOOL_CATEGORIES: Dict[str, list] = {
+    "新闻搜索": [
+        "search_news",
+        "search_china_news",
+        "search_medical_news"
+    ],
+    "公司研究": [
+        "search_roadshows",
+        "search_research_reports"
+    ],
+    "概念板块": [
+        "search_concepts",
+        "get_concept_details",
+        "get_stock_concepts",
+        "get_concept_statistics"
+    ],
+    "股票分析": [
+        "search_limit_up_stocks",
+        "get_daily_stock_analysis"
+    ]
+}
+
+
+# 工具优先级（用于LLM选择工具时的提示）
+TOOL_PRIORITIES: Dict[str, int] = {
+    "search_china_news": 10,  # 最常用
+    "search_concepts": 9,
+    "search_limit_up_stocks": 8,
+    "search_research_reports": 8,
+    "get_stock_concepts": 7,
+    "search_news": 6,
+    "get_daily_stock_analysis": 5,
+    "get_concept_statistics": 5,
+    "search_medical_news": 4,
+    "search_roadshows": 4,
+    "get_concept_details": 3,
+}
+
+
+# 默认参数配置
+DEFAULT_PARAMS = {
+    "top_k": 20,
+    "page_size": 20,
+    "size": 10,
+    "sort_by": "change_pct",
+    "mode": "hybrid",
+    "exact_match": False,
+}