Update agent.py

agent.py

@@ -142,7 +142,7 @@ class BasicAgent:
             model=model,
             tools=tools,
             add_base_tools=False,
-            max_steps=5,
+            max_steps=10,
         )
         self.agent.system_prompt = (
             """
@@ -186,20 +186,4 @@ class BasicAgent:
             
         return final_str
 
-    def evaluate_random_questions(self, csv_path: str = "gaia_qa.csv", sample_size: int = 3):
-        df = pd.read_csv(csv_path)
-        if not {"question", "answer"}.issubset(df.columns):
-            print("CSV must contain 'question' and 'answer' columns.")
-            print("Found columns:", df.columns.tolist())
-            return
-        samples = df.sample(n=sample_size)
-        for _, row in samples.iterrows():
-            question = row["question"].strip()
-            expected = f"FINAL ANSWER: {str(row['answer']).strip()}"
-            result = self(question).strip()
-            print("---")
-            print("Question:", question)
-            print("Expected:", expected)
-            print("Agent:", result)
-            print("Correct:", expected == result)