Update agent.py

agent.py

@@ -7,6 +7,7 @@ from langgraph.prebuilt import ToolNode
 from langchain_community.document_loaders import WikipediaLoader
 from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint
+from langchain_huggingface import ChatHuggingFace
 from langchain.llms import HuggingFaceHub
 import os
 from huggingface_hub import login
@@ -46,12 +47,12 @@ def build_agent():
     #     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
     # )
     
-    llm = HuggingFaceHub(repo_id="distilbert/distilgpt2", task="text-generation",
+    llm = ChatHuggingFace(HuggingFaceHub(repo_id="distilbert/distilgpt2", task="text-generation",
                     model_kwargs={
         "temperature": 0.7,         # Controls randomness (0.0 = deterministic)
         "max_new_tokens": 100       # Optional: max output length
     }, 
-    verbose=True )
+    verbose=True ))
     
     tools = [use_wikipedia_tool, use_search_tool]