@elizaos/sweagent-root 2.0.0-alpha

package/python/tools/edit_anthropic/config.yaml

@@ -0,0 +1,56 @@
+tools:
+  str_replace_editor:
+    signature: |
+      str_replace_editor <command> <path> [<file_text>] [<view_range>] [<old_str>] [<new_str>] [<insert_line>]
+    # This docstrings was taken from openhands:
+    # https://github.com/All-Hands-AI/OpenHands/blob/main/openhands/agenthub/codeact_agent/function_calling.py
+    docstring: >
+      Custom editing tool for viewing, creating and editing files
+      * State is persistent across command calls and discussions with the user
+      * If `path` is a file, `view` displays the result of applying `cat -n`. If `path` is a directory, `view` lists non-hidden files and directories up to 2 levels deep
+      * The `create` command cannot be used if the specified `path` already exists as a file
+      * If a `command` generates a long output, it will be truncated and marked with `<response clipped>`
+      * The `undo_edit` command will revert the last edit made to the file at `path`
+
+      Notes for using the `str_replace` command:
+      * The `old_str` parameter should match EXACTLY one or more consecutive lines from the original file. Be mindful of whitespaces!
+      * If the `old_str` parameter is not unique in the file, the replacement will not be performed. Make sure to include enough context in `old_str` to make it unique
+      * The `new_str` parameter should contain the edited lines that should replace the `old_str`
+    arguments:
+      - name: command
+        type: string
+        description: "The commands to run. Allowed options are: `view`, `create`, `str_replace`, `insert`, `undo_edit`."
+        required: true
+        enum: ["view", "create", "str_replace", "insert", "undo_edit"]
+      - name: path
+        type: string
+        description: "Absolute path to file or directory, e.g. `/testbed/file.py` or `/testbed`."
+        required: true
+      - name: file_text
+        type: string
+        description: "Required parameter of `create` command, with the content of the file to be created."
+        required: false
+        argument_format: "--file_text {{value}}"
+      - name: old_str
+        type: string
+        description: "Required parameter of `str_replace` command containing the string in `path` to replace."
+        required: false
+        argument_format: "--old_str {{value}}"
+      - name: new_str
+        type: string
+        description: "Optional parameter of `str_replace` command containing the new string (if not given, no string will be added). Required parameter of `insert` command containing the string to insert."
+        required: false
+        argument_format: "--new_str {{value}}"
+      - name: insert_line
+        type: integer
+        description: "Required parameter of `insert` command. The `new_str` will be inserted AFTER the line `insert_line` of `path`."
+        required: false
+        argument_format: "--insert_line {{value}}"
+      - name: view_range
+        type: array
+        items:
+          type: integer
+        description: "Optional parameter of `view` command when `path` points to a file. If none is given, the full file is shown. If provided, the file will be shown in the indicated line number range, e.g. [11, 12] will show lines 11 and 12. Indexing at 1 to start. Setting `[start_line, -1]` shows all lines from `start_line` to the end of the file."
+        required: false
+        argument_format: "--view_range {{value|join(' ')}}"
+state_command: "_state_anthropic"

package/python/tools/edit_anthropic/install.sh

@@ -0,0 +1,3 @@
+# Ignore failures, see https://github.com/SWE-agent/SWE-agent/issues/1179
+pip install 'tree-sitter==0.21.3' || true
+pip install 'tree-sitter-languages' || true

package/python/tools/filemap/bin/filemap

