tinyagent-py 0.0.15__py3-none-any.whl → 0.0.16__py3-none-any.whl

tinyagent/code_agent/providers/__init__.py

@@ -1,4 +1,17 @@
 from .base import CodeExecutionProvider
 from .modal_provider import ModalProvider
 
-__all__ = ["CodeExecutionProvider", "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/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/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
         )