tinyagent-py 0.0.5__py3-none-any.whl → 0.0.6__py3-none-any.whl

tinyagent/mcp_client.py

@@ -106,11 +106,6 @@ class MCPClient:
 
     async def close(self):
         """Clean up subprocess and streams."""
-        if self.session:
-            try:
-                await self.session.close()
-            except Exception as e:
-                self.logger.error(f"Error closing session: {e}")
         if self.exit_stack:
             try:
                 await self.exit_stack.aclose()

tinyagent/tiny_agent.py

@@ -746,13 +746,58 @@ class TinyAgent:
 
     
     async def close(self):
-        """Clean up *all* MCP clients."""
+        """
+        Clean up all resources used by the agent including MCP clients and storage.
+        
+        This method should be called when the agent is no longer needed to ensure
+        proper resource cleanup, especially in web frameworks like FastAPI.
+        """
+        cleanup_errors = []
+        
+        # 1. First save any pending state if storage is configured
+        if self.storage:
+            try:
+                self.logger.debug(f"Saving final state before closing (session={self.session_id})")
+                await self.save_agent()
+            except Exception as e:
+                error_msg = f"Error saving final state: {str(e)}"
+                self.logger.error(error_msg)
+                cleanup_errors.append(error_msg)
+        
+        # 2. Close all MCP clients
         for client in self.mcp_clients:
             try:
+                self.logger.debug(f"Closing MCP client: {client}")
                 await client.close()
-            except RuntimeError as e:
-                self.logger.error(f"Error closing MCP client: {str(e)}")
-                # Continue closing other clients even if one fails
+            except Exception as e:
+                error_msg = f"Error closing MCP client: {str(e)}"
+                self.logger.error(error_msg)
+                cleanup_errors.append(error_msg)
+        
+        # 3. Close storage connection if available
+        if self.storage:
+            try:
+                self.logger.debug("Closing storage connection")
+                await self.storage.close()
+            except Exception as e:
+                error_msg = f"Error closing storage: {str(e)}"
+                self.logger.error(error_msg)
+                cleanup_errors.append(error_msg)
+        
+        # 4. Run any cleanup callbacks
+        try:
+            await self._run_callbacks("agent_cleanup")
+        except Exception as e:
+            error_msg = f"Error in cleanup callbacks: {str(e)}"
+            self.logger.error(error_msg)
+            cleanup_errors.append(error_msg)
+        
+        # Log summary of cleanup
+        if cleanup_errors:
+            self.logger.warning(f"TinyAgent cleanup completed with {len(cleanup_errors)} errors")
+        else:
+            self.logger.info(f"TinyAgent cleanup completed successfully (session={self.session_id})")
+
     def clear_conversation(self):
         """
         Clear the conversation history, preserving only the initial system prompt.

{tinyagent_py-0.0.5.dist-info → tinyagent_py-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyagent-py
-Version: 0.0.5
+Version: 0.0.6
 Summary: Tiny Agent with MCP Client
 Author-email: Mahdi Golchin <golchin@askdev.ai>
 Project-URL: Homepage, https://github.com/askbudi/tinyagent

{tinyagent_py-0.0.5.dist-info → tinyagent_py-0.0.6.dist-info}/RECORD

@@ -1,6 +1,6 @@
 tinyagent/__init__.py,sha256=GrD21npMQGzl9ZYKYTP8VxHLzCfJCvA0oTKQZTkmnCw,117
-tinyagent/mcp_client.py,sha256=ZyhIWFQvOveNVhMKkMkUMYdbqRSuqjK_m-ApP2llFVE,6215
-tinyagent/tiny_agent.py,sha256=89OLYj95VUW7uoHpkWXZjmxsQISUMjHffUmDT-3-Nd4,39250
+tinyagent/mcp_client.py,sha256=9dmLtJ8CTwKWKTH6K9z8CaCQuaicOH9ifAuNyX7Kdo0,6030
+tinyagent/tiny_agent.py,sha256=PW_mY0uqKqzVN3ANWauF0RmvXNobR_EoPjGv40m1oKw,41074
 tinyagent/hooks/__init__.py,sha256=UztCHjoqF5JyDolbWwkBsBZkWguDQg23l2GD_zMHt-s,178
 tinyagent/hooks/agno_storage_hook.py,sha256=5qvvjmtraanPa-A46Zstrqq3s1e-sC7Ly0o3zifuw_4,5003
 tinyagent/hooks/gradio_callback.py,sha256=jGsZlObAd6I5lN9cE53dDL_LfiB8I0tBsicuHwwmL-M,44833
@@ -13,8 +13,8 @@ tinyagent/storage/json_file_storage.py,sha256=SYD8lvTHu2-FEHm1tZmsrcgEOirBrlUsUM
 tinyagent/storage/postgres_storage.py,sha256=IGwan8UXHNnTZFK1F8x4kvMDex3GAAGWUg9ePx_5IF4,9018
 tinyagent/storage/redis_storage.py,sha256=hu3y7wHi49HkpiR-AW7cWVQuTVOUk1WaB8TEPGUKVJ8,1742
 tinyagent/storage/sqlite_storage.py,sha256=7lk1XZpr2t4s2bjVr9-AqrI74w4hwkuK3taWtyJZhBc,5769
-tinyagent_py-0.0.5.dist-info/licenses/LICENSE,sha256=YIogcVQnknaaE4K-oaQylFWo8JGRBWnwmGb3fWB_Pww,1064
-tinyagent_py-0.0.5.dist-info/METADATA,sha256=EW9xVFf0vYbe1DQYxQAlNgOXUzuP9g2pqc__yx4V8NQ,9054
-tinyagent_py-0.0.5.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-tinyagent_py-0.0.5.dist-info/top_level.txt,sha256=Ny8aJNchZpc2Vvhp3306L5vjceJakvFxBk-UjjVeA_I,10
-tinyagent_py-0.0.5.dist-info/RECORD,,
+tinyagent_py-0.0.6.dist-info/licenses/LICENSE,sha256=YIogcVQnknaaE4K-oaQylFWo8JGRBWnwmGb3fWB_Pww,1064
+tinyagent_py-0.0.6.dist-info/METADATA,sha256=rL8hpf-PC0CBRcoyDEe71okWlzLi9SEln03ZHRs_ikI,9054
+tinyagent_py-0.0.6.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+tinyagent_py-0.0.6.dist-info/top_level.txt,sha256=Ny8aJNchZpc2Vvhp3306L5vjceJakvFxBk-UjjVeA_I,10
+tinyagent_py-0.0.6.dist-info/RECORD,,