tinyagent-py 0.0.8__py3-none-any.whl → 0.0.9__py3-none-any.whl

tinyagent/code_agent/utils.py

@@ -0,0 +1,120 @@
+import sys
+import cloudpickle
+from typing import Dict, Any
+
+
+def clean_response(resp: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Clean the response from code execution, keeping only relevant fields.
+    
+    Args:
+        resp: Raw response dictionary from code execution
+        
+    Returns:
+        Cleaned response with only essential fields
+    """
+    return {k: v for k, v in resp.items() if k in ['printed_output', 'return_value', 'stderr', 'error_traceback']}
+
+
+def make_session_blob(ns: dict) -> bytes:
+    """
+    Create a serialized blob of the session namespace, excluding unserializable objects.
+    
+    Args:
+        ns: Namespace dictionary to serialize
+        
+    Returns:
+        Serialized bytes of the clean namespace
+    """
+    clean = {}
+    for name, val in ns.items():
+        try:
+            # Try serializing just this one object
+            cloudpickle.dumps(val)
+        except Exception:
+            # drop anything that fails
+            continue
+        else:
+            clean[name] = val
+
+    return cloudpickle.dumps(clean)
+
+
+def _run_python(code: str, globals_dict: Dict[str, Any] = None, locals_dict: Dict[str, Any] = None):
+    """
+    Execute Python code in a controlled environment with proper error handling.
+    
+    Args:
+        code: Python code to execute
+        globals_dict: Global variables dictionary
+        locals_dict: Local variables dictionary
+        
+    Returns:
+        Dictionary containing execution results
+    """
+    import contextlib
+    import traceback
+    import io
+    import ast
+    
+    # Make copies to avoid mutating the original parameters
+    globals_dict = globals_dict or {}
+    locals_dict = locals_dict or {}
+    updated_globals = globals_dict.copy()
+    updated_locals = locals_dict.copy()
+    
+    # Pre-import essential modules into the global namespace
+    # This ensures they're available for imports inside functions
+    essential_modules = ['requests', 'json', 'os', 'sys', 'time', 'datetime', 're', 'random', 'math']
+    
+    for module_name in essential_modules:
+        try:
+            module = __import__(module_name)
+            updated_globals[module_name] = module
+            #print(f"✓ {module_name} module loaded successfully")
+        except ImportError:
+            print(f"⚠️  Warning: {module_name} module not available")
+    
+    tree = ast.parse(code, mode="exec")
+    compiled = compile(tree, filename="<ast>", mode="exec")
+    stdout_buf = io.StringIO()
+    stderr_buf = io.StringIO()
+
+    # Execute with stdout+stderr capture and exception handling
+    error_traceback = None
+    output = None
+
+    with contextlib.redirect_stdout(stdout_buf), contextlib.redirect_stderr(stderr_buf):
+        try:
+            # Merge all variables into globals to avoid scoping issues with generator expressions
+            # When exec() is called with both globals and locals, generator expressions can't
+            # access local variables. By using only globals, everything runs in global scope.
+            merged_globals = updated_globals.copy()
+            merged_globals.update(updated_locals)
+            
+            # Execute with only globals - this fixes generator expression scoping issues
+            output = exec(code, merged_globals)
+            
+            # Update both dictionaries with any new variables created during execution
+            for key, value in merged_globals.items():
+                if key not in updated_globals and key not in updated_locals:
+                    updated_locals[key] = value
+                elif key in updated_locals or key not in updated_globals:
+                    updated_locals[key] = value
+                updated_globals[key] = value
+        except Exception:
+            # Capture the full traceback as a string
+            error_traceback = traceback.format_exc()
+
+    printed_output = stdout_buf.getvalue()
+    stderr_output = stderr_buf.getvalue()
+    error_traceback_output = error_traceback
+
+    return {
+        "printed_output": printed_output, 
+        "return_value": output, 
+        "stderr": stderr_output, 
+        "error_traceback": error_traceback_output,
+        "updated_globals": updated_globals,
+        "updated_locals": updated_locals
+    } 

tinyagent/hooks/__init__.py

@@ -1,4 +1,5 @@
 #from .rich_ui_agent import RichUICallback
 from .rich_ui_callback import RichUICallback
+from .rich_code_ui_callback import RichCodeUICallback
 from .logging_manager import LoggingManager
-__all__ = ["RichUICallback", "LoggingManager"]
+__all__ = ["RichUICallback", "RichCodeUICallback", "LoggingManager"]

{tinyagent_py-0.0.8.dist-info → tinyagent_py-0.0.9.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: tinyagent-py
-Version: 0.0.8
-Summary: Tiny Agent with MCP Client and Extendable Hooks, Tiny but powerful
+Version: 0.0.9
+Summary: TinyAgent with MCP Client, Code Agent (Thinking, Planning, and Executing in Python), and Extendable Hooks, Tiny but powerful
 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
@@ -25,10 +25,19 @@ Provides-Extra: sqlite
 Requires-Dist: aiosqlite>=0.18.0; extra == "sqlite"
 Provides-Extra: gradio
 Requires-Dist: gradio>=3.50.0; extra == "gradio"
+Provides-Extra: code
+Requires-Dist: jinja2; extra == "code"
+Requires-Dist: pyyaml; extra == "code"
+Requires-Dist: cloudpickle; extra == "code"
+Requires-Dist: modal; extra == "code"
 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"
+Requires-Dist: jinja2; extra == "all"
+Requires-Dist: pyyaml; extra == "all"
+Requires-Dist: cloudpickle; extra == "all"
+Requires-Dist: modal; extra == "all"
 Dynamic: license-file
 
 # TinyAgent
@@ -52,7 +61,11 @@ Inspired by:
 - [Build your own Tiny Agent](https://askdev.ai/github/askbudi/tinyagent)
 
 ## Overview
-This is a tiny agent that uses MCP and LiteLLM to interact with a model. You have full control over the agent, you can add any tools you like from MCP and extend the agent using its event system.
+This is a tiny agent framework that uses MCP and LiteLLM to interact with language models. You have full control over the agent, you can add any tools you like from MCP and extend the agent using its event system.
+
+**Two Main Components:**
+- **TinyAgent**: Core agent with MCP tool integration and extensible hooks
+- **TinyCodeAgent**: Specialized agent for secure Python code execution with pluggable providers
 
 ## Installation
 
@@ -64,6 +77,10 @@ pip install tinyagent-py
 # Install with all optional dependencies
 pip install tinyagent-py[all]
 
+# Install with Code Agent support
+pip install tinyagent-py[code]
+
+
 # Install with PostgreSQL support
 pip install tinyagent-py[postgres]
 
@@ -73,6 +90,10 @@ pip install tinyagent-py[sqlite]
 # Install with Gradio UI support
 pip install tinyagent-py[gradio]
 
+
+
+
+
 ```
 
 ### Using uv
@@ -80,6 +101,10 @@ pip install tinyagent-py[gradio]
 # Basic installation
 uv pip install tinyagent-py
 
+# Install with Code Agent support
+uv pip install tinyagent-py[code]
+
+
 # Install with PostgreSQL support
 uv pip install tinyagent-py[postgres]
 
@@ -92,11 +117,11 @@ 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
+
+### TinyAgent (Core Agent)
 [![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)
 
 
@@ -133,6 +158,114 @@ I need accommodation in Toronto between 15th to 20th of May. Give me 5 options f
 await test_agent(task, model="gpt-4.1-mini")
 ```
 
+## TinyCodeAgent - Code Execution Made Easy
+
+TinyCodeAgent is a specialized agent for executing Python code with enterprise-grade reliability and extensible execution providers.
+
+### Quick Start with TinyCodeAgent
+
+```python
+import asyncio
+from tinyagent import TinyCodeAgent
+
+async def main():
+    # Initialize with minimal configuration
+    agent = TinyCodeAgent(
+        model="gpt-4.1-mini",
+        api_key="your-openai-api-key"
+    )
+    
+    try:
+        # Ask the agent to solve a coding problem
+        result = await agent.run("Calculate the factorial of 10 and explain the algorithm")
+        print(result)
+    finally:
+        await agent.close()
+
+asyncio.run(main())
+```
+
+### TinyCodeAgent with Gradio UI
+
+Launch a complete web interface for interactive code execution:
+
+```python
+from tinyagent.code_agent.example import run_example
+import asyncio
+
+# Run the full example with Gradio interface
+asyncio.run(run_example())
+```
+
+### Key Features
+
+- **🔒 Secure Execution**: Sandboxed Python code execution using Modal.com or other providers
+- **🔧 Extensible Providers**: Switch between Modal, Docker, local execution, or cloud functions
+- **🎯 Built for Enterprise**: Production-ready with proper logging, error handling, and resource cleanup  
+- **📁 File Support**: Upload and process files through the Gradio interface
+- **🛠️ Custom Tools**: Add your own tools and functions easily
+- **📊 Session Persistence**: Code state persists across executions
+
+### Provider System
+
+TinyCodeAgent uses a pluggable provider system - change execution backends with minimal code changes:
+
+```python
+# Use Modal (default) - great for production
+agent = TinyCodeAgent(provider="modal")
+
+# Future providers (coming soon)
+# agent = TinyCodeAgent(provider="docker")
+# agent = TinyCodeAgent(provider="local") 
+# agent = TinyCodeAgent(provider="lambda")
+```
+
+### Example Use Cases
+
+**Web Scraping:**
+```python
+result = await agent.run("""
+What are trending spaces on huggingface today?
+""")
+# Agent will create a python tool to request HuggingFace API and find trending spaces
+```
+
+**Use code to solve a task:**
+```python
+response = await agent.run(dedent("""
+Suggest me 13 tags for my Etsy Listing, each tag should be multiworded and maximum 20 characters. Each word should be used only once in the whole corpus, And tags should cover different ways people are searching for the product on Etsy.
+- You should use your coding abilities to check your answer pass the criteria and continue your job until you get to the answer.
+                                
+My Product is **Wedding Invitation Set of 3, in sage green color, with a gold foil border.**
+"""),max_turns=20)
+
+print(response)
+# LLM is not good at this task, counting characters, avoid duplicates, but with the power of code, tiny model like gpt-4.1-mini can do it without any problem.
+```
+
+
+### Configuration Options
+
+```python
+from tinyagent import TinyCodeAgent
+from tinyagent.code_agent.tools import get_weather, get_traffic
+
+# Full configuration example
+agent = TinyCodeAgent(
+    model="gpt-4.1-mini",
+    api_key="your-api-key", 
+    provider="modal",
+    tools=[get_weather, get_traffic],
+    authorized_imports=["requests", "pandas", "numpy"],
+    provider_config={
+        "pip_packages": ["requests", "pandas"],
+        "sandbox_name": "my-code-sandbox"
+    }
+)
+```
+
+For detailed documentation, see the [TinyCodeAgent README](tinyagent/code_agent/README.md).
+
 ## How the TinyAgent Hook System Works
 
 TinyAgent is designed to be **extensible** via a simple, event-driven hook (callback) system. This allows you to add custom logic, logging, UI, memory, or any other behavior at key points in the agent's lifecycle.

tinyagent_py-0.0.9.dist-info/RECORD

@@ -0,0 +1,31 @@
+tinyagent/__init__.py,sha256=-3ZN8unMZDrA366BET1HKp-fnFCyXCAD1fPVbHkJSsY,172
+tinyagent/mcp_client.py,sha256=9dmLtJ8CTwKWKTH6K9z8CaCQuaicOH9ifAuNyX7Kdo0,6030
+tinyagent/memory_manager.py,sha256=tAaZZdxBJ235wJIyr04n3f2Damok4s2UXunTtur_p-4,44916
+tinyagent/tiny_agent.py,sha256=_qHQc2yroSI1Ihn5Ex3FH_ml7aaSNC9-yR540C16Cxs,41125
+tinyagent/code_agent/__init__.py,sha256=YSOblSwRS1QcAYUu--GvF4fKeQX1KRTj9P8CWySY3pY,327
+tinyagent/code_agent/example.py,sha256=qC6i3auUT1YwXS9WK1Ovq-9oDOUgzRxDegYdlVcVcfA,5861
+tinyagent/code_agent/helper.py,sha256=oZnpo-_H3cB12LxNN7Ztd-31EiUcuI2UpWP69xuF8oE,7205
+tinyagent/code_agent/modal_sandbox.py,sha256=3FaTLESkdQmEO9A5dMmMOOkd0w2cnGGpfXngvojQxJ8,16317
+tinyagent/code_agent/tiny_code_agent.py,sha256=D0stvCao0dvrpY2ChxGxf048SEUDW-yHgJ_9Aluw2Ek,23142
+tinyagent/code_agent/utils.py,sha256=rEnrKv9l8knjUroBQhz6Q8q6ehCt9IAsJtM-m3ZMNjY,4231
+tinyagent/code_agent/providers/__init__.py,sha256=myfy9qsBDjNOhcgXJ2E9jO1q5eo6jHp43I2k0k8esLY,136
+tinyagent/code_agent/providers/base.py,sha256=Hm8jrD60QovgQHTwiFE1pKDHPk1cwGbUdzuSwic9Rjc,5832
+tinyagent/code_agent/providers/modal_provider.py,sha256=myq0WKiNi8G0MpUUGo7olbGFVmcZ3iMa8XTmT537PZg,8576
+tinyagent/code_agent/tools/__init__.py,sha256=0XtrgYBgBayOffW50KyrlmrXXs9iu6z1DHu7-D8WGqY,94
+tinyagent/code_agent/tools/example_tools.py,sha256=YbXb7PKuvvxh-LV12Y4n_Ez3RyLA95gWOcZrKsa7UHg,1203
+tinyagent/hooks/__init__.py,sha256=RZow2r0XHLJ3-tnmecScdc0_wrEdmOy5dtXqoiRME5Y,254
+tinyagent/hooks/gradio_callback.py,sha256=TS28i4UUIEWsl_e75tRXyVTOKzH7at8nmIl8d3ZTYag,53493
+tinyagent/hooks/logging_manager.py,sha256=UpdmpQ7HRPyer-jrmQSXcBwi409tV9LnGvXSHjTcYTI,7935
+tinyagent/hooks/rich_code_ui_callback.py,sha256=PLcu5MOSoP4oZR3BtvcV9DquxcIT_d0WzSlkvaDcGOk,19820
+tinyagent/hooks/rich_ui_callback.py,sha256=5iCNOiJmhc1lOL7ZjaOt5Sk3rompko4zu_pAxfTVgJQ,22897
+tinyagent/storage/__init__.py,sha256=7qwfdD4smCl891xaRuiReSUgfOJFy7jJZsN0ul1iQdY,173
+tinyagent/storage/base.py,sha256=GGAMvOoslmm1INLFG_jtwOkRk2Qg39QXx-1LnN7fxDI,1474
+tinyagent/storage/json_file_storage.py,sha256=SYD8lvTHu2-FEHm1tZmsrcgEOirBrlUsUM186X-UPgI,1114
+tinyagent/storage/postgres_storage.py,sha256=IGwan8UXHNnTZFK1F8x4kvMDex3GAAGWUg9ePx_5IF4,9018
+tinyagent/storage/redis_storage.py,sha256=hu3y7wHi49HkpiR-AW7cWVQuTVOUk1WaB8TEPGUKVJ8,1742
+tinyagent/storage/sqlite_storage.py,sha256=ZyOYe0d_oHO1wOIT8FxKIbc67tP_0e_8FnM2Zq8Pwj8,5915
+tinyagent_py-0.0.9.dist-info/licenses/LICENSE,sha256=YIogcVQnknaaE4K-oaQylFWo8JGRBWnwmGb3fWB_Pww,1064
+tinyagent_py-0.0.9.dist-info/METADATA,sha256=5yLZvD5bKtQWg0cORuKGme3XaOy_Cnz7zHx-6QUwJpA,13249
+tinyagent_py-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tinyagent_py-0.0.9.dist-info/top_level.txt,sha256=Ny8aJNchZpc2Vvhp3306L5vjceJakvFxBk-UjjVeA_I,10
+tinyagent_py-0.0.9.dist-info/RECORD,,

tinyagent_py-0.0.8.dist-info/RECORD

@@ -1,20 +0,0 @@
-tinyagent/__init__.py,sha256=GrD21npMQGzl9ZYKYTP8VxHLzCfJCvA0oTKQZTkmnCw,117
-tinyagent/mcp_client.py,sha256=9dmLtJ8CTwKWKTH6K9z8CaCQuaicOH9ifAuNyX7Kdo0,6030
-tinyagent/memory_manager.py,sha256=tAaZZdxBJ235wJIyr04n3f2Damok4s2UXunTtur_p-4,44916
-tinyagent/tiny_agent.py,sha256=_qHQc2yroSI1Ihn5Ex3FH_ml7aaSNC9-yR540C16Cxs,41125
-tinyagent/hooks/__init__.py,sha256=UztCHjoqF5JyDolbWwkBsBZkWguDQg23l2GD_zMHt-s,178
-tinyagent/hooks/gradio_callback.py,sha256=TS28i4UUIEWsl_e75tRXyVTOKzH7at8nmIl8d3ZTYag,53493
-tinyagent/hooks/logging_manager.py,sha256=UpdmpQ7HRPyer-jrmQSXcBwi409tV9LnGvXSHjTcYTI,7935
-tinyagent/hooks/rich_code_ui_callback.py,sha256=PLcu5MOSoP4oZR3BtvcV9DquxcIT_d0WzSlkvaDcGOk,19820
-tinyagent/hooks/rich_ui_callback.py,sha256=5iCNOiJmhc1lOL7ZjaOt5Sk3rompko4zu_pAxfTVgJQ,22897
-tinyagent/storage/__init__.py,sha256=7qwfdD4smCl891xaRuiReSUgfOJFy7jJZsN0ul1iQdY,173
-tinyagent/storage/base.py,sha256=GGAMvOoslmm1INLFG_jtwOkRk2Qg39QXx-1LnN7fxDI,1474
-tinyagent/storage/json_file_storage.py,sha256=SYD8lvTHu2-FEHm1tZmsrcgEOirBrlUsUM186X-UPgI,1114
-tinyagent/storage/postgres_storage.py,sha256=IGwan8UXHNnTZFK1F8x4kvMDex3GAAGWUg9ePx_5IF4,9018
-tinyagent/storage/redis_storage.py,sha256=hu3y7wHi49HkpiR-AW7cWVQuTVOUk1WaB8TEPGUKVJ8,1742
-tinyagent/storage/sqlite_storage.py,sha256=ZyOYe0d_oHO1wOIT8FxKIbc67tP_0e_8FnM2Zq8Pwj8,5915
-tinyagent_py-0.0.8.dist-info/licenses/LICENSE,sha256=YIogcVQnknaaE4K-oaQylFWo8JGRBWnwmGb3fWB_Pww,1064
-tinyagent_py-0.0.8.dist-info/METADATA,sha256=NW-v0Ra2BWiKlI26hYma4lMaHxegGfkFwlR-2C04Qmg,9094
-tinyagent_py-0.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tinyagent_py-0.0.8.dist-info/top_level.txt,sha256=Ny8aJNchZpc2Vvhp3306L5vjceJakvFxBk-UjjVeA_I,10
-tinyagent_py-0.0.8.dist-info/RECORD,,