56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
from dataclasses import dataclass, field
|
|
from typing import Type
|
|
import tyro
|
|
from loguru import logger
|
|
|
|
from deepagents.backends import LocalShellBackend
|
|
|
|
from lang_agent.fs_bkends.base import BaseFilesystemBackend, FilesystemBackendConfig
|
|
|
|
|
|
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
|
@dataclass
|
|
class LocalShellConfig(FilesystemBackendConfig):
|
|
_target:Type = field(default_factory=lambda:LocalShell)
|
|
|
|
workspace_dir:str = "./workspace"
|
|
"""path to workspace directory"""
|
|
|
|
skills_dir:str = "./workspace/skills"
|
|
"""path to directory containing skill files"""
|
|
|
|
rt_skills_dir:str = "/skills"
|
|
"""path to directory with skills in runtime directory"""
|
|
|
|
|
|
class LocalShell(BaseFilesystemBackend):
|
|
def __init__(self, config:LocalShellConfig):
|
|
logger.warning("Caution: The LocalShell backend grants direct access to the local system shell. Improper use can pose significant security and safety risks, including unintended code execution and file access. Use this backend with extreme care.")
|
|
self.config = config
|
|
self._build_backend()
|
|
|
|
def _build_backend(self):
|
|
self.backend = LocalShellBackend(root_dir=self.config.workspace_dir,
|
|
virtual_mode=True,
|
|
# env={"PATH": "/usr/bin:/bin"}
|
|
inherit_env=True)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
# Instantiate a LocalShell instance with the default config
|
|
config = LocalShellConfig()
|
|
shell = LocalShell(config)
|
|
|
|
# Try checking access to 'npx'
|
|
try:
|
|
result = shell.backend.execute("npx --version")
|
|
if result.exit_code == 0:
|
|
print("npx is available, version:", result.output.strip())
|
|
else:
|
|
print("npx returned non-zero exit code:", result.exit_code, file=sys.stderr)
|
|
print("output:", result.output, file=sys.stderr)
|
|
except Exception as e:
|
|
print("Could not access 'npx':", str(e), file=sys.stderr) |