Update writer_agent.py

writer_agent.py

@@ -1,6 +1,12 @@
 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)
+
 
 INSTRUCTIONS = (
     "You are a senior researcher tasked with writing a cohesive report for a research query. "