Update search_agent.py

search_agent.py

@@ -1,6 +1,5 @@
 from agents import Agent, WebSearchTool, ModelSettings, AsyncOpenAI, OpenAIChatCompletionsModel
-
-
+import os
 
 gemini_client = AsyncOpenAI(base_url="https://generativelanguage.googleapis.com/v1beta/openai/", api_key=os.getenv('GOOGLE_API_KEY'))
 
@@ -19,6 +18,6 @@ search_agent = Agent(
     name="Search agent",
     instructions=INSTRUCTIONS,
     tools=[WebSearchTool(search_context_size="low")],
-    model="gpt-4o-mini",
+    model=gemini_model,
     model_settings=ModelSettings(tool_choice="required"),
 )