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
This commit is contained in:
Edwin Jose 2024-10-07 15:42:40 -04:00 committed by GitHub
commit 4ed00cdf0e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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