Update app.py

app.py

@@ -12,7 +12,7 @@ from pydantic import BaseModel, Field
 import nest_asyncio
 import requests
 import os, re
-import litellm
+from agents.extensions.models.litellm_model import LitellmModel
 
 os.getenv("OPENAI_API_KEY")
 os.getenv("GEMINI_API_KEY")
@@ -146,7 +146,11 @@ Follow this cycle until you find the answer:
                 web_search,
                 visit_website
             ],
-            model="gpt-4o-mini"
+            model=LitellmModel(
+                model="gemini/gemini-pro",  # This tells LiteLLM to route to Gemini
+                api_key=os.getenv("GEMINI_API_KEY")
+            )
+
         )
 
         result = Runner.run_sync(