From 4ed00cdf0e03db9df95c15eee3ce4571a7d1c257 Mon Sep 17 00:00:00 2001 From: Edwin Jose Date: Mon, 7 Oct 2024 15:42:40 -0400 Subject: [PATCH] fix: NameError for Vertex and Graph Imports in Flow Tool (#4054) * Update flow_tool.py - Removed unnecessary imports from TYPE_CHECKING block. - Cleaned up import statements for better readability and organization. * Update flow_tool.py --- src/backend/base/langflow/base/tools/flow_tool.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/base/tools/flow_tool.py b/src/backend/base/langflow/base/tools/flow_tool.py index dc592987d..9884ab52b 100644 --- a/src/backend/base/langflow/base/tools/flow_tool.py +++ b/src/backend/base/langflow/base/tools/flow_tool.py @@ -6,6 +6,8 @@ from langchain_core.tools import BaseTool, ToolException from loguru import logger from langflow.base.flow_processing.utils import build_data_from_result_data, format_flow_output_data +from langflow.graph.graph.base import Graph # cannot be a part of TYPE_CHECKING # noqa: TCH001 +from langflow.graph.vertex.base import Vertex # cannot be a part of TYPE_CHECKING # noqa: TCH001 from langflow.helpers.flow import build_schema_from_inputs, get_arg_names, get_flow_inputs, run_flow from langflow.utils.async_helpers import run_until_complete @@ -13,9 +15,6 @@ if TYPE_CHECKING: from langchain_core.runnables import RunnableConfig from pydantic.v1 import BaseModel - from langflow.graph.graph.base import Graph - from langflow.graph.vertex.base import Vertex - class FlowTool(BaseTool): name: str