Update virtual_fs.py

virtual_fs.py

@@ -1,39 +1,58 @@
-"""
-Virtual file system used by the deep agent.
-It stores files in memory and can be dumped to the real file system.
+"""Simple in‑memory virtual file system.
+
+The VFS is just a dictionary mapping file names to string content.
+It is shared between all tools and can be exported to the real
+filesystem when requested.
 """
 
-import os
+from __future__ import annotations
+
 from pathlib import Path
-from typing import Dict, Optional
+from typing import Dict, Iterable
+
 
 class VirtualFileSystem:
-    """A simple in‑memory file system.
+    """In‑memory file system.
 
-    Files are stored as a mapping from a relative path to its content.
+    Attributes
+    ----------
+    files : Dict[str, str]
+        Mapping of file names to file content.
     """
 
-    def __init__(self, base_dir: str = "virtual_fs"):
-        self.base_dir = Path(base_dir)
+    def __init__(self) -> None:
         self.files: Dict[str, str] = {}
 
-    def write_file(self, path: str, content: str) -> None:
-        norm_path = Path(path).as_posix()
-        self.files[norm_path] = content
+    def create_file(self, name: str, content: str) -> None:
+        """Create or overwrite a virtual file."""
+        self.files[name] = content
 
-    def read_file(self, path: str) -> Optional[str]:
-        norm_path = Path(path).as_posix()
-        return self.files.get(norm_path)
+    def list_files(self) -> Iterable[str]:
+        """Return an iterable of file names."""
+        return self.files.keys()
 
-    def list_files(self) -> Dict[str, str]:
-        return dict(self.files)
+    def read_file(self, name: str) -> str:
+        """Return the content of a virtual file.
 
-    def dump_to_real_fs(self, real_root: str = "real_fs") -> None:
-        real_root_path = Path(real_root)
-        for rel_path, content in self.files.items():
-            file_path = real_root_path / rel_path
+        Raises ``KeyError`` if the file does not exist.
+        """
+        return self.files[name]
+
+    def export_to_disk(self, root: str | Path = ".") -> None:
+        """Export all virtual files to the real filesystem.
+
+        Parameters
+        ----------
+        root : str | Path, optional
+            Root directory where the virtual files will be written.
+            Defaults to the current working directory.
+        """
+        root_path = Path(root).expanduser().resolve()
+        root_path.mkdir(parents=True, exist_ok=True)
+        for name, content in self.files.items():
+            file_path = root_path / name
             file_path.parent.mkdir(parents=True, exist_ok=True)
             file_path.write_text(content, encoding="utf-8")
 
-    def __repr__(self) -> str:
-        return f"VirtualFileSystem(files={list(self.files.keys())})"
+    def __repr__(self) -> str:  # pragma: no cover
+        return f"VirtualFileSystem({len(self.files)} files)"