tinyagent-py 0.0.12__py3-none-any.whl → 0.0.13__py3-none-any.whl

tinyagent/code_agent/providers/base.py

@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 from typing import Dict, List, Any, Optional
 from tinyagent.hooks.logging_manager import LoggingManager
+import cloudpickle
 
 
 class CodeExecutionProvider(ABC):
@@ -69,8 +70,6 @@ class CodeExecutionProvider(ABC):
         Args:
             tools: List of tool objects to add
         """
-        import cloudpickle
-        
         tools_str_list = ["import cloudpickle"]
         tools_str_list.append("###########<tools>###########\n")
         for tool in tools:
@@ -82,6 +81,22 @@ class CodeExecutionProvider(ABC):
         tools_str_list.append("\n\n")
         self.code_tools_definitions.extend(tools_str_list)
     
+    def set_code_tools(self, tools: List[Any]) -> None:
+        """
+        Set the code tools available in the execution environment.
+        Replaces any existing tools with the new list.
+        
+        Args:
+            tools: List of tool objects to set
+        """
+        # Clear existing tools
+        self.code_tools = tools.copy()
+        self.code_tools_definitions = []
+        
+        # Add the new tools
+        if tools:
+            self.add_tools(tools)
+    
     def set_user_variables(self, variables: Dict[str, Any]) -> None:
         """
         Set user variables that will be available in the Python environment.
@@ -89,8 +104,6 @@ class CodeExecutionProvider(ABC):
         Args:
             variables: Dictionary of variable name -> value pairs
         """
-        import cloudpickle
-        
         self._user_variables = variables.copy()
         
         # Add variables to the execution environment by serializing them
@@ -149,4 +162,46 @@ class CodeExecutionProvider(ABC):
         Returns:
             Dictionary of current user variables
         """
-        return self._user_variables.copy() 
+        return self._user_variables.copy()
+    
+    def update_user_variables_from_globals(self, globals_dict: Dict[str, Any]) -> None:
+        """
+        Extract and update user variables from the globals dictionary after code execution.
+        This ensures that any modifications to user variables during code execution are preserved.
+        
+        Args:
+            globals_dict: The globals dictionary after code execution
+        """
+        if not globals_dict or not self._user_variables:
+            return
+            
+        # Update user variables with values from globals
+        for var_name in list(self._user_variables.keys()):
+            if var_name in globals_dict:
+                try:
+                    # Try to serialize the value to ensure it's valid
+                    cloudpickle.dumps(globals_dict[var_name])
+                    # Update the user variable with the new value
+                    self._user_variables[var_name] = globals_dict[var_name]
+                except Exception:
+                    # If serialization fails, keep the old value
+                    pass
+                    
+        # Check for new variables that might have been created
+        # This handles cases where LLM creates new variables that should be preserved
+        for var_name, var_value in globals_dict.items():
+            # Skip special variables, modules, and functions
+            if (var_name.startswith('__') or 
+                var_name in ['builtins', 'cloudpickle'] or
+                callable(var_value) or
+                var_name in self._user_variables):
+                continue
+                
+            try:
+                # Try to serialize the value to ensure it's valid
+                cloudpickle.dumps(var_value)
+                # Add the new variable to user variables
+                self._user_variables[var_name] = var_value
+            except Exception:
+                # If serialization fails, skip this variable
+                pass 

tinyagent/code_agent/providers/modal_provider.py

@@ -143,17 +143,28 @@ class ModalProvider(CodeExecutionProvider):
         print(full_code)
         print("#" * 100)
 
-
         
         # Use Modal's native execution methods
         response = self._python_executor(full_code, self._globals_dict, self._locals_dict)
         
         print("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!<response>!!!!!!!!!!!!!!!!!!!!!!!!!")
         
-        # Update the instance globals and locals with the execution results
-        self._globals_dict = cloudpickle.loads(make_session_blob(response["updated_globals"]))
-        self._locals_dict = cloudpickle.loads(make_session_blob(response["updated_locals"]))
-
+        # Always update globals and locals dictionaries, regardless of whether there was an error
+        # This ensures variables are preserved even when code execution fails
+        try:
+            # Update globals and locals from the response
+            if "updated_globals" in response:
+                self._globals_dict = cloudpickle.loads(make_session_blob(response["updated_globals"]))
+                
+            if "updated_locals" in response:
+                self._locals_dict = cloudpickle.loads(make_session_blob(response["updated_locals"]))
+                
+            # Update user variables from the updated globals and locals
+            # This preserves any changes made to variables by the LLM
+            self.update_user_variables_from_globals(self._globals_dict)
+            self.update_user_variables_from_globals(self._locals_dict)
+        except Exception as e:
+            print(f"Warning: Failed to update globals/locals after execution: {str(e)}")
         
         self._log_response(response)
         

