jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor vertex imports in base.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-03 19:44:44 -03:00
commit eba4a0c5a9
1 changed files with 3 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/graph/graph/base.py
View file

@ -12,13 +12,8 @@ from langflow.graph.graph.state_manager import GraphStateManager
from langflow.graph.graph.utils import process_flow from langflow.graph.graph.utils import process_flow
from langflow.graph.schema import INPUT_FIELD_NAME, InterfaceComponentTypes from langflow.graph.schema import INPUT_FIELD_NAME, InterfaceComponentTypes
from langflow.graph.vertex.base import Vertex from langflow.graph.vertex.base import Vertex
from langflow.graph.vertex.types import ( from langflow.graph.vertex.types import (ChatVertex, FileToolVertex, LLMVertex,
ChatVertex, RoutingVertex, ToolkitVertex)
FileToolVertex,
LLMVertex,
RoutingVertex,
ToolkitVertex,
)
from langflow.interface.tools.constants import FILE_TOOLS from langflow.interface.tools.constants import FILE_TOOLS
from langflow.utils import payload from langflow.utils import payload
@ -783,7 +778,7 @@ class Graph:
self.stop_vertex = stop_component_id self.stop_vertex = stop_component_id
vertices = self.sort_up_to_vertex(stop_component_id) vertices = self.sort_up_to_vertex(stop_component_id)
elif start_component_id: elif start_component_id:
vertices = self.sort_up_to_vertex(start_component_id) vertices = self.sort_up_to_vertex(start_component_id, is_start=True)
else: else:
vertices = self.vertices vertices = self.vertices