mcp-use 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

mcp_use/task_managers/base.py

@@ -0,0 +1,151 @@
+"""
+Connection management for MCP implementations.
+
+This module provides an abstract base class for different types of connection
+managers used in MCP connectors.
+"""
+
+import asyncio
+from abc import ABC, abstractmethod
+from typing import Generic, TypeVar
+
+from ..logging import logger
+
+# Type variable for connection types
+T = TypeVar("T")
+
+
+class ConnectionManager(Generic[T], ABC):
+    """Abstract base class for connection managers.
+
+    This class defines the interface for different types of connection managers
+    used with MCP connectors.
+    """
+
+    def __init__(self):
+        """Initialize a new connection manager."""
+        self._ready_event = asyncio.Event()
+        self._done_event = asyncio.Event()
+        self._exception: Exception | None = None
+        self._connection: T | None = None
+        self._task: asyncio.Task | None = None
+
+    @abstractmethod
+    async def _establish_connection(self) -> T:
+        """Establish the connection.
+
+        This method should be implemented by subclasses to establish
+        the specific type of connection needed.
+
+        Returns:
+            The established connection.
+
+        Raises:
+            Exception: If connection cannot be established.
+        """
+        pass
+
+    @abstractmethod
+    async def _close_connection(self, connection: T) -> None:
+        """Close the connection.
+
+        This method should be implemented by subclasses to close
+        the specific type of connection.
+
+        Args:
+            connection: The connection to close.
+        """
+        pass
+
+    async def start(self) -> T:
+        """Start the connection manager and establish a connection.
+
+        Returns:
+            The established connection.
+
+        Raises:
+            Exception: If connection cannot be established.
+        """
+        # Reset state
+        self._ready_event.clear()
+        self._done_event.clear()
+        self._exception = None
+
+        # Create a task to establish and maintain the connection
+        self._task = asyncio.create_task(
+            self._connection_task(), name=f"{self.__class__.__name__}_task"
+        )
+
+        # Wait for the connection to be ready or fail
+        await self._ready_event.wait()
+
+        # If there was an exception, raise it
+        if self._exception:
+            raise self._exception
+
+        # Return the connection
+        if self._connection is None:
+            raise RuntimeError("Connection was not established")
+        return self._connection
+
+    async def stop(self) -> None:
+        """Stop the connection manager and close the connection."""
+        if self._task and not self._task.done():
+            # Cancel the task
+            logger.debug(f"Cancelling {self.__class__.__name__} task")
+            self._task.cancel()
+
+            # Wait for it to complete
+            try:
+                await self._task
+            except asyncio.CancelledError:
+                logger.debug(f"{self.__class__.__name__} task cancelled successfully")
+            except Exception as e:
+                logger.warning(f"Error stopping {self.__class__.__name__} task: {e}")
+
+        # Wait for the connection to be done
+        await self._done_event.wait()
+        logger.debug(f"{self.__class__.__name__} task completed")
+
+    async def _connection_task(self) -> None:
+        """Run the connection task.
+
+        This task establishes and maintains the connection until cancelled.
+        """
+        logger.debug(f"Starting {self.__class__.__name__} task")
+        try:
+            # Establish the connection
+            self._connection = await self._establish_connection()
+            logger.debug(f"{self.__class__.__name__} connected successfully")
+
+            # Signal that the connection is ready
+            self._ready_event.set()
+
+            # Wait indefinitely until cancelled
+            try:
+                # This keeps the connection open until cancelled
+                await asyncio.Event().wait()
+            except asyncio.CancelledError:
+                # Expected when stopping
+                logger.debug(f"{self.__class__.__name__} task received cancellation")
+                pass
+
+        except Exception as e:
+            # Store the exception
+            self._exception = e
+            logger.error(f"Error in {self.__class__.__name__} task: {e}")
+
+            # Signal that the connection is ready (with error)
+            self._ready_event.set()
+
+        finally:
+            # Close the connection if it was established
+            if self._connection is not None:
+                try:
+                    await self._close_connection(self._connection)
+                except Exception as e:
+                    logger.warning(f"Error closing connection in {self.__class__.__name__}: {e}")
+                self._connection = None
+
+            # Signal that the connection is done
+            self._done_event.set()

mcp_use/task_managers/http.py

@@ -0,0 +1,62 @@
+"""
+HTTP connection management for MCP implementations.
+
+This module provides a connection manager for HTTP-based MCP connections.
+"""
+
+import aiohttp
+
+from ..logging import logger
+from .base import ConnectionManager
+
+
+class HttpConnectionManager(ConnectionManager[aiohttp.ClientSession]):
+    """Connection manager for HTTP-based MCP connections.
+
+    This class handles the lifecycle of HTTP client sessions, ensuring proper
+    connection establishment and cleanup.
+    """
+
+    def __init__(
+        self,
+        base_url: str,
+        headers: dict[str, str] | None = None,
+    ):
+        """Initialize a new HTTP connection manager.
+
+        Args:
+            base_url: The base URL for HTTP requests
+            headers: Optional headers to include in all HTTP requests
+        """
+        super().__init__()
+        self.base_url = base_url.rstrip("/")
+        self.headers = headers or {}
+
+    async def _establish_connection(self) -> aiohttp.ClientSession:
+        """Establish an HTTP client session.
+
+        Returns:
+            The established HTTP client session
+
+        Raises:
+            Exception: If session cannot be established
+        """
+        logger.debug(f"Creating HTTP client session for: {self.base_url}")
+        try:
+            session = aiohttp.ClientSession(headers=self.headers)
+            return session
+        except Exception as e:
+            logger.error(f"Failed to create HTTP client session: {e}")
+            raise
+
+    async def _close_connection(self, connection: aiohttp.ClientSession) -> None:
+        """Close the HTTP client session.
+
+        Args:
+            connection: The HTTP client session to close
+        """
+        try:
+            logger.debug("Closing HTTP client session")
+            await connection.close()
+        except Exception as e:
+            logger.warning(f"Error closing HTTP client session: {e}")

mcp_use/task_managers/stdio.py

@@ -0,0 +1,73 @@
+"""
+StdIO connection management for MCP implementations.
+
+This module provides a connection manager for stdio-based MCP connections
+that ensures proper task isolation and resource cleanup.
+"""
+
+import sys
+from typing import Any, TextIO
+
+from mcp import StdioServerParameters
+from mcp.client.stdio import stdio_client
+
+from ..logging import logger
+from .base import ConnectionManager
+
+
+class StdioConnectionManager(ConnectionManager[tuple[Any, Any]]):
+    """Connection manager for stdio-based MCP connections.
+
+    This class handles the proper task isolation for stdio_client context managers
+    to prevent the "cancel scope in different task" error. It runs the stdio_client
+    in a dedicated task and manages its lifecycle.
+    """
+
+    def __init__(
+        self,
+        server_params: StdioServerParameters,
+        errlog: TextIO = sys.stderr,
+    ):
+        """Initialize a new stdio connection manager.
+
+        Args:
+            server_params: The parameters for the stdio server
+            errlog: The error log stream
+        """
+        super().__init__()
+        self.server_params = server_params
+        self.errlog = errlog
+        self._stdio_ctx = None
+
+    async def _establish_connection(self) -> tuple[Any, Any]:
+        """Establish a stdio connection.
+
+        Returns:
+            A tuple of (read_stream, write_stream)
+
+        Raises:
+            Exception: If connection cannot be established.
+        """
+        # Create the context manager
+        self._stdio_ctx = stdio_client(self.server_params, self.errlog)
+
+        # Enter the context manager
+        read_stream, write_stream = await self._stdio_ctx.__aenter__()
+
+        # Return the streams
+        return (read_stream, write_stream)
+
+    async def _close_connection(self, connection: tuple[Any, Any]) -> None:
+        """Close the stdio connection.
+
+        Args:
+            connection: The connection to close (ignored, we use the context manager)
+        """
+        if self._stdio_ctx:
+            # Exit the context manager
+            try:
+                await self._stdio_ctx.__aexit__(None, None, None)
+            except Exception as e:
+                logger.warning(f"Error closing stdio context: {e}")
+            finally:
+                self._stdio_ctx = None

mcp_use/task_managers/websocket.py

@@ -0,0 +1,63 @@
+"""
+WebSocket connection management for MCP implementations.
+
+This module provides a connection manager for WebSocket-based MCP connections.
+"""
+
+import websockets
+from websockets.client import WebSocketClientProtocol
+
+from ..logging import logger
+from .base import ConnectionManager
+
+
+class WebSocketConnectionManager(ConnectionManager[WebSocketClientProtocol]):
+    """Connection manager for WebSocket-based MCP connections.
+
+    This class handles the lifecycle of WebSocket connections, ensuring proper
+    connection establishment and cleanup.
+    """
+
+    def __init__(
+        self,
+        url: str,
+        headers: dict[str, str] | None = None,
+    ):
+        """Initialize a new WebSocket connection manager.
+
+        Args:
+            url: The WebSocket URL to connect to
+            headers: Optional headers to include in the WebSocket connection
+        """
+        super().__init__()
+        self.url = url
+        self.headers = headers or {}
+
+    async def _establish_connection(self) -> WebSocketClientProtocol:
+        """Establish a WebSocket connection.
+
+        Returns:
+            The established WebSocket connection
+
+        Raises:
+            Exception: If connection cannot be established
+        """
+        logger.debug(f"Connecting to WebSocket: {self.url}")
+        try:
+            ws = await websockets.connect(self.url, extra_headers=self.headers)
+            return ws
+        except Exception as e:
+            logger.error(f"Failed to connect to WebSocket: {e}")
+            raise
+
+    async def _close_connection(self, connection: WebSocketClientProtocol) -> None:
+        """Close the WebSocket connection.
+
+        Args:
+            connection: The WebSocket connection to close
+        """
+        try:
+            logger.debug("Closing WebSocket connection")
+            await connection.close()
+        except Exception as e:
+            logger.warning(f"Error closing WebSocket connection: {e}")

