Rename agent

agent.py

@@ -60,7 +60,7 @@ def youtube_analysis_tool(question: str, url: str) -> str | None:
     except Exception as e:
         raise RuntimeError(f"Processing failed: {str(e)}") from e
 
-class BasicAgent:
+class GaiaAgent:
     def _generate_final_answer(self, question: str, answer: str) -> str:
         prompt = f"""
         **Question:** {question}

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from agent import BasicAgent
+from agent import GaiaAgent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -32,7 +32,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = GaiaAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

test.py

@@ -1,7 +1,7 @@
 import sys
-from agent import BasicAgent
+from agent import GaiaAgent
 
-agent = BasicAgent()
+agent = GaiaAgent()
 
 # Get the question from command-line arguments
 if len(sys.argv) > 1: