tinyagent-py 0.0.4__tar.gz → 0.0.6__tar.gz

{tinyagent_py-0.0.4/tinyagent/tinyagent_py.egg-info → tinyagent_py-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyagent-py
-Version: 0.0.4
+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
@@ -31,11 +31,17 @@ Requires-Dist: aiosqlite>=0.18.0; extra == "all"
 Requires-Dist: gradio>=3.50.0; extra == "all"
 Dynamic: license-file
 
-# tinyagent
-Tiny Agent: 100 lines Agent with MCP
+# TinyAgent
+Tiny Agent: 100 lines Agent with MCP and extendable hook system
+
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
+
 ![TinyAgent Logo](https://raw.githubusercontent.com/askbudi/tinyagent/main/public/logo.png)
 
 
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
 
 Inspired by:
 - [Tiny Agents blog post](https://huggingface.co/blog/tiny-agents)
@@ -91,6 +97,8 @@ uv pip install tinyagent-py[dev]
 ```
 
 ## Usage
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
 
 ```python
 from tinyagent import TinyAgent
@@ -239,6 +247,14 @@ if __name__ == "__main__":
 ```
 ---
 
+## Build your own TinyAgent
+
+You can chat with TinyAgent and build your own TinyAgent for your use case.
+
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
+---
+
 ## Contributing Hooks
 
 - Place new hooks in the `tinyagent/hooks/` directory.

{tinyagent_py-0.0.4 → tinyagent_py-0.0.6}/README.md

@@ -1,8 +1,14 @@
-# tinyagent
-Tiny Agent: 100 lines Agent with MCP
+# TinyAgent
+Tiny Agent: 100 lines Agent with MCP and extendable hook system
+
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
+
 ![TinyAgent Logo](https://raw.githubusercontent.com/askbudi/tinyagent/main/public/logo.png)
 
 
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
 
 Inspired by:
 - [Tiny Agents blog post](https://huggingface.co/blog/tiny-agents)
@@ -58,6 +64,8 @@ uv pip install tinyagent-py[dev]
 ```
 
 ## Usage
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
 
 ```python
 from tinyagent import TinyAgent
@@ -206,6 +214,14 @@ if __name__ == "__main__":
 ```
 ---
 
+## Build your own TinyAgent
+
+You can chat with TinyAgent and build your own TinyAgent for your use case.
+
+[![AskDev.AI | Chat with TinyAgent](https://img.shields.io/badge/AskDev.AI-Chat_with_TinyAgent-blue?style=flat-square)](https://askdev.ai/github/askbudi/tinyagent)
+
+---
+
 ## Contributing Hooks
 
 - Place new hooks in the `tinyagent/hooks/` directory.

{tinyagent_py-0.0.4 → tinyagent_py-0.0.6}/pyproject.toml

@@ -3,12 +3,13 @@ requires = ["setuptools>=77.0", "wheel"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
-where = ["tinyagent"]
+where = ["."]
+include = ["tinyagent*"]
 exclude = ["public", "public.*"]
 
 [project]
 name = "tinyagent-py"
-version = "0.0.4"
+version = "0.0.6"
 description = "Tiny Agent with MCP Client"
 readme = "README.md"
 authors = [

tinyagent_py-0.0.6/tinyagent/__init__.py

@@ -0,0 +1,4 @@
+from .tiny_agent import TinyAgent,tool
+from .mcp_client import MCPClient
+
+__all__ = ["TinyAgent", "MCPClient","tool"]

tinyagent_py-0.0.6/tinyagent/mcp_client.py

@@ -0,0 +1,162 @@
+import asyncio
+import json
+import logging
+from typing import Dict, List, Optional, Any, Tuple, Callable
+
+# Keep your MCPClient implementation unchanged
+import asyncio
+from contextlib import AsyncExitStack
+
+# MCP core imports
+from mcp import ClientSession, StdioServerParameters
+from mcp.client.stdio import stdio_client
+
+# Set up logging
+logger = logging.getLogger(__name__)
+
+class MCPClient:
+    def __init__(self, logger: Optional[logging.Logger] = None):
+        self.session = None
+        self.exit_stack = AsyncExitStack()
+        self.logger = logger or logging.getLogger(__name__)
+        
+        # Simplified callback system
+        self.callbacks: List[callable] = []
+        
+        self.logger.debug("MCPClient initialized")
+
+    def add_callback(self, callback: callable) -> None:
+        """
+        Add a callback function to the client.
+        
+        Args:
+            callback: A function that accepts (event_name, client, **kwargs)
+        """
+        self.callbacks.append(callback)
+    
+    async def _run_callbacks(self, event_name: str, **kwargs) -> None:
+        """
+        Run all registered callbacks for an event.
+        
+        Args:
+            event_name: The name of the event
+            **kwargs: Additional data for the event
+        """
+        for callback in self.callbacks:
+            try:
+                logger.debug(f"Running callback: {callback}")
+                if asyncio.iscoroutinefunction(callback):
+                    logger.debug(f"Callback is a coroutine function")
+                    await callback(event_name, self, **kwargs)
+                else:
+                    # Check if the callback is a class with an async __call__ method
+                    if hasattr(callback, '__call__') and asyncio.iscoroutinefunction(callback.__call__):
+                        logger.debug(f"Callback is a class with an async __call__ method")  
+                        await callback(event_name, self, **kwargs)
+                    else:
+                        logger.debug(f"Callback is a regular function")
+                        callback(event_name, self, **kwargs)
+            except Exception as e:
+                logger.error(f"Error in callback for {event_name}: {str(e)}")
+
+    async def connect(self, command: str, args: list[str]):
+        """
+        Launches the MCP server subprocess and initializes the client session.
+        :param command: e.g. "python" or "node"
+        :param args: list of args to pass, e.g. ["my_server.py"] or ["build/index.js"]
+        """
+        # Prepare stdio transport parameters
+        params = StdioServerParameters(command=command, args=args)
+        # Open the stdio client transport
+        self.stdio, self.sock_write = await self.exit_stack.enter_async_context(
+            stdio_client(params)
+        )
+        # Create and initialize the MCP client session
+        self.session = await self.exit_stack.enter_async_context(
+            ClientSession(self.stdio, self.sock_write)
+        )
+        await self.session.initialize()
+
+    async def list_tools(self):
+        resp = await self.session.list_tools()
+        print("Available tools:")
+        for tool in resp.tools:
+            print(f" • {tool.name}: {tool.description}")
+
+    async def call_tool(self, name: str, arguments: dict):
+        """
+        Invokes a named tool and returns its raw content list.
+        """
+        # Notify tool start
+        await self._run_callbacks("tool_start", tool_name=name, arguments=arguments)
+        
+        try:
+            resp = await self.session.call_tool(name, arguments)
+            
+            # Notify tool end
+            await self._run_callbacks("tool_end", tool_name=name, arguments=arguments, 
+                                    result=resp.content, success=True)
+            
+            return resp.content
+        except Exception as e:
+            # Notify tool end with error
+            await self._run_callbacks("tool_end", tool_name=name, arguments=arguments, 
+                                    error=str(e), success=False)
+            raise
+
+    async def close(self):
+        """Clean up subprocess and streams."""
+        if self.exit_stack:
+            try:
+                await self.exit_stack.aclose()
+            except (RuntimeError, asyncio.CancelledError) as e:
+                # Log the error but don't re-raise it
+                self.logger.error(f"Error during client cleanup: {e}")
+            finally:
+                # Always reset these regardless of success or failure
+                self.session = None
+                self.exit_stack = AsyncExitStack()
+
+async def run_example():
+    """Example usage of MCPClient with proper logging."""
+    import sys
+    from tinyagent.hooks.logging_manager import LoggingManager
+    
+    # Create and configure logging manager
+    log_manager = LoggingManager(default_level=logging.INFO)
+    log_manager.set_levels({
+        'tinyagent.mcp_client': logging.DEBUG,  # Debug for this module
+        'tinyagent.tiny_agent': logging.INFO,
+    })
+    
+    # Configure a console handler
+    console_handler = logging.StreamHandler(sys.stdout)
+    log_manager.configure_handler(
+        console_handler,
+        format_string='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        level=logging.DEBUG
+    )
+    
+    # Get module-specific logger
+    mcp_logger = log_manager.get_logger('tinyagent.mcp_client')
+    
+    mcp_logger.debug("Starting MCPClient example")
+    
+    # Create client with our logger
+    client = MCPClient(logger=mcp_logger)
+    
+    try:
+        # Connect to a simple echo server
+        await client.connect("python", ["-m", "mcp.examples.echo_server"])
+        
+        # List available tools
+        await client.list_tools()
+        
+        # Call the echo tool
+        result = await client.call_tool("echo", {"message": "Hello, MCP!"})
+        mcp_logger.info(f"Echo result: {result}")
+        
+    finally:
+        # Clean up
+        await client.close()
+        mcp_logger.debug("Example completed")