Undefined-bot 2.1.0__py3-none-any.whl

Undefined/skills/agents/scheduler_agent/handler.py

@@ -0,0 +1,218 @@
+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:
+    """执行 scheduler_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/scheduler_agent/intro.md

@@ -0,0 +1,17 @@
+# 定时任务助手
+
+## 能力
+- **创建任务**：使用 Crontab 表达式创建定时任务
+- **管理任务**：删除或修改现有的定时任务
+- **查看任务**：列出当前所有活跃的定时任务
+
+## 适用场景
+- 当用户需要定期执行某项操作时
+- 当用户询问“每天早上8点提醒我...”或类似请求时
+- 管理已有的定时任务
+
+## 参数说明
+- `cron_expression`: Crontab 时间表达式（如 `0 8 * * *` 表示每天早上8点）
+- `task_content`: 任务的具体内容描述
+- `task_id`: 任务的唯一标识符
+- `max_executions`: 最大执行次数（可选，默认无限）

Undefined/skills/agents/scheduler_agent/prompt.md

@@ -0,0 +1,67 @@
+你是一个定时任务管理助手，专门帮助用户创建、管理定时任务。
+
+## 你可以做的
+
+1. **创建定时任务** → create_schedule_task
+2. **删除定时任务** → delete_schedule_task
+3. **修改定时任务** → update_schedule_task
+4. **查看所有任务** → list_schedule_tasks
+
+## crontab 表达式说明
+
+crontab 格式：分 时 日 月 周
+
+| 字段 | 含义 | 范围 |
+|------|------|------|
+| 第1个 | 分钟 | 0-59 |
+| 第2个 | 小时 | 0-23 |
+| 第3个 | 日期 | 1-31 |
+| 第4个 | 月份 | 1-12 |
+| 第5个 | 星期 | 0-6 (0=周日) |
+
+### 常用示例
+
+- `* * * * *` - 每分钟执行
+- `30 * * * *` - 每小时30分执行
+- `0 8 * * *` - 每天早上8点执行
+- `0 8 * * 1` - 每周一早上8点执行
+- `0 8 1 * *` - 每月1号早上8点执行
+- `*/5 * * * *` - 每5分钟执行
+- `30 8 * * 1-5` - 工作日早上8点30分执行
+
+### 特殊字符
+
+- `*` - 任意值
+- `,` - 多个值（如 `1,3,5`）
+- `-` - 范围（如 `1-5`）
+- `/` - 间隔（如 `*/15` 表示每15分钟）
+
+## 执行次数限制
+
+使用 `max_executions` 参数可以限制任务执行次数：
+
+- 不设置 → 无限执行
+- `1` → 执行一次后自动删除
+- `N` → 执行 N 次后自动删除
+
+## 任务名称
+
+建议给任务起一个有意义的名称，方便识别和管理。
+
+## 使用流程
+
+1. 用户提出需求
+2. 你选择合适的工具
+3. 用户可以查看当前任务列表
+4. 用户可以删除或修改任务
+
+## 注意事项
+
+- 任务创建后会自动开始计时
+- 修改任务不会重置执行计数
+- 删除任务会立即停止执行
+- 保持回答简洁明了
+
+如果用户需求不明确，请先询问用户澄清问题。
+
+如果问题涉及时间，立刻调用时间工具获取。

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

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

Undefined/skills/agents/scheduler_agent/tools/create_schedule_task/config.json

@@ -0,0 +1,37 @@
+{
+    "type": "function",
+    "function": {
+        "name": "create_schedule_task",
+        "description": "创建一个定时执行的任务。支持 crontab 语法。可以指定要调用的工具及其参数、执行次数限制。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "task_name": {
+                    "type": "string",
+                    "description": "任务名称（用于标识，建议使用有意义的名称）"
+                },
+                "cron_expression": {
+                    "type": "string",
+                    "description": "crontab 表达式 (分 时 日 月 周)。例如 '* * * * *' 表示每分钟执行，'30 8 * * *' 表示每天 8:30 执行。"
+                },
+                "tool_name": {
+                    "type": "string",
+                    "description": "要定时调用的工具名称，例如 'get_current_time', 'send_message' 等。"
+                },
+                "tool_args": {
+                    "type": "object",
+                    "description": "调用工具时传递的参数对象 (JSON)。"
+                },
+                "max_executions": {
+                    "type": "integer",
+                    "description": "最大执行次数（可选）。设置为 1 表示执行一次后自动删除，设置为 N 表示执行 N 次后自动删除，不设置则无限执行。"
+                }
+            },
+            "required": [
+                "cron_expression",
+                "tool_name",
+                "tool_args"
+            ]
+        }
+    }
+}

Undefined/skills/agents/scheduler_agent/tools/create_schedule_task/handler.py

@@ -0,0 +1,68 @@
+from typing import Any, Dict
+import uuid
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    """
+    执行 create_schedule_task 工具
+    创建一个定时执行的任务
+    """
+    task_name = args.get("task_name")
+    cron_expression = args.get("cron_expression")
+    tool_name = args.get("tool_name")
+    tool_args = args.get("tool_args", {})
+    max_executions = args.get("max_executions")
+
+    if max_executions is not None:
+        try:
+            max_executions = int(max_executions)
+            if max_executions < 1:
+                return "max_executions 必须大于 0"
+        except (ValueError, TypeError):
+            return "max_executions 必须是有效的整数"
+
+    task_id = f"task_{uuid.uuid4().hex[:8]}"
+    if task_name:
+        task_id = f"task_{task_name.lower().replace(' ', '_')}_{uuid.uuid4().hex[:4]}"
+
+    target_type = None
+    target_id = None
+
+    scheduler = context.get("scheduler")
+    if not scheduler:
+        return "调度器未在上下文中提供"
+
+    ai_client = context.get("ai_client")
+    if target_id is None and ai_client:
+        if ai_client.current_group_id:
+            target_id = ai_client.current_group_id
+            if not target_type:
+                target_type = "group"
+        elif ai_client.current_user_id:
+            target_id = ai_client.current_user_id
+            if not target_type:
+                target_type = "private"
+
+    if not target_type:
+        target_type = "group"
+
+    success = scheduler.add_task(
+        task_id=task_id,
+        tool_name=tool_name,
+        tool_args=tool_args,
+        cron_expression=cron_expression,
+        target_id=target_id,
+        target_type=target_type,
+        task_name=task_name,
+        max_executions=max_executions,
+    )
+
+    if success:
+        name_info = f" '{task_name}'" if task_name else ""
+        max_info = f"，最多执行 {max_executions} 次" if max_executions else ""
+        return f"定时任务{name_info}已成功添加 (ID: {task_id})。\n将在 '{cron_expression}' 时间执行工具 '{tool_name}'{max_info}。"
+    else:
+        return "添加定时任务失败。请检查 crontab 表达式是否正确。"

Undefined/skills/agents/scheduler_agent/tools/delete_schedule_task/config.json

@@ -0,0 +1,19 @@
+{
+    "type": "function",
+    "function": {
+        "name": "delete_schedule_task",
+        "description": "删除指定的定时任务。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "task_id": {
+                    "type": "string",
+                    "description": "要删除的任务 ID。可通过 list_schedule_tasks 查看所有任务及其 ID。"
+                }
+            },
+            "required": [
+                "task_id"
+            ]
+        }
+    }
+}

Undefined/skills/agents/scheduler_agent/tools/delete_schedule_task/handler.py

@@ -0,0 +1,26 @@
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    """
+    执行 delete_schedule_task 工具
+    删除指定的定时任务
+    """
+    task_id = args.get("task_id")
+
+    if not task_id:
+        "请提供要删除的任务 ID"
+
+    scheduler = context.get("scheduler")
+    if not scheduler:
+        return "调度器未在上下文中提供"
+
+    success = scheduler.remove_task(task_id)
+
+    if success:
+        return f"定时任务 '{task_id}' 已成功删除。"
+    else:
+        return "删除定时任务失败。可能任务不存在。"

Undefined/skills/agents/scheduler_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/scheduler_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/scheduler_agent/tools/list_schedule_tasks/config.json

@@ -0,0 +1,11 @@
+{
+    "type": "function",
+    "function": {
+        "name": "list_schedule_tasks",
+        "description": "列出所有已创建的定时任务，包括任务 ID、名称、工具、表达式、执行次数等信息。",
+        "parameters": {
+            "type": "object",
+            "properties": {}
+        }
+    }
+}

Undefined/skills/agents/scheduler_agent/tools/list_schedule_tasks/handler.py

@@ -0,0 +1,47 @@
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    """
+    执行 list_schedule_tasks 工具
+    列出所有定时任务
+    """
+    scheduler = context.get("scheduler")
+    if not scheduler:
+        return "调度器未在上下文中提供"
+
+    tasks = scheduler.list_tasks()
+
+    if not tasks:
+        return "当前没有定时任务"
+
+    lines = ["定时任务列表：\n"]
+    for task_id, info in tasks.items():
+        task_name = info.get("task_name", "")
+        tool_name = info.get("tool_name", "")
+        cron = info.get("cron", "")
+        tool_args = info.get("tool_args", {})
+        max_exec = info.get("max_executions")
+        current_exec = info.get("current_executions", 0)
+
+        exec_info = ""
+        if max_exec is not None:
+            exec_info = f" ({current_exec}/{max_exec})"
+        else:
+            exec_info = f" ({current_exec}次)"
+
+        name_display = f"【{task_name}】" if task_name else ""
+        args_str = str(tool_args) if tool_args else "{}"
+
+        lines.append(f"- ID: {task_id}")
+        lines.append(f"  名称: {name_display}")
+        lines.append(f"  工具: {tool_name}")
+        lines.append(f"  表达式: {cron}")
+        lines.append(f"  参数: {args_str}")
+        lines.append(f"  已执行: {exec_info}")
+        lines.append("")
+
+    return "\n".join(lines)

Undefined/skills/agents/scheduler_agent/tools/update_schedule_task/config.json

@@ -0,0 +1,39 @@
+{
+    "type": "function",
+    "function": {
+        "name": "update_schedule_task",
+        "description": "修改已存在的定时任务。可修改 crontab 表达式、工具名称、工具参数、任务名称、最大执行次数。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "task_id": {
+                    "type": "string",
+                    "description": "要修改的任务 ID。可通过 list_schedule_tasks 查看所有任务及其 ID。"
+                },
+                "cron_expression": {
+                    "type": "string",
+                    "description": "新的 crontab 表达式 (分 时 日 月 周)。"
+                },
+                "tool_name": {
+                    "type": "string",
+                    "description": "新的工具名称。"
+                },
+                "tool_args": {
+                    "type": "object",
+                    "description": "新的工具参数 (JSON)。"
+                },
+                "task_name": {
+                    "type": "string",
+                    "description": "新的任务名称。"
+                },
+                "max_executions": {
+                    "type": "integer",
+                    "description": "新的最大执行次数（可选）。设置为 1 表示执行一次后自动删除，设置为 N 表示执行 N 次后自动删除，不设置则保持原值。"
+                }
+            },
+            "required": [
+                "task_id"
+            ]
+        }
+    }
+}

Undefined/skills/agents/scheduler_agent/tools/update_schedule_task/handler.py

@@ -0,0 +1,46 @@
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    """
+    执行 update_schedule_task 工具
+    修改已存在的定时任务
+    """
+    task_id = args.get("task_id")
+    cron_expression = args.get("cron_expression")
+    tool_name = args.get("tool_name")
+    tool_args = args.get("tool_args")
+    task_name = args.get("task_name")
+    max_executions = args.get("max_executions")
+
+    if not task_id:
+        return "请提供要修改的任务 ID"
+
+    if max_executions is not None:
+        try:
+            max_executions = int(max_executions)
+            if max_executions < 1:
+                return "max_executions 必须大于 0"
+        except (ValueError, TypeError):
+            return "max_executions 必须是有效的整数"
+
+    scheduler = context.get("scheduler")
+    if not scheduler:
+        return "调度器未在上下文中提供"
+
+    success = scheduler.update_task(
+        task_id=task_id,
+        cron_expression=cron_expression,
+        tool_name=tool_name,
+        tool_args=tool_args,
+        task_name=task_name,
+        max_executions=max_executions,
+    )
+
+    if success:
+        return f"定时任务 '{task_id}' 已成功修改。"
+    else:
+        return "修改定时任务失败。可能任务不存在。"

Undefined/skills/agents/social_agent/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "social_agent",
+        "description": "社交媒体助手，提供 Bilibili、音乐等社交媒体内容的搜索和查询功能。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "用户的搜索需求，例如：'搜索 Python 教学视频'、'找周杰伦的稻香歌词'"
+                }
+            },
+            "required": ["prompt"]
+        }
+    }
+}