Undefined-bot 2.1.0__py3-none-any.whl

Undefined/skills/agents/__init__.py

@@ -0,0 +1,116 @@
+"""
+Agent Registry - Agent 自动发现和注册系统
+"""
+
+import json
+import logging
+from pathlib import Path
+from typing import Any, Dict, List
+
+logger = logging.getLogger(__name__)
+
+
+class AgentRegistry:
+    """Agent 注册表，自动发现和加载 agents"""
+
+    def __init__(self, agents_dir: str | Path | None = None):
+        if agents_dir is None:
+            self.agents_dir = Path(__file__).parent
+        else:
+            self.agents_dir = Path(agents_dir)
+
+        self._agents_schema: List[Dict[str, Any]] = []
+        self._agents_handlers: Dict[str, Any] = {}
+        self.load_agents()
+
+    def load_agents(self) -> None:
+        """自动发现和加载 agents"""
+        self._agents_schema = []
+        self._agents_handlers = {}
+
+        if not self.agents_dir.exists():
+            logger.warning(f"Agent 目录不存在: {self.agents_dir}")
+            return
+
+        for item in self.agents_dir.iterdir():
+            if item.is_dir() and not item.name.startswith("_"):
+                self._load_agent_from_dir(item)
+
+        agent_names = list(self._agents_handlers.keys())
+        logger.info(
+            f"成功加载了 {len(self._agents_schema)} 个 Agent: {', '.join(agent_names)}"
+        )
+
+    def _load_agent_from_dir(self, agent_dir: Path) -> None:
+        """从目录加载单个 agent"""
+        config_path = agent_dir / "config.json"
+        handler_path = agent_dir / "handler.py"
+
+        if not config_path.exists() or not handler_path.exists():
+            logger.debug(
+                f"[Agent加载] 目录 {agent_dir} 缺少 config.json 或 handler.py，跳过"
+            )
+            return
+
+        import importlib.util
+
+        try:
+            with open(config_path, "r", encoding="utf-8") as f:
+                config = json.load(f)
+
+            if "function" not in config or "name" not in config.get("function", {}):
+                logger.error(
+                    f"[Agent错误] Agent 配置无效 {agent_dir}: 缺少 function.name"
+                )
+                return
+
+            agent_name = config["function"]["name"]
+            logger.debug(f"[Agent加载] 正在从 {agent_dir} 加载 Agent: {agent_name}")
+
+            spec = importlib.util.spec_from_file_location(
+                f"agents.{agent_name}", handler_path
+            )
+            if spec is None or spec.loader is None:
+                logger.error(f"从 {handler_path} 加载 Agent 处理器 spec 失败")
+                return
+
+            module = importlib.util.module_from_spec(spec)
+            spec.loader.exec_module(module)
+
+            if not hasattr(module, "execute"):
+                logger.error(f"Agent {agent_dir} 的处理器缺少 'execute' 函数")
+                return
+
+            self._agents_schema.append(config)
+            self._agents_handlers[agent_name] = module.execute
+
+            logger.debug(f"已加载 Agent: {agent_name}")
+
+        except Exception as e:
+            logger.error(f"从 {agent_dir} 加载 Agent 失败: {e}")
+
+    def get_agents_schema(self) -> List[Dict[str, Any]]:
+        """获取所有 agent 的 schema 定义（用于 OpenAI function calling）"""
+        return self._agents_schema
+
+    async def execute_agent(
+        self, agent_name: str, args: Dict[str, Any], context: Dict[str, Any]
+    ) -> str:
+        """执行 agent"""
+        handler = self._agents_handlers.get(agent_name)
+        if not handler:
+            return f"未找到 Agent: {agent_name}"
+
+        try:
+            import asyncio
+
+            start_time = asyncio.get_event_loop().time()
+            if hasattr(handler, "__call__"):
+                result = await handler(args, context)
+                duration = asyncio.get_event_loop().time() - start_time
+                logger.info(f"[Agent执行] {agent_name} 执行成功, 耗时={duration:.4f}s")
+                return str(result)
+            return f"Agent 处理器无效: {agent_name}"
+        except Exception as e:
+            logger.exception(f"[Agent异常] 执行 Agent {agent_name} 时出错")
+            return f"执行 Agent {agent_name} 时出错: {str(e)}"

