microsoft-agents-a365-tooling-extensions-agentframework 0.2.1.dev5__py3-none-any.whl → 0.2.1.dev9__py3-none-any.whl

microsoft_agents_a365/tooling/extensions/agentframework/services/mcp_tool_registration_service.py

@@ -9,6 +9,7 @@ from typing import Any, List, Optional, Sequence, Union
 from agent_framework import ChatAgent, ChatMessage, ChatMessageStoreProtocol, MCPStreamableHTTPTool
 from agent_framework.azure import AzureOpenAIChatClient
 from agent_framework.openai import OpenAIChatClient
+import httpx
 
 from microsoft_agents.hosting.core import Authorization, TurnContext
 
@@ -24,6 +25,10 @@ from microsoft_agents_a365.tooling.utils.utility import (
 )
 
 
+# Default timeout for MCP server HTTP requests (in seconds)
+MCP_HTTP_CLIENT_TIMEOUT_SECONDS = 90.0
+
+
 class McpToolRegistrationService:
     """
     Provides MCP tool registration services for Agent Framework agents.
@@ -46,6 +51,7 @@ class McpToolRegistrationService:
             logger=self._logger
         )
         self._connected_servers = []
+        self._http_clients: List[httpx.AsyncClient] = []
 
     async def add_tool_servers_to_agent(
         self,
@@ -114,11 +120,17 @@ class McpToolRegistrationService:
                         self._orchestrator_name
                     )
 
-                    # Create and configure MCPStreamableHTTPTool
+                    # Create httpx client with auth headers configured
+                    http_client = httpx.AsyncClient(
+                        headers=headers, timeout=MCP_HTTP_CLIENT_TIMEOUT_SECONDS
+                    )
+                    self._http_clients.append(http_client)
+
+                    # Create and configure MCPStreamableHTTPTool with http_client
                     mcp_tools = MCPStreamableHTTPTool(
                         name=server_name,
                         url=config.url,
-                        headers=headers,
+                        http_client=http_client,
                         description=f"MCP tools from {server_name}",
                     )
 
@@ -339,12 +351,21 @@ class McpToolRegistrationService:
     async def cleanup(self):
         """Clean up any resources used by the service."""
         try:
+            # Close MCP server connections
             for plugin in self._connected_servers:
                 try:
                     if hasattr(plugin, "close"):
                         await plugin.close()
                 except Exception as cleanup_ex:
-                    self._logger.debug(f"Error during cleanup: {cleanup_ex}")
+                    self._logger.debug(f"Error during plugin cleanup: {cleanup_ex}")
             self._connected_servers.clear()
+
+            # Close httpx clients to prevent connection/file descriptor leaks
+            for http_client in self._http_clients:
+                try:
+                    await http_client.aclose()
+                except Exception as client_ex:
+                    self._logger.debug(f"Error closing http client: {client_ex}")
+            self._http_clients.clear()
         except Exception as ex:
             self._logger.debug(f"Error during service cleanup: {ex}")

{microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev5.dist-info → microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-a365-tooling-extensions-agentframework
-Version: 0.2.1.dev5
+Version: 0.2.1.dev9
 Summary: Agent Framework integration tools for Agent 365 AI agent tooling
 Author-email: Microsoft <support@microsoft.com>
 License: MIT
@@ -23,6 +23,7 @@ Requires-Dist: microsoft-agents-hosting-core>=0.4.0
 Requires-Dist: agent-framework-azure-ai>=1.0.0b251114
 Requires-Dist: azure-identity>=1.12.0
 Requires-Dist: typing-extensions>=4.0.0
+Requires-Dist: httpx>=0.27.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"

{microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev5.dist-info → microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev9.dist-info}/RECORD

@@ -1,7 +1,7 @@
 microsoft_agents_a365/tooling/extensions/agentframework/__init__.py,sha256=f_v_Uqi-rG_ramG91wrw-u8hzANbn5qS2wbp62S_bNw,905
 microsoft_agents_a365/tooling/extensions/agentframework/services/__init__.py,sha256=Xw_MZvaAiiPCFnTTU59FnG6AS6FxVj5qN2vxazB1I50,363
-microsoft_agents_a365/tooling/extensions/agentframework/services/mcp_tool_registration_service.py,sha256=cKIEJFW8xRI74MF_SY67mxr8Bk2XWP5C7OnUuQXEB9Y,14053
-microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev5.dist-info/METADATA,sha256=5rCdWOAnsKmLTdOUumVvBQzT3lzJ5eONUuKrFpBMw-M,3485
-microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev5.dist-info/top_level.txt,sha256=G3c2_4sy5_EM_BWO67SbK2tKj4G8XFn-QXRbh8g9Lgk,22
-microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev5.dist-info/RECORD,,
+microsoft_agents_a365/tooling/extensions/agentframework/services/mcp_tool_registration_service.py,sha256=9DZiJL_EpVWRojaZlboerl-q5AJl1B17wP_mP7zETpw,14952
+microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev9.dist-info/METADATA,sha256=yz_4lD2jGXfFOjLSOPQ59wUJ4-TElxowL3ay8EmIyg8,3514
+microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev9.dist-info/top_level.txt,sha256=G3c2_4sy5_EM_BWO67SbK2tKj4G8XFn-QXRbh8g9Lgk,22
+microsoft_agents_a365_tooling_extensions_agentframework-0.2.1.dev9.dist-info/RECORD,,