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:
parent
b0bda283ff
commit
4ed00cdf0e
1 changed files with 2 additions and 3 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue