Update planner_agent.py

planner_agent.py

@@ -1,6 +1,13 @@
 from pydantic import BaseModel, Field
 from agents import Agent, AsyncOpenAI, OpenAIChatCompletionsModel
 
+import os
+
+gemini_client = AsyncOpenAI(base_url="https://generativelanguage.googleapis.com/v1beta/openai/", api_key=os.getenv('GOOGLE_API_KEY'))
+
+gemini_model = OpenAIChatCompletionsModel(model="gemini-2.0-flash", openai_client=gemini_client)
+
+
 HOW_MANY_SEARCHES = 5
 
 INSTRUCTIONS = f"You are a helpful research assistant. Given a query, come up with a set of web searches \