@elizaos/sweagent-root 2.0.0-alpha

package/python/tests/tools/test_default_utils.py

@@ -0,0 +1,153 @@
+import json
+import os
+from pathlib import Path
+
+import pytest
+
+from sweagent import TOOLS_DIR
+from tests.utils import make_python_tool_importable
+
+DEFAULT_TOOLS_DIR = TOOLS_DIR / "windowed"
+DEFAULT_TOOLS_BIN = DEFAULT_TOOLS_DIR / "bin"
+
+make_python_tool_importable(DEFAULT_TOOLS_DIR / "lib/windowed_file.py", "windowed_file")
+import windowed_file  # type: ignore
+from windowed_file import TextNotFound, WindowedFile  # type: ignore
+
+
+def test_env_file_override(with_tmp_env_file):
+    assert Path(os.getenv("SWE_AGENT_ENV_FILE")).name == ".swe-agent-env"  # type: ignore
+
+
+def create_test_file_with_content(tmp_env_file: Path, content: str) -> WindowedFile:
+    test_path = tmp_env_file.parent.joinpath("test.py")
+    registry = {
+        "CURRENT_FILE": str(test_path),
+        "FIRST_LINE": "0",
+        "WINDOW": "10",
+    }
+    tmp_env_file.write_text(json.dumps(registry))
+    test_path.write_text(content)
+    wf = WindowedFile(exit_on_exception=False)
+    wf.offset_multiplier = 1 / 4
+    return wf
+
+
+@pytest.fixture
+def windowed_file(with_tmp_env_file) -> WindowedFile:
+    content = "\n".join(map(str, range(100)))
+    return create_test_file_with_content(with_tmp_env_file, content)
+
+
+def test_windowed_file(windowed_file):
+    wfile = windowed_file
+    assert wfile.first_line == 0
+    wfile.first_line = 10
+    assert wfile.first_line == 10
+    assert wfile.window == 10
+    assert wfile.n_lines == 100
+    a, b = wfile.line_range
+    assert b - a == wfile.window - 1
+    assert wfile.line_range == (10, 19)
+    wfile.print_window()
+    wfile.replace_in_window("10", "Hello, world!")
+    assert wfile.n_lines == 100
+    assert wfile.line_range == (7, 16)
+    wfile.first_line = 50
+    wfile.print_window()
+    wfile.replace_in_window("50", "Hello, world!")
+    wfile.print_window()
+    # Line 50 is now the 2nd line of the new window
+    assert wfile.line_range == (47, 56)
+    with pytest.raises(TextNotFound):
+        wfile.replace_in_window("asdf", "Hello, world!")
+
+
+def test_windowed_file_goto(windowed_file):
+    wfile = windowed_file
+    assert wfile.first_line == 0
+    wfile.first_line = 10
+    wfile.goto(0, mode="top")
+    assert wfile.line_range[0] == 0
+    wfile.goto(50, mode="top")
+    # first line is 50 - 10//4 = 47
+    assert wfile.line_range[0] == 47
+    wfile.goto(100, mode="top")
+    assert wfile.line_range[1] == 99
+
+
+def test_windowed_file_scroll(windowed_file):
+    wfile = windowed_file
+    assert wfile.first_line == 0
+    wfile.first_line = 10
+    wfile.scroll(10)
+    assert wfile.first_line == 20
+    wfile.scroll(-10)
+    assert wfile.first_line == 10
+    wfile.scroll(-100)
+    # Can't go lower than the middle of the lowest window
+    assert wfile.first_line == 0
+    wfile.scroll(100)
+    assert wfile.line_range[1] == 99
+
+
+# def test_goto_command(windowed_file):
+#     wfile = windowed_file
+#     assert wfile.current_line == 10
+#     assert (DEFAULT_TOOLS_BIN / "goto").exists()
+
+#     # Use sys.executable to get the correct Python interpreter
+#     import sys
+#     cmd = f"{sys.executable} {DEFAULT_TOOLS_BIN}/goto 50"
+#     print(
+#         subprocess.check_output(
+#             cmd,
+#             shell=True,
+#             universal_newlines=True,
+#             stderr=subprocess.STDOUT,
+#             env=os.environ.copy(),  # Ensure we pass the current environment
+#         )
+#     )
+#     assert wfile.current_line == 50
+
+
+_DEFAULT_WINDOW_OUTPUT = """[File: {path} (100 lines total)]
+(10 more lines above)
+11:10
+12:11
+13:12
+14:13
+15:14
+16:15
+17:16
+18:17
+19:18
+20:19
+(80 more lines below)
+"""
+
+
+def test_print_window(windowed_file, capsys):
+    wfile = windowed_file
+    wfile.first_line = 10
+    wfile.print_window()
+    captured = capsys.readouterr()
+    print(captured.out)
+    expected = _DEFAULT_WINDOW_OUTPUT.format(path=wfile.path.resolve())
+    assert captured.out == expected
+
+
+_DEFAULT_WINDOW_OUTPUT_NEW_FILE = """[File: {path} (1 lines total)]
+1:
+"""
+
+
+def test_print_window_new_file(with_tmp_env_file, capsys):
+    wfile = create_test_file_with_content(with_tmp_env_file, "\n")
+    assert wfile.n_lines == 1
+    assert wfile.line_range == (0, 0)
+    wfile.print_window()
+    captured = capsys.readouterr()
+    print(captured.out)
+    expected = _DEFAULT_WINDOW_OUTPUT_NEW_FILE.format(path=wfile.path.resolve())
+    assert captured.out == expected

