Merge pull request #1376 from Masao-Taketani/master

bugfix. third argument of ResearchAgent has to be tone

Author: assafelovic

multi_agents/agents/editor.py

@@ -13,9 +13,10 @@
 class EditorAgent:
     """Agent responsible for editing and managing code."""
 
-    def __init__(self, websocket=None, stream_output=None, headers=None):
+    def __init__(self, websocket=None, stream_output=None, tone=None, headers=None):
         self.websocket = websocket
         self.stream_output = stream_output
+        self.tone = tone
         self.headers = headers or {}
 
     async def plan_research(self, research_state: Dict[str, any]) -> Dict[str, any]:
@@ -117,7 +118,7 @@ def _format_planning_instructions(self, initial_research: str, include_human_fee
     def _initialize_agents(self) -> Dict[str, any]:
         """Initialize the research, reviewer, and reviser skills."""
         return {
-            "research": ResearchAgent(self.websocket, self.stream_output, self.headers),
+            "research": ResearchAgent(self.websocket, self.stream_output, self.tone, self.headers),
             "reviewer": ReviewerAgent(self.websocket, self.stream_output, self.headers),
             "reviser": ReviserAgent(self.websocket, self.stream_output, self.headers),
         }

multi_agents/agents/orchestrator.py

@@ -43,7 +43,7 @@ def _create_output_directory(self):
     def _initialize_agents(self):
         return {
             "writer": WriterAgent(self.websocket, self.stream_output, self.headers),
-            "editor": EditorAgent(self.websocket, self.stream_output, self.headers),
+            "editor": EditorAgent(self.websocket, self.stream_output, self.tone, self.headers),
             "research": ResearchAgent(self.websocket, self.stream_output, self.tone, self.headers),
             "publisher": PublisherAgent(self.output_dir, self.websocket, self.stream_output, self.headers),
             "human": HumanAgent(self.websocket, self.stream_output, self.headers)