vf/vf_react
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新ios

Browse Source
This commit is contained in:
价小前
2026-01-17 18:38:55 +08:00
parent 5ed4eaf048
commit f3426e7c64
1 changed files with 109 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

128
mcp_server.py
View File

@@ -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_description": True
"include_reason": args.get("include_reason", True) # V4 新增：返回股票关联原因
}
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,