Spaces:
Running
Running
File size: 3,412 Bytes
9d0fd45 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | """Virtual path mapping — hide host filesystem paths from agents."""
from __future__ import annotations
import os
from pathlib import Path
# ── Virtual → Physical mapping ──────────────────────────────────────────
# Virtual root that agents see
VIRTUAL_PREFIX = "/mnt/agent"
# Pre-rename virtual root: old sessions/traces still reference it, so reads
# keep resolving. Never emitted in new output.
LEGACY_VIRTUAL_PREFIX = "/mnt/frontier_agent"
# Virtual path segments
VIRTUAL_WORKSPACE = f"{VIRTUAL_PREFIX}/workspace"
VIRTUAL_OUTPUTS = f"{VIRTUAL_PREFIX}/outputs"
VIRTUAL_SKILLS = f"{VIRTUAL_PREFIX}/skills"
VIRTUAL_DATA = f"{VIRTUAL_PREFIX}/data"
# Physical roots (resolved at runtime)
_PROJECT_ROOT = str(Path(__file__).resolve().parent.parent.parent)
def _get_physical_roots() -> dict[str, str]:
"""Map virtual prefixes to physical directories."""
return {
VIRTUAL_WORKSPACE: os.path.join(_PROJECT_ROOT, "data", "workspace"),
VIRTUAL_OUTPUTS: "/tmp/agent-outputs",
VIRTUAL_SKILLS: os.path.join(_PROJECT_ROOT, "plugins", "skills"),
VIRTUAL_DATA: os.path.join(_PROJECT_ROOT, "data"),
}
def virtual_to_physical(virtual_path: str) -> str:
"""Translate a virtual path to physical path.
Args:
virtual_path: Path starting with /mnt/agent/...
Returns:
Physical path on host filesystem.
If not a virtual path, returns as-is.
"""
if virtual_path.startswith(LEGACY_VIRTUAL_PREFIX):
virtual_path = VIRTUAL_PREFIX + virtual_path[len(LEGACY_VIRTUAL_PREFIX):]
if not virtual_path.startswith(VIRTUAL_PREFIX):
return virtual_path
for v_prefix, p_prefix in _get_physical_roots().items():
if virtual_path.startswith(v_prefix):
relative = virtual_path[len(v_prefix):].lstrip("/")
return os.path.join(p_prefix, relative)
return virtual_path
def physical_to_virtual(physical_path: str) -> str:
"""Translate a physical path back to virtual path.
Args:
physical_path: Actual host filesystem path.
Returns:
Virtual path if it maps to a known root, otherwise as-is.
"""
resolved = str(Path(physical_path).resolve()) if physical_path else physical_path
for v_prefix, p_prefix in _get_physical_roots().items():
p_resolved = str(Path(p_prefix).resolve()) if os.path.exists(p_prefix) else p_prefix
if resolved.startswith(p_resolved):
relative = resolved[len(p_resolved):].lstrip("/")
return f"{v_prefix}/{relative}" if relative else v_prefix
return physical_path
def mask_paths_in_output(output: str) -> str:
"""Replace any physical paths in tool output with virtual equivalents.
Scans for known physical root patterns and replaces them.
This prevents the LLM from seeing host filesystem structure.
Args:
output: Raw tool output string.
Returns:
Output with physical paths replaced by virtual paths.
"""
if not output:
return output
for v_prefix, p_prefix in _get_physical_roots().items():
if p_prefix in output:
output = output.replace(p_prefix, v_prefix)
# Also mask the project root itself
if _PROJECT_ROOT in output:
output = output.replace(_PROJECT_ROOT, VIRTUAL_PREFIX)
return output
|