added a few models

app.py

@@ -25,9 +25,10 @@ class BasicAgent:
         #                          api_key=os.getenv("GEMINI_API_KEY"))
         #model=HfApiModel(api_key=os.getenv('HUGGING_FACE_API_KEY'))
         # Instantiate the agent
+        web_search = GoogleSearchTool(provider="serper")
         self.agent = CodeAgent(
                     tools=[
-                           GoogleSearchTool(provider="serper"), 
+                           web_search, 
                            ChessSolver(),
                            #DuckDuckGoSearchTool(),
                            VegetableFruitClassification(), 
@@ -46,11 +47,11 @@ class BasicAgent:
         If you are asked for a number, don't use comma to write your number neither use units such as $ or
         percent sign unless specified otherwise.
         If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the
-        digits in plain text unless specified otherwise. Never include currency symbols in the response.
+        digits in plain text unless specified otherwise. Never include currency symbols in the response. 
         If you are asked for a comma separated list, apply the above rules depending of whether the element
         to be put in the list is a number or a string. For question that contain phrases like `what is the number` or 
         `what is the highest number` return just the number, e.g., 2. For questions around currency, 
-        include just the number, not the currency sign. 
+        include just the number, not the currency sign.
         """
         self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + SYSTEM_PROMPT
     def __call__(self, question: str) -> str:
@@ -98,11 +99,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return f"An unexpected error occurred fetching questions: {e}", None
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        model = LiteLLMModel(model_id= "gemini/gemini-2.0-flash",
+        gemini_model = LiteLLMModel(model_id= "gemini/gemini-2.0-flash",
                                   api_key=os.getenv("GEMINI_API_KEY"))
-        # model = OpenAIServerModel(model_id="gpt-4o",
-        #                          api_key=os.getenv("OPEN_AI_KEY"))
-        agent = BasicAgent(model=model)
+        gpt_model = OpenAIServerModel(model_id="gpt-4o",
+                                 api_key=os.getenv("OPEN_AI_KEY"))
+        qwen_model=HfApiModel(api_key=os.getenv('HUGGING_FACE_API_KEY'))
+        chat_gpt_agent = BasicAgent(model=gpt_model)
+        gemini_agent = BasicAgent(model=gemini_model)
+        qwen_agent = BasicAgent(model=qwen_model)
         print('Initialized the agent.')
     except Exception as e:
         print(f"Error instantiating agent: {e}")
@@ -115,10 +119,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    #questions_data = [i for i in questions_data if i['task_id'] == '5a0c1adf-205e-4841-a666-7c3ef95def9d']
+    #questions_data = [i for i in questions_data if i['task_id'] == '7bd855d8-463d-4ed5-93ca-5fe35145f733']
     images = []
     #added limit for testing
     for req_num, item in enumerate(questions_data):
+        if item.get("task_id") in ['3f57289b-8c60-48be-bd80-01f8099ca449', '8e867cd7-cff9-4e6c-867a-ff5ddc2550be',
+                                   '3cef3a44-215e-4aed-8e3b-b1e3f08063b7']:
+            executing_agent = chat_gpt_agent
+        else:
+            executing_agent = gemini_agent
         task_id = item.get("task_id")
         question_text = item.get("question") + ' You can use wikipedia. Do not change original names or omit name parts.'
         file_name = item.get('file_name')
@@ -130,14 +139,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = executing_agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:
             print(f"Error running agent on task {task_id}: {e}")
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
         time.sleep(30)
-        print('Number of answered questions:', len(results_log))
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")