Undefined-bot 2.1.0__py3-none-any.whl

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

@@ -0,0 +1,207 @@
+import os
+import aiohttp
+import logging
+from typing import Dict, Any, cast
+
+logger = logging.getLogger(__name__)
+
+API_KEY = os.getenv("WEATHER_API_KEY", "")
+BASE_URL = "https://api.seniverse.com/v3"
+
+
+async def fetch_data(url: str, params: Dict[str, Any]) -> Dict[str, Any]:
+    if not API_KEY:
+        logger.error("WEATHER_API_KEY 未设置")
+        return {"error": "天气服务未配置API密钥"}
+
+    params["key"] = API_KEY
+    params["language"] = "zh-Hans"
+    params["unit"] = "c"
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, params=params) as response:
+            if response.status != 200:
+                error_text = await response.text()
+                logger.error(f"天气 API 错误: {response.status} - {error_text}")
+                return {"error": f"API请求失败: {response.status}"}
+            json_data = await response.json()
+            return cast(Dict[str, Any], json_data)
+
+
+async def get_weather_now(location: str) -> str:
+    url = f"{BASE_URL}/weather/now.json"
+    data = await fetch_data(url, {"location": location})
+
+    if "error" in data:
+        return str(data["error"])
+
+    if not data.get("results"):
+        return "未找到该城市的天气信息。"
+
+    results = cast(list[dict[str, Any]], data["results"])
+    result = results[0]
+    loc = cast(dict[str, Any], result["location"])
+    now = cast(dict[str, Any], result["now"])
+    update_time = str(result["last_update"]).split("T")[1].split("+")[0]
+
+    msg = [f"【{loc['name']} 天气实况】"]
+
+    if "text" in now:
+        msg.append(f"天气: {now['text']}")
+    if "temperature" in now:
+        msg.append(f"温度: {now['temperature']}°C")
+
+    if "feels_like" in now:
+        msg.append(f"体感: {now['feels_like']}°C")
+    if "humidity" in now:
+        msg.append(f"湿度: {now['humidity']}%")
+    if "wind_direction" in now:
+        wind_dir = str(now["wind_direction"])
+        if "wind_direction_degree" in now:
+            wind_dir += f" ({now['wind_direction_degree']}°)"
+        msg.append(f"风向: {wind_dir}")
+    elif "wind_direction_degree" in now:
+        msg.append(f"风向: {now['wind_direction_degree']}°")
+
+    if "wind_speed" in now:
+        msg.append(f"风速: {now['wind_speed']}km/h")
+    if "wind_scale" in now:
+        msg.append(f"风力等级: {now['wind_scale']}")
+    if "visibility" in now:
+        msg.append(f"能见度: {now['visibility']}km")
+
+    msg.append(f"更新时间: {update_time}")
+
+    return "\n".join(msg)
+
+
+async def get_weather_forecast(location: str) -> str:
+    url = f"{BASE_URL}/weather/daily.json"
+    # start=0 包括今天。days=5 为请求的范围。
+    data = await fetch_data(url, {"location": location, "start": "0", "days": "5"})
+
+    if "error" in data:
+        return str(data["error"])
+
+    if not data.get("results"):
+        return "未找到该城市的天气预报。"
+
+    results = cast(list[dict[str, Any]], data["results"])
+    result = results[0]
+    loc_name = str(result["location"]["name"])
+    daily_data = cast(list[dict[str, Any]], result["daily"])
+
+    msg = [f"【{loc_name} 未来天气预报】"]
+    for day in daily_data:
+        date = day.get("date", "未知日期")
+        text_day = day.get("text_day")
+        text_night = day.get("text_night")
+        high = day.get("high")
+        low = day.get("low")
+        precip = day.get("precip")
+        wind_scale = day.get("wind_scale")
+
+        day_info = [f"{date}:"]
+        if text_day and text_night:
+            weather_str = (
+                text_day if text_day == text_night else f"{text_day}转{text_night}"
+            )
+            day_info.append(str(weather_str))
+        elif text_day:
+            day_info.append(str(text_day))
+
+        if low is not None and high is not None:
+            day_info.append(f"{low}~{high}°C")
+
+        if precip:
+            day_info.append(f"降水概率{precip}%")
+
+        if wind_scale:
+            day_info.append(f"风力{wind_scale}级")
+
+        msg.append(" ".join(day_info))
+
+    return "\n".join(msg)
+
+
+async def get_life_suggestion(location: str) -> str:
+    url = f"{BASE_URL}/life/suggestion.json"
+    data = await fetch_data(url, {"location": location})
+
+    if "error" in data:
+        return str(data["error"])
+
+    if not data.get("results"):
+        return "未找到该城市的生活指数信息。"
+
+    results = cast(list[dict[str, Any]], data["results"])
+    result = results[0]
+    loc_name = str(result["location"]["name"])
+    suggestion = cast(dict[str, Any], result.get("suggestion", {}))
+
+    label_map = {
+        "ac": "空调开启",
+        "air_pollution": "空气扩散",
+        "airing": "晾晒",
+        "allergy": "过敏",
+        "beer": "啤酒",
+        "boating": "划船",
+        "car_washing": "洗车",
+        "chill": "风寒",
+        "comfort": "舒适度",
+        "dating": "约会",
+        "dressing": "穿衣",
+        "fishing": "钓鱼",
+        "flu": "感冒",
+        "hair_dressing": "美发",
+        "kiteflying": "放风筝",
+        "makeup": "化妆",
+        "mood": "心情",
+        "morning_sport": "晨练",
+        "night_life": "夜生活",
+        "road_condition": "路况",
+        "shopping": "购物",
+        "sport": "运动",
+        "sunscreen": "防晒",
+        "traffic": "交通",
+        "travel": "旅游",
+        "umbrella": "雨伞",
+        "uv": "紫外线",
+    }
+
+    msg = [f"【{loc_name} 生活指数】"]
+
+    for key, info_raw in suggestion.items():
+        info = cast(dict[str, Any], info_raw)
+        if not info or not info.get("brief"):
+            continue
+
+        label = label_map.get(key, key)
+        brief = info["brief"]
+        msg.append(f"{label}: {brief}")
+
+    if len(msg) == 1:
+        return f"【{loc_name}】暂无生活指数信息。"
+
+    return "\n".join(msg)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    location = args.get("location")
+    query_type = args.get("query_type", "now")
+
+    if not location:
+        return "请提供城市名称。"
+
+    try:
+        if query_type == "forecast":
+            return await get_weather_forecast(location)
+        elif query_type == "life":
+            return await get_life_suggestion(location)
+        else:
+            # 默认为当前天气
+            return await get_weather_now(location)
+
+    except Exception as e:
+        logger.exception(f"天气查询工具出错: {e}")
+        return f"查询出错: {str(e)}"

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

