Update crew.py

crew.py

@@ -7,17 +7,17 @@ from crewai_tools import (
     WebsiteSearchTool
 )
 from google import genai
-from openai import OpenAI
+#from openai import OpenAI
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from util import get_final_answer
 
 ## LLMs
 
-MANAGER_MODEL      = "gpt-4.1"
-AGENT_MODEL        = "gpt-4.1"
+MANAGER_MODEL      = "gemini-2.0-flash"
+AGENT_MODEL        = "gemini-2.0-flash"
 
-FINAL_ANSWER_MODEL = "gpt-4.5-preview"
+FINAL_ANSWER_MODEL = "gemini-2.0-flash"
 
 IMAGE_MODEL        = "gemini-2.0-flash"
 AUDIO_MODEL        = "gemini-2.0-flash"
@@ -87,7 +87,7 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            client = genai.Client(api_key="GOOGLE_API_KEY")
+            client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
             file = client.files.upload(file=file_path)
             
@@ -114,7 +114,7 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            client = genai.Client(api_key="GOOGLE_API_KEY")
+            client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
             file = client.files.upload(file=file_path)
             
@@ -141,7 +141,7 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            client = genai.Client(api_key="GOOGLE_API_KEY")
+            client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
             return client.models.generate_content(
                 model=YOUTUBE_MODEL,
@@ -168,7 +168,7 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            client = genai.Client(api_key="GOOGLE_API_KEY")
+            client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
             file = client.files.upload(file=file_path)