tinyagent/code_agent/tiny_code_agent.py

@@ -261,7 +261,16 @@ class TinyCodeAgent:
         async def run_python(code_lines: List[str], timeout: int = 120) -> str:
             """Execute Python code using the configured provider."""
             try:
+                # Before execution, ensure provider has the latest user variables
+                if self.user_variables:
+                    self.code_provider.set_user_variables(self.user_variables)
+                    
                 result = await self.code_provider.execute_python(code_lines, timeout)
+                
+                # After execution, update TinyCodeAgent's user_variables from the provider
+                # This ensures they stay in sync
+                self.user_variables = self.code_provider.get_user_variables()
+                
                 return str(result)
             except Exception as e:
                 print("!"*100)
@@ -272,6 +281,11 @@ class TinyCodeAgent:
                 print(f"{COLOR['RED']}{str(e)}{COLOR['ENDC']}")
                 print(f"{COLOR['RED']}{traceback.format_exc()}{COLOR['ENDC']}")
                 print("!"*100)
+                
+                # Even after an exception, update user_variables from the provider
+                # This ensures any variables that were successfully created/modified are preserved
+                self.user_variables = self.code_provider.get_user_variables()
+                
                 return f"Error executing code: {str(e)}"
         
         self.agent.add_tool(run_python)

tinyagent/code_agent/utils.py

@@ -116,22 +116,34 @@ def _run_python(
     #updated_globals['print'] = custom_print
     
     # Parse the code
-    tree = ast.parse(code, mode="exec")
-    compiled = compile(tree, filename="<ast>", mode="exec")
+    try:
+        tree = ast.parse(code, mode="exec")
+        compiled = compile(tree, filename="<ast>", mode="exec")
+    except SyntaxError as e:
+        # Return syntax error without executing
+        return {
+            "printed_output": "", 
+            "return_value": None, 
+            "stderr": "", 
+            "error_traceback": f"Syntax error: {str(e)}",
+            "updated_globals": updated_globals,
+            "updated_locals": updated_locals
+        }
+    
     stdout_buf = io.StringIO()
     stderr_buf = io.StringIO()   
     # Execute with exception handling
     error_traceback = None
     output = None
 
+    # 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)
+
     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)
-            
             # Add 'exec' to authorized_functions for internal use
             internal_authorized_functions = ['exec','eval']
             if authorized_functions is not None and not isinstance(authorized_functions, bool):