@@ -0,0 +1,18 @@
+{
+    "type": "function",
+    "function": {
+        "name": "weibohot",
+        "description": "获取微博热搜排行榜，实时返回当前热门话题和趋势。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "limit": {
+                    "type": "integer",
+                    "description": "返回的热搜数量，默认10",
+                    "default": 10
+                }
+            },
+            "required": []
+        }
+    }
+}

Undefined/skills/agents/info_agent/tools/weibohot/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/weibohot")
+            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/whois/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "whois",
+        "description": "查询域名的Whois信息，包括注册商、注册人、注册时间、到期时间、DNS服务器等信息。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "domain": {
+                    "type": "string",
+                    "description": "需要查询的域名，例如：example.com"
+                }
+            },
+            "required": ["domain"]
+        }
+    }
+}

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

@@ -0,0 +1,63 @@
+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:
+    domain = args.get("domain")
+    
+    if not domain:
+        return "❌ 域名不能为空"
+    
+    try:
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            params = {"domain": domain}
+            logger.info(f"查询Whois信息: {domain}")
+            
+            response = await client.get("https://v2.xxapi.cn/api/whois", params=params)
+            response.raise_for_status()
+            data = response.json()
+            
+            if data.get("code") != 200:
+                return f"查询Whois失败: {data.get('msg')}"
+            
+            whois_data = data.get("data", {})
+            result = f"【{domain} Whois信息】\n\n"
+            
+            domain_name = whois_data.get("Domain Name", "")
+            registrar = whois_data.get("Sponsoring Registrar", "")
+            registrar_url = whois_data.get("Registrar URL", "")
+            registrant = whois_data.get("Registrant", "")
+            registrant_email = whois_data.get("Registrant Contact Email", "")
+            registration_time = whois_data.get("Registration Time", "")
+            expiration_time = whois_data.get("Expiration Time", "")
+            dns_servers = whois_data.get("DNS Serve", [])
+            
+            if domain_name:
+                result += f"域名: {domain_name}\n"
+            if registrar:
+                result += f"注册商: {registrar}\n"
+            if registrar_url:
+                result += f"注册商URL: {registrar_url}\n"
+            if registrant:
+                result += f"注册人: {registrant}\n"
+            if registrant_email:
+                result += f"注册人邮箱: {registrant_email}\n"
+            if registration_time:
+                result += f"注册时间: {registration_time}\n"
+            if expiration_time:
+                result += f"到期时间: {expiration_time}\n"
+            if dns_servers:
+                result += f"DNS服务器: {', '.join(dns_servers)}\n"
+            
+            return result
+            
+    except httpx.TimeoutException:
+        return "请求超时，请稍后重试"
+    except httpx.HTTPStatusError as e:
+        logger.error(f"HTTP 错误: {e}")
+        return f"查询Whois失败: {e}"
+    except Exception as e:
+        logger.exception(f"查询Whois失败: {e}")
+        return f"查询Whois失败: {e}"

