Update agent.py

agent.py

@@ -282,13 +282,10 @@ def create_agent_executor(provider: str = "groq"):
     print(f"Initializing agent with provider: {provider}")
 
     # Step 1: Build LLMs - Use Google for vision capabilities
-    if provider == "google": 
-        main_llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro-latest", temperature=0)
-        vision_llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro-latest", temperature=0)
-    elif provider == "groq": 
+    if provider == "groq": 
         main_llm = ChatGroq(model_name="meta-llama/llama-4-maverick-17b-128e-instruct", temperature=0)
         # Use Google for vision since Groq's vision support may be limited
-        main_llm = ChatGroq(model_name="meta-llama/llama-4-maverick-17b-128e-instruct", temperature=0)
+        vision_llm = ChatGroq(model_name="meta-llama/llama-4-maverick-17b-128e-instruct", temperature=0)
     elif provider == "huggingface": 
         main_llm = ChatHuggingFace(llm=HuggingFaceEndpoint(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", temperature=0.1))
         vision_llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro-latest", temperature=0)