diff --git a/__pycache__/mcp_database.cpython-310.pyc b/__pycache__/mcp_database.cpython-310.pyc index e6d84f78..026c6759 100644 Binary files a/__pycache__/mcp_database.cpython-310.pyc and b/__pycache__/mcp_database.cpython-310.pyc differ diff --git a/__pycache__/mcp_server.cpython-310.pyc b/__pycache__/mcp_server.cpython-310.pyc index 42430b61..81c425da 100644 Binary files a/__pycache__/mcp_server.cpython-310.pyc and b/__pycache__/mcp_server.cpython-310.pyc differ diff --git a/mcp_database.py b/mcp_database.py index 4c868afb..0f467013 100644 --- a/mcp_database.py +++ b/mcp_database.py @@ -1030,3 +1030,51 @@ async def get_stock_intraday_statistics( except Exception as e: logger.error(f"[ClickHouse] 日内统计失败: {e}", exc_info=True) return {"success": False, "error": str(e)} + + +async def get_stock_code_by_name(stock_name: str) -> Dict[str, Any]: + """ + 根据股票名称查询股票代码 + + Args: + stock_name: 股票名称(支持模糊匹配) + + Returns: + 匹配的股票列表,包含代码和名称 + """ + pool = await get_pool() + + async with pool.acquire() as conn: + async with conn.cursor(aiomysql.DictCursor) as cursor: + # 使用 LIKE 进行模糊匹配 + query = """ + SELECT DISTINCT + SECCODE as code, + SECNAME as name, + F030V as industry + FROM ea_baseinfo + WHERE SECNAME LIKE %s + OR SECNAME = %s + ORDER BY + CASE WHEN SECNAME = %s THEN 0 ELSE 1 END, + SECCODE + LIMIT 10 + """ + + # 精确匹配和模糊匹配 + like_pattern = f"%{stock_name}%" + + await cursor.execute(query, (like_pattern, stock_name, stock_name)) + results = await cursor.fetchall() + + if not results: + return { + "success": False, + "error": f"未找到名称包含 '{stock_name}' 的股票" + } + + return { + "success": True, + "data": results, + "count": len(results) + } diff --git a/mcp_server.py b/mcp_server.py index 6cda9cd7..ac55ace4 100644 --- a/mcp_server.py +++ b/mcp_server.py @@ -87,7 +87,7 @@ MODEL_CONFIGS = { }, "deepmoney": { "api_key": "", # 空值 - "base_url": "http://111.62.35.51:8000/v1", + "base_url": "http://111.62.35.50:8000/v1", "model": "deepmoney", "max_tokens": 32768, # DeepMoney 本地托管,上下文 65536,输出限制 32768 }, @@ -495,6 +495,20 @@ TOOLS: List[ToolDefinition] = [ "required": ["query"] } ), + ToolDefinition( + name="get_stock_code_by_name", + description="根据股票名称查询股票代码,支持模糊匹配。当只知道股票名称不知道代码时使用。", + parameters={ + "type": "object", + "properties": { + "stock_name": { + "type": "string", + "description": "股票名称,例如:'贵州茅台'、'舒泰神'、'比亚迪'" + } + }, + "required": ["stock_name"] + } + ), ToolDefinition( name="get_stock_basic_info", description="获取股票基本信息,包括公司名称、行业、地址、主营业务、高管等基础数据。", @@ -1578,9 +1592,24 @@ async def handle_search_research_reports(args: Dict[str, Any]) -> Any: response.raise_for_status() return response.json() +async def handle_get_stock_code_by_name(args: Dict[str, Any]) -> Any: + """根据股票名称查询股票代码""" + # 兼容不同的参数名: stock_name, name + stock_name = args.get("stock_name") or args.get("name") + if not stock_name: + return {"success": False, "error": "缺少股票名称参数 (stock_name/name)"} + + logger.info(f"[get_stock_code_by_name] 查询股票名称: {stock_name}") + result = await db.get_stock_code_by_name(stock_name) + return result + async def handle_get_stock_basic_info(args: Dict[str, Any]) -> Any: """处理股票基本信息查询""" - seccode = args["seccode"] + # 兼容不同的参数名: seccode, stock_code, code + seccode = args.get("seccode") or args.get("stock_code") or args.get("code") + if not seccode: + return {"success": False, "error": "缺少股票代码参数 (seccode/stock_code/code)"} + result = await db.get_stock_basic_info(seccode) if result: return {"success": True, "data": result} @@ -1808,6 +1837,7 @@ TOOL_HANDLERS = { "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, + "get_stock_code_by_name": handle_get_stock_code_by_name, "get_stock_basic_info": handle_get_stock_basic_info, "get_stock_financial_index": handle_get_stock_financial_index, "get_stock_trade_data": handle_get_stock_trade_data, @@ -3466,7 +3496,7 @@ MEETING_MODEL_CONFIGS = { }, "deepmoney": { "api_key": "", - "base_url": "http://111.62.35.51:8000/v1", + "base_url": "http://111.62.35.50:8000/v1", "model": "deepmoney", }, }