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,99 @@
+import json
+from typing import Any, Dict
+
+import httpx
+from langchain_core.tools import BaseTool
+
+from config import Config
+
+
+class SrApiQueryTool(BaseTool):
+    """调用 SR API 执行 SQL 查询"""
+
+    name: str = "sr_api_query"
+    description: str = (
+        "调用 SR API 执行 SQL 查询。输入为 JSON 字符串，示例："
+        '{"sql":"SELECT * FROM table","page":1,"rows":10,"orderBySelect":true,"timeout":30}'
+    )
+
+    def _run(self, payload: str) -> str:
+        """执行 SQL 查询"""
+        try:
+            data = json.loads(payload)
+        except Exception as e:
+            return f"请求参数解析失败: {e}"
+
+        sql = data.get("sql")
+        page = int(data.get("page", 1))
+        rows = int(data.get("rows", 10))
+        order_by_select = bool(data.get("orderBySelect", True))
+        timeout = float(data.get("timeout", 30))
+
+        if not sql:
+            return "缺少 sql"
+
+        cfg = Config.get_section("sr_api")
+        url = cfg.get("url")
+        app_key = cfg.get("llzappkey")
+        secret_key = cfg.get("llzsercret")
+
+        if not url or not app_key or not secret_key:
+            return "sr_api 配置缺失 url/llzAppkey/llzSercret"
+
+        headers = {"llzAppkey": app_key, "llzSercret": secret_key}
+        body = {"sql": sql, "page": page, "rows": rows, "orderBySelect": order_by_select}
+
+        try:
+            with httpx.Client(timeout=timeout) as client:
+                response = client.post(url, json=body, headers=headers)
+            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}"
+
+        sql = data.get("sql")
+        page = int(data.get("page", 1))
+        rows = int(data.get("rows", 10))
+        order_by_select = bool(data.get("orderBySelect", True))
+        timeout = float(data.get("timeout", 30))
+
+        if not sql:
+            return "缺少 sql"
+
+        cfg = Config.get_section("sr_api")
+        url = cfg.get("url")
+        app_key = cfg.get("llzappkey")
+        secret_key = cfg.get("llzsercret")
+
+        if not url or not app_key or not secret_key:
+            return "sr_api 配置缺失 url/llzAppkey/llzSercret"
+
+        headers = {"llzAppkey": app_key, "llzSercret": secret_key}
+        body = {"sql": sql, "page": page, "rows": rows, "orderBySelect": order_by_select}
+
+        try:
+            async with httpx.AsyncClient(timeout=timeout) as client:
+                response = await client.post(url, json=body, headers=headers)
+            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}"