@@ -0,0 +1,45 @@
+#!/root/miniconda3/bin/python
+
+import argparse
+import warnings
+
+# tree_sitter is throwing a FutureWarning
+warnings.simplefilter("ignore", category=FutureWarning)
+from tree_sitter_languages import get_language, get_parser
+
+parser = argparse.ArgumentParser(
+    description="Print the contents of a Python file, skipping lengthy function and method definitions."
+)
+parser.add_argument("file_path", type=str, help="The path to the file to be read")
+args = parser.parse_args()
+
+# We assume that all input files are Python.
+parser = get_parser("python")
+language = get_language("python")
+file_contents = open(args.file_path).read()
+
+# We assume that files are utf8 encoded.
+tree = parser.parse(bytes(file_contents, "utf8"))
+
+# See https://tree-sitter.github.io/tree-sitter/using-parsers#pattern-matching-with-queries.
+query = language.query("""
+(function_definition
+  body: (_) @body)
+""")
+
+# TODO: consider special casing docstrings such that they are not elided. This
+# could be accomplished by checking whether `body.text.decode('utf8')` starts
+# with `"""` or `'''`.
+elide_line_ranges = [
+    (node.start_point[0], node.end_point[0])
+    for node, _ in query.captures(tree.root_node)
+    # Only elide if it's sufficiently long
+    if node.end_point[0] - node.start_point[0] >= 5
+]
+# Note that tree-sitter line numbers are 0-indexed, but we display 1-indexed.
+elide_lines = {line for start, end in elide_line_ranges for line in range(start, end + 1)}
+elide_messages = [(start, f"... eliding lines {start+1}-{end+1} ...") for start, end in elide_line_ranges]
+for i, line in sorted(
+    elide_messages + [(i, line) for i, line in enumerate(file_contents.splitlines()) if i not in elide_lines]
+):
+    print(f"{i+1:6d} {line}")

package/python/tools/filemap/config.yaml

@@ -0,0 +1,9 @@
+tools:
+  filemap:
+    signature: "filemap <file_path>"
+    docstring: "Print the contents of a Python file, skipping lengthy function and method definitions."
+    arguments:
+      - name: file_path
+        type: string
+        description: The path to the file to be read
+        required: true

package/python/tools/filemap/install.sh

@@ -0,0 +1,2 @@
+pip install 'tree-sitter==0.21.3'
+pip install 'tree-sitter-languages'

package/python/tools/forfeit/bin/exit_forfeit

@@ -0,0 +1,5 @@
+main() {
+    echo "###SWE-AGENT-EXIT-FORFEIT###"
+}
+
+main "$@"

package/python/tools/forfeit/config.yaml

@@ -0,0 +1,5 @@
+tools:
+  exit_forfeit:
+    signature: "exit_forfeit"
+    docstring: "Give up on the current challenge and terminate the session."
+    arguments: []

package/python/tools/image_tools/bin/view_image

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+# view_image – view an image file as a base64-encoded markdown image
+import base64
+import mimetypes
+import pathlib
+import sys
+
+VALID_MIME_TYPES = {
+    "image/png",
+    "image/jpeg",
+    "image/webp",
+}
+
+if len(sys.argv) != 2:
+    sys.exit(f"usage: {pathlib.Path(sys.argv[0]).name} <image-file>")
+
+img_path = pathlib.Path(sys.argv[1])
+
+if not img_path.exists():
+    sys.exit(f"Error: File '{img_path}' does not exist")
+
+if not img_path.is_file():
+    sys.exit(f"Error: '{img_path}' is not a file")
+
+try:
+    mime = mimetypes.guess_type(img_path.name)[0]
+    if mime not in VALID_MIME_TYPES:
+        sys.exit(f"Error: Unsupported image type: {mime}. Valid types are: {', '.join(VALID_MIME_TYPES)}")
+
+    # read the file, base64-encode, and convert bytes → str
+    b64 = base64.b64encode(img_path.read_bytes()).decode("ascii")
+
+    # write the exact markdown snippet to stdout
+    print(f"![{img_path.as_posix()}](data:{mime};base64,{b64})")
+except Exception as e:
+    sys.exit(f"Error processing image: {e}")

package/python/tools/image_tools/config.yaml

@@ -0,0 +1,9 @@
+tools:
+  view_image:
+    signature: "view_image <image_file>"
+    docstring: "view an image file"
+    arguments:
+      - name: image_file
+        type: string
+        description: "the path to the image file to view"
+        required: true

