MCP Development Framework

{ "sourceFile": "mcp_simple_tool/tools/web_tool.py", "activeCommit": 0, "commits": [ { "activePatchIndex": 0, "patches": [ { "date": 1741332226149, "content": "Index: \n===================================================================\n--- \n+++ \n" } ], "date": 1741332226149, "name": "Commit-0", "content": "\"\"\"\n网页内容获取工具\n\"\"\"\n\nimport httpx\nfrom typing import Dict, List, Any\nimport mcp.types as types\nfrom .base import BaseTool\n\n\nclass WebTool(BaseTool):\n \"\"\"\n 用于获取网页内容的工具\n \"\"\"\n \n @property\n def name(self) -> str:\n return \"url\"\n \n @property\n def description(self) -> str:\n return \"获取指定URL的网页内容\"\n \n @property\n def input_schema(self) -> Dict[str, Any]:\n return {\n \"type\": \"object\",\n \"required\": [\"url\"],\n \"properties\": {\n \"url\": {\n \"type\": \"string\",\n \"description\": \"要获取内容的网站URL\",\n }\n },\n }\n \n async def execute(self, arguments: Dict[str, Any]) -> List[types.TextContent | types.ImageContent | types.EmbeddedResource]:\n \"\"\"\n 获取网页内容\n \n Args:\n arguments: 参数字典，必须包含'url'键\n \n Returns:\n 网页内容列表\n \"\"\"\n if \"url\" not in arguments:\n return [types.TextContent(\n type=\"text\",\n text=\"错误: 缺少必要参数 'url'\"\n )]\n \n return await self._fetch_website(arguments[\"url\"])\n \n async def _fetch_website(self, url: str) -> List[types.TextContent | types.ImageContent | types.EmbeddedResource]:\n \"\"\"\n 获取指定URL的网页内容\n \n Args:\n url: 要获取内容的网站URL\n \n Returns:\n 网页内容列表\n \"\"\"\n headers = {\n \"User-Agent\": \"MCP Development Framework (github.com/modelcontextprotocol/python-sdk)\"\n }\n \n try:\n timeout = httpx.Timeout(10.0, connect=5.0)\n async with httpx.AsyncClient(\n follow_redirects=True, \n headers=headers,\n timeout=timeout\n ) as client:\n response = await client.get(url)\n response.raise_for_status()\n return [types.TextContent(type=\"text\", text=response.text)]\n except httpx.TimeoutException:\n return [types.TextContent(\n type=\"text\",\n text=\"错误: 请求超时，无法获取网站内容。\"\n )]\n except httpx.HTTPStatusError as e:\n return [types.TextContent(\n type=\"text\",\n text=f\"错误: HTTP {e.response.status_code} 错误，无法获取网站内容。\"\n )]\n except Exception as e:\n return [types.TextContent(\n type=\"text\",\n text=f\"错误: 获取网站内容失败: {str(e)}\"\n )] " } ] }