更新ios

mcp_server.py

@@ -48,7 +48,7 @@ class ServiceEndpoints:
     """API服务端点配置"""
     NEWS_API = "http://222.128.1.157:21891"  # 新闻API
     ROADSHOW_API = "http://222.128.1.157:19800"  # 路演API
-    CONCEPT_API = "http://222.128.1.157:16801"  # 概念API V2（concept_api_v2.py，端口16801）
+    CONCEPT_API = "http://222.128.1.157:16801"  # 概念API V4（concept_api_v4.py，端口映射到16801）
     STOCK_ANALYSIS_API = "http://222.128.1.157:8811"  # 涨停分析+研报API
     MAIN_APP_API = "http://127.0.0.1:5001"  # 主应用API（自选股、自选事件等）
 
@@ -315,7 +315,7 @@ TOOLS: List[ToolDefinition] = [
     ),
     ToolDefinition(
         name="search_concepts",
-        description="搜索股票概念板块，返回概念详情及相关股票列表。",
+        description="搜索股票概念板块，支持语义搜索和原因搜索，返回概念详情及相关股票列表。",
         parameters={
             "type": "object",
             "properties": {
@@ -326,6 +326,24 @@ TOOLS: List[ToolDefinition] = [
                 "trade_date": {
                     "type": "string",
                     "description": "交易日期，格式：YYYY-MM-DD，不传则使用今天"
+                },
+                "filter_lv1": {
+                    "type": "string",
+                    "description": "一级分类过滤，如：'科技'、'消费'、'金融'"
+                },
+                "filter_lv2": {
+                    "type": "string",
+                    "description": "二级分类过滤"
+                },
+                "search_reason": {
+                    "type": "boolean",
+                    "description": "是否同时搜索股票被纳入概念的原因",
+                    "default": False
+                },
+                "include_stock_reasons": {
+                    "type": "boolean",
+                    "description": "是否在返回的股票列表中包含原因",
+                    "default": False
                 }
             },
             "required": ["query"]
@@ -333,7 +351,7 @@ TOOLS: List[ToolDefinition] = [
     ),
     ToolDefinition(
         name="get_concept_details",
-        description="根据概念ID获取详细信息，包括描述、相关股票、涨跌幅数据等。",
+        description="根据概念ID获取详细信息，包括描述、insight、相关股票（含关联原因）、涨跌幅数据、层级结构等。",
         parameters={
             "type": "object",
             "properties": {
@@ -351,7 +369,7 @@ TOOLS: List[ToolDefinition] = [
     ),
     ToolDefinition(
         name="get_stock_concepts",
-        description="查询指定股票的所有相关概念板块，包括涨跌幅信息。",
+        description="查询指定股票的所有相关概念板块，包括涨跌幅信息和该股票被纳入概念的原因。",
         parameters={
             "type": "object",
             "properties": {
@@ -364,15 +382,14 @@ TOOLS: List[ToolDefinition] = [
                     "description": "返回概念数量",
                     "default": 50
                 },
-                "sort_by": {
-                    "type": "string",
-                    "description": "排序方式",
-                    "enum": ["stock_count", "concept_name", "recent"],
-                    "default": "stock_count"
-                },
                 "trade_date": {
                     "type": "string",
                     "description": "交易日期，格式：YYYY-MM-DD"
+                },
+                "include_reason": {
+                    "type": "boolean",
+                    "description": "是否返回股票被纳入概念的原因",
+                    "default": True
                 }
             },
             "required": ["stock_code"]
@@ -380,7 +397,7 @@ TOOLS: List[ToolDefinition] = [
     ),
     ToolDefinition(
         name="get_concept_statistics",
-        description="获取概念板块统计数据，包括涨幅榜、跌幅榜、活跃榜、波动榜、连涨榜。",
+        description="获取概念板块统计数据，包括涨幅榜、跌幅榜、活跃榜、波动榜、连涨榜。支持按概念类型（叶子/母概念）过滤。",
         parameters={
             "type": "object",
             "properties": {
@@ -400,11 +417,41 @@ TOOLS: List[ToolDefinition] = [
                     "type": "integer",
                     "description": "最少股票数量过滤",
                     "default": 3
+                },
+                "concept_type": {
+                    "type": "string",
+                    "description": "概念类型过滤：leaf(叶子概念)、lv1(一级母概念)、lv2(二级母概念)、lv3(三级母概念)",
+                    "enum": ["leaf", "lv1", "lv2", "lv3"],
+                    "default": "leaf"
                 }
             },
             "required": []
         }
     ),
+    ToolDefinition(
+        name="search_concepts_by_reason",
+        description="按股票关联原因搜索概念。可以找到所有reason中提到特定关键词的概念和股票，例如搜索'AI芯片'可以找到所有因AI芯片相关业务被纳入概念的股票。",
+        parameters={
+            "type": "object",
+            "properties": {
+                "query": {
+                    "type": "string",
+                    "description": "搜索关键词，会在股票关联原因中搜索"
+                },
+                "size": {
+                    "type": "integer",
+                    "description": "返回结果数量",
+                    "default": 20
+                },
+                "include_stock_details": {
+                    "type": "boolean",
+                    "description": "是否返回匹配的股票详情（含高亮原因）",
+                    "default": True
+                }
+            },
+            "required": ["query"]
+        }
+    ),
     ToolDefinition(
         name="search_limit_up_stocks",
         description="搜索涨停股票，支持按日期、关键词、板块等条件搜索。包括混合语义搜索。",
@@ -1468,8 +1515,9 @@ async def handle_search_roadshows(args: Dict[str, Any]) -> Any:
     return response.json()
 
 async def handle_search_concepts(args: Dict[str, Any]) -> Any:
-    """处理概念搜索
+    """处理概念搜索（适配 concept_api_v4）
 
+    支持语义搜索、层级过滤、原因搜索等 V4 新特性
     参数写死：size=12, page=1, sort_by="_score"
     trade_date 如果没传则使用今天的日期
     """
@@ -1485,21 +1533,38 @@ async def handle_search_concepts(args: Dict[str, Any]) -> Any:
         "sort_by": "_score",  # 写死，按相关度排序
         "trade_date": trade_date,
         "search_size": 100,
-        "use_knn": True
+        "use_knn": True,
+        # V4 新增参数
+        "search_reason": args.get("search_reason", False),  # 是否同时搜索股票原因
+        "include_stock_reasons": args.get("include_stock_reasons", False)  # 是否在返回的股票列表中包含原因
     }
 
+    # V4 新增：层级过滤
+    if args.get("filter_lv1"):
+        payload["filter_lv1"] = args["filter_lv1"]
+    if args.get("filter_lv2"):
+        payload["filter_lv2"] = args["filter_lv2"]
+
     logger.info(f"[search_concepts] 请求参数: {payload}")
     response = await HTTP_CLIENT.post(f"{ServiceEndpoints.CONCEPT_API}/search", json=payload)
     response.raise_for_status()
     return response.json()
 
 async def handle_get_concept_details(args: Dict[str, Any]) -> Any:
-    """处理概念详情获取"""
+    """处理概念详情获取（适配 concept_api_v4）
+
+    V4 返回内容包含：
+    - 概念基本信息（concept_id, concept, description, insight）
+    - 完整股票列表（含关联原因 reason）
+    - 层级结构信息（hierarchy: lv1/lv2/lv3）
+    - 涨跌幅数据（price_info）
+    """
     concept_id = args["concept_id"]
     params = {}
     if args.get("trade_date"):
         params["trade_date"] = args["trade_date"]
 
+    logger.info(f"[get_concept_details] 查询概念 {concept_id} 的详情")
     response = await HTTP_CLIENT.get(
         f"{ServiceEndpoints.CONCEPT_API}/concept/{concept_id}",
         params=params
@@ -1508,7 +1573,7 @@ async def handle_get_concept_details(args: Dict[str, Any]) -> Any:
     return response.json()
 
 async def handle_get_stock_concepts(args: Dict[str, Any]) -> Any:
-    """处理股票概念获取"""
+    """处理股票概念获取（适配 concept_api_v4）"""
     # 兼容不同的参数名: stock_code, seccode, code
     stock_code = args.get("stock_code") or args.get("seccode") or args.get("code")
     if not stock_code:
@@ -1516,13 +1581,12 @@ async def handle_get_stock_concepts(args: Dict[str, Any]) -> Any:
 
     params = {
         "size": args.get("size", 50),
-        "sort_by": args.get("sort_by", "stock_count"),
+        "include_reason": args.get("include_reason", True)  # V4 新增：返回股票关联原因
-        "include_description": True
     }
     if args.get("trade_date"):
         params["trade_date"] = args["trade_date"]
 
-    logger.info(f"[get_stock_concepts] 查询股票 {stock_code} 的概念")
+    logger.info(f"[get_stock_concepts] 查询股票 {stock_code} 的概念 (include_reason={params['include_reason']})")
     response = await HTTP_CLIENT.get(
         f"{ServiceEndpoints.CONCEPT_API}/stock/{stock_code}/concepts",
         params=params
@@ -1531,7 +1595,10 @@ async def handle_get_stock_concepts(args: Dict[str, Any]) -> Any:
     return response.json()
 
 async def handle_get_concept_statistics(args: Dict[str, Any]) -> Any:
-    """处理概念统计获取"""
+    """处理概念统计获取（适配 concept_api_v4）
+
+    V4 新增 concept_type 参数，支持按概念类型过滤统计数据
+    """
     params = {}
     if args.get("days"):
         params["days"] = args["days"]
@@ -1541,11 +1608,33 @@ async def handle_get_concept_statistics(args: Dict[str, Any]) -> Any:
         params["end_date"] = args["end_date"]
     if args.get("min_stock_count"):
         params["min_stock_count"] = args["min_stock_count"]
+    # V4 新增：概念类型过滤
+    if args.get("concept_type"):
+        params["concept_type"] = args["concept_type"]
 
+    logger.info(f"[get_concept_statistics] 请求参数: {params}")
     response = await HTTP_CLIENT.get(f"{ServiceEndpoints.CONCEPT_API}/statistics", params=params)
     response.raise_for_status()
     return response.json()
 
+async def handle_search_concepts_by_reason(args: Dict[str, Any]) -> Any:
+    """按股票关联原因搜索概念（concept_api_v4 新增接口）
+
+    调用 /search/reason 接口，可以找到所有reason中提到特定关键词的概念和股票
+    例如搜索"AI芯片"可以找到所有因AI芯片相关业务被纳入概念的股票
+    """
+    payload = {
+        "query": args["query"],
+        "size": args.get("size", 20),
+        "page": 1,  # 写死第一页
+        "include_stock_details": args.get("include_stock_details", True)
+    }
+
+    logger.info(f"[search_concepts_by_reason] 请求参数: {payload}")
+    response = await HTTP_CLIENT.post(f"{ServiceEndpoints.CONCEPT_API}/search/reason", json=payload)
+    response.raise_for_status()
+    return response.json()
+
 async def handle_search_limit_up_stocks(args: Dict[str, Any]) -> Any:
     """处理涨停股票搜索"""
     payload = {
@@ -1835,6 +1924,7 @@ TOOL_HANDLERS = {
     "get_concept_details": handle_get_concept_details,
     "get_stock_concepts": handle_get_stock_concepts,
     "get_concept_statistics": handle_get_concept_statistics,
+    "search_concepts_by_reason": handle_search_concepts_by_reason,
     "search_limit_up_stocks": handle_search_limit_up_stocks,
     "get_daily_stock_analysis": handle_get_daily_stock_analysis,
     "search_research_reports": handle_search_research_reports,