eb0c52b720
- Add PyInstaller spec file and build script for creating standalone executables - Add install.sh for curl | sh installation from GitHub releases - Add GitHub Actions workflow for multi-platform builds (macOS, Linux, Windows) - Move sandbox-only deps (playwright, ipython, libtmux, etc.) to optional extras - Make google-cloud-aiplatform optional ([vertex] extra) to reduce binary size - Use lazy imports in tool actions to avoid loading sandbox deps at startup - Add -v/--version flag to CLI - Add website and Discord links to completion message - Binary size: ~97MB (down from ~120MB with all deps)
145 lines
3.9 KiB
Python
145 lines
3.9 KiB
Python
import json
|
|
import re
|
|
from pathlib import Path
|
|
from typing import Any, cast
|
|
|
|
from strix.tools.registry import register_tool
|
|
|
|
|
|
def _parse_file_editor_output(output: str) -> dict[str, Any]:
|
|
try:
|
|
pattern = r"<oh_aci_output_[^>]+>\n(.*?)\n</oh_aci_output_[^>]+>"
|
|
match = re.search(pattern, output, re.DOTALL)
|
|
|
|
if match:
|
|
json_str = match.group(1)
|
|
data = json.loads(json_str)
|
|
return cast("dict[str, Any]", data)
|
|
return {"output": output, "error": None}
|
|
except (json.JSONDecodeError, AttributeError):
|
|
return {"output": output, "error": None}
|
|
|
|
|
|
@register_tool
|
|
def str_replace_editor(
|
|
command: str,
|
|
path: str,
|
|
file_text: str | None = None,
|
|
view_range: list[int] | None = None,
|
|
old_str: str | None = None,
|
|
new_str: str | None = None,
|
|
insert_line: int | None = None,
|
|
) -> dict[str, Any]:
|
|
from openhands_aci import file_editor
|
|
|
|
try:
|
|
path_obj = Path(path)
|
|
if not path_obj.is_absolute():
|
|
path = str(Path("/workspace") / path_obj)
|
|
|
|
result = file_editor(
|
|
command=command,
|
|
path=path,
|
|
file_text=file_text,
|
|
view_range=view_range,
|
|
old_str=old_str,
|
|
new_str=new_str,
|
|
insert_line=insert_line,
|
|
)
|
|
|
|
parsed = _parse_file_editor_output(result)
|
|
|
|
if parsed.get("error"):
|
|
return {"error": parsed["error"]}
|
|
|
|
return {"content": parsed.get("output", result)}
|
|
|
|
except (OSError, ValueError) as e:
|
|
return {"error": f"Error in {command} operation: {e!s}"}
|
|
|
|
|
|
@register_tool
|
|
def list_files(
|
|
path: str,
|
|
recursive: bool = False,
|
|
) -> dict[str, Any]:
|
|
from openhands_aci.utils.shell import run_shell_cmd
|
|
|
|
try:
|
|
path_obj = Path(path)
|
|
if not path_obj.is_absolute():
|
|
path = str(Path("/workspace") / path_obj)
|
|
path_obj = Path(path)
|
|
|
|
if not path_obj.exists():
|
|
return {"error": f"Directory not found: {path}"}
|
|
|
|
if not path_obj.is_dir():
|
|
return {"error": f"Path is not a directory: {path}"}
|
|
|
|
cmd = f"find '{path}' -type f -o -type d | head -500" if recursive else f"ls -1a '{path}'"
|
|
|
|
exit_code, stdout, stderr = run_shell_cmd(cmd)
|
|
|
|
if exit_code != 0:
|
|
return {"error": f"Error listing directory: {stderr}"}
|
|
|
|
items = stdout.strip().split("\n") if stdout.strip() else []
|
|
|
|
files = []
|
|
dirs = []
|
|
|
|
for item in items:
|
|
item_path = item if recursive else str(Path(path) / item)
|
|
item_path_obj = Path(item_path)
|
|
|
|
if item_path_obj.is_file():
|
|
files.append(item)
|
|
elif item_path_obj.is_dir():
|
|
dirs.append(item)
|
|
|
|
return {
|
|
"files": sorted(files),
|
|
"directories": sorted(dirs),
|
|
"total_files": len(files),
|
|
"total_dirs": len(dirs),
|
|
"path": path,
|
|
"recursive": recursive,
|
|
}
|
|
|
|
except (OSError, ValueError) as e:
|
|
return {"error": f"Error listing directory: {e!s}"}
|
|
|
|
|
|
@register_tool
|
|
def search_files(
|
|
path: str,
|
|
regex: str,
|
|
file_pattern: str = "*",
|
|
) -> dict[str, Any]:
|
|
from openhands_aci.utils.shell import run_shell_cmd
|
|
|
|
try:
|
|
path_obj = Path(path)
|
|
if not path_obj.is_absolute():
|
|
path = str(Path("/workspace") / path_obj)
|
|
|
|
if not Path(path).exists():
|
|
return {"error": f"Directory not found: {path}"}
|
|
|
|
escaped_regex = regex.replace("'", "'\"'\"'")
|
|
|
|
cmd = f"rg --line-number --glob '{file_pattern}' '{escaped_regex}' '{path}'"
|
|
|
|
exit_code, stdout, stderr = run_shell_cmd(cmd)
|
|
|
|
if exit_code not in {0, 1}:
|
|
return {"error": f"Error searching files: {stderr}"}
|
|
return {"output": stdout if stdout else "No matches found"}
|
|
|
|
except (OSError, ValueError) as e:
|
|
return {"error": f"Error searching files: {e!s}"}
|
|
|
|
|
|
# ruff: noqa: TRY300
|