Update agents/agents_nodes.py

agents/agents_nodes.py

@@ -1,11 +1,13 @@
 import json
+from transformers import pipeline
+import torch
 from langchain_core.messages import AIMessage, ToolMessage
-from langgraph.prebuilt import ToolNode
 from langchain_huggingface import HuggingFacePipeline
+from langchain_core.runnables import RunnableLambda
+from langgraph.prebuilt import ToolNode
 from utils.state_utils import AgentState
 from tools.financial_tools import time_value_tool
-from transformers import pipeline
-import torch
+
 
 
 # LLL instantation
@@ -23,10 +25,17 @@ text_generator = pipeline(
 )
 
 
-llm = HuggingFacePipeline(pipeline=text_generator)
-llm_instantiated = llm.bind_tools(    
-    [time_value_tool],
-    tool_choice={"type": "function", "function": {"name": "time_value_tool"}}
+# llm = HuggingFacePipeline(pipeline=text_generator)
+# llm_instantiated = llm.bind_tools(    
+#     [time_value_tool],
+#     tool_choice={"type": "function", "function": {"name": "time_value_tool"}}
+# )
+
+llm_instantiated = llm | RunnableLambda(
+    lambda x: x.bind_tools(
+        [time_value_tool],
+        tool_choice={"type": "function", "function": {"name": "time_value_tool"}}
+    )
 )
 
 def agent_node(state: AgentState):