Undefined/skills/agents/naga_code_analysis_agent/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "naga_code_analysis_agent",
+        "description": "NagaAgent代码分析助手，提供针对NagaAgent项目的文件读取、目录浏览、内容搜索、文件匹配等功能，用于理解和分析代码库。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "分析需求'"
+                }
+            },
+            "required": ["prompt"]
+        }
+    }
+}

Undefined/skills/agents/naga_code_analysis_agent/handler.py

@@ -0,0 +1,222 @@
+from typing import Any, Dict, Callable
+from pathlib import Path
+import importlib.util
+import json
+import asyncio
+import aiofiles
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class AgentToolRegistry:
+    """Agent 内部的工具注册表"""
+
+    def __init__(self, tools_dir: Path) -> None:
+        self.tools_dir: 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:
+    """执行 code_analysis_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}\n\n请选择合适的工具来满足用户需求。",
+        },
+    ]
+
+    max_iterations: int = 20
+    iteration: int = 0
+    conversation_ended: bool = False
+
+    while iteration < max_iterations and not conversation_ended:
+        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/naga_code_analysis_agent/intro.md

@@ -0,0 +1,17 @@
+# NagaAgent代码分析助手
+
+## 能力
+- **文件读取**：读取指定文件的内容
+- **目录列表**：列出指定目录下的结构
+- **内容搜索**：支持正则表达式的文件内容搜索
+- **文件查找**：使用 glob 模式查找匹配文件
+- **项目介绍**：获取 NagaAgent 项目的详细文档
+
+## 适用场景
+- **仅当**用户明确询问有关 "NagaAgent" 自身架构、代码或实现细节时使用
+- 一般代码分析请使用 `file_analysis_agent`
+
+## 参数说明
+- `file_path`: 文件路径（相对工作目录）
+- `dir_path`: 目录路径
+- `pattern`: 搜索的正则模式或 Glob 模式

Undefined/skills/agents/naga_code_analysis_agent/prompt.md

@@ -0,0 +1,19 @@
+你是一个专业的代码分析助手，专门帮助用户理解和分析代码库。
+
+你的任务是根据用户需求，选择合适的工具来完成任务：
+
+1. **读取文件内容** → read_file
+2. **列出目录结构** → list_directory
+3. **搜索文件内容** → search_file_content
+4. **按模式查找文件** → glob
+5. **了解 NagaAgent 项目** → read_naga_intro
+
+注意事项：
+- 文件路径都是相对于工作目录的
+- 搜索支持正则表达式
+- 如果文件内容过长会被截断
+- 保持回答简洁，聚焦于用户需要的信息
+
+如果用户需求不明确，请先询问用户澄清问题。
+
+如果问题涉及时间，立刻调用时间工具获取。

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

@@ -0,0 +1 @@
+# Code Analysis Agent Tools

Undefined/skills/agents/naga_code_analysis_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/naga_code_analysis_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/naga_code_analysis_agent/tools/glob/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "glob",
+        "description": "查找匹配模式的文件。路径是相对于当前工作目录的。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "pattern": {
+                    "type": "string",
+                    "description": "文件模式（如 **/*.py, code/**/*.py）"
+                }
+            },
+            "required": ["pattern"]
+        }
+    }
+}