Update app.py

app.py

@@ -12,7 +12,7 @@ import json
 import re
 import uuid
 import time
-import ast # <-- Import ast module
+import ast
 
 # --- Pydantic Import ---
 from pydantic import BaseModel, Field
@@ -29,7 +29,7 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, END, StateGraph
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_core.tools import tool
-from langchain_groq import ChatGroq
+from langchain_google_genai import ChatGoogleGenerativeAI # <-- For Gemini
 
 # --- RAG Imports ---
 from langchain_text_splitters import RecursiveCharacterTextSplitter
@@ -696,9 +696,11 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent (Single LLM) initializing...")
         
-        GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-        if not GROQ_API_KEY:
-            raise ValueError("GROQ_API_KEY environment variable is not set!")
+        # --- *** UPDATED: Check for Google API Key *** ---
+        GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+        if not GEMINI_API_KEY:
+            raise ValueError("GEMINI_API_KEY environment variable is not set!")
+        # --- *** END UPDATE *** ---
 
         self.tools = defined_tools
         
@@ -767,21 +769,22 @@ Your goal: Provide the EXACT answer in the EXACT format requested.
 **REMEMBER:** One step at a time. Use tools. Format JSON correctly.
 """
 
-        print("Initializing Groq LLM...")
+        # --- *** UPDATED: Initialize Google Gemini LLM *** ---
+        print("Initializing Google Gemini LLM...")
         try:
             # --- Initialize ONE Powerful LLM for all tasks ---
-            self.llm_with_tools = ChatGroq(
+            self.llm_with_tools = ChatGoogleGenerativeAI(
+                model="gemini-1.5-pro-latest", # Use Gemini 1.5 Pro
+                google_api_key=GEMINI_API_KEY,
                 temperature=0,
-                groq_api_key=GROQ_API_KEY,
-                model_name="llama-3.3-70b-versatile", # <-- Use the powerful model
-                max_tokens=4096,
-                timeout=60
+                convert_system_message_to_human=True # Helps with compatibility
             ).bind_tools(self.tools)
-            print("✅ Main LLM (llama-3.3-70b-versatile with tools) initialized.")
+            print("✅ Main LLM (Gemini 1.5 Pro with tools) initialized.")
 
         except Exception as e:
-            print(f"❌ Error initializing Groq: {e}")
+            print(f"❌ Error initializing Gemini: {e}")
             raise
+        # --- *** END UPDATE *** ---
         
         # --- Node 1: The Agent ---
         def agent_node(state: AgentState):