package/python/tools/multilingual_setup/bin/do_nothing

@@ -0,0 +1,2 @@
+# This will never be called, but having a file here
+# ensures that chmod +x commands on the bin directory don't fail

package/python/tools/multilingual_setup/config.yaml

@@ -0,0 +1 @@
+tools: {}

package/python/tools/multilingual_setup/install.sh

@@ -0,0 +1,45 @@
+#!/usr/bin/env bash
+
+# Define variables to exclude
+EXCLUDE_VARS="PWD|LANG|PYTHONPATH|ROOT|PS0|PS1|PS2|_|OLDPWD|LC_ALL|LANG|LSCOLORS|SHLVL"
+
+
+echo "Original Environment Variables:"
+env | sort
+
+# Only add Python 3.11 to PATH if no python exists
+if ! command -v python &> /dev/null; then
+    echo -e "\nNo Python found in system, adding Python 3.11 to PATH"
+    export PATH="/root/python3.11/bin:$PATH"
+
+    # Create python/pip aliases
+    ln -s "/root/python3.11/bin/python3" "/root/python3.11/bin/python"
+    ln -s "/root/python3.11/bin/pip3" "/root/python3.11/bin/pip"
+    echo "Created symlinks: python -> python3, pip -> pip3"
+else
+    echo -e "\nPython already exists in system, skipping Python 3.11 setup"
+fi
+
+# Attempt to read and set process 1 environment
+echo -e "\nSetting environment variables from /proc/1/environ..."
+if [ -r "/proc/1/environ" ]; then
+    while IFS= read -r -d '' var; do
+        # Skip excluded variables
+        if ! echo "$var" | grep -qE "^(${EXCLUDE_VARS})="; then
+            # If the variable is PATH, append and deduplicate
+            if [[ "$var" =~ ^PATH= ]]; then
+                # Combine paths and remove duplicates while preserving order
+                export PATH="$(echo "${PATH}:${var#PATH=}" | tr ':' '\n' | awk '!seen[$0]++' | tr '\n' ':' | sed 's/:$//')"
+            else
+                export "$var"
+            fi
+        fi
+    done < /proc/1/environ
+    echo "Successfully imported environment from /proc/1/environ"
+else
+    echo "Cannot access /proc/1/environ - Permission denied"
+fi
+
+# Print updated environment variables
+echo -e "\nUpdated Environment Variables:"
+env | sort

package/python/tools/registry/bin/_read_env

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+
+import sys
+
+from registry import registry  # type: ignore
+
+if __name__ == "__main__":
+    var_name = sys.argv[1]
+    default_value = sys.argv[2] if len(sys.argv) > 2 else ""
+    print(registry.get(var_name, default_value))

package/python/tools/registry/bin/_write_env

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+
+import sys
+
+from registry import registry  # type: ignore
+
+if __name__ == "__main__":
+    var_name = sys.argv[1]
+    var_value = sys.argv[2] if len(sys.argv) > 2 else ""
+    registry[var_name] = var_value

package/python/tools/registry/config.yaml

@@ -0,0 +1 @@
+tools: {}

package/python/tools/registry/install.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+
+# script_dir=$(dirname "$(readlink -f "$0")")
+bundle_dir=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+
+export PYTHONPATH="$bundle_dir/lib":$PYTHONPATH

package/python/tools/registry/lib/registry.py

@@ -0,0 +1,56 @@
+import json
+import os
+from pathlib import Path
+from typing import Any
+
+
+class EnvRegistry:
+    """Read and write variables into a file. This is used to persist state between tool
+    calls without using environment variables (which are problematic because you cannot
+    set them in a subprocess).
+
+    The default file location is `/root/.swe-agent-env`, though this can be overridden
+    by the `env_file` argument or the `SWE_AGENT_ENV_FILE` environment variable.
+    """
+
+    def __init__(self, env_file: Path | None = None):
+        self._env_file = env_file
+
+    @property
+    def env_file(self) -> Path:
+        if self._env_file is None:
+            env_file = Path(os.environ.get("SWE_AGENT_ENV_FILE", "/root/.swe-agent-env"))
+        else:
+            env_file = self._env_file
+        if not env_file.exists():
+            env_file.write_text("{}")
+        return env_file
+
+    def __getitem__(self, key: str) -> str:
+        return json.loads(self.env_file.read_text())[key]
+
+    def get(self, key: str, default_value: Any = None, fallback_to_env: bool = True) -> Any:
+        """Get a value from registry:
+
+        Args:
+            key: The key to get the value for.
+            default_value: The default value to return if the key is not found in the registry.
+            fallback_to_env: If True, fallback to environment variables if the key is not found in the registry.
+                If there's no environment variable, return the default value.
+        """
+        if fallback_to_env and key in os.environ:
+            default_value = os.environ[key]
+        return json.loads(self.env_file.read_text()).get(key, default_value)
+
+    def get_if_none(self, value: Any, key: str, default_value: Any = None) -> Any:
+        if value is not None:
+            return value
+        return self.get(key, default_value)
+
+    def __setitem__(self, key: str, value: Any):
+        env = json.loads(self.env_file.read_text())
+        env[key] = value
+        self.env_file.write_text(json.dumps(env))
+
+
+registry = EnvRegistry()

package/python/tools/review_on_submit_m/README.md

@@ -0,0 +1,6 @@
+# Review on submit.
+
+Provides an alternative for `submit` that does not immediately submit, but asks the
+agent to perform additional reviewing steps.
+
+Only `submit -f` will trigger the real submit.

package/python/tools/review_on_submit_m/bin/submit

@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+
+import argparse
+from pathlib import Path
+import subprocess
+import sys
+import os
+import io
+
+from registry import registry
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Submit changes for review")
+    parser.add_argument("-f", "--force", action="store_true", help="Force submit without review")
+    args = parser.parse_args()
+
+    repo_root = registry.get("ROOT", os.getenv("ROOT"))
+    assert repo_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")
+
+    submit_review_messages = registry.get("SUBMIT_REVIEW_MESSAGES", [])
+    n_stages = len(submit_review_messages)
+    current_stage = registry.get("SUBMIT_STAGE", 0)
+    if not args.force and current_stage != n_stages:
+        message = submit_review_messages[current_stage]
+        message = message.replace("{{diff}}", patch)
+        message = message.replace("{{problem_statement}}", registry.get("PROBLEM_STATEMENT", ""))
+        registry["SUBMIT_STAGE"] = current_stage + 1
+        print(message)
+        sys.exit(0)
+
+    print("<<SWE_AGENT_SUBMISSION>>")
+    print(patch)
+    print("<<SWE_AGENT_SUBMISSION>>")
+
+
+if __name__ == "__main__":
+    # There are some super strange "ascii can't decode x" errors when printing to the terminal
+    # that can be solved with setting the default encoding for stdout
+    # (note that python3.6 doesn't have the reconfigure method)
+    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
+    main()

package/python/tools/review_on_submit_m/config.yaml

@@ -0,0 +1,6 @@
+tools:
+  submit:
+    signature: "submit"
+    docstring: "submits the current file"
+    # Do not actually show the -f argument to the model, only
+    # use it from the agent for submission after error

package/python/tools/search/bin/find_file

@@ -0,0 +1,31 @@
+main() {
+    if [ $# -eq 1 ]; then
+        local file_name="$1"
+        local dir="./"
+    elif [ $# -eq 2 ]; then
+        local file_name="$1"
+        if [ -d "$2" ]; then
+            local dir="$2"
+        else
+            echo "Directory $2 not found"
+            return
+        fi
+    else
+        echo "Usage: find_file <file_name> [<dir>]"
+        return
+    fi
+
+    dir=$(realpath "$dir")
+    local matches=$(find "$dir" -type f -name "$file_name")
+    # if no matches, return
+    if [ -z "$matches" ]; then
+        echo "No matches found for \"$file_name\" in $dir"
+        return
+    fi
+    # Calculate total number of matches
+    local num_matches=$(echo "$matches" | wc -l | awk '{$1=$1; print $0}')
+    echo "Found $num_matches matches for \"$file_name\" in $dir:"
+    echo "$matches" | awk '{print $0}'
+}
+
+main "$@"

package/python/tools/search/bin/search_dir

@@ -0,0 +1,39 @@
+main() {
+    if [ $# -eq 1 ]; then
+        local search_term="$1"
+        local dir="./"
+    elif [ $# -eq 2 ]; then
+        local search_term="$1"
+        if [ -d "$2" ]; then
+            local dir="$2"
+        else
+            echo "Directory $2 not found"
+            return
+        fi
+    else
+        echo "Usage: search_dir <search_term> [<dir>]"
+        return
+    fi
+    dir=$(realpath "$dir")
+    local matches=$(find "$dir" -type f ! -path '*/.*' -exec grep -nIH -- "$search_term" {} + | cut -d: -f1 | sort | uniq -c)
+    # if no matches, return
+    if [ -z "$matches" ]; then
+        echo "No matches found for \"$search_term\" in $dir"
+        return
+    fi
+    # Calculate total number of matches
+    local num_matches=$(echo "$matches" | awk '{sum+=$1} END {print sum}')
+    # calculate total number of files matched
+    local num_files=$(echo "$matches" | wc -l | awk '{$1=$1; print $0}')
+    # if num_files is > 100, print an error
+    if [ $num_files -gt 100 ]; then
+        echo "More than $num_files files matched for \"$search_term\" in $dir. Please narrow your search."
+        return
+    fi
+
+    echo "Found $num_matches matches for \"$search_term\" in $dir:"
+    echo "$matches" | awk '{$2=$2; gsub(/^\.+\/+/, "./", $2); print $2 " ("$1" matches)"}'
+    echo "End of matches for \"$search_term\" in $dir"
+}
+
+main "$@"

package/python/tools/search/bin/search_file

@@ -0,0 +1,55 @@
+main() {
+    # Check if the first argument is provided
+    local search_term="${1:-}"
+    if [ -z "${search_term}" ]; then
+        echo "Usage: search_file <search_term> [<file>]"
+        return
+    fi
+    # Check if the second argument is provided
+    if [ $# -ge 2 ]; then
+        # Check if the provided argument is a valid file
+        if [ -f "$2" ]; then
+            local file="$2"  # Set file if valid
+        else
+            echo "Usage: search_file <search_term> [<file>]"
+            echo "Error: File name $2 not found. Please provide a valid file name."
+            return  # Exit if the file is not valid
+        fi
+    else
+        local CURRENT_FILE=$(_read_env CURRENT_FILE)
+        # Check if a file is open
+        if [ -z "${CURRENT_FILE:-}" ]; then
+            echo "No file open. Use the open command first."
+            return  # Exit if no file is open
+        fi
+        local file="$CURRENT_FILE"  # Set file to the current open file
+    fi
+    local search_term="$1"
+    file=$(realpath "$file")
+    # Use grep to directly get the desired formatted output
+    local matches=$(grep -nH -- "$search_term" "$file")
+    # Check if no matches were found
+    if [ -z "${matches:-}" ]; then
+        echo "No matches found for \"$search_term\" in $file"
+        return
+    fi
+    # Calculate total number of matches
+    local num_matches=$(echo "$matches" | wc -l | awk '{$1=$1; print $0}')
+
+    # calculate total number of lines matched
+    local num_lines=$(echo "$matches" | cut -d: -f1 | sort | uniq | wc -l | awk '{$1=$1; print $0}')
+    # if num_lines is > 100, print an error
+    if [ $num_lines -gt 100 ]; then
+        echo "More than $num_lines lines matched for \"$search_term\" in $file. Please narrow your search."
+        return
+    fi
+
+    # Print the total number of matches and the matches themselves
+    echo "Found $num_matches matches for \"$search_term\" in $file:"
+    echo "$matches" | cut -d: -f1-2 | sort -u -t: -k2,2n | while IFS=: read -r filename line_number; do
+        echo "Line $line_number:$(sed -n "${line_number}p" "$file")"
+    done
+    echo "End of matches for \"$search_term\" in $file"
+}
+
+main "$@"

package/python/tools/search/config.yaml

@@ -0,0 +1,37 @@
+tools:
+  find_file:
+    signature: "find_file <file_name> [<dir>]"
+    docstring: "finds all files with the given name or pattern in dir. If dir is not provided, searches in the current directory"
+    arguments:
+      - name: file_name
+        type: string
+        description: "the name of the file or pattern to search for. supports shell-style wildcards (e.g. *.py)"
+        required: true
+      - name: dir
+        type: string
+        description: "the directory to search in (if not provided, searches in the current directory)"
+        required: false
+  search_dir:
+    signature: "search_dir <search_term> [<dir>]"
+    docstring: "searches for search_term in all files in dir. If dir is not provided, searches in the current directory"
+    arguments:
+      - name: search_term
+        type: string
+        description: "the term to search for"
+        required: true
+      - name: dir
+        type: string
+        description: "the directory to search in (if not provided, searches in the current directory)"
+        required: false
+  search_file:
+    signature: "search_file <search_term> [<file>]"
+    docstring: "searches for search_term in file. If file is not provided, searches in the current open file"
+    arguments:
+      - name: search_term
+        type: string
+        description: "the term to search for"
+        required: true
+      - name: file
+        type: string
+        description: "the file to search in (if not provided, searches in the current open file)"
+        required: false

package/python/tools/search/install.sh

@@ -0,0 +1,3 @@
+_write_env SEARCH_RESULTS "()"
+_write_env SEARCH_FILES "()"
+_write_env SEARCH_INDEX 0

package/python/tools/submit/bin/submit

@@ -0,0 +1,17 @@
+main() {
+    cd $ROOT
+
+    # Check if the patch file exists and is non-empty
+    if [ -s "/root/test.patch" ]; then
+        # Apply the patch in reverse
+        git apply -R < "/root/test.patch"
+    fi
+
+    git add -A
+    git diff --cached > /root/model.patch
+    echo "<<SWE_AGENT_SUBMISSION>>"
+    cat /root/model.patch
+    echo "<<SWE_AGENT_SUBMISSION>>"
+}
+
+main "$@"

package/python/tools/submit/config.yaml

@@ -0,0 +1,5 @@
+tools:
+  submit:
+    signature: "submit"
+    docstring: "submits the current file"
+    arguments: []

package/python/tools/web_browser/bin/click_mouse

@@ -0,0 +1,41 @@
+#!/root/python3.11/bin/python3
+from __future__ import annotations
+
+import sys
+from argparse import ArgumentParser
+from pathlib import Path
+
+lib_path = str(Path(__file__).resolve().parent.parent / "lib")
+sys.path.insert(0, lib_path)
+
+from web_browser_config import ClientConfig
+from web_browser_utils import (
+    _autosave_screenshot_from_response,
+    _print_response_with_metadata,
+    send_request,
+)
+
+config = ClientConfig()
+
+
+def click(x, y, button):
+    """Click at the specified coordinates (x, y)."""
+    response = send_request(
+        config.port,
+        "click",
+        "POST",
+        {"x": x, "y": y, "button": button, "return_screenshot": config.autoscreenshot},
+    )
+    if response is None:
+        return
+    _print_response_with_metadata(response)
+    _autosave_screenshot_from_response(response, config.screenshot_mode)
+
+
+if __name__ == "__main__":
+    parser = ArgumentParser()
+    parser.add_argument("x", type=int)
+    parser.add_argument("y", type=int)
+    parser.add_argument("button", nargs="?", default="left", choices=["left", "right"])
+    args = parser.parse_args()
+    click(args.x, args.y, args.button)