tinyagent-py 0.0.15__tar.gz → 0.0.16__tar.gz

{tinyagent_py-0.0.15 → tinyagent_py-0.0.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyagent-py
-Version: 0.0.15
+Version: 0.0.16
 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
@@ -274,6 +274,30 @@ agent = TinyCodeAgent(
 )
 ```
 
+### Automatic Git Checkpoints
+
+TinyCodeAgent can automatically create Git checkpoints after each successful shell command execution. This helps track changes made by the agent and provides a safety net for reverting changes if needed.
+
+```python
+# Enable automatic Git checkpoints during initialization
+agent = TinyCodeAgent(
+    model="gpt-4.1-mini",
+    auto_git_checkpoint=True  # Enable automatic Git checkpoints
+)
+
+# Or enable/disable it later
+agent.enable_auto_git_checkpoint(True)  # Enable
+agent.enable_auto_git_checkpoint(False)  # Disable
+
+# Check current status
+is_enabled = agent.get_auto_git_checkpoint_status()
+```
+
+Each checkpoint includes:
+- Descriptive commit message with the command description
+- Timestamp of when the command was executed
+- The actual command that was run
+
 For detailed documentation, see the [TinyCodeAgent README](tinyagent/code_agent/README.md).
 
 ## How the TinyAgent Hook System Works

{tinyagent_py-0.0.15 → tinyagent_py-0.0.16}/README.md

@@ -232,6 +232,30 @@ agent = TinyCodeAgent(
 )
 ```
 
+### Automatic Git Checkpoints
+
+TinyCodeAgent can automatically create Git checkpoints after each successful shell command execution. This helps track changes made by the agent and provides a safety net for reverting changes if needed.
+
+```python
+# Enable automatic Git checkpoints during initialization
+agent = TinyCodeAgent(
+    model="gpt-4.1-mini",
+    auto_git_checkpoint=True  # Enable automatic Git checkpoints
+)
+
+# Or enable/disable it later
+agent.enable_auto_git_checkpoint(True)  # Enable
+agent.enable_auto_git_checkpoint(False)  # Disable
+
+# Check current status
+is_enabled = agent.get_auto_git_checkpoint_status()
+```
+
+Each checkpoint includes:
+- Descriptive commit message with the command description
+- Timestamp of when the command was executed
+- The actual command that was run
+
 For detailed documentation, see the [TinyCodeAgent README](tinyagent/code_agent/README.md).
 
 ## How the TinyAgent Hook System Works

{tinyagent_py-0.0.15 → tinyagent_py-0.0.16}/pyproject.toml

@@ -12,7 +12,7 @@ tinyagent = ["prompts/*.yaml"]
 
 [project]
 name = "tinyagent-py"
-version = "0.0.15"
+version = "0.0.16"
 description = "TinyAgent with MCP Client, Code Agent (Thinking, Planning, and Executing in Python), and Extendable Hooks, Tiny but powerful"
 readme = "README.md"
 authors = [

tinyagent_py-0.0.16/tinyagent/code_agent/providers/__init__.py

@@ -0,0 +1,17 @@
+from .base import CodeExecutionProvider
+from .modal_provider import ModalProvider
+
+# Import SeatbeltProvider conditionally to avoid errors on non-macOS systems
+import platform
+if platform.system() == "Darwin":
+    try:
+        from .seatbelt_provider import SeatbeltProvider
+    except ImportError:
+        # If there's an issue importing, just don't make it available
+        pass
+
+__all__ = ["CodeExecutionProvider", "ModalProvider"]
+
+# Add SeatbeltProvider to __all__ if it was successfully imported
+if platform.system() == "Darwin" and "SeatbeltProvider" in globals():
+    __all__.append("SeatbeltProvider") 

{tinyagent_py-0.0.15 → tinyagent_py-0.0.16}/tinyagent/code_agent/providers/base.py

@@ -21,6 +21,9 @@ class CodeExecutionProvider(ABC):
         pip_packages: List[str] = None,
         secrets: Dict[str, Any] = None,
         lazy_init: bool = True,
+        bypass_shell_safety: bool = False,
+        additional_safe_shell_commands: Optional[List[str]] = None,
+        additional_safe_control_operators: Optional[List[str]] = None,
         **kwargs
     ):
         self.log_manager = log_manager
@@ -35,15 +38,36 @@ class CodeExecutionProvider(ABC):
         self._locals_dict = kwargs.get("locals_dict", {})
         self._user_variables = {}
         self.code_tools_definitions = []
+        
+        # Shell safety configuration
+        self.bypass_shell_safety = bypass_shell_safety
+        
         # Safe shell commands that don't modify the system or access sensitive data
         self.safe_shell_commands: Set[str] = {
             "ls", "cat", "grep", "find", "echo", "pwd", "whoami", "date", 
             "head", "tail", "wc", "sort", "uniq", "tr", "cut", "sed", "awk",
-            "ps", "df", "du", "uname", "which", "type", "file", "stat","rg","if",
+            "ps", "df", "du", "uname", "which", "type", "file", "stat", "rg", "if",
             "tree"
         }
+        
+        # Add additional safe shell commands if provided
+        if additional_safe_shell_commands:
+            if "*" in additional_safe_shell_commands:
+                # If wildcard is provided, allow all commands (effectively bypassing the check)
+                self.bypass_shell_safety = True
+            else:
+                self.safe_shell_commands.update(additional_safe_shell_commands)
+        
         # Safe control operators for shell commands
         self.safe_control_operators: Set[str] = {"&&", "||", ";", "|"}
+        
+        # Add additional safe control operators if provided
+        if additional_safe_control_operators:
+            if "*" in additional_safe_control_operators:
+                # If wildcard is provided, allow all operators
+                self.safe_control_operators = set("*")
+            else:
+                self.safe_control_operators.update(additional_safe_control_operators)
     
     @abstractmethod
     async def execute_python(
@@ -102,6 +126,10 @@ class CodeExecutionProvider(ABC):
             - safe: Boolean indicating if command is safe
             - reason: Reason why command is not safe (if applicable)
         """
+        # If shell safety checks are bypassed, consider all commands safe
+        if self.bypass_shell_safety:
+            return {"safe": True}
+            
         if type(command) == str:
             command = command.split(" ")
         if not command or not isinstance(command, list) or len(command) == 0:

{tinyagent_py-0.0.15 → tinyagent_py-0.0.16}/tinyagent/code_agent/providers/modal_provider.py

@@ -47,6 +47,9 @@ class ModalProvider(CodeExecutionProvider):
         sandbox_name: str = "tinycodeagent-sandbox",
         local_execution: bool = False,
         check_string_obfuscation: bool = True,
+        bypass_shell_safety: bool = False,  # Default to False for ModalProvider
+        additional_safe_shell_commands: Optional[List[str]] = None,
+        additional_safe_control_operators: Optional[List[str]] = None,
         **kwargs
     ):
         """
@@ -67,6 +70,9 @@ class ModalProvider(CodeExecutionProvider):
             sandbox_name: Name of the Modal sandbox
             local_execution: Whether to execute code locally
             check_string_obfuscation: If True (default), check for string obfuscation techniques. Set to False to allow legitimate use of base64 encoding and other string manipulations.
+            bypass_shell_safety: If True, bypass shell command safety checks (default: False for modal)
+            additional_safe_shell_commands: Additional shell commands to consider safe
+            additional_safe_control_operators: Additional shell control operators to consider safe
             **kwargs: Additional keyword arguments
         
         Note:
@@ -114,6 +120,9 @@ class ModalProvider(CodeExecutionProvider):
             pip_packages=final_packages,
             secrets=modal_secrets or {},
             lazy_init=lazy_init,
+            bypass_shell_safety=bypass_shell_safety,
+            additional_safe_shell_commands=additional_safe_shell_commands,
+            additional_safe_control_operators=additional_safe_control_operators,
             **kwargs
         )