查找股票代码

Instructions

Implementation Reference

• mcp_aktools/__init__.py:28-41 (handler)

  Main handler function for the 'search' MCP tool. Decorated with @mcp.tool, accepts keyword and market parameters, delegates to ak_search() helper, and formats the result.

  @mcp.tool(
      title="查找股票代码",
      description="根据股票名称、公司名称等关键词查找股票代码, 不支持加密货币。"
                  "该工具比较耗时，当你知道股票代码或用户已指定股票代码时，建议直接通过股票代码使用其他工具",
  )
  def search(
      keyword: str = Field(description="搜索关键词，公司名称、股票名称、股票代码、证券简称"),
      market: str = field_market,
  ):
      info = ak_search(None, keyword, market)
      if info is not None:
          suffix = f"交易市场: {market}"
          return "\n".join([info.to_string(), suffix])
      return f"Not Found for {keyword}"

• mcp_aktools/__init__.py:34-35 (schema)

  Input schema for the search tool: keyword (required string) and market (optional, defaults to 'sh').

  keyword: str = Field(description="搜索关键词，公司名称、股票名称、股票代码、证券简称"),
  market: str = field_market,

• mcp_aktools/__init__.py:527-561 (helper)

  Helper function that searches across multiple markets (SH, SZ, HK, US) using akshare APIs. Supports exact code match, keyword match (code or name), substring match (keyword length >= 4), and prefix match.

  def ak_search(symbol=None, keyword=None, market=None):
      markets = [
          ["sh", ak.stock_info_a_code_name, "code", "name"],
          ["sh", ak.stock_info_sh_name_code, "证券代码", "证券简称"],
          ["sz", ak.stock_info_sz_name_code, "A股代码", "A股简称"],
          ["hk", ak.stock_hk_spot, "代码", "中文名称"],
          ["hk", ak.stock_hk_spot_em, "代码", "名称"],
          ["us", ak.get_us_stock_name, "symbol", "cname"],
          ["us", ak.get_us_stock_name, "symbol", "name"],
          ["sh", ak.fund_etf_spot_ths, "基金代码", "基金名称"],
          ["sz", ak.fund_etf_spot_ths, "基金代码", "基金名称"],
          ["sh", ak.fund_info_index_em, "基金代码", "基金名称"],
          ["sz", ak.fund_info_index_em, "基金代码", "基金名称"],
          ["sh", ak.fund_etf_spot_em, "代码", "名称"],
          ["sz", ak.fund_etf_spot_em, "代码", "名称"],
      ]
      for m in markets:
          if market and market != m[0]:
              continue
          all = ak_cache(m[1], ttl=86400, ttl2=86400*7)
          if all is None or all.empty:
              continue
          for _, v in all.iterrows():
              code, name = str(v[m[2]]).upper(), str(v[m[3]]).upper()
              if symbol and symbol.upper() == code:
                  return v
              if keyword and keyword.upper() in [code, name]:
                  return v
          for _, v in all.iterrows() if keyword else []:
              name = str(v[m[3]])
              if len(keyword) >= 4 and keyword in name:
                  return v
              if name.startswith(keyword):
                  return v
      return None

• mcp_aktools/__init__.py:564-579 (helper)

  Caching helper that wraps akshare API calls with a two-level cache (in-memory TTL cache + disk cache) to avoid redundant network requests.

  def ak_cache(fun, *args, **kwargs) -> pd.DataFrame | None:
      key = kwargs.pop("key", None)
      if not key:
          key = f"{fun.__name__}-{args}-{kwargs}"
      ttl1 = kwargs.pop("ttl", 86400)
      ttl2 = kwargs.pop("ttl2", None)
      cache = CacheKey.init(key, ttl1, ttl2)
      all = cache.get()
      if all is None:
          try:
              _LOGGER.info("Request akshare: %s", [key, args, kwargs])
              all = fun(*args, **kwargs)
              cache.set(all)
          except Exception as exc:
              _LOGGER.exception(str(exc))
      return all

• mcp_aktools/__init__.py:28-31 (registration)

  Registration of the 'search' tool via the @mcp.tool decorator on the FastMCP instance, with title '查找股票代码'.

  @mcp.tool(
      title="查找股票代码",
      description="根据股票名称、公司名称等关键词查找股票代码, 不支持加密货币。"
                  "该工具比较耗时，当你知道股票代码或用户已指定股票代码时，建议直接通过股票代码使用其他工具",