sdk/python: Support execution environments

Author: ndeodhar

sdk/python/polos/__init__.py

@@ -29,6 +29,24 @@
     get_workflow,
     workflow,
 )
+from .execution import (
+    DockerEnvironment,
+    DockerEnvironmentConfig,
+    E2BEnvironmentConfig,
+    EnvironmentInfo,
+    ExecOptions,
+    ExecResult,
+    ExecToolConfig,
+    ExecutionEnvironment,
+    GlobOptions,
+    GrepMatch,
+    GrepOptions,
+    LocalEnvironment,
+    LocalEnvironmentConfig,
+    SandboxToolsConfig,
+    SandboxToolsResult,
+    sandbox_tools,
+)
 from .features import events, schedules
 from .features.events import BatchEventPayload, EventPayload
 from .features.schedules import SchedulePayload
@@ -38,7 +56,16 @@
 from .runtime.client import ExecutionHandle, PolosClient
 from .runtime.queue import Queue, queue
 from .runtime.worker import Worker
-from .tools.tool import Tool, tool
+from .tools.ask_user import AskUserField, AskUserFieldOption, AskUserInput, create_ask_user_tool
+from .tools.tool import Tool, ToolApproval, tool
+from .tools.web_search import (
+    WebSearchFunction,
+    WebSearchOptions,
+    WebSearchResult,
+    WebSearchResultItem,
+    WebSearchToolConfig,
+    create_web_search_tool,
+)
 from .types.types import (
     AgentConfig,
     BatchStepResult,
@@ -76,6 +103,7 @@
     "StopConditionContext",
     "tool",
     "Tool",
+    "ToolApproval",
     "hook",
     "HookContext",
     "HookResult",
@@ -102,4 +130,32 @@
     "WorkflowTimeoutError",
     "StepExecutionError",
     "BatchWorkflowInput",
+    # Execution framework
+    "sandbox_tools",
+    "SandboxToolsResult",
+    "SandboxToolsConfig",
+    "ExecutionEnvironment",
+    "DockerEnvironment",
+    "LocalEnvironment",
+    "DockerEnvironmentConfig",
+    "E2BEnvironmentConfig",
+    "LocalEnvironmentConfig",
+    "ExecToolConfig",
+    "ExecOptions",
+    "ExecResult",
+    "GlobOptions",
+    "GrepOptions",
+    "GrepMatch",
+    "EnvironmentInfo",
+    # Tools
+    "create_ask_user_tool",
+    "AskUserInput",
+    "AskUserField",
+    "AskUserFieldOption",
+    "create_web_search_tool",
+    "WebSearchToolConfig",
+    "WebSearchResult",
+    "WebSearchResultItem",
+    "WebSearchOptions",
+    "WebSearchFunction",
 ]

sdk/python/polos/agents/stop_conditions.py

@@ -211,7 +211,7 @@ async def max_tokens(ctx: StopConditionContext, config: MaxTokensConfig) -> bool
 class MaxStepsConfig(BaseModel):
     """Configuration for max_steps stop condition."""
 
-    count: int = 5
+    count: int = 20
 
 
 @stop_condition

sdk/python/polos/execution/__init__.py

@@ -0,0 +1,79 @@
+"""Execution framework -- sandbox tools for AI agents.
+
+Provides tools for running commands, reading/writing files, and searching
+codebases inside isolated environments (Docker, E2B, Local).
+"""
+
+# Main entry point
+# Environment implementations
+from .docker import DockerEnvironment
+from .environment import ExecutionEnvironment
+from .local import LocalEnvironment
+
+# Output utilities
+from .output import is_binary, parse_grep_output, strip_ansi, truncate_output
+from .sandbox_tools import SandboxToolsResult, sandbox_tools
+
+# Security utilities
+from .security import assert_safe_path, evaluate_allowlist, is_within_restriction
+
+# Tool factories
+from .tools.edit import create_edit_tool
+from .tools.exec import create_exec_tool
+from .tools.glob import create_glob_tool
+from .tools.grep import create_grep_tool
+from .tools.read import create_read_tool
+from .tools.write import create_write_tool
+
+# Types
+from .types import (
+    DockerEnvironmentConfig,
+    E2BEnvironmentConfig,
+    EnvironmentInfo,
+    ExecOptions,
+    ExecResult,
+    ExecToolConfig,
+    GlobOptions,
+    GrepMatch,
+    GrepOptions,
+    LocalEnvironmentConfig,
+    SandboxToolsConfig,
+)
+
+__all__ = [
+    # Main entry point
+    "sandbox_tools",
+    "SandboxToolsResult",
+    # Types
+    "ExecutionEnvironment",
+    "ExecOptions",
+    "ExecResult",
+    "GlobOptions",
+    "GrepOptions",
+    "GrepMatch",
+    "EnvironmentInfo",
+    "DockerEnvironmentConfig",
+    "E2BEnvironmentConfig",
+    "LocalEnvironmentConfig",
+    "ExecToolConfig",
+    "SandboxToolsConfig",
+    # Environments
+    "DockerEnvironment",
+    "LocalEnvironment",
+    # Security
+    "evaluate_allowlist",
+    "assert_safe_path",
+    "is_within_restriction",
+    # Output
+    "truncate_output",
+    "is_binary",
+    "parse_grep_output",
+    "strip_ansi",
+    # Tool factories
+    "create_exec_tool",
+    "create_read_tool",
+    "create_write_tool",
+    "create_edit_tool",
+    "create_glob_tool",
+    "create_grep_tool",
+]