Undefined-bot 2.1.0__py3-none-any.whl

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

@@ -0,0 +1,17 @@
+# 社交媒体助手
+
+## 能力
+- **Bilibili 服务**：搜索视频、番剧、UP主，查询用户详细信息
+- **音乐服务**：全网搜索音乐，获取歌曲详情及歌词
+- **视频推荐**：随机获取视频推荐内容
+
+## 适用场景
+- 用户想找 B 站视频或用户信息时
+- 用户想听歌、找歌词或了解歌曲信息时
+- 用户无聊想要随机视频推荐时
+
+## 参数说明
+- `keyword` (搜索): 搜索关键词
+- `buid` (B站): B站用户 UID
+- `music_name` (音乐): 歌曲名称
+- `index` (通用): 结果序号选择

Undefined/skills/agents/social_agent/prompt.md

@@ -0,0 +1,19 @@
+你是一个社交媒体助手，专门帮助用户搜索和获取社交媒体内容。
+
+你的任务是根据用户需求，选择合适的工具来完成任务：
+
+1. **搜索 B 站视频/番剧/用户** → bilibili_search
+2. **获取 B 站用户详情** → bilibili_user_info
+3. **搜索音乐** → music_global_search
+4. **获取音乐详情** → music_info_get
+5. **获取歌词** → music_lyrics
+6. **获取随机视频推荐** → video_random_recommend
+
+注意事项：
+- 搜索结果可能较多，建议用户指定更精确的需求
+- 音乐搜索支持按序号选择具体歌曲
+- 保持回答简洁，突出关键信息
+
+如果用户需求不明确，请先询问用户澄清问题。
+
+如果问题涉及时间，立刻调用时间工具获取。

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

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

Undefined/skills/agents/social_agent/tools/bilibili_search/config.json

@@ -0,0 +1,21 @@
+{
+    "type": "function",
+    "function": {
+        "name": "bilibili_search",
+        "description": "搜索 Bilibili 视频、番剧、用户等。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "msg": {
+                    "type": "string",
+                    "description": "搜索内容"
+                },
+                "n": {
+                    "type": "integer",
+                    "description": "返回数据数量 (默认5)"
+                }
+            },
+            "required": ["msg"]
+        }
+    }
+}

Undefined/skills/agents/social_agent/tools/bilibili_search/handler.py

@@ -0,0 +1,68 @@
+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:
+    msg = args.get("msg")
+    n = args.get("n", 5)
+
+    url = "https://api.xingzhige.com/API/b_search/"
+
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            response = await client.get(url, params={"msg": msg, "n": n})
+            response.raise_for_status()
+            data = response.json()
+
+            # API 返回一个以 0, 1, 2... 为键的字典来表示列表？
+            # 或者是一个字典列表？
+            # 根据 'n'，它可能返回一个列表或字典索引。
+            # 让我们同时处理这两种情况。
+
+            results = []
+            if isinstance(data, list):
+                results = data
+            elif isinstance(data, dict):
+                # 检查是否为索引键 "0", "1" 等。
+                if "0" in data:
+                    for i in range(len(data)):
+                        key = str(i)
+                        if key in data:
+                            results.append(data[key])
+                elif "code" in data and data["code"] != 200:
+                    return f"搜索失败: {data.get('msg', '未知错误')}"
+                else:
+                    # 单个结果或非预期格式
+                    results = [data]
+
+            output = f"🔍 B站搜索 '{msg}' 结果:\n"
+            for item in results:
+                title = item.get("title")
+                linktype = item.get("linktype")
+                name = item.get("name")
+                bvid = item.get("bvid")
+
+                item_str = ""
+                if linktype and title:
+                    item_str += f"- [{linktype}] {title}\n"
+                elif title:
+                    item_str += f"- {title}\n"
+
+                if name:
+                    item_str += f"  UP主: {name}\n"
+
+                if bvid:
+                    url_link = f"https://www.bilibili.com/video/{bvid}"
+                    item_str += f"  链接: {url_link}\n"
+
+                if item_str:
+                    output += item_str + "\n"
+
+            return output
+
+    except Exception as e:
+        logger.exception(f"B站搜索失败: {e}")
+        return f"B站搜索失败: {e}"

Undefined/skills/agents/social_agent/tools/bilibili_user_info/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "bilibili_user_info",
+        "description": "查询 Bilibili 用户信息。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "mid": {
+                    "type": "integer",
+                    "description": "用户 UID (mid)"
+                }
+            },
+            "required": ["mid"]
+        }
+    }
+}

Undefined/skills/agents/social_agent/tools/bilibili_user_info/handler.py

@@ -0,0 +1,68 @@
+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:
+    mid = args.get("mid")
+    url = "https://api.xingzhige.com/API/b_personal/"
+
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            response = await client.get(url, params={"mid": mid})
+            response.raise_for_status()
+            data = response.json()
+
+            if isinstance(data, dict):
+                if "code" in data and data["code"] != 0 and data["code"] != 200:
+                    # 某些 API 返回 code 0 表示成功，其他返回 200。
+                    # 但是检查是否有错误信息。
+                    if "message" in data:
+                        # 可能是一个错误
+                        pass
+
+                name = data.get("name")
+                level = data.get("level")
+                sex = data.get("sex")
+                desc = data.get("desc")
+                follower = data.get("follower")
+                following = data.get("following")
+                roomid = data.get("roomid")
+                face = data.get("face")
+
+                output_lines = []
+
+                header = "📺 B站用户"
+                if name:
+                    header += f": {name}"
+                if data.get("mid"):
+                    header += f" (UID: {data.get('mid')})"
+                output_lines.append(header)
+
+                if level is not None:
+                    output_lines.append(f"🆙 等级: Lv{level}")
+
+                if sex:
+                    output_lines.append(f"⚧ 性别: {sex}")
+
+                if desc:
+                    output_lines.append(f"📝 简介: {desc}")
+
+                if follower is not None and following is not None:
+                    output_lines.append(f"👥 粉丝: {follower} | 关注: {following}")
+
+                if roomid:
+                    output_lines.append(f"🎥 直播间: {roomid}")
+
+                if face:
+                    output_lines.append(f"🖼️ 头像: {face}")
+
+                return "\n".join(output_lines)
+
+            return str(data)
+
+    except Exception as e:
+        logger.exception(f"B站用户查询失败: {e}")
+        return f"B站用户查询失败: {e}"

Undefined/skills/agents/social_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/social_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/social_agent/tools/music_global_search/config.json

@@ -0,0 +1,21 @@
+{
+    "type": "function",
+    "function": {
+        "name": "music_global_search",
+        "description": "全网音乐搜索。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "msg": {
+                    "type": "string",
+                    "description": "歌曲名称"
+                },
+                "n": {
+                    "type": "integer",
+                    "description": "序号 (可选)"
+                }
+            },
+            "required": ["msg"]
+        }
+    }
+}

Undefined/skills/agents/social_agent/tools/music_global_search/handler.py

@@ -0,0 +1,47 @@
+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:
+    msg = args.get("msg")
+    n = args.get("n", 1)
+    
+    url = "https://api.jkyai.top/API/qsyyjs.php"
+    
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            response = await client.get(url, params={"msg": msg, "n": n, "type": "json"})
+            response.raise_for_status()
+            data = response.json()
+            
+            if isinstance(data, dict):
+                output_lines = []
+                
+                title = data.get("title")
+                if title:
+                    output_lines.append(f"🎵 音乐搜索: {title}")
+                
+                singer = data.get("singer")
+                if singer:
+                    output_lines.append(f"👤 歌手: {singer}")
+                
+                music_url = data.get("music")
+                if music_url:
+                    output_lines.append(f"🔗 链接: {music_url}")
+                
+                cover = data.get("cover")
+                if cover:
+                    output_lines.append(f"🖼️ 封面: {cover}")
+                    
+                if output_lines:
+                    return "\n".join(output_lines)
+                else:
+                    return "未找到相关音乐信息。"
+            
+            return str(data)
+
+    except Exception as e:
+        logger.exception(f"音乐搜索失败: {e}")
+        return f"音乐搜索失败: {e}"

