From 2635971078808b88c01c9d77236be9cdf8b1f906 Mon Sep 17 00:00:00 2001 From: goulustis Date: Thu, 13 Nov 2025 00:23:44 +0800 Subject: [PATCH] use graph state --- lang_agent/graphs/tool_nodes.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/lang_agent/graphs/tool_nodes.py b/lang_agent/graphs/tool_nodes.py index 539b1c5..216a5f5 100644 --- a/lang_agent/graphs/tool_nodes.py +++ b/lang_agent/graphs/tool_nodes.py @@ -1,9 +1,3 @@ -from __future__ import annotations # optional but recommended -from typing import TYPE_CHECKING - -if TYPE_CHECKING: - from lang_agent.graphs.routing import State# only imported for type hints - from dataclasses import dataclass, field, is_dataclass from typing import Type, TypedDict, Literal, Dict, List, Tuple import tyro @@ -14,6 +8,7 @@ import os.path as osp from lang_agent.config import InstantiateConfig, KeyConfig from lang_agent.tool_manager import ToolManager from lang_agent.base import ToolNodeBase +from lang_agent.graphs.graph_state import State from langchain_core.language_models import BaseChatModel from langchain_core.messages import SystemMessage