agentrun-inner-test 0.0.46__py3-none-any.whl

agentrun/toolset/api/control.py

@@ -0,0 +1,262 @@
+"""
+This file is auto generated by the code generation script.
+Do not modify this file manually.
+Use the `make codegen` command to regenerate.
+
+当前文件为自动生成的控制 API 客户端代码。请勿手动修改此文件。
+使用 `make codegen` 命令重新生成。
+
+source: codegen/configs/toolset_control_api.yaml
+
+
+Tool 管控链路 API
+"""
+
+from typing import Dict, Optional
+
+from alibabacloud_devs20230714.models import (
+    ListToolsetsRequest,
+    ListToolsetsResponseBody,
+    Toolset,
+)
+from alibabacloud_tea_openapi.exceptions._client import ClientException
+from alibabacloud_tea_openapi.exceptions._server import ServerException
+from alibabacloud_tea_util.models import RuntimeOptions
+import pydash
+
+from agentrun.utils.config import Config
+from agentrun.utils.control_api import ControlAPI
+from agentrun.utils.exception import ClientError, ServerError
+from agentrun.utils.log import logger
+
+
+class ToolControlAPI(ControlAPI):
+    """Tool 管控链路 API"""
+
+    def __init__(self, config: Optional[Config] = None):
+        """初始化 API 客户端
+
+        Args:
+            config: 全局配置对象
+        """
+        super().__init__(config)
+
+    def get_toolset(
+        self,
+        name: str,
+        headers: Optional[Dict[str, str]] = None,
+        config: Optional[Config] = None,
+    ) -> Toolset:
+        """获取工具
+
+        Args:
+            name: Tool 名称
+
+            headers: 请求头
+            config: 配置
+
+        Returns:
+            Toolset: ToolSet 对象
+
+        Raises:
+            AgentRuntimeError: 调用失败时抛出
+            ClientError: 客户端错误
+            ServerError: 服务器错误
+            APIError: 运行时错误
+        """
+
+        try:
+            client = self._get_devs_client(config)
+            response = client.get_toolset_with_options(
+                name,
+                headers=headers or {},
+                runtime=RuntimeOptions(),
+            )
+
+            logger.debug(
+                "request api get_toolset, request Request ID:"
+                f" {response.headers['x-acs-request-id'] if response.headers else ''}\n"
+                f"  request: {[name,]}\n  response: {response.body}"
+            )
+
+            return response.body
+        except ClientException as e:
+            raise ClientError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+                request=[
+                    name,
+                ],
+            ) from e
+        except ServerException as e:
+            raise ServerError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+            ) from e
+
+    async def get_toolset_async(
+        self,
+        name: str,
+        headers: Optional[Dict[str, str]] = None,
+        config: Optional[Config] = None,
+    ) -> Toolset:
+        """获取工具
+
+        Args:
+            name: Tool 名称
+
+            headers: 请求头
+            config: 配置
+
+        Returns:
+            Toolset: ToolSet 对象
+
+        Raises:
+            AgentRuntimeError: 调用失败时抛出
+            ClientError: 客户端错误
+            ServerError: 服务器错误
+            APIError: 运行时错误
+        """
+
+        try:
+            client = self._get_devs_client(config)
+            response = await client.get_toolset_with_options_async(
+                name,
+                headers=headers or {},
+                runtime=RuntimeOptions(),
+            )
+
+            logger.debug(
+                "request api get_toolset, request Request ID:"
+                f" {response.headers['x-acs-request-id'] if response.headers else ''}\n"
+                f"  request: {[name,]}\n  response: {response.body}"
+            )
+
+            return response.body
+        except ClientException as e:
+            raise ClientError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+                request=[
+                    name,
+                ],
+            ) from e
+        except ServerException as e:
+            raise ServerError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+            ) from e
+
+    def list_toolsets(
+        self,
+        input: ListToolsetsRequest,
+        headers: Optional[Dict[str, str]] = None,
+        config: Optional[Config] = None,
+    ) -> ListToolsetsResponseBody:
+        """枚举 ToolSets
+
+        Args:
+            input: 枚举的配置
+
+            headers: 请求头
+            config: 配置
+
+        Returns:
+            ListToolsetsResponseBody: ToolSet 列表
+
+        Raises:
+            AgentRuntimeError: 调用失败时抛出
+            ClientError: 客户端错误
+            ServerError: 服务器错误
+            APIError: 运行时错误
+        """
+
+        try:
+            client = self._get_devs_client(config)
+            response = client.list_toolsets_with_options(
+                input,
+                headers=headers or {},
+                runtime=RuntimeOptions(),
+            )
+
+            logger.debug(
+                "request api list_toolsets, request Request ID:"
+                f" {response.headers['x-acs-request-id'] if response.headers else ''}\n"
+                f"  request: {[input.to_map(),]}\n  response: {response.body}"
+            )
+
+            return response.body
+        except ClientException as e:
+            raise ClientError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+                request=[
+                    input,
+                ],
+            ) from e
+        except ServerException as e:
+            raise ServerError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+            ) from e
+
+    async def list_toolsets_async(
+        self,
+        input: ListToolsetsRequest,
+        headers: Optional[Dict[str, str]] = None,
+        config: Optional[Config] = None,
+    ) -> ListToolsetsResponseBody:
+        """枚举 ToolSets
+
+        Args:
+            input: 枚举的配置
+
+            headers: 请求头
+            config: 配置
+
+        Returns:
+            ListToolsetsResponseBody: ToolSet 列表
+
+        Raises:
+            AgentRuntimeError: 调用失败时抛出
+            ClientError: 客户端错误
+            ServerError: 服务器错误
+            APIError: 运行时错误
+        """
+
+        try:
+            client = self._get_devs_client(config)
+            response = await client.list_toolsets_with_options_async(
+                input,
+                headers=headers or {},
+                runtime=RuntimeOptions(),
+            )
+
+            logger.debug(
+                "request api list_toolsets, request Request ID:"
+                f" {response.headers['x-acs-request-id'] if response.headers else ''}\n"
+                f"  request: {[input.to_map(),]}\n  response: {response.body}"
+            )
+
+            return response.body
+        except ClientException as e:
+            raise ClientError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+                request=[
+                    input,
+                ],
+            ) from e
+        except ServerException as e:
+            raise ServerError(
+                e.status_code,
+                pydash.get(e, "data.message", pydash.get(e, "message", "")),
+                request_id=e.request_id,
+            ) from e

agentrun/toolset/api/mcp.py

@@ -0,0 +1,100 @@
+"""MCP协议处理 / MCP Protocol Handler
+
+处理MCP(Model Context Protocol)协议的工具调用。
+Handles tool invocations for MCP (Model Context Protocol).
+"""
+
+from typing import Any, Dict, Optional
+
+from agentrun.utils.config import Config
+from agentrun.utils.log import logger
+
+
+class MCPSession:
+
+    def __init__(self, url: str, config: Optional[Config] = None):
+        self.url = url
+        self.config = Config.with_configs(config)
+
+    async def __aenter__(self):
+        from mcp import ClientSession
+        from mcp.client.sse import sse_client
+
+        timeout = self.config.get_timeout()
+        self.client = sse_client(
+            url=self.url,
+            headers=self.config.get_headers(),
+            timeout=timeout if timeout else 60,
+        )
+        (read, write) = await self.client.__aenter__()
+
+        self.client_session = ClientSession(read, write)
+        session = await self.client_session.__aenter__()
+        await session.initialize()
+
+        return session
+
+    async def __aexit__(self, *args):
+        await self.client_session.__aexit__(*args)
+        await self.client.__aexit__(*args)
+
+    def toolsets(self, config: Optional[Config] = None):
+        return MCPToolSet(url=self.url + "/toolsets", config=config)
+
+
+class MCPToolSet:
+
+    def __init__(self, url: str, config: Optional[Config] = None):
+        try:
+            __import__("mcp")
+        except ImportError:
+            logger.warning(
+                "MCPToolSet requires Python 3.10 or higher and install 'mcp'"
+                " package."
+            )
+
+        self.url = url
+        self.config = Config.with_configs(config)
+
+    def new_session(self, config: Optional[Config] = None):
+        cfg = Config.with_configs(self.config, config)
+        return MCPSession(url=self.url, config=cfg)
+
+    async def tools_async(self, config: Optional[Config] = None):
+        async with self.new_session(config=config) as session:
+            results = await session.list_tools()
+            return results.tools
+
+    def tools(self, config: Optional[Config] = None):
+        import asyncio
+
+        return asyncio.run(self.tools_async(config=config))
+
+    async def call_tool_async(
+        self,
+        name: str,
+        arguments: Optional[Dict[str, Any]] = None,
+        config: Optional[Config] = None,
+    ):
+        async with self.new_session(config=config) as session:
+            result = await session.call_tool(
+                name=name,
+                arguments=arguments,
+            )
+            return [item.model_dump() for item in result.content]
+
+    def call_tool(
+        self,
+        name: str,
+        arguments: Optional[Dict[str, Any]] = None,
+        config: Optional[Config] = None,
+    ):
+        import asyncio
+
+        return asyncio.run(
+            self.call_tool_async(
+                name=name,
+                arguments=arguments,
+                config=config,
+            )
+        )