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, ChatHuggingFace
+from langchain_huggingface import HuggingFacePipeline, ChatHuggingFace,HuggingFaceEndpoint
 from langchain_core.runnables import RunnableLambda
 from langgraph.prebuilt import ToolNode
 from utils.state_utils import AgentState
@@ -10,7 +10,7 @@ from tools.financial_tools import time_value_tool
 
 
 
-# LLL instantation
+# LLM instantation
 
 # text_generator = pipeline(
 #     "text-generation",  # Task type
@@ -24,17 +24,14 @@ from tools.financial_tools import time_value_tool
 #     max_new_tokens=700  # Limit output length
 # )
 
-
-text_generator = HuggingFaceEndpoint(
-    repo_id="Qwen/Qwen3-Embedding-0.6B",
+llm_endpoint = HuggingFaceEndpoint(
+    endpoint_url="<your_endpoint_url>",
     task="text-generation",
-    max_new_tokens=512,
-    do_sample=False,
+    max_new_tokens=1024,
+    do_sample=False
 )
-
-
 # llm = HuggingFacePipeline(pipeline=text_generator)
-llm =ChatHuggingFace(llm=text_generator)
+llm = ChatHuggingFace(llm=llm_endpoint)
 
 llm_instantiated = llm.bind_tools(    
     [time_value_tool],