From ad7a9ed3249f62e7389541fb62997ee230d3d7cb Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Fri, 19 Sep 2025 09:17:14 +0300 Subject: [PATCH] Provide register_tools to the base research agent --- src/haiku/rag/research/base.py | 1 - src/haiku/rag/research/evaluation_agent.py | 4 ---- src/haiku/rag/research/orchestrator.py | 5 ----- src/haiku/rag/research/synthesis_agent.py | 6 ------ 4 files changed, 16 deletions(-) diff --git a/src/haiku/rag/research/base.py b/src/haiku/rag/research/base.py index 1a0796c9..6f612b7a 100644 --- a/src/haiku/rag/research/base.py +++ b/src/haiku/rag/research/base.py @@ -75,7 +75,6 @@ class BaseResearchAgent[T](ABC): """Return the system prompt for this agent.""" pass - @abstractmethod def register_tools(self) -> None: """Register agent-specific tools.""" pass diff --git a/src/haiku/rag/research/evaluation_agent.py b/src/haiku/rag/research/evaluation_agent.py index 8d3f5541..519f36f7 100644 --- a/src/haiku/rag/research/evaluation_agent.py +++ b/src/haiku/rag/research/evaluation_agent.py @@ -36,7 +36,3 @@ class AnalysisEvaluationAgent(BaseResearchAgent[EvaluationResult]): def get_system_prompt(self) -> str: return EVALUATION_AGENT_PROMPT - - def register_tools(self) -> None: - """No additional tools needed - uses LLM capabilities directly.""" - pass diff --git a/src/haiku/rag/research/orchestrator.py b/src/haiku/rag/research/orchestrator.py index 1274f384..5fe788c3 100644 --- a/src/haiku/rag/research/orchestrator.py +++ b/src/haiku/rag/research/orchestrator.py @@ -53,11 +53,6 @@ class ResearchOrchestrator(BaseResearchAgent[ResearchPlan]): def get_system_prompt(self) -> str: return ORCHESTRATOR_PROMPT - def register_tools(self) -> None: - """Register orchestration tools.""" - # Tools are no longer needed - orchestrator directly calls agents - pass - def _format_context_for_prompt(self, context: ResearchContext) -> str: """Format the research context as XML for inclusion in prompts.""" diff --git a/src/haiku/rag/research/synthesis_agent.py b/src/haiku/rag/research/synthesis_agent.py index e0dbd0a3..e99ee719 100644 --- a/src/haiku/rag/research/synthesis_agent.py +++ b/src/haiku/rag/research/synthesis_agent.py @@ -32,9 +32,3 @@ class SynthesisAgent(BaseResearchAgent[ResearchReport]): def get_system_prompt(self) -> str: return SYNTHESIS_AGENT_PROMPT - - def register_tools(self) -> None: - """Register synthesis-specific tools.""" - # The agent will use its LLM capabilities directly for synthesis - # The structured output will guide the report generation - pass