@@ -152,6 +164,18 @@ def _run_python(
         except Exception:
             # Capture the full traceback as a string
             error_traceback = traceback.format_exc()
+            
+            # CRITICAL FIX: Even when an exception occurs, we need to update the globals and locals
+            # with any variables that were successfully created/modified before the exception
+            for key, value in merged_globals.items():
+                # Skip special variables and modules
+                if key.startswith('__') or key in ['builtins', 'traceback', 'contextlib', 'io', 'ast', 'sys']:
+                    continue
+                    
+                # Update both dictionaries with the current state
+                if key in updated_locals or key not in updated_globals:
+                    updated_locals[key] = value
+                updated_globals[key] = value
 
     # Join all captured output
     #printed_output = ''.join(output_buffer)  

tinyagent/hooks/gradio_callback.py

@@ -845,8 +845,9 @@ class GradioCallback:
 
                     # Footer
                     gr.Markdown(
-                        "<div style='text-align: center; margin-top: 20px;'>"
-                        "Powered by <a href='https://github.com/askbudi/tinyagent' target='_blank'>TinyAgent</a>"
+                         "<div style='text-align: center; margin-top: 20px;'>"
+                        "Built with ❤️ by <a href='https://github.com/askbudi/tinyagent' target='_blank'>TinyAgent</a>"
+                        "<br>Start building your own AI agents with TinyAgent"
                         "</div>"
                     )
 

{tinyagent_py-0.0.12.dist-info → tinyagent_py-0.0.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyagent-py
-Version: 0.0.12
+Version: 0.0.13
 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
@@ -60,6 +60,16 @@ Inspired by:
 ## Quick Links
 - [Build your own Tiny Agent](https://askdev.ai/github/askbudi/tinyagent)
 
+
+## Live Projects using TinyAgent (🔥)
+- [AskDev.AI](https://askdev.ai) - Understand, chat, and summarize codebase of any project on GitHub.
+- [HackBuddy AI](https://huggingface.co/spaces/ask-dev/HackBuddyAI) - A Hackathon Assistant Agent, built with TinyCodeAgent and Gradio. Match invdividuals to teams based on their skills, interests and organizer preferences.
+
+- [TinyCodeAgent Demo](https://huggingface.co/spaces/ask-dev/TinyCodeAgent) - A playground for TinyCodeAgent, built with tinyagent, Gradio and Modal.com
+
+** Building something with TinyAgent? Let us know and I'll add it here!**
+
+
 ## Overview
 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.
 

{tinyagent_py-0.0.12.dist-info → tinyagent_py-0.0.13.dist-info}/RECORD

@@ -7,15 +7,15 @@ tinyagent/code_agent/example.py,sha256=qC6i3auUT1YwXS9WK1Ovq-9oDOUgzRxDegYdlVcVc
 tinyagent/code_agent/helper.py,sha256=oZnpo-_H3cB12LxNN7Ztd-31EiUcuI2UpWP69xuF8oE,7205
 tinyagent/code_agent/modal_sandbox.py,sha256=RcQ5a-UFyqV7xSHnttpgAQQ-mNWk-9Z0tb836ua7C0E,16381
 tinyagent/code_agent/safety.py,sha256=WHad2ypzfsdKnXG6FcgXcgGiMC-H4KTmOzhP9S9i3Zw,22209
-tinyagent/code_agent/tiny_code_agent.py,sha256=r3c_irn5RJp9qNewhlOLFaGGTAjLWqHjTwM8QewN54I,27426
-tinyagent/code_agent/utils.py,sha256=CAg8LKc_sxnXmwF0L2nRP13XsftYl3ZiD5qWO5D7QeQ,6842
+tinyagent/code_agent/tiny_code_agent.py,sha256=UrNjmJSrDC493bXGARrt0tDd1Bn56FgNlxuQSu4J614,28194
+tinyagent/code_agent/utils.py,sha256=FxHnpb06S2o2-xPRHgf9zAnzbXvGWs5QApNI4DEH__U,7870
 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=4uhmDA2pLYbbBi3actrBkHD_-pt3gt2incKBj12hab0,10084
+tinyagent/code_agent/providers/base.py,sha256=LfmahpulNbnivn5m91GTAo6ityjidq05dC3qx9EtJ80,8203
+tinyagent/code_agent/providers/modal_provider.py,sha256=R0qt8XlTMFMbznMHN32pQxupDE9KR18NpQ3l1wJJP0w,10799
 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=3vKfGknn7XOHtcoF6DkLKDCJmC_oCRhu1-kqWmAAOc4,56821
+tinyagent/hooks/gradio_callback.py,sha256=78x2x9AbYoLV-qwCxn2sH4s39DLlhNCzL7qCkVR-vy4,56911
 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
@@ -26,8 +26,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=ZyOYe0d_oHO1wOIT8FxKIbc67tP_0e_8FnM2Zq8Pwj8,5915
-tinyagent_py-0.0.12.dist-info/licenses/LICENSE,sha256=YIogcVQnknaaE4K-oaQylFWo8JGRBWnwmGb3fWB_Pww,1064
-tinyagent_py-0.0.12.dist-info/METADATA,sha256=Q05U9Z6UJNPoHtmAbpMzxB2HfcnQbGOHHz3W1EfgWt0,13250
-tinyagent_py-0.0.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tinyagent_py-0.0.12.dist-info/top_level.txt,sha256=Ny8aJNchZpc2Vvhp3306L5vjceJakvFxBk-UjjVeA_I,10
-tinyagent_py-0.0.12.dist-info/RECORD,,
+tinyagent_py-0.0.13.dist-info/licenses/LICENSE,sha256=YIogcVQnknaaE4K-oaQylFWo8JGRBWnwmGb3fWB_Pww,1064
+tinyagent_py-0.0.13.dist-info/METADATA,sha256=qr6akWcOHEZNUtQ34H_XalRWtJD-mK6WLiwmlcTH_N0,13848
+tinyagent_py-0.0.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tinyagent_py-0.0.13.dist-info/top_level.txt,sha256=Ny8aJNchZpc2Vvhp3306L5vjceJakvFxBk-UjjVeA_I,10
+tinyagent_py-0.0.13.dist-info/RECORD,,