first

agent_runtime/image_source_map.py

@@ -0,0 +1,32 @@
+import threading
+from pathlib import Path
+from typing import Dict
+
+_MAP_LOCK = threading.Lock()
+_SAFE_TO_ORIGINAL: Dict[str, str] = {}
+MAX_RECORDS = 2048
+
+
+def _normalize_path(path_or_uri: str) -> str:
+    raw = path_or_uri.strip()
+    if raw.startswith('file://'):
+        raw = raw[len('file://'):]
+    return str(Path(raw).expanduser().resolve())
+
+
+def register_safe_image(safe_path: str, original_path: str) -> None:
+    safe_abs = _normalize_path(safe_path)
+    original_abs = _normalize_path(original_path)
+    with _MAP_LOCK:
+        _SAFE_TO_ORIGINAL[safe_abs] = original_abs
+        if len(_SAFE_TO_ORIGINAL) <= MAX_RECORDS:
+            return
+        overflow = len(_SAFE_TO_ORIGINAL) - MAX_RECORDS
+        for key in list(_SAFE_TO_ORIGINAL.keys())[:overflow]:
+            del _SAFE_TO_ORIGINAL[key]
+
+
+def resolve_original_image(path_or_uri: str) -> str:
+    safe_abs = _normalize_path(path_or_uri)
+    with _MAP_LOCK:
+        return _SAFE_TO_ORIGINAL.get(safe_abs, safe_abs)