Update agents/agents_nodes.py
Browse files- 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",
|