fixing formatting

Signed-off-by: Jerry Guan <jerryguan777@gmail.com>

Author: Jerryguan777

examples/sandbox_agent/src/nat_sandbox_agent/__init__.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """
 Sandbox Agent - A general-purpose agent with sandboxed execution capabilities.
 

examples/sandbox_agent/src/nat_sandbox_agent/prompts/__init__.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Prompts module - System prompts for the sandbox agent."""
 
 from nat_sandbox_agent.prompts.system_prompt import SANDBOX_AGENT_SYSTEM_PROMPT

examples/sandbox_agent/src/nat_sandbox_agent/prompts/system_prompt.py

@@ -12,10 +12,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """System prompts for the sandbox agent."""
 
-
 SANDBOX_AGENT_SYSTEM_PROMPT = """You are a powerful AI assistant with access to an isolated \
 sandbox environment where you can execute code, browse the web, and manipulate files.
 
@@ -121,6 +119,7 @@
 
 Remember: You are operating in a sandboxed environment. ALWAYS use tools to solve problems - never guess or assume."""
 
+
 def get_system_prompt(
     additional_instructions: str | None = None,
     available_tools: list[str] | None = None,

examples/sandbox_agent/src/nat_sandbox_agent/register.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Sandbox Agent workflow registration for NAT framework."""
 
 import logging
@@ -45,6 +44,7 @@
 
 # ============ Configuration Classes ============
 
+
 class SandboxAgentWorkflowConfig(FunctionBaseConfig, name="sandbox_agent"):
     """Configuration for the Sandbox Agent workflow.
 
@@ -76,8 +76,7 @@ class SandboxAgentWorkflowConfig(FunctionBaseConfig, name="sandbox_agent"):
             "image": "python:3.12-slim",
             "memory_limit": "1g",
             "cpu_limit": 2.0,
-            "network_enabled": True,
-        },
+            "network_enabled": True, },
         description="Sandbox configuration dictionary.",
     )
 
@@ -97,17 +96,21 @@ class SandboxAgentWorkflowConfig(FunctionBaseConfig, name="sandbox_agent"):
         description="Additional instructions to append to the system prompt.",
     )
 
+
 # ============ Agent State ============
 
+
 class AgentState(TypedDict):
     """State for the sandbox agent graph."""
 
     messages: Annotated[list, add_messages]
     iteration_count: int
     sandbox_id: str
 
+
 # ============ Workflow Registration ============
 
