langchain-mcp-tools 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

langchain_mcp_tools/langchain_mcp_tools.py

@@ -50,7 +50,8 @@ The key aspects are:
    - Initializing multiple MCP servers in parallel requires a dedicated
      `asyncio.Task` per server
    - Server cleanup can be initiated later by a task other than the one that
-     initialized the resources
+     initialized the resources, whereas `AsyncExitStack.aclose()` must be
+     called from the same task that created the context
 
 2. Solution:
 
@@ -151,12 +152,16 @@ async def spawn_mcp_server_tools_task(
             env=env
         )
 
+        # Use an intermediate `asynccontextmanager` to log the cleanup message
         @asynccontextmanager
         async def log_before_aexit(context_manager, message):
             yield await context_manager.__aenter__()
-            logger.info(message)
-            await context_manager.__aexit__(None, None, None)
+            try:
+                logger.info(message)
+            finally:
+                await context_manager.__aexit__(None, None, None)
 
+        # Initialize the MCP server
         exit_stack = AsyncExitStack()
 
         stdio_transport = await exit_stack.enter_async_context(
@@ -174,8 +179,10 @@ async def spawn_mcp_server_tools_task(
         await session.initialize()
         logger.info(f'MCP server "{server_name}": connected')
 
+        # Get MCP tools
         tools_response = await session.list_tools()
 
+        # Wrap MCP tools to into LangChain tools
         for tool in tools_response.tools:
             class McpToLangChainAdapter(BaseTool):
                 name: str = tool.name or 'NO NAME'
@@ -211,10 +218,13 @@ async def spawn_mcp_server_tools_task(
         logger.error(f'Error getting response: {str(e)}')
         raise
 
+    # Set ready_event; signals tools are ready
     ready_event.set()
 
+    # Keep this task alive until cleanup is requested
     await cleanup_event.wait()
 
+    # Cleanup the resources
     await exit_stack.aclose()
 
 
@@ -258,6 +268,7 @@ async def convert_mcp_to_langchain_tools(
     ready_event_list = []
     cleanup_event_list = []
 
+    # Concurrently initialize all the MCP servers
     tasks = []
     for server_name, server_config in server_configs.items():
         server_tools_accumulator: List[BaseTool] = []
@@ -276,12 +287,16 @@ async def convert_mcp_to_langchain_tools(
         ))
         tasks.append(task)
 
+    # Wait for all tasks to finish filling in the `server_tools_accumulator`
     await asyncio.gather(*(event.wait() for event in ready_event_list))
 
+    # Flatten the tools list
     langchain_tools = [
         item for sublist in per_server_tools for item in sublist
     ]
 
+    # Define a cleanup callback to set cleanup_event and signal that
+    # it is time to clean up the resources
     async def mcp_cleanup() -> None:
         for event in cleanup_event_list:
             event.set()

{langchain_mcp_tools-0.1.0.dist-info → langchain_mcp_tools-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: langchain-mcp-tools
-Version: 0.1.0
+Version: 0.1.1
 Summary: Model Context Protocol (MCP) To LangChain Tools Conversion Utility
 Project-URL: Bug Tracker, https://github.com/hideya/langchain-mcp-tools-py/issues
 Project-URL: Source Code, https://github.com/hideya/langchain-mcp-tools-py
@@ -28,7 +28,7 @@ This package is intended to simplify the use of
 server tools with LangChain / Python.
 
 It contains a utility function `convert_mcp_to_langchain_tools()`.  
-This function handles parallel initialization of specified multiple MCP servers
+This async function handles parallel initialization of specified multiple MCP servers
 and converts their available tools into a list of LangChain-compatible tools.
 
 A typescript equivalent of this utility library is available
@@ -130,7 +130,8 @@ Any comments pointing out something I am missing would be greatly appreciated!
    - Initializing multiple MCP servers in parallel requires a dedicated
      `asyncio.Task` per server
    - Server cleanup can be initiated later by a task other than the one
-     that initialized the resources
+     that initialized the resources, whereas `AsyncExitStack.aclose()` must be
+     called from the same task that created the context
 
 2. Solution:
 

langchain_mcp_tools-0.1.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+langchain_mcp_tools/__init__.py,sha256=Xtv2VphhrWB_KlxTIofHZqtCIGtNEl0MxugnrNXTERA,94
+langchain_mcp_tools/langchain_mcp_tools.py,sha256=fBswDX7s7mY5w5DEVMB6ObuUboNivy1_hbyaN4X07ls,11104
+langchain_mcp_tools/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langchain_mcp_tools-0.1.1.dist-info/LICENSE,sha256=CRC91e8v116gCpnp7h49oIa6_zjhxqnHFTREeoZFJwA,1072
+langchain_mcp_tools-0.1.1.dist-info/METADATA,sha256=RNN7QFR1lQ5QBg4pnRQWRymoj4IKSY8GxmQpkPefzA4,6899
+langchain_mcp_tools-0.1.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+langchain_mcp_tools-0.1.1.dist-info/top_level.txt,sha256=aR_9V2A1Yt-Bca60KmndmGLUWb2wiM5IOG-Gkaf1dxY,20
+langchain_mcp_tools-0.1.1.dist-info/RECORD,,

langchain_mcp_tools-0.1.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-langchain_mcp_tools/__init__.py,sha256=Xtv2VphhrWB_KlxTIofHZqtCIGtNEl0MxugnrNXTERA,94
-langchain_mcp_tools/langchain_mcp_tools.py,sha256=PxkOAwQ8WUd12N-hZNfy2u5hawvhlEQhs0CsyPTWch0,10374
-langchain_mcp_tools/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_mcp_tools-0.1.0.dist-info/LICENSE,sha256=CRC91e8v116gCpnp7h49oIa6_zjhxqnHFTREeoZFJwA,1072
-langchain_mcp_tools-0.1.0.dist-info/METADATA,sha256=FuNglrw844NXdrNWNgkK7WNIUIIdW7a1sdJKRSmdT4o,6794
-langchain_mcp_tools-0.1.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-langchain_mcp_tools-0.1.0.dist-info/top_level.txt,sha256=aR_9V2A1Yt-Bca60KmndmGLUWb2wiM5IOG-Gkaf1dxY,20
-langchain_mcp_tools-0.1.0.dist-info/RECORD,,