Update email_agent.py

email_agent.py

@@ -5,6 +5,11 @@ import sendgrid
 from sendgrid.helpers.mail import Email, Mail, Content, To
 from agents import Agent, function_tool, AsyncOpenAI, OpenAIChatCompletionsModel
 
+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)
+
+
 
 @function_tool
 def send_email(subject: str, html_body: str) -> Dict[str, str]:
@@ -26,5 +31,5 @@ email_agent = Agent(
     name="Email agent",
     instructions=INSTRUCTIONS,
     tools=[send_email],
-    model="gpt-4o-mini",
+    model=gemini_model,
 )