+
 @register_function(
     config_type=SandboxAgentWorkflowConfig,
     framework_wrappers=[LLMFrameworkEnum.LANGCHAIN],
@@ -171,7 +174,6 @@ async def sandbox_agent_workflow(config: SandboxAgentWorkflowConfig, builder: Bu
         )
 
         # ============ Define Agent Nodes ============
-
         async def agent_node(state: AgentState) -> dict:
             """Agent reasoning node - decides what action to take."""
             messages = state["messages"]
@@ -182,10 +184,8 @@ async def agent_node(state: AgentState) -> dict:
                 logger.warning(f"Reached max iterations ({config.max_iterations})")
                 return {
                     "messages": [
-                        AIMessage(
-                            content="I've reached the maximum number of steps. "
-                            "Here's a summary of what was accomplished so far."
-                        )
+                        AIMessage(content="I've reached the maximum number of steps. "
+                                  "Here's a summary of what was accomplished so far.")
                     ],
                     "iteration_count": iteration,
                 }
@@ -229,7 +229,9 @@ def should_continue(state: AgentState) -> str:
         graph.add_conditional_edges(
             "agent",
             should_continue,
-            {"tools": "tools", END: END},
+            {
+                "tools": "tools", END: END
+            },
         )
         graph.add_edge("tools", "agent")
 
@@ -244,7 +246,6 @@ def should_continue(state: AgentState) -> str:
         logger.info("Agent graph compiled successfully")
 
         # ============ Response Function ============
-
         async def _response_fn(input_message: str) -> str:
             """Process user input and return agent response.
 

examples/sandbox_agent/src/nat_sandbox_agent/sandbox/__init__.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Sandbox module - Provides isolated execution environments."""
 
 from nat_sandbox_agent.sandbox.base import BaseSandbox

examples/sandbox_agent/src/nat_sandbox_agent/sandbox/base.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Base sandbox interface - Abstract base class for sandbox implementations."""
 
 from abc import ABC
@@ -31,9 +30,7 @@
 WORKSPACE_DOWNLOADS = f"{WORKSPACE_ROOT}/downloads"
 
 # Command to initialize workspace directories
-WORKSPACE_INIT_COMMAND = (
-    f"mkdir -p {WORKSPACE_INPUT} {WORKSPACE_OUTPUT} {WORKSPACE_TEMP} {WORKSPACE_DOWNLOADS}"
-)
+WORKSPACE_INIT_COMMAND = (f"mkdir -p {WORKSPACE_INPUT} {WORKSPACE_OUTPUT} {WORKSPACE_TEMP} {WORKSPACE_DOWNLOADS}")
 
 # Default script path for Python execution
 DEFAULT_SCRIPT_PATH = f"{WORKSPACE_TEMP}/_script.py"

examples/sandbox_agent/src/nat_sandbox_agent/sandbox/daytona_sandbox.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Daytona cloud-based sandbox implementation."""
 
 import asyncio
@@ -38,15 +37,15 @@ class DaytonaSandbox(BaseSandbox):
     DEFAULT_IMAGE = "daytonaio/workspace:latest"
 
     def __init__(
-        self,
-        api_key: str,
-        server_url: str = "https://api.daytona.io",
-        target: str = "us",
-        image: str = DEFAULT_IMAGE,
-        cpu: int = 2,
-        memory: int = 4,  # GB
-        disk: int = 10,  # GB
-        auto_stop_interval: int = 30,  # minutes
+            self,
+            api_key: str,
+            server_url: str = "https://api.daytona.io",
+            target: str = "us",
+            image: str = DEFAULT_IMAGE,
+            cpu: int = 2,
+            memory: int = 4,  # GB
+            disk: int = 10,  # GB
+            auto_stop_interval: int = 30,  # minutes
     ):
         """Initialize Daytona sandbox configuration.
 
@@ -79,15 +78,11 @@ def _get_client(self):
                 from daytona_sdk import Daytona
                 from daytona_sdk import DaytonaConfig
 
-                config = DaytonaConfig(
-                    api_key=self._api_key,
-                )
+                config = DaytonaConfig(api_key=self._api_key, )
                 self._client = Daytona(config)
             except ImportError:
-                raise ImportError(
-                    "daytona-sdk package is required for DaytonaSandbox. "
-                    "Install it with: pip install daytona-sdk"
-                )
+                raise ImportError("daytona-sdk package is required for DaytonaSandbox. "
+                                  "Install it with: pip install daytona-sdk")
         return self._client
 
     async def start(self) -> None:
@@ -112,9 +107,7 @@ async def start(self) -> None:
             )
 
             # Create and start the sandbox (wrap sync call to avoid blocking)
-            self._sandbox = await asyncio.get_running_loop().run_in_executor(
-                None, lambda: client.create(params)
-            )
+            self._sandbox = await asyncio.get_running_loop().run_in_executor(None, lambda: client.create(params))
 
             # Initialize workspace directories
             await self.run_command(WORKSPACE_INIT_COMMAND)
@@ -131,9 +124,7 @@ async def cleanup(self) -> None:
             logger.info(f"Cleaning up Daytona sandbox: {self._sandbox.id}")
             try:
                 # Wrap sync call to avoid blocking
-                await asyncio.get_running_loop().run_in_executor(
-                    None, self._sandbox.delete
-                )
+                await asyncio.get_running_loop().run_in_executor(None, self._sandbox.delete)
                 self._sandbox = None
             except Exception as e:
                 logger.error(f"Failed to cleanup Daytona sandbox: {e}")
@@ -163,8 +154,7 @@ async def run_command(
                         command,
                         cwd=working_dir,
                         env=env,
-                        timeout=int(timeout),
-                    ),
+                        timeout=int(timeout), ),
                 ),
                 timeout=timeout,
             )
@@ -199,9 +189,7 @@ async def read_file(self, path: str) -> str:
         try:
             # Wrap sync call to avoid blocking
             # Daytona SDK uses download_file() which returns bytes
-            data = await asyncio.get_running_loop().run_in_executor(
-                None, lambda: self._sandbox.fs.download_file(path)
-            )
+            data = await asyncio.get_running_loop().run_in_executor(None, lambda: self._sandbox.fs.download_file(path))
             return data.decode("utf-8", errors="replace")
         except Exception as e:
             if "not found" in str(e).lower():
@@ -223,8 +211,7 @@ async def write_file(self, path: str, content: str) -> None:
             # Wrap sync call to avoid blocking
             # Daytona SDK uses upload_file(data, path) which takes bytes
             await asyncio.get_running_loop().run_in_executor(
-                None, lambda: self._sandbox.fs.upload_file(content.encode("utf-8"), path)
-            )
+                None, lambda: self._sandbox.fs.upload_file(content.encode("utf-8"), path))
         except Exception as e:
             logger.error(f"Failed to write file {path}: {e}")
             raise

examples/sandbox_agent/src/nat_sandbox_agent/sandbox/docker_sandbox.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Docker-based sandbox implementation."""
 
 import asyncio
@@ -22,11 +21,11 @@
 import tarfile
 import uuid
 
+import docker
 from docker.errors import ContainerError
 from docker.errors import ImageNotFound
 from docker.errors import NotFound
 
-import docker
 from nat_sandbox_agent.sandbox.base import WORKSPACE_INIT_COMMAND
 from nat_sandbox_agent.sandbox.base import WORKSPACE_ROOT
 from nat_sandbox_agent.sandbox.base import BaseSandbox
@@ -96,9 +95,7 @@ async def start(self) -> None:
                 self._client.images.get(self._image)
             except ImageNotFound:
                 logger.info(f"Pulling image: {self._image}")
-                await asyncio.get_running_loop().run_in_executor(
-                    None, self._client.images.pull, self._image
-                )
+                await asyncio.get_running_loop().run_in_executor(None, self._client.images.pull, self._image)
 
             # Container configuration
             container_config = {
@@ -123,12 +120,9 @@ async def start(self) -> None:
 
             # Create and start container
             self._container = await asyncio.get_running_loop().run_in_executor(
-                None, lambda: self._client.containers.create(**container_config)
-            )
+                None, lambda: self._client.containers.create(**container_config))
 
-            await asyncio.get_running_loop().run_in_executor(
-                None, self._container.start
-            )
+            await asyncio.get_running_loop().run_in_executor(None, self._container.start)
 
             # Initialize workspace directories
             await self.run_command(WORKSPACE_INIT_COMMAND)
@@ -148,9 +142,7 @@ async def cleanup(self) -> None:
         if self._container:
             logger.info(f"Cleaning up Docker sandbox: {self._container_name}")
             try:
-                await asyncio.get_running_loop().run_in_executor(
-                    None, lambda: self._container.remove(force=True)
-                )
+                await asyncio.get_running_loop().run_in_executor(None, lambda: self._container.remove(force=True))
                 self._container = None
             except NotFound:
                 # Container already removed
@@ -190,8 +182,7 @@ async def run_command(
                         cmd=wrapped_command,
                         workdir=working_dir,
                         environment=env,
-                        demux=True,
-                    ),
+                        demux=True, ),
                 ),
                 timeout=timeout + 5,  # Give container timeout a chance to fire first
             )
