Update agent.py

agent.py

@@ -10,6 +10,7 @@ from serpapi import GoogleSearch
 # 1️⃣ Switch Graph → StateGraph
 from langgraph.graph import StateGraph
 #from langchain_core.language_models.llms import LLM
+from langchain_groq import ChatGroq
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
@@ -132,18 +133,8 @@ class AgentState(TypedDict):
 # ────────────────
 # 6️⃣ Build the StateGraph
 # ────────────────
-def build_graph(provider: str = "huggingface") -> StateGraph:
-    # Instantiate LLM
-    endpoint_url = "https://api-inference.huggingface.co/models/Qwen/Qwen-7B"
-    hf_token = os.getenv("HF_TOKEN")
-    if not hf_token:
-        raise ValueError("HF_TOKEN missing in env")
-    llm = ChatHuggingFace(
-    llm=HuggingFaceEndpoint(
-        endpoint_url=endpoint_url,
-        temperature=0.0,
-        huggingfacehub_api_token= hf_token,
-        
+def build_graph(provider: str = "groq") -> StateGraph:
+     llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
     )
 )