tinyagent-py 0.0.3__tar.gz → 0.0.5__tar.gz

{tinyagent_py-0.0.3 → tinyagent_py-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyagent-py
-Version: 0.0.3
+Version: 0.0.5
 Summary: Tiny Agent with MCP Client
 Author-email: Mahdi Golchin <golchin@askdev.ai>
 Project-URL: Homepage, https://github.com/askbudi/tinyagent
@@ -12,15 +12,36 @@ License-File: LICENSE
 Requires-Dist: mcp
 Requires-Dist: litellm
 Requires-Dist: openai
+Requires-Dist: tiktoken
+Requires-Dist: uuid
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Provides-Extra: postgres
+Requires-Dist: asyncpg>=0.27.0; extra == "postgres"
+Provides-Extra: sqlite
+Requires-Dist: aiosqlite>=0.18.0; extra == "sqlite"
+Provides-Extra: gradio
+Requires-Dist: gradio>=3.50.0; extra == "gradio"
+Provides-Extra: all
+Requires-Dist: asyncpg>=0.27.0; extra == "all"
+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)
@@ -37,15 +58,47 @@ This is a tiny agent that uses MCP and LiteLLM to interact with a model. You hav
 
 ### Using pip
 ```bash
+# Basic installation
 pip install tinyagent-py
+
+# Install with all optional dependencies
+pip install tinyagent-py[all]
+
+# Install with PostgreSQL support
+pip install tinyagent-py[postgres]
+
+# Install with SQLite support
+pip install tinyagent-py[sqlite]
+
+# Install with Gradio UI support
+pip install tinyagent-py[gradio]
+
 ```
 
 ### Using uv
 ```bash
+# Basic installation
 uv pip install tinyagent-py
+
+# Install with PostgreSQL support
+uv pip install tinyagent-py[postgres]
+
+# Install with SQLite support
+uv pip install tinyagent-py[sqlite]
+
+# Install with Gradio UI support
+uv pip install tinyagent-py[gradio]
+
+# Install with all optional dependencies
+uv pip install tinyagent-py[all]
+
+# Install with development tools
+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
@@ -158,7 +211,7 @@ The `GradioCallback` hook lets you spin up a full-featured web chat interface fo
 Features:
 - **Browser-based chat** with streaming updates  
 - **File uploads** (\*.pdf, \*.docx, \*.txt) that the agent can reference  
-- **Live “thinking” view** so you see intermediate thoughts  
+- **Live "thinking" view** so you see intermediate thoughts  
 - **Collapsible tool-call sections** showing inputs & outputs  
 - **Real-time token usage** (prompt, completion, total)  
 - **Toggleable display options** for thinking & tool calls  
@@ -194,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.3/tinyagent/tinyagent_py.egg-info/PKG-INFO → tinyagent_py-0.0.5/README.md

@@ -1,26 +1,14 @@
-Metadata-Version: 2.4
-Name: tinyagent-py
-Version: 0.0.3
-Summary: Tiny Agent with MCP Client
-Author-email: Mahdi Golchin <golchin@askdev.ai>
-Project-URL: Homepage, https://github.com/askbudi/tinyagent
-Project-URL: Bug Tracker, https://github.com/askbudi/tinyagent/issues
-Project-URL: Chat, https://askdev.ai/github/askbudi/tinyagent
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: mcp
-Requires-Dist: litellm
-Requires-Dist: openai
-Provides-Extra: dev
-Requires-Dist: pytest; extra == "dev"
-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)
@@ -37,15 +25,47 @@ This is a tiny agent that uses MCP and LiteLLM to interact with a model. You hav
 
 ### Using pip
 ```bash
+# Basic installation
 pip install tinyagent-py
+
+# Install with all optional dependencies
+pip install tinyagent-py[all]
+
+# Install with PostgreSQL support
+pip install tinyagent-py[postgres]
+
+# Install with SQLite support
+pip install tinyagent-py[sqlite]
+
+# Install with Gradio UI support
+pip install tinyagent-py[gradio]
+
 ```
 
 ### Using uv
 ```bash
+# Basic installation
 uv pip install tinyagent-py
+
+# Install with PostgreSQL support
+uv pip install tinyagent-py[postgres]
+
+# Install with SQLite support
+uv pip install tinyagent-py[sqlite]
+
+# Install with Gradio UI support
+uv pip install tinyagent-py[gradio]
+
+# Install with all optional dependencies
+uv pip install tinyagent-py[all]
+
+# Install with development tools
+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
@@ -158,7 +178,7 @@ The `GradioCallback` hook lets you spin up a full-featured web chat interface fo
 Features:
 - **Browser-based chat** with streaming updates  
 - **File uploads** (\*.pdf, \*.docx, \*.txt) that the agent can reference  
-- **Live “thinking” view** so you see intermediate thoughts  
+- **Live "thinking" view** so you see intermediate thoughts  
 - **Collapsible tool-call sections** showing inputs & outputs  
 - **Real-time token usage** (prompt, completion, total)  
 - **Toggleable display options** for thinking & tool calls  
@@ -194,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.3 → tinyagent_py-0.0.5}/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.3"
+version = "0.0.5"
 description = "Tiny Agent with MCP Client"
 readme = "README.md"
 authors = [
@@ -18,15 +19,32 @@ requires-python = ">=3.8"
 dependencies = [
   "mcp",
   "litellm",
-  "openai"
+  "openai",
+  "tiktoken",
+  "uuid",
   # add whatever else you need…
 ]
 
 [project.optional-dependencies]
 dev = [
   "pytest",
-  
-  
+  "black",
+  "isort",
+  "mypy"
+]
+postgres = [
+  "asyncpg>=0.27.0"
+]
+sqlite = [
+  "aiosqlite>=0.18.0"
+]
+gradio = [
+  "gradio>=3.50.0"
+]
+all = [
+  "asyncpg>=0.27.0",
+  "aiosqlite>=0.18.0",
+  "gradio>=3.50.0"
 ]
 [project.urls]
 "Homepage" = "https://github.com/askbudi/tinyagent"

tinyagent_py-0.0.5/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.5/tinyagent/mcp_client.py

@@ -0,0 +1,167 @@
+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.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()
+            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")