fix some bugs

Author: zhiwei

src/agent/agent.py

@@ -51,13 +51,14 @@ def create_agent():
                 raise ValueError(f"Tool ID '{tool_id}' is not registered.")
             tools.append(REGISTED_TOOLS[tool_id]())
 
-        tools = tools + sub_agent_tools
-            
+        #tools = tools + sub_agent_tools
+
         agent = REGISTED_AGENTS["planning_agent"](
             config=planning_agent_config,
             model=model_manager.registed_models[planning_agent_config.model_id],
             tools=tools,
             max_steps=planning_agent_config.max_steps,
+            managed_agents=sub_agents,
             description=planning_agent_config.description,
             name=planning_agent_config.name,
             provide_run_summary=True,

src/base/async_multistep_agent.py

@@ -380,7 +380,7 @@ async def _run(
                 self.step_number += 1
 
         if final_answer is None and self.step_number == max_steps + 1:
-            final_answer = self._handle_max_steps_reached(task, images, step_start_time)
+            final_answer = await self._handle_max_steps_reached(task, images, step_start_time)
             yield action_step
         yield FinalAnswerStep(final_answer)
 
@@ -407,8 +407,8 @@ def _finalize_step(self, memory_step: ActionStep, step_start_time: float):
                 memory_step, agent=self
             )
 
-    def _handle_max_steps_reached(self, task: str, images: list["PIL.Image.Image"], step_start_time: float) -> Any:
-        final_answer = self.provide_final_answer(task, images)
+    async def _handle_max_steps_reached(self, task: str, images: list["PIL.Image.Image"], step_start_time: float) -> Any:
+        final_answer = await self.provide_final_answer(task, images)
         final_memory_step = ActionStep(
             step_number=self.step_number, error=AgentMaxStepsError("Reached max steps.", self.logger)
         )
@@ -577,7 +577,10 @@ async def provide_final_answer(self, task: str, images: list["PIL.Image.Image"]
         ]
         if images:
             messages[0]["content"].append({"type": "image"})
-        messages += await self.write_memory_to_messages()[1:]
+        messages = await self.write_memory_to_messages()
+        messages += messages[1:]
+
+
         messages += [
             {
                 "role": MessageRole.USER,
@@ -592,7 +595,7 @@ async def provide_final_answer(self, task: str, images: list["PIL.Image.Image"]
             }
         ]
         try:
-            chat_message: ChatMessage = self.model(messages)
+            chat_message: ChatMessage = await self.model(messages)
             return chat_message.content
         except Exception as e:
             return f"Error in generating final LLM output:\n{e}"