{mcp_use-0.0.3.dist-info → mcp_use-0.0.4.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: mcp_use
-Version: 0.0.3
-Summary: Model-Agnostic MCP Library for LLMs
-Home-page: https://github.com/pietrozullo/mcp_use
-Author: Pietro Zullo
-Author-email: pietro.zullo@gmail.com
+Version: 0.0.4
+Summary: MCP Library for LLMs
+Author-email: Pietro Zullo <pietro.zullo@gmail.com>
+License: MIT
+License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
@@ -14,40 +14,28 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: mcp
-Requires-Dist: langchain>=0.1.0
-Requires-Dist: langchain-community>=0.0.10
-Requires-Dist: websockets>=12.0
 Requires-Dist: aiohttp>=3.9.0
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: typing-extensions>=4.8.0
 Requires-Dist: jsonschema-pydantic>=0.1.0
+Requires-Dist: langchain-community>=0.0.10
+Requires-Dist: langchain>=0.1.0
+Requires-Dist: mcp
+Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv>=1.0.0
-Provides-Extra: dev
-Requires-Dist: pytest>=7.4.0; extra == "dev"
-Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
-Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
-Requires-Dist: black>=23.9.0; extra == "dev"
-Requires-Dist: isort>=5.12.0; extra == "dev"
-Requires-Dist: mypy>=1.5.0; extra == "dev"
-Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: typing-extensions>=4.8.0
+Requires-Dist: websockets>=12.0
 Provides-Extra: anthropic
-Requires-Dist: anthropic>=0.15.0; extra == "anthropic"
+Requires-Dist: anthropic>=0.15.0; extra == 'anthropic'
+Provides-Extra: dev
+Requires-Dist: black>=23.9.0; extra == 'dev'
+Requires-Dist: isort>=5.12.0; extra == 'dev'
+Requires-Dist: mypy>=1.5.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21.0; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1.0; extra == 'dev'
+Requires-Dist: pytest>=7.4.0; extra == 'dev'
+Requires-Dist: ruff>=0.1.0; extra == 'dev'
 Provides-Extra: openai
-Requires-Dist: openai>=1.10.0; extra == "openai"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
+Requires-Dist: openai>=1.10.0; extra == 'openai'
+Description-Content-Type: text/markdown
 
 <picture>
   <source media="(prefers-color-scheme: dark)" srcset="./static/mcp-use-dark.png">
@@ -55,7 +43,7 @@ Dynamic: summary
   <img alt="Shows a black MCP-Use Logo in light color mode and a white one in dark color mode." src="./static/mcp-use.png"  width="full">
 </picture>
 
-<h1 align="center">Model-Agnostic MCP Library for LLMs 🤖</h1>
+<h1 align="center">Use MCPs directly from python 🤖</h1>
 
 [![GitHub stars](https://img.shields.io/github/stars/pietrozullo/mcp-use?style=social)](https://github.com/pietrozullo/mcp-use/stargazers)
 
@@ -356,7 +344,7 @@ If you use MCP-Use in your research or project, please cite:
 ```bibtex
 @software{mcp_use2024,
   author = {Zullo, Pietro},
-  title = {MCP-Use: Model-Agnostic MCP Library for LLMs},
+  title = {MCP-Use: MCP Library for Python},
   year = {2024},
   publisher = {GitHub},
   url = {https://github.com/pietrozullo/mcp-use}

mcp_use-0.0.4.dist-info/RECORD

@@ -0,0 +1,24 @@
+mcp_use/__init__.py,sha256=wH2KewPklX8TjKnuECJhUCPL6BqOngmFjTtpleKi46s,699
+mcp_use/client.py,sha256=7dBjkRPqyUlp-8NuLcYbW-4ctLDp329DsAvRM1HlVus,7807
+mcp_use/config.py,sha256=ZoOdKX6qVR5SD7PfO-iqzKxXOb4NMcnpHpZquGgMas4,2578
+mcp_use/logging.py,sha256=2-hSB7ZWcHEx_OFHNg8GIbSGCZx3MW4mZGGWxi2Ew3E,2690
+mcp_use/session.py,sha256=Z4EZTUnQUX0QyGMzkJIrMRTX4SDk6qQUoBld408LIJE,3449
+mcp_use/agents/__init__.py,sha256=ukchMTqCOID6ikvLmJ-6sldWTVFIzztGQo4BX6QeQr8,312
+mcp_use/agents/base.py,sha256=bfuldi_89AbSbNc8KeTiCArRT9V62CNxHOWYkLHWjyA,1605
+mcp_use/agents/langchain_agent.py,sha256=RaQEo-tm3b4SBeu3DxE9YYdTuTAHBqEd5vSZ6Qi3g5Q,10320
+mcp_use/agents/mcpagent.py,sha256=87NwMYr4gnAj-fEntAPo7AWbHI60JvnJM5dLIbls-i0,12260
+mcp_use/agents/prompts/default.py,sha256=msFRqH5ZPlcpVh5lzT9Bq5gu5m9wH819KO-xYiFs01g,392
+mcp_use/connectors/__init__.py,sha256=jnd-7pPPJMb0UNJ6aD9lInj5Tlamc8lA_mFyG8RWJpo,385
+mcp_use/connectors/base.py,sha256=caUaTfsODUOik8JF9mPtcZDyZhoIz2X12I_BhAfZK10,1616
+mcp_use/connectors/http.py,sha256=KqVf0HXouFoeQ_bBUr6KQifiUjTo7K6EOCRkqVpFx4Q,7763
+mcp_use/connectors/stdio.py,sha256=aEeZ-OZS6yknFPEy-YpwvwFILGIaAsA48uB1I4j8wog,6752
+mcp_use/connectors/websocket.py,sha256=4xqxl9UncrfU6NitvKfB80Hk2g7o0Gc0G5sm6sY3RAk,9534
+mcp_use/task_managers/__init__.py,sha256=6VVe5ceSxXmQvBpjH-6aFud5dRJMNA6pu0qpAnfxpIA,460
+mcp_use/task_managers/base.py,sha256=ksNdxTwq8N-zqymxVoKGnWXq9iqkLYC61uB91o6Mh-4,4888
+mcp_use/task_managers/http.py,sha256=XhrF73RGRnVctBVW2FlFrFTJR2pIGXhtNvfJFiW0Olw,1881
+mcp_use/task_managers/stdio.py,sha256=DEISpXv4mo3d5a-WT8lkWbrXJwUh7QW0nMT_IM3fHGg,2269
+mcp_use/task_managers/websocket.py,sha256=SVgTLFogiynb48eyi6ZioWIKLLWiVBCNE59rXi6GrCM,1943
+mcp_use-0.0.4.dist-info/METADATA,sha256=A63pM3FRoTjjjQl9zDGgUSls-_vX6N7Du4o5VfeuqXY,9141
+mcp_use-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_use-0.0.4.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
+mcp_use-0.0.4.dist-info/RECORD,,

{mcp_use-0.0.3.dist-info → mcp_use-0.0.4.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any
-

mcp_use-0.0.3.dist-info/RECORD

@@ -1,19 +0,0 @@
-mcp_use/__init__.py,sha256=hZhaqPZ72buzDZARZvMFvWx37E7aqjQ1SjI3U4A4ia4,738
-mcp_use/client.py,sha256=yPjc50n4q0uumOOMykLN5X3OKIDzr-UtU8Byff_Aex8,6922
-mcp_use/config.py,sha256=ZoOdKX6qVR5SD7PfO-iqzKxXOb4NMcnpHpZquGgMas4,2578
-mcp_use/logging.py,sha256=2-hSB7ZWcHEx_OFHNg8GIbSGCZx3MW4mZGGWxi2Ew3E,2690
-mcp_use/session.py,sha256=Z4EZTUnQUX0QyGMzkJIrMRTX4SDk6qQUoBld408LIJE,3449
-mcp_use/agents/__init__.py,sha256=ukchMTqCOID6ikvLmJ-6sldWTVFIzztGQo4BX6QeQr8,312
-mcp_use/agents/base.py,sha256=bfuldi_89AbSbNc8KeTiCArRT9V62CNxHOWYkLHWjyA,1605
-mcp_use/agents/langchain_agent.py,sha256=JiZQ4KRgtLCETvVubCIEo0egPUyTc0M56OR6pGtrEf0,8879
-mcp_use/agents/mcpagent.py,sha256=R1cjgx2sjbCrr0lHJSOEnoA8UIH_aTk1xyuf6z_RGFU,5434
-mcp_use/connectors/__init__.py,sha256=jnd-7pPPJMb0UNJ6aD9lInj5Tlamc8lA_mFyG8RWJpo,385
-mcp_use/connectors/base.py,sha256=caUaTfsODUOik8JF9mPtcZDyZhoIz2X12I_BhAfZK10,1616
-mcp_use/connectors/http.py,sha256=ryNOJFof98rAKkIzccMQNLYtzu1V2KC9s8R2PKP4EFY,4702
-mcp_use/connectors/stdio.py,sha256=LjbrPu8FtB1FLMLBdK7NpRQrYMhx_QITdZWH3e47fW8,4483
-mcp_use/connectors/websocket.py,sha256=nRMaURCP1d1dGTAHCn49RFOWKLtKAqdWot04STgtodg,5301
-mcp_use-0.0.3.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
-mcp_use-0.0.3.dist-info/METADATA,sha256=xb7TGK6dasApSS3tgdk3MTR0iEDLqf_MgpLCfxxOBEM,9461
-mcp_use-0.0.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-mcp_use-0.0.3.dist-info/top_level.txt,sha256=xK9hG-ab5BJu7z6yGSxpeT3eVVTyzbpFiwJ0h8oknBM,8
-mcp_use-0.0.3.dist-info/RECORD,,

mcp_use-0.0.3.dist-info/top_level.txt

@@ -1 +0,0 @@
-mcp_use