Update final_agent.py

final_agent.py

@@ -20,6 +20,7 @@ from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
 from langchain_core.messages import BaseMessage, HumanMessage, AIMessage, ToolMessage
 from langchain_core.tools import tool
+from langchain_groq import ChatGroq
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_community.tools.tavily_search import TavilySearchResults
 
@@ -29,6 +30,7 @@ from langchain_community.tools.tavily_search import TavilySearchResults
 load_dotenv()
 gemini_api_key = os.getenv("GEMINI_API_KEY")
 tavily_api_key = os.getenv("TAVILY_API_KEY")
+groq_api_key = os,getenv("GROQ_API_KEY")
 
 # --- Optional: Tesseract Path ---
 # If Tesseract OCR is not in your system's PATH environment variable,
@@ -47,12 +49,17 @@ if not tavily_api_key:
 
 # --- Initialize LLM ---
 # Using the model specified in the user's code block
-llm = ChatGoogleGenerativeAI(
-    model="gemini-2.0-flash-lite", # As per user's last provided code
-    google_api_key=gemini_api_key,
-    temperature=0.1 # Low temperature for factual tasks
+# llm = ChatGoogleGenerativeAI(
+#     model="gemini-2.0-flash-lite", # As per user's last provided code
+#     google_api_key=gemini_api_key,
+#     temperature=0.1 # Low temperature for factual tasks
+# )
+llm = ChatGroq(
+    model="gemma2-9b-it",
+    api_key=groq_api_key,
+    temperature = 0.1
 )
-print(f"LLM Initialized: {llm.model}")
+# print(f"LLM Initialized: {llm.model}")
 
 # ==============================================================================
 # State Definition