package/python/tests/tools/test_split_string.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+from tests.utils import make_python_tool_importable
+
+make_python_tool_importable("tools/windowed/lib/flake8_utils.py", "flake8_utils")
+from flake8_utils import Flake8Error, format_flake8_output  # type: ignore
+
+
+def test_partition_flake8_line():
+    assert Flake8Error.from_line("existing_lint error.py:12:41: E999 SyntaxError: invalid syntax") == Flake8Error(
+        "existing_lint error.py", 12, 41, "E999 SyntaxError: invalid syntax"
+    )
+
+
+# def test_update_previous_errors():
+#     previous_errors = [
+#         Flake8Error("existing_lint_error.py", 12, 41, "E999 SyntaxError: invalid syntax"),
+#         Flake8Error("existing_lint_error.py", 15, 41, "E999 SyntaxError: invalid syntax"),
+#         Flake8Error("existing_lint_error.py", 20, 41, "E999 SyntaxError: invalid syntax"),
+#     ]
+#     assert _update_previous_errors(previous_errors, (15, 18), 3) == [
+#         Flake8Error("existing_lint_error.py", 12, 41, "E999 SyntaxError: invalid syntax"),
+#         Flake8Error("existing_lint_error.py", 19, 41, "E999 SyntaxError: invalid syntax"),
+#     ]
+#     assert _update_previous_errors([], (15, 18), 3) == []
+
+
+def test_flake8_format_no_error_1():
+    assert (
+        format_flake8_output(
+            "a:12:41: e", previous_errors_string="a:12:41: e", replacement_window=(50, 51), replacement_n_lines=10
+        )
+        == ""
+    )
+
+
+def test_flake8_format_no_error_2():
+    assert (
+        format_flake8_output(
+            "a:12:41: e", previous_errors_string="a:13:41: e", replacement_window=(1, 2), replacement_n_lines=1
+        )
+        == ""
+    )
+
+
+def test_flake8_format_no_error_3():
+    assert (
+        format_flake8_output(
+            "a:12:41: e",
+            previous_errors_string="a:13:41: e",
+            replacement_window=(1, 2),
+            replacement_n_lines=1,
+            show_line_numbers=False,
+        )
+        == ""
+    )
+
+
+def test_flake8_format_error_1():
+    assert (
+        format_flake8_output(
+            "a:12:41: e",
+            previous_errors_string="a:13:41: e",
+            replacement_window=(12, 13),
+            replacement_n_lines=10,
+            show_line_numbers=False,
+        )
+        == "- e"
+    )
+
+
+def test_flake8_format_error_1_linenumbers():
+    assert (
+        format_flake8_output(
+            "a:12:41: e",
+            previous_errors_string="a:13:41: e",
+            replacement_window=(12, 13),
+            replacement_n_lines=10,
+            show_line_numbers=True,
+        )
+        == "- line 12 col 41: e"
+    )

package/python/tests/utils.py

@@ -0,0 +1,29 @@
+import importlib.machinery
+import importlib.util
+import sys
+from pathlib import Path
+
+
+def make_python_tool_importable(file_path: str | Path, module_name: str | None = None) -> None:
+    """Import a Python executable script and add it to sys.modules.
+
+    Args:
+        file_path: Path to the executable script (relative or absolute)
+        module_name: Name of the module to import the script as. If None, the filename stem is used.
+    """
+    # Convert to absolute path if needed
+    abs_path = Path(file_path).resolve().as_posix()
+    module_name = Path(file_path).stem if module_name is None else module_name
+
+    if module_name in sys.modules:
+        return  # Module already imported
+
+    loader = importlib.machinery.SourceFileLoader(module_name, abs_path)
+    spec = importlib.util.spec_from_file_location(module_name, abs_path, loader=loader)
+    if spec is None or spec.loader is None:
+        msg = f"Could not load module spec for {file_path}"
+        raise ImportError(msg)
+
+    module = importlib.util.module_from_spec(spec)
+    sys.modules[module_name] = module
+    spec.loader.exec_module(module)

package/python/tools/diff_state/bin/_state_diff_state

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+
+def main() -> None:
+    import json
+    import os
+    from pathlib import Path
+    import subprocess
+
+    from registry import registry
+
+    state_path = Path("/root/state.json")
+    if state_path.exists():
+        state = json.loads(state_path.read_text())
+    else:
+        state = {}
+
+    repo_root = registry.get("ROOT", os.getenv("ROOT"))
+
+    patch_path = Path("/root/model.patch")
+
+    subprocess.run(
+        f"git add -A && git diff --cached > {patch_path}",
+        shell=True,
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+        cwd=repo_root,
+    )
+
+    patch = patch_path.read_text(errors="backslashreplace")
+    state["diff"] = patch.strip()
+
+    state_path.write_text(json.dumps(state))
+
+
+def _del_diff():
+    from pathlib import Path
+    import json
+
+    state_path = Path("/root/state.json")
+    if state_path.exists():
+        state = json.loads(state_path.read_text())
+    else:
+        state = {}
+    state["diff"] = ""
+    state_path.write_text(json.dumps(state))
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except Exception as e:
+        _del_diff()

package/python/tools/diff_state/config.yaml

@@ -0,0 +1,2 @@
+tools: {}
+state_command: "_state_diff_state"

package/python/tools/edit_anthropic/bin/_state_anthropic

@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+
+import json
+import os
+from pathlib import Path
+
+
+def main():
+    state_path = Path("/root/state.json")
+    if state_path.exists():
+        state = json.loads(state_path.read_text())
+    else:
+        state = {}
+
+    state["working_dir"] = os.getcwd()
+
+    state_path.write_text(json.dumps(state))
+
+
+if __name__ == "__main__":
+    main()