TestTest

app.py

@@ -212,10 +212,6 @@ def save_dataset_to_hub(df: pd.DataFrame, dataset_name: str) -> tuple[bool, str]
         
         # Add metadata with explicit string types for all columns
         dataset.info.description = "Agent steps data from evaluation run"
-        dataset.info.features = {
-            col: datasets.Value('string') for col in df_to_save.columns
-        }
-        
         # Save to hub with token
         dataset.push_to_hub(
             dataset_name,

mini_agents.py

@@ -66,9 +66,9 @@ audio_model = InferenceClientModel(
 audio_agent = CodeAgent(
     model=audio_model,
     tools=[transcribe_audio_tool, get_audio_from_file_path, noise_reduction, audio_segmentation, speaker_diarization],
-    max_steps=6,
+    max_steps=4,
     additional_authorized_imports=AUTHORIZED_IMPORTS,
-    planning_interval=6,
+    planning_interval=4,
     name="audio_agent",
     description="This agent is responsible for processing audio, loading mp3 audio and converting it to base64, reducing noise, segmenting audio and transcribing audio (in base64 format). It cannot process videos."
 )
@@ -82,9 +82,9 @@ vlm_model = InferenceClientModel(
 vlm_agent = CodeAgent(
     model=vlm_model,
     tools=[image_processing, object_detection_tool, ocr_scan_tool, extract_images_from_video, get_image_from_file_path, get_video_from_file_path],
-    max_steps=6,
+    max_steps=4,
     additional_authorized_imports=AUTHORIZED_IMPORTS,
-    planning_interval=6,
+    planning_interval=4,
     name="vlm_agent",
     description="This agent is responsible for downloading images or videos, processing images or videos, detecting objects in them and extracting text from them. It cannot process audios."
 )
@@ -167,7 +167,7 @@ class MasterAgentWrapper:
             managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
             tools=self.base_tools,  # Initialize without browser tools
             add_base_tools=False,
-            max_steps=20,
+            max_steps=8,
             additional_authorized_imports=AUTHORIZED_IMPORTS,
             verbosity_level=logging.INFO,
             planning_interval=4,