fix circular import

Author: SentienceDEV

sentience/tools/context.py

@@ -7,8 +7,7 @@
 
 if TYPE_CHECKING:
     from ..agent_runtime import AgentRuntime
-
-from .filesystem import FileSandbox
+    from .filesystem import FileSandbox
 
 
 class BackendCapabilities(BaseModel):
@@ -38,12 +37,14 @@ class ToolContext:
     def __init__(
         self,
         runtime: AgentRuntime,
-        files: FileSandbox | None = None,
+        files: "FileSandbox" | None = None,
         base_dir: Path | None = None,
     ) -> None:
         self.runtime = runtime
         if files is None:
             root = base_dir or (Path.cwd() / ".sentience" / "files")
+            from .filesystem import FileSandbox
+
             files = FileSandbox(root)
         self.files = files
 

sentience/tools/filesystem.py

@@ -3,8 +3,10 @@
 from pathlib import Path
 
 from pydantic import BaseModel, Field
+from typing import TYPE_CHECKING
 
-from .context import ToolContext
+if TYPE_CHECKING:
+    from .context import ToolContext
 from .registry import ToolRegistry
 
 
@@ -92,7 +94,7 @@ def register_filesystem_tools(
 ) -> ToolRegistry:
     """Register sandboxed filesystem tools."""
 
-    def _get_files(ctx: ToolContext | None) -> FileSandbox:
+    def _get_files(ctx: "ToolContext" | None) -> FileSandbox:
         if ctx is not None:
             return ctx.files
         if sandbox is not None:
@@ -105,7 +107,7 @@ def _get_files(ctx: ToolContext | None) -> FileSandbox:
         output_model=ReadFileOutput,
         description="Read a file from the sandbox.",
     )
-    async def read_file(ctx: ToolContext | None, params: ReadFileInput) -> ReadFileOutput:
+    async def read_file(ctx: "ToolContext" | None, params: ReadFileInput) -> ReadFileOutput:
         files = _get_files(ctx)
         return ReadFileOutput(content=files.read_text(params.path))
 
@@ -115,7 +117,7 @@ async def read_file(ctx: ToolContext | None, params: ReadFileInput) -> ReadFileO
         output_model=WriteFileOutput,
         description="Write a file to the sandbox.",
     )
-    async def write_file(ctx: ToolContext | None, params: WriteFileInput) -> WriteFileOutput:
+    async def write_file(ctx: "ToolContext" | None, params: WriteFileInput) -> WriteFileOutput:
         files = _get_files(ctx)
         written = files.write_text(params.path, params.content, overwrite=params.overwrite)
         return WriteFileOutput(path=params.path, bytes_written=written)
@@ -126,7 +128,7 @@ async def write_file(ctx: ToolContext | None, params: WriteFileInput) -> WriteFi
         output_model=AppendFileOutput,
         description="Append text to a file in the sandbox.",
     )
-    async def append_file(ctx: ToolContext | None, params: AppendFileInput) -> AppendFileOutput:
+    async def append_file(ctx: "ToolContext" | None, params: AppendFileInput) -> AppendFileOutput:
         files = _get_files(ctx)
         written = files.append_text(params.path, params.content)
         return AppendFileOutput(path=params.path, bytes_written=written)
@@ -137,7 +139,7 @@ async def append_file(ctx: ToolContext | None, params: AppendFileInput) -> Appen
         output_model=ReplaceFileOutput,
         description="Replace text in a file in the sandbox.",
     )
-    async def replace_file(ctx: ToolContext | None, params: ReplaceFileInput) -> ReplaceFileOutput:
+    async def replace_file(ctx: "ToolContext" | None, params: ReplaceFileInput) -> ReplaceFileOutput:
         files = _get_files(ctx)
         replaced = files.replace_text(params.path, params.old, params.new)
         return ReplaceFileOutput(path=params.path, replaced=replaced)