Update agent.py

agent.py

@@ -9,7 +9,8 @@ from serpapi import GoogleSearch
 
 # 1️⃣ Switch Graph → StateGraph
 from langgraph.graph import StateGraph
-from langchain_core.language_models.llms import LLM
+#from langchain_core.language_models.llms import LLM
+from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
@@ -136,7 +137,15 @@ def build_graph(provider: str = "huggingface") -> StateGraph:
     hf_token = os.getenv("HF_TOKEN")
     if not hf_token:
         raise ValueError("HF_TOKEN missing in env")
-    llm = LLM(provider=provider, token=hf_token, model="meta-llama/Llama-2-7b-chat-hf")
+    llm = ChatHuggingFace(
+    llm=HuggingFaceEndpoint(
+        url=f"https://api-inference.huggingface.co/models/{os.getenv('HF_MODEL_ID')}",
+        temperature=0.0,
+        huggingfacehub_api_token= hf_token,
+        
+    )
+)
+
 
     # 6.1) Node: init → seed system prompt
     def init_node(_: AgentState) -> AgentState: