Undefined-bot 2.1.0__py3-none-any.whl

Undefined/skills/agents/entertainment_agent/tools/horoscope/handler.py

@@ -0,0 +1,141 @@
+from typing import Any, Dict
+import logging
+import httpx
+
+logger = logging.getLogger(__name__)
+
+# 星座名称映射（中文 -> 英文）
+CONSTELLATION_MAP = {
+    "白羊座": "aries",
+    "金牛座": "taurus",
+    "双子座": "gemini",
+    "巨蟹座": "cancer",
+    "狮子座": "leo",
+    "处女座": "virgo",
+    "天秤座": "libra",
+    "天蝎座": "scorpio",
+    "射手座": "sagittarius",
+    "摩羯座": "capricorn",
+    "水瓶座": "aquarius",
+    "双鱼座": "pisces",
+}
+
+# 时间类型映射（中文 -> 英文）
+TIME_TYPE_MAP = {
+    "今日": "today",
+    "本周": "week",
+    "本月": "month",
+    "本年": "year",
+}
+
+# 星座星级显示
+STAR_MAP = {1: "★", 2: "★★", 3: "★★★", 4: "★★★★", 5: "★★★★★"}
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    constellation = args.get("constellation")
+    time_type = args.get("time_type", "today")
+    
+    if not constellation:
+        return "❌ 星座不能为空"
+    
+    # 转换星座名称为英文
+    constellation_en = CONSTELLATION_MAP.get(constellation, constellation)
+    if constellation_en not in CONSTELLATION_MAP.values():
+        return f"❌ 不支持的星座: {constellation}\n支持的星座: {', '.join(CONSTELLATION_MAP.keys())}"
+    
+    # 转换时间类型为英文
+    time_type_en = TIME_TYPE_MAP.get(time_type, time_type)
+    if time_type_en not in TIME_TYPE_MAP.values():
+        return f"❌ 不支持的时间类型: {time_type}\n支持的时间类型: {', '.join(TIME_TYPE_MAP.keys())}"
+    
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            params = {
+                "type": constellation_en,
+                "time": time_type_en
+            }
+            logger.info(f"获取星座运势: {constellation} ({constellation_en}), 时间: {time_type} ({time_type_en})")
+            
+            response = await client.get("https://v2.xxapi.cn/api/horoscope", params=params)
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200:
+                return f"获取运势失败: {data.get('msg')}"
+            
+            fortune_data = data.get("data", {})
+            
+            # 格式化运势信息
+            title = fortune_data.get("title", constellation)
+            time_text = fortune_data.get("type", time_type)
+            short_comment = fortune_data.get("shortcomment", "")
+            date_text = fortune_data.get("time", "")
+            
+            # 运势评分
+            fortune = fortune_data.get("fortune", {})
+            fortune_stars = {
+                "综合": STAR_MAP.get(fortune.get("all", 0), ""),
+                "健康": STAR_MAP.get(fortune.get("health", 0), ""),
+                "爱情": STAR_MAP.get(fortune.get("love", 0), ""),
+                "财运": STAR_MAP.get(fortune.get("money", 0), ""),
+                "工作": STAR_MAP.get(fortune.get("work", 0), ""),
+            }
+            
+            # 运势指数
+            index = fortune_data.get("index", {})
+            
+            # 运势文本
+            fortunetext = fortune_data.get("fortunetext", {})
+            
+            # 幸运信息
+            lucky_color = fortune_data.get("luckycolor", "")
+            lucky_constellation = fortune_data.get("luckyconstellation", "")
+            lucky_number = fortune_data.get("luckynumber", "")
+            
+            # 宜忌
+            todo = fortune_data.get("todo", {})
+            todo_yi = todo.get("ji", "")
+            todo_ji = todo.get("yi", "")
+            
+            # 构建结果
+            result = f"【{title} {time_text}】{date_text}\n"
+            result += f"短评：{short_comment}\n\n"
+            
+            result += "【运势评分】\n"
+            for name, stars in fortune_stars.items():
+                result += f"{name}：{stars}\n"
+            result += "\n"
+            
+            result += "【运势指数】\n"
+            for name, value in index.items():
+                result += f"{name}：{value}\n"
+            result += "\n"
+            
+            result += "【运势详情】\n"
+            for name, text in fortunetext.items():
+                result += f"{name}：{text}\n"
+            result += "\n"
+            
+            result += "【幸运信息】\n"
+            result += f"幸运色：{lucky_color}\n"
+            result += f"幸运星座：{lucky_constellation}\n"
+            result += f"幸运数字：{lucky_number}\n"
+            result += "\n"
+            
+            if todo_yi or todo_ji:
+                result += "【宜忌】\n"
+                if todo_yi:
+                    result += f"宜：{todo_yi}\n"
+                if todo_ji:
+                    result += f"忌：{todo_ji}\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/entertainment_agent/tools/minecraft_skin/config.json

@@ -0,0 +1,43 @@
+{
+    "type": "function",
+    "function": {
+        "name": "minecraft_skin",
+        "description": "获取 Minecraft 玩家皮肤/头像。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "name": {
+                    "type": "string",
+                    "description": "正版 ID"
+                },
+                "type": {
+                    "type": "string",
+                    "description": "渲染部位 (头像, 头颅, 身体, 皮肤, 披风)",
+                    "enum": ["头像", "头颅", "身体", "皮肤", "披风"]
+                },
+                "overlay": {
+                    "type": "boolean",
+                    "description": "是否渲染多层皮肤"
+                },
+                "size": {
+                    "type": "integer",
+                    "description": "头像大小 (1-512)"
+                },
+                "scale": {
+                    "type": "integer",
+                    "description": "体型大小 (1-10)"
+                },
+                "target_id": {
+                    "type": "integer",
+                    "description": "发送目标的 ID (群号或 QQ 号)"
+                },
+                "message_type": {
+                    "type": "string",
+                    "description": "消息类型 (group 或 private)",
+                    "enum": ["group", "private"]
+                }
+            },
+            "required": ["name", "target_id", "message_type"]
+        }
+    }
+}

Undefined/skills/agents/entertainment_agent/tools/minecraft_skin/handler.py

@@ -0,0 +1,55 @@
+from typing import Any, Dict
+import httpx
+import logging
+from pathlib import Path
+import uuid
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    name = args.get("name")
+    render_type = args.get("type", "头像")
+    overlay = args.get("overlay", True)
+    size = args.get("size", 160)
+    scale = args.get("scale", 6)
+    target_id = args.get("target_id")
+    message_type = args.get("message_type")
+
+    url = "https://api.xingzhige.com/API/get_Minecraft_skins/"
+    params = {
+        "name": name,
+        "type": render_type,
+        "overlay": str(overlay).lower(),
+        "size": size,
+        "scale": scale,
+    }
+
+    try:
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.get(url, params=params)
+
+            # 检查内容类型
+            content_type = response.headers.get("content-type", "")
+
+            if "application/json" in content_type:
+                return f"获取失败: {response.text}"
+
+            # 假设是图片
+            filename = f"mc_skin_{uuid.uuid4().hex[:8]}.png"
+            filepath = Path.cwd() / "img" / filename
+            filepath.parent.mkdir(exist_ok=True)
+
+            with open(filepath, "wb") as f:
+                f.write(response.content)
+
+            send_image_callback = context.get("send_image_callback")
+            if send_image_callback:
+                await send_image_callback(target_id, message_type, str(filepath))
+                return f"Minecraft 皮肤/头像已发送给 {message_type} {target_id}"
+            else:
+                return "发送图片回调未设置，图片已保存但无法发送。"
+
+    except Exception as e:
+        logger.exception(f"Minecraft 皮肤获取失败: {e}")
+        return f"Minecraft 皮肤获取失败: {e}"

Undefined/skills/agents/entertainment_agent/tools/novel_search/config.json

@@ -0,0 +1,25 @@
+{
+    "type": "function",
+    "function": {
+        "name": "novel_search",
+        "description": "小说搜索与阅读。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "name": {
+                    "type": "string",
+                    "description": "小说名 (搜索用)"
+                },
+                "href": {
+                    "type": "string",
+                    "description": "章节链接/ID (阅读用)"
+                },
+                "num": {
+                    "type": "string",
+                    "description": "页码 (阅读用)"
+                }
+            },
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/entertainment_agent/tools/novel_search/handler.py

@@ -0,0 +1,31 @@
+from typing import Any, Dict
+import httpx
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    name = args.get("name")
+    href = args.get("href")
+    num = args.get("num")
+
+    url = "https://api.jkyai.top/API/fqmfxs.php"
+    params = {}
+    if name:
+        params["name"] = name
+    if href:
+        params["href"] = href
+    if num:
+        params["num"] = num
+
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            response = await client.get(url, params=params)
+
+            # API 返回文本
+            return response.text
+
+    except Exception as e:
+        logger.exception(f"小说工具操作失败: {e}")
+        return f"小说工具操作失败: {e}"

Undefined/skills/agents/entertainment_agent/tools/renjian/config.json

@@ -0,0 +1,12 @@
+{
+    "type": "function",
+    "function": {
+        "name": "renjian",
+        "description": "随机返回“在人间凑数的日子”中的幽默句子，带给用户轻松幽默的日常感悟。",
+        "parameters": {
+            "type": "object",
+            "properties": {},
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/entertainment_agent/tools/renjian/handler.py

@@ -0,0 +1,30 @@
+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:
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            logger.info("获取人间凑数语录")
+            
+            response = await client.get("https://v2.xxapi.cn/api/renjian")
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200:
+                return f"获取语录失败: {data.get('msg')}"
+            
+            quote = data.get("data", "")
+            
+            return f"【在人间凑数的日子】\n{quote}"
+            
+    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/entertainment_agent/tools/wenchang_dijun/config.json

@@ -0,0 +1,12 @@
+{
+    "type": "function",
+    "function": {
+        "name": "wenchang_dijun",
+        "description": "文昌帝君灵签抽签工具，随机抽取文昌帝君签文，适用于求签、运势查询和文化应用。返回签文、诗句、签号和签文图片。",
+        "parameters": {
+            "type": "object",
+            "properties": {},
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/entertainment_agent/tools/wenchang_dijun/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:
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            logger.info("抽取文昌帝君灵签")
+            
+            response = await client.get("https://v2.xxapi.cn/api/wenchangdijunrandom")
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200:
+                return f"抽签失败: {data.get('msg')}"
+            
+            fortune_data = data.get("data", {})
+            title = fortune_data.get("title", "")
+            poem = fortune_data.get("poem", "")
+            content = fortune_data.get("content", "")
+            pic = fortune_data.get("pic", "")
+            fortune_id = fortune_data.get("id", "")
+            
+            result = "【文昌帝君灵签】\n"
+            result += f"签号：{fortune_id}\n"
+            result += f"签名：{title}\n\n"
+            result += f"【签诗】\n{poem}\n\n"
+            result += f"【签文】\n{content}\n"
+            
+            if pic:
+                result += f"\n签文图片：{pic}"
+            
+            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/file_analysis_agent/__init__.py

@@ -0,0 +1 @@
+# File Analysis Agent - 支持解析各种文件格式

Undefined/skills/agents/file_analysis_agent/config.json

@@ -0,0 +1,21 @@
+{
+    "type": "function",
+    "function": {
+        "name": "file_analysis_agent",
+        "description": "文件分析助手，支持解析各种文件格式：文档（PDF、Word、PPT、Excel）、代码、压缩包、图片、音频、视频等。参数为 URL 或 file_id（QQ 内的文件 ID）。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "file_source": {
+                    "type": "string",
+                    "description": "文件源，可以是 URL 或 QQ 的 file_id"
+                },
+                "prompt": {
+                    "type": "string",
+                    "description": "额外的分析需求，例如：'提取所有邮件地址'、'统计代码行数'、'解压并查看内容'"
+                }
+            },
+            "required": ["file_source"]
+        }
+    }
+}

Undefined/skills/agents/file_analysis_agent/handler.py

@@ -0,0 +1,248 @@
+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 专属工具
+
+        遍历 tools 目录，读取 config.json 和 handler.py，
+        将合法的工具注册到内存中。
+        """
+        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
+
+        返回:
+            OpenAI function calling 格式的工具定义列表
+        """
+        return self._tools_schema
+
+    async def execute_tool(
+        self, tool_name: str, args: dict[str, Any], context: dict[str, Any]
+    ) -> str:
+        """执行工具
+
+        参数:
+            tool_name: 工具名称
+            args: 工具参数
+            context: 执行上下文
+
+        返回:
+            工具执行结果字符串
+        """
+        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:
+    """执行 file_analysis_agent"""
+    file_source: str = args.get("file_source", "")
+    user_prompt: str = args.get("prompt", "")
+
+    if not file_source:
+        return "请提供文件 URL 或 file_id"
+
+    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()
+
+    if user_prompt:
+        user_content = f"文件源：{file_source}\n\n用户需求：{user_prompt}"
+    else:
+        user_content = f"请分析这个文件：{file_source}"
+
+    messages: list[dict[str, Any]] = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": user_content},
+    ]
+
+    max_iterations: int = 30
+    iteration: int = 0
+
+    while iteration < max_iterations:
+        iteration += 1
+
+        try:
+            # 1. 调用 AI 模型获取决策 (Action)
+            #    模型根据当前的 message history (包含用户需求、之前的 Action 和 Observation)
+            #    决定是调用工具 (Action) 还是输出最终回复 (Final Answer)
+            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", [])
+
+            # 兼容性处理：如果 content 不为空但有 tool_calls，清空 content 以避免模型自言自语
+            if content.strip() and tool_calls:
+                content = ""
+
+            # 2. 如果模型没有调用工具，说明任务已完成或需要进一步交互
+            if not tool_calls:
+                return content
+
+            # 将模型的决策添加到历史记录
+            messages.append(
+                {"role": "assistant", "content": content, "tool_calls": tool_calls}
+            )
+
+            # 3. 执行模型选择的工具 (Observation)
+            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 "达到最大迭代次数"