Undefined-bot 2.1.0__py3-none-any.whl

Undefined/skills/agents/file_analysis_agent/tools/read_text_file/handler.py

@@ -0,0 +1,90 @@
+import chardet
+from pathlib import Path
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    file_path: str = args.get("file_path", "")
+    max_lines: int | None = args.get("max_lines")
+
+    path = Path(file_path)
+
+    if not path.exists():
+        return f"错误：文件不存在 {file_path}"
+
+    if not path.is_file():
+        return f"错误：{file_path} 不是文件"
+
+    try:
+        file_size = path.stat().st_size
+
+        with open(path, "rb") as binary_file:
+            raw_data: bytes = binary_file.read()
+
+        detected = chardet.detect(raw_data)
+        encoding: str | None = detected.get("encoding") if detected else None
+        confidence: float = detected.get("confidence", 0.0) if detected else 0.0
+
+        encodings_to_try: list[str] = []
+        if encoding and confidence > 0.5:
+            encodings_to_try.append(encoding)
+        encodings_to_try.extend(
+            [
+                "utf-8",
+                "gbk",
+                "gb2312",
+                "gb18030",
+                "big5",
+                "shift_jis",
+                "euc-kr",
+                "latin-1",
+            ]
+        )
+
+        file_content: str = ""
+        actual_encoding: str = "unknown"
+        decode_error: Exception | None = None
+
+        for enc in encodings_to_try:
+            try:
+                file_content = raw_data.decode(enc)
+                actual_encoding = enc
+                break
+            except (UnicodeDecodeError, LookupError) as e:
+                decode_error = e
+                continue
+
+        if not file_content and decode_error:
+            return f"错误：无法解码文件，可能是二进制文件。文件大小：{file_size} 字节"
+
+        control_chars = sum(
+            1 for c in file_content if ord(c) < 32 and c not in "\t\n\r"
+        )
+        if control_chars > len(file_content) * 0.05:
+            return f"错误：文件包含大量控制字符，可能是二进制文件。文件大小：{file_size} 字节"
+
+        total_lines = file_content.count("\n")
+        if max_lines and total_lines > max_lines:
+            lines = file_content.split("\n")[:max_lines]
+            file_content = "\n".join(lines)
+            truncated = True
+        else:
+            truncated = False
+
+        info_lines: list[str] = []
+        info_lines.append(f"文件大小：{file_size} 字节")
+        info_lines.append(f"检测编码：{encoding} (置信度: {confidence:.2f})")
+        info_lines.append(f"实际编码：{actual_encoding}")
+        info_lines.append(f"内容行数：{total_lines}")
+        if truncated:
+            info_lines.append(f"已截断到前 {max_lines} 行")
+        info_lines.append("")
+
+        return "\n".join(info_lines) + file_content
+
+    except Exception as e:
+        logger.exception(f"读取文件失败: {e}")
+        return f"读取文件失败: {e}"

Undefined/skills/agents/info_agent/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "info_agent",
+        "description": "信息查询助手，提供天气、热搜、新闻、历史、网络诊断、编码工具等各类实用信息查询功能。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "用户的查询需求，例如：'北京天气'、'今日热搜'、'查询 qq123456 的等级'"
+                }
+            },
+            "required": ["prompt"]
+        }
+    }
+}

Undefined/skills/agents/info_agent/handler.py

@@ -0,0 +1,220 @@
+from typing import Any, Dict, Callable
+import importlib.util
+import json
+import asyncio
+import aiofiles
+import logging
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+
+class AgentToolRegistry:
+    """Agent 内部的工具注册表"""
+
+    def __init__(self, tools_dir: Path) -> None:
+        self.tools_dir: Path = (
+            tools_dir if isinstance(tools_dir, Path) else Path(tools_dir)
+        )
+        self._tools_schema: list[dict[str, Any]] = []
+        self._tools_handlers: dict[str, Callable[..., Any]] = {}
+        self.load_tools()
+
+    def load_tools(self) -> None:
+        """加载 agent 专属工具"""
+        if not self.tools_dir.exists():
+            logger.warning(f"Agent 工具目录不存在: {self.tools_dir}")
+            return
+
+        for item in self.tools_dir.iterdir():
+            if item.is_dir() and not item.name.startswith("_"):
+                self._load_tool_from_dir(item)
+
+        logger.info(
+            f"Agent 加载了 {len(self._tools_schema)} 个工具: {list(self._tools_handlers.keys())}"
+        )
+
+    def _load_tool_from_dir(self, tool_dir: Path) -> None:
+        """从目录加载工具"""
+        config_path: Path = tool_dir / "config.json"
+        handler_path: Path = tool_dir / "handler.py"
+
+        if not config_path.exists() or not handler_path.exists():
+            return
+
+        try:
+            with open(config_path, "r", encoding="utf-8") as f:
+                config: dict[str, Any] = json.load(f)
+
+            if "function" not in config or "name" not in config.get("function", {}):
+                return
+
+            tool_name: str = config["function"]["name"]
+
+            spec = importlib.util.spec_from_file_location(
+                f"agent_tools.{tool_name}", handler_path
+            )
+            if spec is None or spec.loader is None:
+                return
+
+            module = importlib.util.module_from_spec(spec)
+            spec.loader.exec_module(module)
+
+            if not hasattr(module, "execute"):
+                return
+
+            self._tools_schema.append(config)
+            self._tools_handlers[tool_name] = module.execute
+
+        except Exception as e:
+            logger.error(f"从 {tool_dir} 加载工具失败: {e}")
+
+    def get_tools_schema(self) -> list[dict[str, Any]]:
+        """获取工具 schema"""
+        return self._tools_schema
+
+    async def execute_tool(
+        self, tool_name: str, args: dict[str, Any], context: dict[str, Any]
+    ) -> str:
+        """执行工具"""
+        handler = self._tools_handlers.get(tool_name)
+        if not handler:
+            return f"未找到工具: {tool_name}"
+
+        try:
+            if asyncio.iscoroutinefunction(handler):
+                result = await handler(args, context)
+            else:
+                result = handler(args, context)
+            return str(result)
+        except Exception as e:
+            logger.exception(f"执行工具 {tool_name} 时出错")
+            return f"执行工具 {tool_name} 时出错: {str(e)}"
+
+
+async def _load_prompt() -> str:
+    """从 prompt.md 文件加载系统提示词"""
+    prompt_path: Path = Path(__file__).parent / "prompt.md"
+    if prompt_path.exists():
+        async with aiofiles.open(prompt_path, "r", encoding="utf-8") as f:
+            return await f.read()
+    return _get_default_prompt()
+
+
+def _get_default_prompt() -> str:
+    """默认提示词"""
+    return "你是一个信息查询助手..."
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    """执行 info_agent"""
+    user_prompt: str = args.get("prompt", "")
+
+    if not user_prompt:
+        return "请提供您的查询需求"
+
+    agent_tools_dir: Path = Path(__file__).parent / "tools"
+    tool_registry = AgentToolRegistry(agent_tools_dir)
+
+    tools: list[dict[str, Any]] = tool_registry.get_tools_schema()
+
+    ai_client = context.get("ai_client")
+    if not ai_client:
+        return "AI client 未在上下文中提供"
+
+    agent_config = ai_client.agent_config
+
+    system_prompt: str = await _load_prompt()
+
+    messages: list[dict[str, Any]] = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": f"用户需求：{user_prompt}"},
+    ]
+
+    max_iterations: int = 20
+    iteration: int = 0
+
+    while iteration < max_iterations:
+        iteration += 1
+
+        try:
+            response = await ai_client._http_client.post(
+                agent_config.api_url,
+                headers={
+                    "Authorization": f"Bearer {agent_config.api_key}",
+                    "Content-Type": "application/json",
+                },
+                json=ai_client._build_request_body(
+                    model_config=agent_config,
+                    messages=messages,
+                    max_tokens=agent_config.max_tokens,
+                    tools=tools if tools else None,
+                    tool_choice="auto",
+                ),
+            )
+            response.raise_for_status()
+            result: dict[str, Any] = response.json()
+
+            choice: dict[str, Any] = result.get("choices", [{}])[0]
+            message: dict[str, Any] = choice.get("message", {})
+            content: str = message.get("content") or ""
+            tool_calls: list[dict[str, Any]] = message.get("tool_calls", [])
+
+            if content.strip() and tool_calls:
+                content = ""
+
+            if not tool_calls:
+                return content
+
+            messages.append(
+                {"role": "assistant", "content": content, "tool_calls": tool_calls}
+            )
+
+            # 准备并发执行工具
+            tool_tasks = []
+            tool_call_ids = []
+
+            for tool_call in tool_calls:
+                call_id: str = tool_call.get("id", "")
+                function: dict[str, Any] = tool_call.get("function", {})
+                function_name: str = function.get("name", "")
+                function_args_str: str = function.get("arguments", "{}")
+
+                logger.info(f"Agent 正在准备工具: {function_name}")
+
+                try:
+                    function_args: dict[str, Any] = json.loads(function_args_str)
+                except json.JSONDecodeError:
+                    function_args = {}
+
+                tool_call_ids.append(call_id)
+                tool_tasks.append(
+                    tool_registry.execute_tool(function_name, function_args, context)
+                )
+
+            # 并发执行
+            if tool_tasks:
+                logger.info(f"Agent 正在并发执行 {len(tool_tasks)} 个工具")
+                results = await asyncio.gather(*tool_tasks, return_exceptions=True)
+
+                for i, tool_result in enumerate(results):
+                    call_id = tool_call_ids[i]
+                    content_str: str = ""
+                    if isinstance(tool_result, Exception):
+                        content_str = f"错误: {str(tool_result)}"
+                    else:
+                        content_str = str(tool_result)
+
+                    messages.append(
+                        {
+                            "role": "tool",
+                            "tool_call_id": call_id,
+                            "content": content_str,
+                        }
+                    )
+
+        except Exception as e:
+            logger.exception(f"Agent 执行失败: {e}")
+            return f"处理失败: {e}"
+
+    return "达到最大迭代次数"

Undefined/skills/agents/info_agent/intro.md

@@ -0,0 +1,22 @@
+# 信息查询助手
+
+## 能力
+- **生活服务**：查询指定城市天气、查询实时黄金价格
+- **热搜榜单**：获取百度、微博、抖音热搜榜，获取腾讯新闻
+- **文化历史**：查询历史上的今天
+- **网络工具**：网络诊断(TCP Ping/网速测试/连通性检查)、Whois域名查询、Base64编解码、Hash计算
+- **其他查询**：查询QQ等级
+
+## 适用场景
+- 用户查询天气、金价时
+- 用户想看新闻或热搜时
+- 用户查询历史事件时
+- 用户需要测试网络连通性、查询域名信息时
+- 用户需要进行 Base64 编解码或计算 Hash 时
+- 用户查询 QQ 等级时
+
+## 参数说明
+- `city` (天气): 城市名称（如：北京）
+- `qq` (QQ等级): 目标 QQ 号码
+- `target`/`host` (网络工具): 目标域名或 IP
+- `text` (编码/哈希): 需要处理的文本内容

Undefined/skills/agents/info_agent/prompt.md

@@ -0,0 +1,27 @@
+你是一个信息查询助手，专门帮助用户获取各种实用信息。
+
+你的任务是根据用户需求，选择合适的工具来完成任务：
+
+1. **天气查询** → weather_query
+2. **黄金价格** → gold_price
+3. **百度热搜** → baiduhot
+4. **微博热搜** → weibohot
+5. **抖音热搜** → douyinhot
+6. **腾讯新闻** → news_tencent
+7. **历史今天** → history
+8. **文昌帝君** → wenchang_dijun
+9. **QQ 等级** → qq_level_query
+10. **网络诊断** → tcping / speed / net_check
+11. **Base64 编码/解码** → base64
+12. **哈希计算** → hash
+13. **Whois 查询** → whois
+
+注意事项：
+- 天气查询支持城市名称
+- QQ 等级查询需要提供目标 QQ 号
+- 网络诊断可能需要一定时间
+- 保持回答简洁明了
+
+如果用户需求不明确，请先询问用户澄清问题。
+
+如果问题涉及时间，立刻调用时间工具获取。

Undefined/skills/agents/info_agent/tools/__init__.py

@@ -0,0 +1 @@
+# Info Agent Tools

Undefined/skills/agents/info_agent/tools/baiduhot/config.json

@@ -0,0 +1,18 @@
+{
+    "type": "function",
+    "function": {
+        "name": "baiduhot",
+        "description": "获取百度热搜排行榜，实时返回百度搜索引擎上的热门搜索词和趋势。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "limit": {
+                    "type": "integer",
+                    "description": "返回的热搜数量，默认10",
+                    "default": 10
+                }
+            },
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/info_agent/tools/baiduhot/handler.py

@@ -0,0 +1,49 @@
+from typing import Any, Dict
+import logging
+import httpx
+
+logger = logging.getLogger(__name__)
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    limit = args.get("limit", 10)
+    
+    if limit < 1 or limit > 50:
+        return "❌ 热搜数量必须在 1-50 之间"
+    
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            logger.info(f"获取百度热搜，数量: {limit}")
+            
+            response = await client.get("https://v2.xxapi.cn/api/baiduhot")
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200:
+                return f"获取百度热搜失败: {data.get('msg')}"
+            
+            hot_list = data.get("data", [])
+            if not hot_list:
+                return "暂无热搜数据"
+            
+            result = f"【百度热搜 TOP {min(limit, len(hot_list))}】\n\n"
+            
+            for idx, item in enumerate(hot_list[:limit], 1):
+                title = item.get("title", "")
+                hot = item.get("hot", "")
+                url = item.get("url", "")
+                result += f"{idx}. {title}\n"
+                result += f"   热度: {hot}\n"
+                if url:
+                    result += f"   链接: {url}\n"
+                result += "\n"
+            
+            return result
+            
+    except httpx.TimeoutException:
+        return "请求超时，请稍后重试"
+    except httpx.HTTPStatusError as e:
+        logger.error(f"HTTP 错误: {e}")
+        return f"获取百度热搜失败: {e}"
+    except Exception as e:
+        logger.exception(f"获取百度热搜失败: {e}")
+        return f"获取百度热搜失败: {e}"

Undefined/skills/agents/info_agent/tools/base64/config.json

@@ -0,0 +1,22 @@
+{
+    "type": "function",
+    "function": {
+        "name": "base64",
+        "description": "Base64加密与解密工具，支持对字符串进行Base64编码（加密）和解码（解密）操作。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "text": {
+                    "type": "string",
+                    "description": "需要加密或解密的字符串"
+                },
+                "operation": {
+                    "type": "string",
+                    "description": "操作类型：encode（加密）或 decode（解密）",
+                    "enum": ["encode", "decode"]
+                }
+            },
+            "required": ["text", "operation"]
+        }
+    }
+}

Undefined/skills/agents/info_agent/tools/base64/handler.py

@@ -0,0 +1,44 @@
+from typing import Any, Dict
+import logging
+import httpx
+
+logger = logging.getLogger(__name__)
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    text = args.get("text")
+    operation = args.get("operation")
+    
+    if not text:
+        return "❌ 文本不能为空"
+    if not operation:
+        return "❌ 操作类型不能为空"
+    if operation not in ["encode", "decode"]:
+        return "❌ 操作类型必须是 encode（加密）或 decode（解密）"
+    
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            params = {
+                "type": operation,
+                "text": text
+            }
+            logger.info(f"Base64 {operation}: {text[:50]}...")
+            
+            response = await client.get("https://v2.xxapi.cn/api/base64", params=params)
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200 and data.get("code") != "200":
+                return f"Base64 {operation} 失败: {data.get('msg')}"
+            
+            result = data.get("data")
+            operation_text = "加密" if operation == "encode" else "解密"
+            return f"Base64{operation_text}结果：\n{result}"
+            
+    except httpx.TimeoutException:
+        return "请求超时，请稍后重试"
+    except httpx.HTTPStatusError as e:
+        logger.error(f"HTTP 错误: {e}")
+        return f"请求失败: {e}"
+    except Exception as e:
+        logger.exception(f"Base64操作失败: {e}")
+        return f"操作失败: {e}"

Undefined/skills/agents/info_agent/tools/douyinhot/config.json

@@ -0,0 +1,18 @@
+{
+    "type": "function",
+    "function": {
+        "name": "douyinhot",
+        "description": "获取抖音热榜，实时返回抖音平台上的热门视频、话题和排行榜数据。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "limit": {
+                    "type": "integer",
+                    "description": "返回的热榜数量，默认10",
+                    "default": 10
+                }
+            },
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/info_agent/tools/douyinhot/handler.py

@@ -0,0 +1,53 @@
+from typing import Any, Dict
+import logging
+import httpx
+
+logger = logging.getLogger(__name__)
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    limit = args.get("limit", 10)
+    
+    if limit < 1 or limit > 50:
+        return "❌ 热榜数量必须在 1-50 之间"
+    
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            logger.info(f"获取抖音热榜，数量: {limit}")
+            
+            response = await client.get("https://v2.xxapi.cn/api/douyinhot")
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200:
+                return f"获取抖音热榜失败: {data.get('msg')}"
+            
+            hot_list = data.get("data", [])
+            if not hot_list:
+                return "暂无热榜数据"
+            
+            result = f"【抖音热榜 TOP {min(limit, len(hot_list))}】\n\n"
+            
+            for idx, item in enumerate(hot_list[:limit], 1):
+                word = item.get("word", "")
+                hot_value = item.get("hot_value", 0)
+                position = item.get("position", idx)
+                video_count = item.get("video_count", 0)
+                
+                # 将热度值转换为万
+                hot_str = f"{hot_value / 10000:.1f}万" if hot_value >= 10000 else str(hot_value)
+                
+                result += f"{position}. {word}\n"
+                result += f"   热度: {hot_str}\n"
+                result += f"   视频数: {video_count}\n"
+                result += "\n"
+            
+            return result
+            
+    except httpx.TimeoutException:
+        return "请求超时，请稍后重试"
+    except httpx.HTTPStatusError as e:
+        logger.error(f"HTTP 错误: {e}")
+        return f"获取抖音热榜失败: {e}"
+    except Exception as e:
+        logger.exception(f"获取抖音热榜失败: {e}")
+        return f"获取抖音热榜失败: {e}"

Undefined/skills/agents/info_agent/tools/get_current_time/config.json

@@ -0,0 +1,12 @@
+{
+    "type": "function",
+    "function": {
+        "name": "get_current_time",
+        "description": "获取当前系统时间。",
+        "parameters": {
+            "type": "object",
+            "properties": {},
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/info_agent/tools/get_current_time/handler.py

@@ -0,0 +1,5 @@
+from typing import Any, Dict
+from datetime import datetime
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    return datetime.now().strftime("%Y-%m-%d %H:%M:%S")

Undefined/skills/agents/info_agent/tools/gold_price/config.json

@@ -0,0 +1,12 @@
+{
+    "type": "function",
+    "function": {
+        "name": "gold_price",
+        "description": "获取今日黄金价格，包括银行投资金条价格、黄金回收价格和贵金属品牌价格。",
+        "parameters": {
+            "type": "object",
+            "properties": {},
+            "required": []
+        }
+    }
+}