zhenxun_bot/zhenxun/services/ai/tools/providers/builtin/native.py

from zhenxun.services.ai.tools.core.tool import ServerSideTool


class WebSearchTool(ServerSideTool):
    """原生网页搜索工具"""

    type_id = "web_search"

    def __init__(
        self,
        name: str = "google_search",
        description: str = "使用 Google 搜索查找实时信息。",
        dynamic_threshold: float | None = None,
        domain_filters: list[str] | None = None,
    ):
        super().__init__(name=name, description=description)
        self.dynamic_threshold = dynamic_threshold
        self.domain_filters = domain_filters


class CodeExecutionTool(ServerSideTool):
    """原生代码沙箱执行工具"""

    type_id = "code_execution"

    def __init__(self, name: str = "code_execution", timeout: int | None = None):
        super().__init__(
            name=name, description="执行 Python 代码来解决数学或数据问题。"
        )
        self.timeout = timeout


class ComputerUseTool(ServerSideTool):
    """原生的桌面环境控制工具"""

    type_id = "computer_use"

    def __init__(self, display_width_px: int = 1024, display_height_px: int = 768):
        super().__init__(name="computer_use", description="控制电脑用户界面。")
        self.display_width_px = display_width_px
        self.display_height_px = display_height_px


class FileSearchTool(ServerSideTool):
    """原生文件检索工具"""

    type_id = "file_search"

    def __init__(self, name: str = "file_search"):
        super().__init__(name=name, description="搜索已索引的文件。")


class GoogleMapsTool(ServerSideTool):
    """原生谷歌地图检索工具"""

    type_id = "google_map"

    def __init__(self):
        super().__init__(name="google_map", description="搜索 Google 地图数据。")


class UrlContextTool(ServerSideTool):
    """原生 URL 上下文拉取工具"""

    type_id = "url_context"

    def __init__(self):
        super().__init__(name="url_context", description="从指定 URL 获取上下文信息。")


class Native:
    """
    云端原生工具命名空间工厂 (Namespace Factory)。

    为开发者提供统一的云端内置工具调用入口，享受顶级 IDE 补全体验。
    此类工具仅会向大模型提供描述，物理执行发生在各大模型厂商的服务端。
    """

    @classmethod
    def web_search(
        cls,
        name: str = "google_search",
        description: str = "使用 Google 搜索查找实时信息。",
        dynamic_threshold: float | None = None,
        domain_filters: list[str] | None = None,
    ) -> WebSearchTool:
        """
        原生网页搜索引擎工具 (如 Google Search, Bing)。
        """
        return WebSearchTool(name, description, dynamic_threshold, domain_filters)

    @classmethod
    def code_execution(
        cls, name: str = "code_execution", timeout: int | None = None
    ) -> CodeExecutionTool:
        """
        原生代码沙箱执行工具 (如 Gemini Code Execution, OpenAI Advanced Data Analysis)。
        """
        return CodeExecutionTool(name, timeout)

    @classmethod
    def computer_use(
        cls, display_width_px: int = 1024, display_height_px: int = 768
    ) -> ComputerUseTool:
        """
        原生桌面环境控制工具 (如 Claude Computer Use)。
        """
        return ComputerUseTool(display_width_px, display_height_px)

    @classmethod
    def file_search(cls, name: str = "file_search") -> FileSearchTool:
        """
        原生文件检索工具。
        """
        return FileSearchTool(name)

    @classmethod
    def google_map(cls) -> GoogleMapsTool:
        """
        原生谷歌地图检索工具 (Gemini 独占)。
        """
        return GoogleMapsTool()

    @classmethod
    def url_context(cls) -> UrlContextTool:
        """
        原生 URL 上下文拉取工具。
        """
        return UrlContextTool()