From e4d4c21f344dc52e95af612cb8ec0c3bea843160 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 26 Feb 2024 17:10:52 -0300 Subject: [PATCH] Use display_name in exception --- src/backend/langflow/graph/graph/base.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py index de126374d..1ea34fd51 100644 --- a/src/backend/langflow/graph/graph/base.py +++ b/src/backend/langflow/graph/graph/base.py @@ -9,13 +9,8 @@ from langflow.graph.graph.constants import lazy_load_vertex_dict from langflow.graph.graph.utils import process_flow from langflow.graph.schema import InterfaceComponentTypes from langflow.graph.vertex.base import Vertex -from langflow.graph.vertex.types import ( - ChatVertex, - FileToolVertex, - LLMVertex, - RoutingVertex, - ToolkitVertex, -) +from langflow.graph.vertex.types import (ChatVertex, FileToolVertex, LLMVertex, + RoutingVertex, ToolkitVertex) from langflow.interface.tools.constants import FILE_TOOLS from langflow.utils import payload @@ -307,7 +302,7 @@ class Graph: for vertex in self.vertices: if not self._validate_vertex(vertex): raise ValueError( - f"{vertex.vertex_type} is not connected to any other components" + f"{vertex.display_name} is not connected to any other components" ) def _validate_vertex(self, vertex: Vertex) -> bool: