Add files via upload

Author: answeryt

MCP/__init__.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+from ._mcp_function import MCPToolFunction
+from ._client_base import MCPClientBase
+from ._stateful_client_base import StatefulClientBase
+from ._http_stateless_client import HttpStatelessClient
+from ._stdio_stateful_client import StdIOStatefulClient
+
+__all__ = [
+    'MCPToolFunction',
+    'MCPClientBase', 
+    'StatefulClientBase',
+    'HttpStatelessClient',
+    'StdIOStatefulClient'
+]

MCP/__pycache__/__init__.cpython-313.pyc

@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+"""The base class for MCP clients in AgentScope."""
+from abc import abstractmethod
+from dataclasses import dataclass
+from typing import Callable, List
+
+import mcp.types
+
+from _logging import logger
+
+
+@dataclass
+class ImageBlock:
+    type: str
+    source: "Base64Source"
+
+
+@dataclass
+class Base64Source:
+    type: str
+    media_type: str
+    data: str
+
+
+@dataclass
+class AudioBlock:
+    type: str
+    source: Base64Source
+
+
+@dataclass
+class TextBlock:
+    type: str
+    text: str
+
+
+class MCPClientBase:
+    """Base class for MCP clients."""
+
+    def __init__(self, name: str) -> None:
+        """Initialize the MCP client with a name.
+
+        Args:
+            name (`str`):
+                The name to identify the MCP server, which should be unique
+                across the MCP servers.
+        """
+        self.name = name
+
+    @abstractmethod
+    async def get_callable_function(
+        self,
+        func_name: str,
+        wrap_tool_result: bool = True,
+    ) -> Callable:
+        """Get a tool function by its name."""
+
+    @staticmethod
+    def _convert_mcp_content_to_as_blocks(
+        mcp_content_blocks: list,
+    ) -> List[TextBlock | ImageBlock | AudioBlock]:
+        """Convert MCP content to AgentScope blocks."""
+
+        as_content: list = []
+        for content in mcp_content_blocks:
+            if isinstance(content, mcp.types.TextContent):
+                as_content.append(
+                    TextBlock(
+                        type="text",
+                        text=content.text,
+                    ),
+                )
+            elif isinstance(content, mcp.types.ImageContent):
+                as_content.append(
+                    ImageBlock(
+                        type="image",
+                        source=Base64Source(
+                            type="base64",
+                            media_type=content.mimeType,
+                            data=content.data,
+                        ),
+                    ),
+                )
+            elif isinstance(content, mcp.types.AudioContent):
+                as_content.append(
+                    AudioBlock(
+                        type="audio",
+                        source=Base64Source(
+                            type="base64",
+                            media_type=content.mimeType,
+                            data=content.data,
+                        ),
+                    ),
+                )
+            elif isinstance(content, mcp.types.EmbeddedResource):
+                if isinstance(
+                    content.resource,
+                    mcp.types.TextResourceContents,
+                ):
+                    as_content.append(
+                        TextBlock(
+                            type="text",
+                            text=content.resource.model_dump_json(indent=2),
+                        ),
+                    )
+                else:
+                    # TODO: support the BlobResourceContents in the future,
+                    #  which is a base64-encoded string representing the
+                    #  binary data
+                    logger.error(
+                        "Unsupported EmbeddedResource content type: %s. "
+                        "Skipping this content.",
+                        type(content.resource),
+                    )
+            else:
+                logger.warning(
+                    "Unsupported content type: %s. Skipping this content.",
+                    type(content),
+                )
+        return as_content