Update agents/agents_nodes.py

agents/agents_nodes.py

@@ -2,7 +2,7 @@ import json
 from transformers import pipeline
 import torch
 from langchain_core.messages import AIMessage, ToolMessage
-from langchain_huggingface import HuggingFacePipeline
+from langchain_huggingface import HuggingFacePipeline, ChatHuggingFace
 from langchain_core.runnables import RunnableLambda
 from langgraph.prebuilt import ToolNode
 from utils.state_utils import AgentState
@@ -25,7 +25,8 @@ text_generator = pipeline(
 )
 
 
-llm = HuggingFacePipeline(pipeline=text_generator)
+# llm = HuggingFacePipeline(pipeline=text_generator)
+llm =ChatHuggingFace(llm=text_generator)
 
 llm_instantiated = llm.bind_tools(    
     [time_value_tool],