Undefined/skills/agents/social_agent/tools/music_info_get/config.json

@@ -0,0 +1,22 @@
+{
+    "type": "function",
+    "function": {
+        "name": "music_info_get",
+        "description": "获取音乐详细信息 (通过 ID)。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "id": {
+                    "type": "string",
+                    "description": "歌曲 ID"
+                },
+                "type": {
+                    "type": "string",
+                    "description": "音乐平台 (wy, qq, kw, mg, qi)",
+                    "enum": ["wy", "qq", "kw", "mg", "qi"]
+                }
+            },
+            "required": ["id", "type"]
+        }
+    }
+}

Undefined/skills/agents/social_agent/tools/music_info_get/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:
+    song_id = args.get("id")
+    platform = args.get("type")
+
+    url = "https://api.jkyai.top/API/yyjhss.php"
+
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            response = await client.get(url, params={"id": song_id, "type": platform})
+            response.raise_for_status()
+            data = response.json()
+
+            # 数据结构: code, msg, data{name, artist, url, ...}
+            if data.get("code") == 1:
+                info = data.get("data", {})
+                return (
+                    f"🎵 歌曲信息: {info.get('name')}\n"
+                    f"👤 歌手: {info.get('artist')}\n"
+                    f"💿 专辑: {info.get('album')}\n"
+                    f"🔗 链接: {info.get('url')}\n"
+                    f"🖼️ 图片: {info.get('pic')}"
+                )
+            else:
+                return f"获取失败: {data.get('msg')}"
+
+    except Exception as e:
+        logger.exception(f"获取歌曲详情失败: {e}")
+        return f"获取歌曲详情失败: {e}"

Undefined/skills/agents/social_agent/tools/music_lyrics/config.json

@@ -0,0 +1,22 @@
+{
+    "type": "function",
+    "function": {
+        "name": "music_lyrics",
+        "description": "获取歌曲歌词。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "id": {
+                    "type": "string",
+                    "description": "歌曲 ID"
+                },
+                "msg": {
+                    "type": "string",
+                    "description": "音乐平台 (qq, wy, kw, qi)",
+                    "enum": ["qq", "wy", "kw", "qi"]
+                }
+            },
+            "required": ["id", "msg"]
+        }
+    }
+}

Undefined/skills/agents/social_agent/tools/music_lyrics/handler.py

@@ -0,0 +1,26 @@
+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:
+    song_id = args.get("id")
+    platform = args.get("msg")
+
+    url = "https://api.jkyai.top/API/jhlrcgc.php"
+
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            response = await client.get(
+                url, params={"id": song_id, "msg": platform, "type": "text"}
+            )
+            # API 文档说明 type 是可选的，默认为 text。
+            # 如果是 text，它可能直接返回歌词。
+
+            return f"🎵 歌词内容:\n{response.text}"
+
+    except Exception as e:
+        logger.exception(f"获取歌词失败: {e}")
+        return f"获取歌词失败: {e}"

Undefined/skills/agents/social_agent/tools/video_random_recommend/config.json

@@ -0,0 +1,21 @@
+{
+    "type": "function",
+    "function": {
+        "name": "video_random_recommend",
+        "description": "获取随机视频推荐（返回视频链接）。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "target_id": {
+                    "type": "integer",
+                    "description": "目标 ID (用于日志记录)"
+                },
+                 "message_type": {
+                    "type": "string",
+                    "description": "消息类型 (group 或 private)"
+                }
+            },
+            "required": []
+        }
+    }
+}