init

2026-02-26 13:43:44 +08:00
parent 2c2db92ae9
commit 68200cdfe6
51 changed files with 2107 additions and 351 deletions
@@ -0,0 +1,89 @@
+import json
+from typing import Any, Dict
+
+import httpx
+from langchain_core.tools import BaseTool
+
+
+class RestApiTool(BaseTool):
+    """RESTful API 调用工具"""
+
+    name: str = "rest_api"
+    description: str = (
+        "调用 RESTful API。输入为 JSON 字符串，示例："
+        '{"method":"GET","url":"https://example.com/api","params":{},"headers":{},"body":null,"timeout":30}'
+    )
+
+    def _run(self, payload: str) -> str:
+        """执行 RESTful API 请求"""
+        try:
+            data = json.loads(payload)
+        except Exception as e:
+            return f"请求参数解析失败: {e}"
+
+        method = str(data.get("method", "GET")).upper()
+        url = data.get("url")
+        params: Dict[str, Any] = data.get("params") or {}
+        headers: Dict[str, Any] = data.get("headers") or {}
+        body = data.get("body")
+        timeout = float(data.get("timeout", 30))
+
+        if not url:
+            return "缺少 url"
+
+        try:
+            with httpx.Client(timeout=timeout) as client:
+                response = client.request(
+                    method=method,
+                    url=url,
+                    params=params,
+                    headers=headers,
+                    json=body,
+                )
+            return json.dumps(
+                {
+                    "status_code": response.status_code,
+                    "headers": dict(response.headers),
+                    "text": response.text,
+                },
+                ensure_ascii=False,
+            )
+        except Exception as e:
+            return f"请求失败: {e}"
+
+    async def _arun(self, payload: str) -> str:
+        """工具的异步版本"""
+        try:
+            data = json.loads(payload)
+        except Exception as e:
+            return f"请求参数解析失败: {e}"
+
+        method = str(data.get("method", "GET")).upper()
+        url = data.get("url")
+        params: Dict[str, Any] = data.get("params") or {}
+        headers: Dict[str, Any] = data.get("headers") or {}
+        body = data.get("body")
+        timeout = float(data.get("timeout", 30))
+
+        if not url:
+            return "缺少 url"
+
+        try:
+            async with httpx.AsyncClient(timeout=timeout) as client:
+                response = await client.request(
+                    method=method,
+                    url=url,
+                    params=params,
+                    headers=headers,
+                    json=body,
+                )
+            return json.dumps(
+                {
+                    "status_code": response.status_code,
+                    "headers": dict(response.headers),
+                    "text": response.text,
+                },
+                ensure_ascii=False,
+            )
+        except Exception as e:
+            return f"请求失败: {e}"