33 lines
978 B
Python
33 lines
978 B
Python
import asyncio
|
|
from dataclasses import dataclass
|
|
|
|
from haiku.rag.client import HaikuRAG
|
|
from haiku.rag.research.dependencies import ResearchContext
|
|
from haiku.rag.research.models import EvaluationResult, InsightAnalysis
|
|
from haiku.rag.research.stream import ResearchStream
|
|
from rich.console import Console
|
|
|
|
|
|
@dataclass
|
|
class ResearchDeps:
|
|
client: HaikuRAG
|
|
console: Console | None = None
|
|
stream: ResearchStream | None = None
|
|
semaphore: asyncio.Semaphore | None = None
|
|
|
|
def emit_log(self, message: str, state: "ResearchState | None" = None) -> None:
|
|
if self.console:
|
|
self.console.print(message)
|
|
if self.stream:
|
|
self.stream.log(message, state)
|
|
|
|
|
|
@dataclass
|
|
class ResearchState:
|
|
context: ResearchContext
|
|
iterations: int = 0
|
|
max_iterations: int = 3
|
|
confidence_threshold: float = 0.8
|
|
max_concurrency: int = 1
|
|
last_eval: EvaluationResult | None = None
|
|
last_analysis: InsightAnalysis | None = None
|