Change managers to MultiStepAgent

mini_agents.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, InferenceClientModel
+from smolagents import CodeAgent, InferenceClientModel, MultiStepAgent
 from tools import sort_list, operate_two_numbers, convert_number
 from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
 from vlm_tools import download_image, image_processing, object_detection, ocr_scan
@@ -71,7 +71,7 @@ pandas_agent = CodeAgent(
     description="This agent is responsible for converting data to a dataframe, performing pandas operations on such dataframe and converting the dataframe back to a json or a csv file."
 )
 
-multimodal_manager = CodeAgent(
+multimodal_manager = MultiStepAgent(
     model=MODEL_CHOICES["code"][0],
     managed_agents=[audio_agent, vlm_agent],
     tools=[sort_list],
@@ -81,7 +81,7 @@ multimodal_manager = CodeAgent(
     description="This agent is responsible for managing the audio and vlm agents."
 )
 
-operation_manager = CodeAgent(
+operation_manager = MultiStepAgent(
     model=MODEL_CHOICES["code"][0],
     managed_agents=[arithmetic_agent, pandas_agent],
     tools=[sort_list],
@@ -91,7 +91,7 @@ operation_manager = CodeAgent(
     description="This agent is responsible for managing the arithmetic and pandas agents."
 )
 
-master_agent = CodeAgent(
+master_agent = MultiStepAgent(
     model=MODEL_CHOICES["master"][0],
     managed_agents=[multimodal_manager, operation_manager],
     tools=[sort_list, *community_tools],