Undefined/skills/agents/entertainment_agent/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "entertainment_agent",
+        "description": "娱乐助手，提供 AI 绘画、学习答疑、星座运势、Minecraft 皮肤、小说阅读等娱乐功能。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "用户的娱乐需求，例如：'画一只可爱的猫咪'、'白羊座今日运势'、'搜索斗破苍穹'"
+                }
+            },
+            "required": ["prompt"]
+        }
+    }
+}

Undefined/skills/agents/entertainment_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:
+    """执行 entertainment_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/entertainment_agent/intro.md

@@ -0,0 +1,25 @@
+# 娱乐助手
+
+## 能力
+- **AI 绘画**：根据描述生成图片
+- **AI 学习助手**：解答学习问题，支持深度思考模式
+- **星座运势**：查询十二星座的今日运势
+- **文昌帝君**：调用文昌帝君工具查询升学运势
+- **Minecraft 皮肤**：生成 Minecraft 玩家皮肤图片
+- **小说搜索与阅读**：搜索网络小说并阅读章节内容
+- **人间小句**：提供“在人间凑数的日子”中的幽默句子，带给用户轻松幽默的日常感悟
+
+## 适用场景
+- 用户想要生成图片时
+- 用户需要解答题目或学习辅导时
+- 用户查询星座运势时
+- 用户需要 Minecraft 皮肤时
+- 用户想看小说时
+- 用户查询人间评分时
+
+## 参数说明
+- `prompt` (AI 绘画): 图片描述词，越详细越好
+- `question` (学习助手): 具体的学习问题
+- `astrology_name` (星座): 星座名称（如：白羊座）
+- `username` (Minecraft): 玩家名称
+- `novel_name` (小说): 小说名称

Undefined/skills/agents/entertainment_agent/prompt.md

@@ -0,0 +1,20 @@
+你是一个娱乐助手，专门帮助用户获取各种娱乐内容和功能。
+
+你的任务是根据用户需求，选择合适的工具来完成任务：
+
+1. **AI 绘图** → ai_draw_one
+2. **学习答疑** → ai_study_helper
+3. **星座运势** → horoscope
+4. **Minecraft 皮肤** → minecraft_skin
+5. **小说阅读** → novel_search
+6. **人间小句** → renjian
+7. **文昌帝君** → wenchang_dijun
+
+注意事项：
+- 对于 AI 生成的内容，可以适当调整参数获得更好效果
+- 小说阅读支持分页
+- 保持回答轻松友好的风格
+
+如果用户需求不明确，请先询问用户澄清问题。
+
+如果问题涉及时间，立刻调用时间工具获取。

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

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

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

@@ -0,0 +1,34 @@
+{
+    "type": "function",
+    "function": {
+        "name": "ai_draw_one",
+        "description": "AI 绘图工具 (DrawOne)。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "绘图提示词"
+                },
+                "model": {
+                    "type": "string",
+                    "description": "绘图模型 (例如: anylora, anything-v5, etc. 可选)"
+                },
+                "size": {
+                    "type": "string",
+                    "description": "比例 (例如 2:3)"
+                },
+                "target_id": {
+                    "type": "integer",
+                    "description": "发送目标的 ID"
+                },
+                "message_type": {
+                    "type": "string",
+                    "description": "消息类型 (group 或 private)",
+                    "enum": ["group", "private"]
+                }
+            },
+            "required": ["prompt", "target_id", "message_type"]
+        }
+    }
+}

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

@@ -0,0 +1,62 @@
+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:
+    prompt = args.get("prompt")
+    model = args.get("model", "anything-v5")  # 默认模型猜测
+    size = args.get("size", "1:1")
+    target_id = args.get("target_id")
+    message_type = args.get("message_type")
+
+    url = "https://api.xingzhige.com/API/DrawOne/"
+    params = {"prompt": prompt, "model": model, "size": size}
+
+    try:
+        async with httpx.AsyncClient(timeout=60.0) as client:  # AI 绘画可能需要一些时间
+            response = await client.get(url, params=params)
+
+            try:
+                data = response.json()
+            except Exception:
+                return f"API 返回错误 (非JSON): {response.text[:100]}"
+
+            # 解析响应
+            # 文档说 "返回: Json"。它可能包含 "url" 或 "image" 字段。
+            # 我将假设是 'url' 或类似字段。
+            # 如果找不到 URL，我将返回 JSON 给用户以进行调试。
+
+            image_url = data.get("url") or data.get("image") or data.get("img")
+
+            if not image_url and "data" in data and isinstance(data["data"], str):
+                image_url = data["data"]
+
+            if not image_url:
+                return f"未找到图片链接: {data}"
+
+            # 下载图片
+            img_response = await client.get(image_url)
+            img_response.raise_for_status()
+
+            filename = f"ai_draw_{uuid.uuid4().hex[:8]}.jpg"
+            filepath = Path.cwd() / "img" / filename
+            filepath.parent.mkdir(exist_ok=True)
+
+            with open(filepath, "wb") as f:
+                f.write(img_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"AI 绘图已发送给 {message_type} {target_id}"
+            else:
+                return "发送图片回调未设置"
+
+    except Exception as e:
+        logger.exception(f"AI 绘图失败: {e}")
+        return f"AI 绘图失败: {e}"

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

@@ -0,0 +1,22 @@
+{
+    "type": "function",
+    "function": {
+        "name": "ai_study_helper",
+        "description": "AI 学习助手/解题工具。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "question": {
+                    "type": "string",
+                    "description": "问题内容"
+                },
+                "content": {
+                    "type": "string",
+                    "description": "是否开启深度思考 (yes/no)",
+                    "enum": ["yes", "no"]
+                }
+            },
+            "required": ["question"]
+        }
+    }
+}

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

@@ -0,0 +1,35 @@
+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:
+    question = args.get("question")
+    content = args.get("content", "yes")
+
+    url = "https://api.jkyai.top/API/wnjtzs.php"
+
+    try:
+        async with httpx.AsyncClient(timeout=60.0) as client:
+            response = await client.get(
+                url, params={"question": question, "content": content, "type": "json"}
+            )
+            response.raise_for_status()
+            data = response.json()
+
+            # 格式化
+            status = data.get("status")
+            if status != "success":
+                return f"AI 响应失败: {status}"
+
+            q = data.get("question", "")
+            ans = data.get("answer", "")
+            model = data.get("model", "")
+
+            return f"🤖 AI 解答 ({model}):\n❓ 问题: {q}\n💡 答案: {ans}"
+
+    except Exception as e:
+        logger.exception(f"AI 助手请求失败: {e}")
+        return f"AI 助手请求失败: {e}"

Undefined/skills/agents/entertainment_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/entertainment_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/entertainment_agent/tools/horoscope/config.json

@@ -0,0 +1,24 @@
+{
+    "type": "function",
+    "function": {
+        "name": "horoscope",
+        "description": "获取指定星座的运势信息，支持今日、本周、本月和本年的运势查询。返回综合运势、健康、爱情、财运、工作等详细运势信息，包括运势评分、运势描述、幸运色、幸运数字等。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "constellation": {
+                    "type": "string",
+                    "description": "星座名称（中文或英文）",
+                    "enum": ["白羊座", "金牛座", "双子座", "巨蟹座", "狮子座", "处女座", "天秤座", "天蝎座", "射手座", "摩羯座", "水瓶座", "双鱼座", "aries", "taurus", "gemini", "cancer", "leo", "virgo", "libra", "scorpio", "sagittarius", "capricorn", "aquarius", "pisces"]
+                },
+                "time_type": {
+                    "type": "string",
+                    "description": "运势时间类型",
+                    "enum": ["today", "week", "month", "year", "今日", "本周", "本月", "本年"],
+                    "default": "today"
+                }
+            },
+            "required": ["constellation"]
+        }
+    }
+}