chore: import upstream snapshot with attribution

python/tests/agents/langchain/test_backend.py

@@ -0,0 +1,123 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+import pytest
+
+from mirage import MountMode, RAMResource, Workspace
+from mirage.agents.langchain.backend import LangchainWorkspace
+
+
+@pytest.fixture
+def workspace():
+    return Workspace({"/": RAMResource()}, mode=MountMode.WRITE)
+
+
+@pytest.fixture
+def backend(workspace):
+    return LangchainWorkspace(workspace)
+
+
+def test_id(backend):
+    assert backend.id == "mirage"
+
+
+def test_custom_id(workspace):
+    b = LangchainWorkspace(workspace, sandbox_id="custom")
+    assert b.id == "custom"
+
+
+@pytest.mark.asyncio
+async def test_aexecute_echo(backend):
+    resp = await backend.aexecute("echo hello")
+    assert resp.exit_code == 0
+    assert "hello" in resp.output
+
+
+@pytest.mark.asyncio
+async def test_aexecute_failing_command(backend):
+    resp = await backend.aexecute("cat /nonexistent")
+    assert resp.exit_code != 0
+
+
+@pytest.mark.asyncio
+async def test_awrite_and_aread(backend):
+    result = await backend.awrite("/test.txt", "hello world")
+    assert result.error is None
+
+    content = await backend.aread("/test.txt")
+    assert "hello world" in content
+
+
+@pytest.mark.asyncio
+async def test_awrite_existing_file_errors(backend):
+    await backend.awrite("/exists.txt", "first")
+    result = await backend.awrite("/exists.txt", "second")
+    assert result.error is not None
+
+
+@pytest.mark.asyncio
+async def test_aedit(backend):
+    await backend.awrite("/edit.txt", "foo bar baz")
+    result = await backend.aedit("/edit.txt", "bar", "qux")
+    assert result.error is None
+    content = await backend.aread("/edit.txt")
+    assert "qux" in content
+    assert "bar" not in content
+
+
+@pytest.mark.asyncio
+async def test_als_info(backend):
+    await backend.awrite("/dir/a.txt", "a")
+    await backend.awrite("/dir/b.txt", "b")
+    entries = await backend.als_info("/dir")
+    paths = [e["path"] for e in entries]
+    assert len(paths) == 2
+
+
+@pytest.mark.asyncio
+async def test_agrep_raw(backend):
+    await backend.awrite("/search.txt",
+                         "hello world\ngoodbye world\nhello again")
+    result = await backend.agrep_raw("hello", path="/")
+    assert isinstance(result, list)
+    assert len(result) >= 2
+
+
+@pytest.mark.asyncio
+async def test_aglob_info(backend):
+    await backend.awrite("/data/a.txt", "a")
+    await backend.awrite("/data/b.py", "b")
+    entries = await backend.aglob_info("*.txt", path="/data")
+    paths = [e["path"] for e in entries]
+    assert any("a.txt" in p for p in paths)
+    assert not any("b.py" in p for p in paths)
+
+
+@pytest.mark.asyncio
+async def test_execute_pipe(backend):
+    await backend.awrite("/pipe.txt", "aaa\nbbb\nccc\naaa\n")
+    resp = await backend.aexecute("cat /pipe.txt | sort | uniq | wc -l")
+    assert resp.exit_code == 0
+    assert "3" in resp.output
+
+
+@pytest.mark.asyncio
+async def test_upload_and_download(backend):
+    files = [("/up1.txt", b"content1"), ("/up2.txt", b"content2")]
+    up_results = await backend.aupload_files(files)
+    assert all(r.error is None for r in up_results)
+
+    down_results = await backend.adownload_files(["/up1.txt", "/up2.txt"])
+    assert down_results[0].content == b"content1"
+    assert down_results[1].content == b"content2"

python/tests/agents/langchain/test_convert.py

@@ -0,0 +1,72 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.agents.langchain._convert import (io_to_execute_response,
+                                              io_to_file_infos,
+                                              io_to_grep_matches)
+from mirage.io.types import IOResult
+
+
+def test_io_to_execute_response_basic():
+    io = IOResult(stdout=b"hello world\n", exit_code=0)
+    resp = io_to_execute_response(io)
+    assert resp.output == "hello world\n"
+    assert resp.exit_code == 0
+    assert resp.truncated is False
+
+
+def test_io_to_execute_response_none_stdout():
+    io = IOResult(stdout=None, exit_code=1)
+    resp = io_to_execute_response(io)
+    assert resp.output == ""
+    assert resp.exit_code == 1
+
+
+def test_io_to_execute_response_stderr_appended():
+    io = IOResult(stdout=b"out", stderr=b"err", exit_code=1)
+    resp = io_to_execute_response(io)
+    assert "out" in resp.output
+    assert "err" in resp.output
+
+
+def test_io_to_grep_matches():
+    io = IOResult(stdout=b"/a.txt:1:hello world\n/b.txt:5:hello there\n",
+                  exit_code=0)
+    matches = io_to_grep_matches(io)
+    assert len(matches) == 2
+    assert matches[0]["path"] == "/a.txt"
+    assert matches[0]["line"] == 1
+    assert matches[0]["text"] == "hello world"
+
+
+def test_io_to_grep_matches_empty():
+    io = IOResult(stdout=b"", exit_code=1)
+    matches = io_to_grep_matches(io)
+    assert matches == []
+
+
+def test_io_to_file_infos():
+    io = IOResult(stdout=b"/foo/bar.txt\n/foo/baz.py\n/foo/sub/\n",
+                  exit_code=0)
+    infos = io_to_file_infos(io)
+    assert len(infos) == 3
+    dirs = [i for i in infos if i.get("is_dir")]
+    assert len(dirs) == 1
+    assert dirs[0]["path"] == "/foo/sub"
+
+
+def test_io_to_file_infos_empty():
+    io = IOResult(stdout=b"", exit_code=0)
+    infos = io_to_file_infos(io)
+    assert infos == []

python/tests/agents/langchain/test_messages.py

@@ -0,0 +1,70 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.agents.langchain._messages import extract_text
+
+
+def test_extract_text_from_string():
+
+    class Msg:
+        content = "hello world"
+
+    assert extract_text([Msg()]) == ["hello world"]
+
+
+def test_extract_text_from_blocks():
+
+    class Msg:
+        content = [
+            {
+                "type": "text",
+                "text": "hello"
+            },
+            {
+                "type": "tool_use",
+                "name": "ls",
+                "input": {}
+            },
+            {
+                "type": "text",
+                "text": "goodbye"
+            },
+        ]
+
+    assert extract_text([Msg()]) == ["hello", "goodbye"]
+
+
+def test_extract_text_skips_empty():
+
+    class Msg:
+        content = [
+            {
+                "type": "text",
+                "text": "  "
+            },
+            {
+                "type": "text",
+                "text": "real"
+            },
+        ]
+
+    assert extract_text([Msg()]) == ["real"]
+
+
+def test_extract_text_no_content():
+
+    class Msg:
+        pass
+
+    assert extract_text([Msg()]) == []