stirrup 0.1.4__tar.gz → 0.1.5__tar.gz

{stirrup-0.1.4 → stirrup-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: stirrup
-Version: 0.1.4
+Version: 0.1.5
 Summary: The lightweight foundation for building agents
 Keywords: ai,agent,llm,openai,anthropic,tools,framework
 Author: Artificial Analysis, Inc.
@@ -77,20 +77,20 @@ Description-Content-Type: text/markdown
 <br>
 </div>
 
-
 <p align="center">
   <a href="https://pypi.python.org/pypi/stirrup"><img src="https://img.shields.io/pypi/v/stirrup" alt="PyPI version" /></a>&nbsp;<!--
   --><a href="https://github.com/ArtificialAnalysis/Stirrup/blob/main/LICENSE"><img src="https://img.shields.io/github/license/ArtificialAnalysis/Stirrup" alt="License" /></a>&nbsp;<!--
   --><a href="https://stirrup.artificialanalysis.ai"><img src="https://img.shields.io/badge/MkDocs-4F46E5?logo=materialformkdocs&logoColor=fff" alt="MkDocs" /></a>
 </p>
 
-
 Stirrup is a lightweight framework, or starting point template, for building agents. It differs from other agent frameworks by:
 
 - **Working with the model, not against it:** Stirrup gets out of the way and lets the model choose its own approach to completing tasks (similar to Claude Code). Many frameworks impose rigid workflows that can degrade results.
 - **Best practices and tools built-in:** We analyzed the leading agents (Claude Code, Codex, and others) to understand and incorporate best practices relating to topics like context management and foundational tools (e.g., code execution).
 - **Fully customizable:** Use Stirrup as a package or as a starting template to build your own fully customized agents.
 
+> **Note:** This is the Python implementation, [StirrupJS](https://github.com/ArtificialAnalysis/StirrupJS) is the Typescript implementation.
+
 ## Features
 
 - 🧪 **Code execution:** Run code locally, in Docker, or in an E2B sandbox

{stirrup-0.1.4 → stirrup-0.1.5}/README.md

@@ -9,20 +9,20 @@
 <br>
 </div>
 
-
 <p align="center">
   <a href="https://pypi.python.org/pypi/stirrup"><img src="https://img.shields.io/pypi/v/stirrup" alt="PyPI version" /></a>&nbsp;<!--
   --><a href="https://github.com/ArtificialAnalysis/Stirrup/blob/main/LICENSE"><img src="https://img.shields.io/github/license/ArtificialAnalysis/Stirrup" alt="License" /></a>&nbsp;<!--
   --><a href="https://stirrup.artificialanalysis.ai"><img src="https://img.shields.io/badge/MkDocs-4F46E5?logo=materialformkdocs&logoColor=fff" alt="MkDocs" /></a>
 </p>
 
-
 Stirrup is a lightweight framework, or starting point template, for building agents. It differs from other agent frameworks by:
 
 - **Working with the model, not against it:** Stirrup gets out of the way and lets the model choose its own approach to completing tasks (similar to Claude Code). Many frameworks impose rigid workflows that can degrade results.
 - **Best practices and tools built-in:** We analyzed the leading agents (Claude Code, Codex, and others) to understand and incorporate best practices relating to topics like context management and foundational tools (e.g., code execution).
 - **Fully customizable:** Use Stirrup as a package or as a starting template to build your own fully customized agents.
 
+> **Note:** This is the Python implementation, [StirrupJS](https://github.com/ArtificialAnalysis/StirrupJS) is the Typescript implementation.
+
 ## Features
 
 - 🧪 **Code execution:** Run code locally, in Docker, or in an E2B sandbox

{stirrup-0.1.4 → stirrup-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stirrup"
-version = "0.1.4"
+version = "0.1.5"
 description = "The lightweight foundation for building agents"
 readme = "README.md"
 license = { file = "LICENSE" }

{stirrup-0.1.4 → stirrup-0.1.5}/src/stirrup/tools/code_backends/base.py

@@ -245,6 +245,39 @@ class CodeExecToolProvider(ToolProvider, ABC):
         """
         ...
 
+    @abstractmethod
+    async def is_directory(self, path: str) -> bool:
+        """Check if a path is a directory in this execution environment.
+
+        Args:
+            path: Path within this execution environment.
+
+        Returns:
+            True if the path exists and is a directory, False otherwise.
+
+        Raises:
+            RuntimeError: If execution environment not started.
+
+        """
+        ...
+
+    @abstractmethod
+    async def list_files(self, path: str) -> list[str]:
+        """List all files recursively in a directory within this execution environment.
+
+        Args:
+            path: Directory path within this execution environment.
+
+        Returns:
+            List of file paths (relative to the given path) for all files in the directory.
+            Returns an empty list if the path is a file or doesn't exist.
+
+        Raises:
+            RuntimeError: If execution environment not started.
+
+        """
+        ...
+
     async def save_output_files(
         self,
         paths: list[str],
@@ -334,18 +367,44 @@ class CodeExecToolProvider(ToolProvider, ABC):
             try:
                 if source_env:
                     # Cross-environment transfer: read from source_env
-                    content = await source_env.read_file_bytes(path_str)
-                    filename = Path(path_str).name
-                    dest_path = f"{dest_dir_str}/{filename}" if dest_dir_str else filename
-                    logger.debug(
-                        "UPLOAD CROSS-ENV: %s (%d bytes) from %s -> %s",
-                        path_str,
-                        len(content),
-                        type(source_env).__name__,
-                        dest_path,
-                    )
-                    await self.write_file_bytes(dest_path, content)
-                    result.uploaded.append(UploadedFile(Path(path_str), dest_path, len(content)))
+                    # Check if it's a directory first
+                    if await source_env.is_directory(path_str):
+                        # Handle directory recursively
+                        # Preserve directory name when dest_dir not specified
+                        dir_name = Path(path_str).name
+                        files = await source_env.list_files(path_str)
+                        for rel_file_path in files:
+                            src_file_path = f"{path_str}/{rel_file_path}"
+                            # If dest_dir specified, put files directly there
+                            # Otherwise, preserve the source directory name
+                            if dest_dir_str:
+                                dest_path = f"{dest_dir_str}/{rel_file_path}"
+                            else:
+                                dest_path = f"{dir_name}/{rel_file_path}"
+                            content = await source_env.read_file_bytes(src_file_path)
+                            logger.debug(
+                                "UPLOAD CROSS-ENV (dir): %s (%d bytes) from %s -> %s",
+                                src_file_path,
+                                len(content),
+                                type(source_env).__name__,
+                                dest_path,
+                            )
+                            await self.write_file_bytes(dest_path, content)
+                            result.uploaded.append(UploadedFile(Path(src_file_path), dest_path, len(content)))
+                    else:
+                        # Single file transfer
+                        content = await source_env.read_file_bytes(path_str)
+                        filename = Path(path_str).name
+                        dest_path = f"{dest_dir_str}/{filename}" if dest_dir_str else filename
+                        logger.debug(
+                            "UPLOAD CROSS-ENV: %s (%d bytes) from %s -> %s",
+                            path_str,
+                            len(content),
+                            type(source_env).__name__,
+                            dest_path,
+                        )
+                        await self.write_file_bytes(dest_path, content)
+                        result.uploaded.append(UploadedFile(Path(path_str), dest_path, len(content)))
                 else:
                     # Local filesystem upload - must be handled by subclass
                     # This is a fallback that reads from local fs and writes to env

{stirrup-0.1.4 → stirrup-0.1.5}/src/stirrup/tools/code_backends/docker.py

@@ -505,6 +505,53 @@ class DockerCodeExecToolProvider(CodeExecToolProvider):
         host_path = self._container_path_to_host(path)
         return host_path.exists() and host_path.is_file()
 
+    async def is_directory(self, path: str) -> bool:
+        """Check if a path is a directory in the container.
+
+        Since files are volume-mounted, checks directly on the host temp directory.
+
+        Args:
+            path: Path (relative or absolute container path).
+
+        Returns:
+            True if the path exists and is a directory, False otherwise.
+
+        Raises:
+            RuntimeError: If environment not started.
+            ValueError: If path is outside mounted directory.
+
+        """
+        host_path = self._container_path_to_host(path)
+        return host_path.exists() and host_path.is_dir()
+
+    async def list_files(self, path: str) -> list[str]:
+        """List all files recursively in a directory within the container.
+
+        Since files are volume-mounted, lists directly from the host temp directory.
+
+        Args:
+            path: Directory path (relative or absolute container path).
+
+        Returns:
+            List of file paths (relative to the given path) for all files in the directory.
+            Returns an empty list if the path is a file or doesn't exist.
+
+        Raises:
+            RuntimeError: If environment not started.
+            ValueError: If path is outside mounted directory.
+
+        """
+        host_path = self._container_path_to_host(path)
+        if not host_path.exists() or not host_path.is_dir():
+            return []
+
+        files = []
+        for file_path in host_path.rglob("*"):
+            if file_path.is_file():
+                rel_path = file_path.relative_to(host_path)
+                files.append(str(rel_path))
+        return files
+
     async def run_command(self, cmd: str, *, timeout: int = SHELL_TIMEOUT) -> CommandResult:
         """Execute a shell command in the Docker container.
 

{stirrup-0.1.4 → stirrup-0.1.5}/src/stirrup/tools/code_backends/e2b.py

@@ -150,6 +150,68 @@ class E2BCodeExecToolProvider(CodeExecToolProvider):
 
         return await self._sbx.files.exists(path)
 
+    async def is_directory(self, path: str) -> bool:
+        """Check if a path is a directory in the E2B sandbox.
+
+        Args:
+            path: Path within the sandbox.
+
+        Returns:
+            True if the path exists and is a directory, False otherwise.
+
+        Raises:
+            RuntimeError: If environment not started.
+
+        """
+        if self._sbx is None:
+            raise RuntimeError("ExecutionEnvironment not started.")
+
+        if not await self._sbx.files.exists(path):
+            return False
+
+        info = await self._sbx.files.get_info(path)
+        return info.type == FileType.DIR
+
+    async def list_files(self, path: str) -> list[str]:
+        """List all files recursively in a directory within the E2B sandbox.
+
+        Args:
+            path: Directory path within the sandbox.
+
+        Returns:
+            List of file paths (relative to the given path) for all files in the directory.
+            Returns an empty list if the path is a file or doesn't exist.
+
+        Raises:
+            RuntimeError: If environment not started.
+
+        """
+        if self._sbx is None:
+            raise RuntimeError("ExecutionEnvironment not started.")
+
+        if not await self._sbx.files.exists(path):
+            return []
+
+        info = await self._sbx.files.get_info(path)
+        if info.type != FileType.DIR:
+            return []
+
+        # Use find command to list all files recursively
+        result = await self.run_command(f"find {path} -type f")
+        if result.exit_code != 0:
+            return []
+
+        files = []
+        for line in result.stdout.strip().split("\n"):
+            if line:
+                # Convert absolute path to relative path
+                rel_path = line.removeprefix(f"{path}/").removeprefix(path)
+                if rel_path.startswith("/"):
+                    rel_path = rel_path[1:]
+                if rel_path:
+                    files.append(rel_path)
+        return files
+
     async def run_command(self, cmd: str, *, timeout: int = SHELL_TIMEOUT) -> CommandResult:
         """Execute command in E2B execution environment, returning raw CommandResult."""
         if self._sbx is None:

{stirrup-0.1.4 → stirrup-0.1.5}/src/stirrup/tools/code_backends/local.py

@@ -222,6 +222,49 @@ class LocalCodeExecToolProvider(CodeExecToolProvider):
         resolved = self._resolve_and_validate_path(path)
         return resolved.exists() and resolved.is_file()
 
+    async def is_directory(self, path: str) -> bool:
+        """Check if a path is a directory in the temp directory.
+
+        Args:
+            path: Path (relative or absolute within the temp dir).
+
+        Returns:
+            True if the path exists and is a directory, False otherwise.
+
+        Raises:
+            RuntimeError: If environment not started.
+            ValueError: If path is outside temp directory.
+
+        """
+        resolved = self._resolve_and_validate_path(path)
+        return resolved.exists() and resolved.is_dir()
+
+    async def list_files(self, path: str) -> list[str]:
+        """List all files recursively in a directory within the temp directory.
+
+        Args:
+            path: Directory path (relative or absolute within the temp dir).
+
+        Returns:
+            List of file paths (relative to the given path) for all files in the directory.
+            Returns an empty list if the path is a file or doesn't exist.
+
+        Raises:
+            RuntimeError: If environment not started.
+            ValueError: If path is outside temp directory.
+
+        """
+        resolved = self._resolve_and_validate_path(path)
+        if not resolved.exists() or not resolved.is_dir():
+            return []
+
+        files = []
+        for file_path in resolved.rglob("*"):
+            if file_path.is_file():
+                rel_path = file_path.relative_to(resolved)
+                files.append(str(rel_path))
+        return files
+
     async def run_command(self, cmd: str, *, timeout: int = SHELL_TIMEOUT) -> CommandResult:
         """Execute command in the temp directory.