MCP2Tavily

Instructions

Implementation Reference

• mcp2tavily.py:106-109 (handler)

  The handler function for the 'search_web_info' tool. It is decorated with @mcp.tool() for registration and delegates the search logic to the _do_search helper function.

  @mcp.tool()
  def search_web_info(query: str) -> str:
      """从网络搜索用户查询的信息"""
      return _do_search(query)

• mcp2tavily.py:32-61 (helper)

  Supporting utility function that performs the actual web search using TavilyClient, processes the response including answer and top 3 sources, handles UTF-8 encoding issues, and formats the output.

  def _do_search(query: str) -> str:
      """Internal function to handle the search logic with UTF-8 support"""
      try:
          # 确保查询字符串是UTF-8编码
          query = query.encode('utf-8').decode('utf-8')
          tavily_client = TavilyClient(api_key=API_KEY)
          response = tavily_client.search(
              query=query,
              search_depth="basic",
              include_answer=True,
              include_raw_content=False
          )
          
          # 确保响应文本是UTF-8编码
          answer = response.get('answer', 'No answer found').encode('utf-8').decode('utf-8')
          sources = response.get('sources', [])
          
          result = f"Answer: {answer}\n\nSources:"
          for source in sources[:3]:
              title = source.get('title', 'No title').encode('utf-8').decode('utf-8')
              url = source.get('url', 'No URL')
              result += f"\n- {title}: {url}"
              
          return result
      except UnicodeError as e:
          logger.error(f"Encoding error: {str(e)}")
          return "Error: Unicode encoding issue occurred"
      except Exception as e:
          logger.error(f"Search error: {str(e)}")
          return f"Error performing search: {str(e)}"