@@ -263,9 +254,7 @@ def _extract_from_archive():
 
         try:
             # Wrap all blocking I/O in run_in_executor
-            return await asyncio.get_running_loop().run_in_executor(
-                None, _extract_from_archive
-            )
+            return await asyncio.get_running_loop().run_in_executor(None, _extract_from_archive)
 
         except NotFound:
             raise FileNotFoundError(f"File not found: {path}") from None

examples/sandbox_agent/src/nat_sandbox_agent/sandbox/factory.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Factory for creating sandbox instances."""
 
 import os
@@ -159,7 +158,9 @@ def create_sandbox(config: SandboxConfig) -> BaseSandbox:
         volumes = None
         if config.volumes:
             volumes = {
-                host_path: {"bind": container_path, "mode": "rw"}
+                host_path: {
+                    "bind": container_path, "mode": "rw"
+                }
                 for host_path, container_path in config.volumes.items()
             }
 

examples/sandbox_agent/src/nat_sandbox_agent/tools/__init__.py

@@ -12,7 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Tools module - Modular tool implementations.
 
 Tool Architecture:
@@ -31,13 +30,10 @@
 __all__ = [
     # Factory
     "create_all_tools",
-    "get_tool_descriptions",
-    # Sandbox
+    "get_tool_descriptions",  # Sandbox
     "create_sandbox_tools",
-    "SandboxToolExecutor",
-    # Host
-    "create_host_tools",
-    # Utilities
+    "SandboxToolExecutor",  # Host
+    "create_host_tools",  # Utilities
     "DEFAULT_MAX_OUTPUT_CHARS",
     "truncate_output",
 ]