Larger model.

app.py

@@ -6,7 +6,7 @@ import mimetypes
 import requests
 import pandas as pd
 from llama_index.core.llms import ChatMessage, TextBlock, ImageBlock, AudioBlock
-from llama_index.llms.gemini import Gemini
+from llama_index.llms.google_genai import GoogleGenAI
 from llama_index.core.agent.workflow import ReActAgent, AgentOutput
 from llama_index.core.tools import FunctionTool
 from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
@@ -37,7 +37,7 @@ class BasicAgent:
         wikipedia_search_tool = FunctionTool.from_defaults(WikipediaToolSpec().search_data)
         self._tools = [search_tool, wikipedia_load_tool, wikipedia_search_tool]
 
-        self._llm = Gemini(api_key=GOOGLE_API_KEY, model="models/gemini-2.0-flash-lite", max_tokens=1600)
+        self._llm = GoogleGenAI(api_key=GOOGLE_API_KEY, model="gemini-2.0-flash", max_tokens=1600)
         self._agent = ReActAgent(tools=self._tools, llm=self._llm)
         # Modify the react prompt.
         self._agent.update_prompts({"react_header": SYSTEM_PROMPT})
@@ -86,9 +86,9 @@ def get_media_content(item):
                 return ImageBlock(image=file_response.content)
             elif media_type == 'text':
                 return TextBlock(text=file_response.content)
-            # Audio currently not supported
-            #elif media_type == 'audio':
-            #    return AudioBlock(audio=file_response.content)
+            # Audio currently not supported?
+            elif media_type == 'audio':
+                return AudioBlock(audio=file_response.content)
 
 
 def create_question_message(item):