Shago commited on
Commit
bf3b3ab
·
verified ·
1 Parent(s): 8eaf921

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +9 -0
agents/agents_nodes.py CHANGED
@@ -1,3 +1,12 @@
 
 
 
 
 
 
 
 
 
1
  text_generator = pipeline(
2
  "text-generation", # Task type
3
  #model="deepseek-ai/DeepSeek-R1-0528-Qwen3-8B",
 
1
+ import json
2
+ from langchain_core.messages import AIMessage, ToolMessage
3
+ from langgraph.prebuilt import ToolNode
4
+ from langchain_huggingface import HuggingFacePipeline
5
+ from utils.state_utils import AgentState
6
+ from tools.financial_tools import time_value_tool
7
+ from transformers import pipeline
8
+ import torch
9
+
10
  text_generator = pipeline(
11
  "text-generation", # Task type
12
  #model="deepseek-ai/DeepSeek-R1-0528-Qwen3-8B",