Simplify deps

haiku_rag_slim/haiku/rag/agents/rlm/dependencies.py

@@ -6,8 +6,6 @@ from haiku.rag.store.models import Document, SearchResult
 if TYPE_CHECKING:
     from haiku.rag.agents.rlm.docker_sandbox import DockerSandbox
     from haiku.rag.agents.rlm.models import CodeExecution
-    from haiku.rag.client import HaikuRAG
-    from haiku.rag.config.models import AppConfig
 
 
 @dataclass
@@ -24,7 +22,5 @@ class RLMContext:
 class RLMDeps:
     """Dependencies for RLM agent."""
 
-    client: "HaikuRAG"
-    config: "AppConfig"
     sandbox: "DockerSandbox"
     context: RLMContext = field(default_factory=RLMContext)

haiku_rag_slim/haiku/rag/agents/rlm/docker_sandbox.py

@@ -36,6 +36,12 @@ class DockerSandbox:
 
     DEFAULT_IMAGE = "ghcr.io/ggozad/haiku.rag-slim:latest"
 
+    haiku_client: "HaikuRAG"
+    config: RLMConfig
+    context: RLMContext
+    image: str
+    _process: subprocess.Popen[bytes] | None
+
     def __init__(
         self,
         client: "HaikuRAG",
@@ -47,7 +53,7 @@ class DockerSandbox:
         self.config = config
         self.context = context
         self.image = image or self.DEFAULT_IMAGE
-        self._process: subprocess.Popen[bytes] | None = None
+        self._process = None
 
     def _build_docker_cmd(self) -> list[str]:
         """Build the docker run command."""

haiku_rag_slim/haiku/rag/client.py

@@ -1364,8 +1364,6 @@ class HaikuRAG:
             image=self._config.rlm.docker_image,
         ) as sandbox:
             deps = RLMDeps(
-                client=self,
-                config=self._config,
                 sandbox=sandbox,
                 context=context,
             )