Switched to using Google gemini-2.0-flash

app.py

@@ -34,10 +34,15 @@ class BasicAgent:
         print("BasicAgent initialized.")
         
         # Initialize the agent with a model and tools
+        # model = OpenAIServerModel(
+        #     model_id=os.environ["MODEL_ID"],
+        #     api_base="https://api.openai.com/v1",
+        #     api_key=os.environ["OPENAI_API_KEY"],
+        # )
         model = OpenAIServerModel(
             model_id=os.environ["MODEL_ID"],
-            api_base="https://api.openai.com/v1",
-            api_key=os.environ["OPENAI_API_KEY"],
+            api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+            api_key=os.environ["GEMINI_API_KEY"],
         )
 
         web_agent = ToolCallingAgent(
@@ -49,15 +54,6 @@ class BasicAgent:
             description="This agent can search the web and visit webpages to gather information.",
         )
 
-        python_agent = ToolCallingAgent(
-            verbosity_level=1,
-            tools=[PythonInterpreterTool()],
-            max_steps=5,
-            model=model,
-            name="python_agent",
-            description="This agent can run Python code snippets.",
-        )
-
         stt_agent = ToolCallingAgent(
             verbosity_level=1,
             tools=[SpeechToTextTool()],
@@ -70,8 +66,7 @@ class BasicAgent:
         manager_agent = CodeAgent(
             tools=[],
             model=model,
-            managed_agents=[web_agent, stt_agent, python_agent],
-            # managed_agents=[web_agent],
+            managed_agents=[web_agent, stt_agent],
             additional_authorized_imports=["time", "numpy", "pandas"],
         )
 
@@ -92,11 +87,11 @@ class BasicAgent:
             answer = self.agent.run(system_prompt + question, images=[image_bytes])
         elif file_type == "audio":
             arguments = {"audio": file_name}
-            answer = self.agent.run(system_prompt + question, arguments=arguments)
+            answer = self.agent.run(system_prompt + question, additional_args=arguments)
         elif file_type == "python":
             with open(file_name, "r") as file:
                 python_code = file.read()
-            answer = self.agent.run(system_prompt + question, code=python_code)
+            answer = self.agent.run(system_prompt + question, additional_args={"code": python_code})
         else:
             answer = self.agent.run(system_prompt + question)
 
@@ -169,10 +164,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         question_text = item.get("question")
         file_name = item.get("file_name")
         file_type = "unknown"
-        print(f"Fetching file content '{file_name}' for task ID: {task_id}")
 
         if file_name:
-            print(f"Fetching file content for task ID: {task_id}")
+            print(f"Fetching file content '{file_name}' for task ID: {task_id}")
             try:
                 file_url = f"{api_url}/files/{task_id}"
                 file_response = requests.get(file_url, timeout=15)