voxagent 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

voxagent/_version.py

@@ -1,2 +1,2 @@
-__version__ = "0.2.3"
+__version__ = "0.2.4"
 __version_info__ = tuple(int(x) for x in __version__.split("."))

voxagent/agent/core.py

@@ -125,6 +125,68 @@ class Agent(Generic[DepsT, OutputT]):
         # Toolsets (MCP servers, etc.) - store for later
         self._toolsets = toolsets or []
 
+        # MCP connection caching (persistent across run() calls)
+        self._mcp_manager: MCPServerManager | None = None
+        self._mcp_tools: list[ToolDefinition] = []
+        self._mcp_connected: bool = False
+
+    # -------------------------------------------------------------------------
+    # MCP Connection Management
+    # -------------------------------------------------------------------------
+
+    async def connect_mcp(self) -> list[ToolDefinition]:
+        """Connect to MCP servers and cache the connection.
+
+        This method connects to all MCP servers in toolsets and caches the
+        connection for reuse across multiple run() calls. Call this during
+        initialization/warmup to avoid connection overhead on first message.
+
+        Returns:
+            List of ToolDefinition objects from connected MCP servers.
+        """
+        if self._mcp_connected:
+            return self._mcp_tools
+
+        if self._toolsets:
+            self._mcp_manager = MCPServerManager()
+            await self._mcp_manager.add_servers(self._toolsets)
+            self._mcp_tools = await self._mcp_manager.connect_all()
+            self._mcp_connected = True
+
+        return self._mcp_tools
+
+    async def disconnect_mcp(self) -> None:
+        """Disconnect from MCP servers.
+
+        Call this when the agent is no longer needed to clean up MCP
+        server connections. This is called automatically when using
+        the agent as an async context manager.
+        """
+        if self._mcp_manager and self._mcp_connected:
+            await self._mcp_manager.disconnect_all()
+            self._mcp_manager = None
+            self._mcp_connected = False
+            self._mcp_tools = []
+
+    @property
+    def mcp_connected(self) -> bool:
+        """Check if MCP servers are currently connected."""
+        return self._mcp_connected
+
+    async def __aenter__(self) -> "Agent[DepsT, OutputT]":
+        """Enter async context manager - connect MCP servers."""
+        await self.connect_mcp()
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: Any,
+    ) -> None:
+        """Exit async context manager - disconnect MCP servers."""
+        await self.disconnect_mcp()
+
     @staticmethod
     def _parse_model_string(model_string: str) -> ModelConfig:
         """Parse 'provider:model' string into ModelConfig.
@@ -495,19 +557,22 @@ class Agent(Generic[DepsT, OutputT]):
         timeout_handler: TimeoutHandler | None = None
         timed_out = False
         error_message: str | None = None
-        mcp_manager: MCPServerManager | None = None
+        # Track if we connected MCP in this run (for cleanup)
+        mcp_connected_in_this_run = False
 
         if timeout_ms:
             timeout_handler = TimeoutHandler(timeout_ms)
             await timeout_handler.start(abort_controller)
 
         try:
-            # Connect to MCP servers if any
-            mcp_tools: list[ToolDefinition] = []
-            if self._toolsets:
-                mcp_manager = MCPServerManager()
-                await mcp_manager.add_servers(self._toolsets)
-                mcp_tools = await mcp_manager.connect_all()
+            # Use cached MCP connection if available, otherwise connect
+            if self._mcp_connected:
+                mcp_tools = self._mcp_tools
+            elif self._toolsets:
+                mcp_tools = await self.connect_mcp()
+                mcp_connected_in_this_run = True
+            else:
+                mcp_tools = []
 
             # Get all tools (native + MCP)
             all_tools = self._get_all_tools(mcp_tools)
@@ -652,9 +717,10 @@ class Agent(Generic[DepsT, OutputT]):
             )
 
         finally:
-            # Disconnect MCP servers
-            if mcp_manager:
-                await mcp_manager.disconnect_all()
+            # Only disconnect MCP servers if we connected them in this run
+            # (not if using cached connection from connect_mcp())
+            if mcp_connected_in_this_run and not self._mcp_connected:
+                await self.disconnect_mcp()
             if timeout_handler:
                 timeout_handler.cancel()
             abort_controller.cleanup()
@@ -685,19 +751,22 @@ class Agent(Generic[DepsT, OutputT]):
         abort_controller = AbortController()
         timeout_handler: TimeoutHandler | None = None
         timed_out = False
-        mcp_manager: MCPServerManager | None = None
+        # Track if we connected MCP in this run (for cleanup)
+        mcp_connected_in_this_run = False
 
         if timeout_ms:
             timeout_handler = TimeoutHandler(timeout_ms)
             await timeout_handler.start(abort_controller)
 
         try:
-            # Connect to MCP servers if any
-            mcp_tools: list[ToolDefinition] = []
-            if self._toolsets:
-                mcp_manager = MCPServerManager()
-                await mcp_manager.add_servers(self._toolsets)
-                mcp_tools = await mcp_manager.connect_all()
+            # Use cached MCP connection if available, otherwise connect
+            if self._mcp_connected:
+                mcp_tools = self._mcp_tools
+            elif self._toolsets:
+                mcp_tools = await self.connect_mcp()
+                mcp_connected_in_this_run = True
+            else:
+                mcp_tools = []
 
             # Get all tools (native + MCP)
             all_tools = self._get_all_tools(mcp_tools)
@@ -838,9 +907,10 @@ class Agent(Generic[DepsT, OutputT]):
             )
 
         finally:
-            # Disconnect MCP servers
-            if mcp_manager:
-                await mcp_manager.disconnect_all()
+            # Only disconnect MCP servers if we connected them in this run
+            # (not if using cached connection from connect_mcp())
+            if mcp_connected_in_this_run and not self._mcp_connected:
+                await self.disconnect_mcp()
             if timeout_handler:
                 timeout_handler.cancel()
             abort_controller.cleanup()

voxagent/providers/chatgpt.py

@@ -206,7 +206,7 @@ class ChatGPTProvider(BaseProvider):
                     input_msgs.append({
                         "type": "message",
                         "role": "assistant",
-                        "content": [{"type": "input_text", "text": msg.content}],
+                        "content": [{"type": "output_text", "text": msg.content}],
                     })
                 # Add function_call items for each tool call
                 for tc in msg.tool_calls:
@@ -242,25 +242,28 @@ class ChatGPTProvider(BaseProvider):
                             "output": block.get("content", ""),
                         })
                     elif hasattr(block, "text"):
-                        # TextBlock
+                        # TextBlock - use output_text for assistant, input_text for user
+                        content_type = "output_text" if msg.role == "assistant" else "input_text"
                         input_msgs.append({
                             "type": "message",
                             "role": "user" if msg.role == "user" else "assistant",
-                            "content": [{"type": "input_text", "text": block.text}],
+                            "content": [{"type": content_type, "text": block.text}],
                         })
                     elif isinstance(block, dict) and "text" in block:
+                        content_type = "output_text" if msg.role == "assistant" else "input_text"
                         input_msgs.append({
                             "type": "message",
                             "role": "user" if msg.role == "user" else "assistant",
-                            "content": [{"type": "input_text", "text": block["text"]}],
+                            "content": [{"type": content_type, "text": block["text"]}],
                         })
             elif isinstance(msg.content, str):
-                # Simple string content
+                # Simple string content - use output_text for assistant, input_text for user
                 role = "user" if msg.role == "user" else "assistant"
+                content_type = "output_text" if msg.role == "assistant" else "input_text"
                 input_msgs.append({
                     "type": "message",
                     "role": role,
-                    "content": [{"type": "input_text", "text": msg.content}],
+                    "content": [{"type": content_type, "text": msg.content}],
                 })
 
         return input_msgs
@@ -335,7 +338,13 @@ class ChatGPTProvider(BaseProvider):
                     if response.status_code == 401:
                         yield ErrorChunk(error="Authentication failed - token may be expired")
                         return
-                    response.raise_for_status()
+                    if response.status_code >= 400:
+                        # Read error body before raising
+                        error_body = await response.aread()
+                        error_text = error_body.decode("utf-8", errors="replace")
+                        logger.error("ChatGPT API error %d: %s", response.status_code, error_text)
+                        yield ErrorChunk(error=f"HTTP {response.status_code}: {error_text[:500]}")
+                        return
 
                     async for line in response.aiter_lines():
                         if abort_signal and abort_signal.aborted:

{voxagent-0.2.3.dist-info → voxagent-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: voxagent
-Version: 0.2.3
+Version: 0.2.4
 Summary: A lightweight, model-agnostic LLM provider abstraction with streaming and tool support
 Project-URL: Homepage, https://github.com/lensator/voxagent
 Project-URL: Documentation, https://github.com/lensator/voxagent#readme

{voxagent-0.2.3.dist-info → voxagent-0.2.4.dist-info}/RECORD

@@ -1,9 +1,9 @@
 voxagent/__init__.py,sha256=YMYC95iwWXK26hicGYmd2erNOInrYtohwUVOuNmpTCs,3927
-voxagent/_version.py,sha256=avoaFz2ddynhpdQn-udgPxrhqWOcXWc2liIjyh_vgkI,87
+voxagent/_version.py,sha256=uG0jH_KsTBSDgsjnJdMe2gyLmoUNsNl07RdN-sUtP8Y,87
 voxagent/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 voxagent/agent/__init__.py,sha256=eASoU7Zhvw8BtJ-iUqVN06S4fMLkHwDgUZbHeH2AUOM,755
 voxagent/agent/abort.py,sha256=2Wnnxq8Dcn7wQkKPHrba2o0OeOdzF4NNsl-usgE4CJw,5191
-voxagent/agent/core.py,sha256=zXDUubx6oWQCj1V997s2zJ0Xj0XCaWIPZNinTNMz2zY,30581
+voxagent/agent/core.py,sha256=ddjDoWcDqiacRtNzef5wORR2f5uHqL1Y1gOmzoeXRFY,33339
 voxagent/code/__init__.py,sha256=MzbrYReislAB-lCZEZn_lBEPGjYZyPK-c9RFCq1nSm0,1379
 voxagent/code/agent.py,sha256=fVaOlJNvnHJedPCCW5Rmm4_-YRvgcYpaLicXA-8HWxU,9634
 voxagent/code/sandbox.py,sha256=LP2cwXchDk6mtiYGRb_RmkGNoyPv5OEKQC2h4M89dC0,11669
@@ -17,7 +17,7 @@ voxagent/providers/anthropic.py,sha256=A9aJUBc2nwqFdtvh5CyHp0GSQR82DaxwJQsgA79cf
 voxagent/providers/augment.py,sha256=K8ORZsOh88FBh4d7pYZUgMQHtQXnhJizpjwGkbWf53Y,8623
 voxagent/providers/auth.py,sha256=X3yDSHzj7gP-XxiYVpTSpDZs608gxnd-0UYX5aNNsEw,3650
 voxagent/providers/base.py,sha256=f_tPz4LFWX2nU9q1aa__Z3UpT35VpQnDVr40Z3VM6uk,6933
-voxagent/providers/chatgpt.py,sha256=DBjoc8UTONxwz2BGhsPVevSJBlLduai4gjzt62_lZrE,15620
+voxagent/providers/chatgpt.py,sha256=YW7VVXiPyqmeWKwbXraRfKakYCdb0tlTfqB6RqRK0hY,16445
 voxagent/providers/claudecode.py,sha256=4pUnWO9IJlXnagnqu3bl6SIDdyjpDohjOwJqRe96g6U,4331
 voxagent/providers/cli_base.py,sha256=0N4psjsH9ekgfq6ynspPQ48iB5BmN6qiOSDPMHjJdBA,7554
 voxagent/providers/codex.py,sha256=NUtCH1fqf1OTI2JEOlNClofm6spWq4ZvlaPaHbyotyE,5397
@@ -52,6 +52,6 @@ voxagent/tools/registry.py,sha256=MNJzgcmKT0AoMWIky9TJY4WVhzn5dkmjIHsUiZ3mv3U,25
 voxagent/types/__init__.py,sha256=3VunuprKKEpOR9Cg-UITHJXds_xQ-tfqQb4S7wD3nP4,933
 voxagent/types/messages.py,sha256=c6hNi9w6C8gbFoFm5fFge35vwJGywaoR_OiPQprfyVs,3494
 voxagent/types/run.py,sha256=4vYq0pCqH7_7SWbMb1SplWj4TLiE3DELDYMi0HefFmo,5071
-voxagent-0.2.3.dist-info/METADATA,sha256=lxBJ9wIDJhCCLGb4P_ps9C0d5kfYVdTAHZNyDer9vYU,5685
-voxagent-0.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-voxagent-0.2.3.dist-info/RECORD,,
+voxagent-0.2.4.dist-info/METADATA,sha256=MwN72uka42N7nHRbI6F6d3YvqFklZmcAEDDyg093e04,5685
+voxagent-0.2.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+voxagent-0.2.4.dist-info/RECORD,,