langchain-agentcore-codeinterpreter 0.0.2__tar.gz → 0.0.3__tar.gz

{langchain_agentcore_codeinterpreter-0.0.2 → langchain_agentcore_codeinterpreter-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-agentcore-codeinterpreter
-Version: 0.0.2
+Version: 0.0.3
 Summary: Amazon Bedrock AgentCore Code Interpreter sandbox integration for Deep Agents
 Project-URL: Source Code, https://github.com/langchain-ai/langchain-aws/tree/main/libs/agentcore-codeinterpreter
 Project-URL: Repository, https://github.com/langchain-ai/langchain-aws

{langchain_agentcore_codeinterpreter-0.0.2 → langchain_agentcore_codeinterpreter-0.0.3}/langchain_agentcore_codeinterpreter/sandbox.py

@@ -32,6 +32,21 @@ _AGENTCORE_EXECUTOR = ThreadPoolExecutor(
 )
 
 
+def _normalize_relative_path(path: str) -> str:
+    """Strip leading slashes and ``./`` prefixes to a canonical relative path.
+
+    Args:
+        path: File path (absolute or relative).
+
+    Returns:
+        Canonical relative path string with no leading ``/`` or ``./``.
+    """
+    path = path.lstrip("/")
+    while path.startswith("./"):
+        path = path[2:]
+    return path
+
+
 class SessionExpiredError(Exception):
     """Raised when the AgentCore session has expired or been terminated."""
 
@@ -104,8 +119,7 @@ def _extract_files_from_stream(
     """
     path_lookup: dict[str, str] = {}
     for path in requested_paths:
-        stripped = path.lstrip("/")
-        path_lookup[stripped] = path
+        path_lookup[_normalize_relative_path(path)] = path
 
     files: dict[str, bytes] = {}
 
@@ -117,13 +131,16 @@ def _extract_files_from_stream(
                 continue
             resource = item.get("resource", {})
             uri = resource.get("uri", "")
-            file_path = uri.replace("file://", "").lstrip("/")
+            file_path = _normalize_relative_path(uri.replace("file://", ""))
 
             content: bytes | None = None
             if "text" in resource:
                 content = resource["text"].encode("utf-8")
             elif "blob" in resource:
-                content = base64.b64decode(resource["blob"])
+                blob = resource["blob"]
+                # The AgentCore stream may deliver blob as already-decoded bytes.
+                # Only base64-decode when it arrives as encoded text.
+                content = blob if isinstance(blob, bytes) else base64.b64decode(blob)
 
             if content is not None:
                 original_path = path_lookup.get(file_path, file_path)
@@ -178,7 +195,7 @@ class AgentCoreSandbox(BaseSandbox):
 
     @staticmethod
     def _to_relative_path(path: str) -> str:
-        """Strip leading slashes so paths are relative for AgentCore APIs.
+        """Strip leading slashes and ``./`` prefixes for AgentCore APIs.
 
         Args:
             path: File path (absolute or relative).
@@ -186,7 +203,7 @@ class AgentCoreSandbox(BaseSandbox):
         Returns:
             Relative path string.
         """
-        return path.lstrip("/")
+        return _normalize_relative_path(path)
 
     def _invoke(self, method: str, params: dict[str, Any]) -> dict[str, Any]:
         """Invoke the interpreter and eagerly consume the response stream.

{langchain_agentcore_codeinterpreter-0.0.2 → langchain_agentcore_codeinterpreter-0.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langchain-agentcore-codeinterpreter"
-version = "0.0.2"
+version = "0.0.3"
 description = "Amazon Bedrock AgentCore Code Interpreter sandbox integration for Deep Agents"
 readme = "README.md"
 license = "MIT"

{langchain_agentcore_codeinterpreter-0.0.2 → langchain_agentcore_codeinterpreter-0.0.3}/tests/unit_tests/test_sandbox.py

@@ -228,6 +228,36 @@ def test_download_files_handles_session_expiry() -> None:
     assert results[0].error == "permission_denied"
 
 
+def test_download_files_dot_slash_path() -> None:
+    """./-prefixed paths must round-trip through readFiles and lookup."""
+    fake_png = b"\x89PNG\r\n\x1a\n" + b"\x00" * 100
+    sandbox, mock = _make_sandbox(
+        {
+            "stream": [
+                {
+                    "result": {
+                        "content": [
+                            {
+                                "type": "resource",
+                                "resource": {
+                                    "uri": "file:///data/foo.png",
+                                    "blob": fake_png,
+                                },
+                            }
+                        ]
+                    }
+                }
+            ]
+        }
+    )
+    results = sandbox.download_files(["./data/foo.png"])
+    mock.invoke.assert_called_once_with(
+        method="readFiles", params={"paths": ["data/foo.png"]}
+    )
+    assert results[0].error is None
+    assert results[0].content == fake_png
+
+
 # ------------------------------------------------------------------
 # _to_relative_path()
 # ------------------------------------------------------------------
@@ -240,6 +270,13 @@ def test_relative_path_stripping() -> None:
     assert AgentCoreSandbox._to_relative_path("///triple.txt") == "triple.txt"
 
 
+def test_relative_path_strips_dot_slash() -> None:
+    """./ and repeated ././ prefixes should be stripped."""
+    assert AgentCoreSandbox._to_relative_path("./data/foo.png") == "data/foo.png"
+    assert AgentCoreSandbox._to_relative_path("././foo.png") == "foo.png"
+    assert AgentCoreSandbox._to_relative_path("/./data/foo.png") == "data/foo.png"
+
+
 # ------------------------------------------------------------------
 # Constructor
 # ------------------------------------------------------------------

{langchain_agentcore_codeinterpreter-0.0.2 → langchain_agentcore_codeinterpreter-0.0.3}/tests/unit_tests/test_stream_parsing.py

@@ -182,6 +182,54 @@ def test_extract_files_blob_resource() -> None:
     assert files["/data.bin"] == b"binary data"
 
 
+def test_extract_files_blob_already_bytes() -> None:
+    """Blob delivered as raw bytes must not be base64-decoded again."""
+    png_magic = b"\x89PNG\r\n\x1a\n"
+    fake_png = png_magic + b"\x00" * 100
+    response: dict[str, Any] = {
+        "stream": [
+            {
+                "result": {
+                    "content": [
+                        {
+                            "type": "resource",
+                            "resource": {
+                                "uri": "file:///test.png",
+                                "blob": fake_png,
+                            },
+                        }
+                    ]
+                }
+            }
+        ]
+    }
+    files = _extract_files_from_stream(response, ["test.png"])
+    assert files["test.png"] == fake_png
+
+
+def test_extract_files_dot_slash_path() -> None:
+    """./-prefixed requested paths should match URIs without the prefix."""
+    response: dict[str, Any] = {
+        "stream": [
+            {
+                "result": {
+                    "content": [
+                        {
+                            "type": "resource",
+                            "resource": {
+                                "uri": "file:///data/foo.png",
+                                "text": "x",
+                            },
+                        }
+                    ]
+                }
+            }
+        ]
+    }
+    files = _extract_files_from_stream(response, ["./data/foo.png"])
+    assert files["./data/foo.png"] == b"x"
+
+
 def test_extract_files_path_normalization() -> None:
     """Relative requested paths should still match file:/// URIs."""
     response: dict[str, Any] = {