23 lines
513 B
Python
23 lines
513 B
Python
from dataclasses import dataclass, field
|
|
from typing import TYPE_CHECKING
|
|
|
|
from haiku.rag.store.models import Document
|
|
|
|
if TYPE_CHECKING:
|
|
from haiku.rag.agents.rlm.sandbox import Sandbox
|
|
|
|
|
|
@dataclass
|
|
class RLMContext:
|
|
"""Mutable context accumulating data during RLM execution."""
|
|
|
|
documents: list[Document] | None = None
|
|
filter: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class RLMDeps:
|
|
"""Dependencies for RLM agent."""
|
|
|
|
sandbox: "Sandbox"
|
|
context: RLMContext = field(default_factory=RLMContext)
|