diff --git a/haiku_rag_slim/haiku/rag/agents/rlm/dependencies.py b/haiku_rag_slim/haiku/rag/agents/rlm/dependencies.py index aca51ebf..1644b291 100644 --- a/haiku_rag_slim/haiku/rag/agents/rlm/dependencies.py +++ b/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) diff --git a/haiku_rag_slim/haiku/rag/agents/rlm/docker_sandbox.py b/haiku_rag_slim/haiku/rag/agents/rlm/docker_sandbox.py index 4500586c..9f77fe51 100644 --- a/haiku_rag_slim/haiku/rag/agents/rlm/docker_sandbox.py +++ b/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.""" diff --git a/haiku_rag_slim/haiku/rag/client.py b/haiku_rag_slim/haiku/rag/client.py index 67a6a349..0ddc0db6 100644 --- a/haiku_rag_slim/haiku/rag/client.py +++ b/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, )