From 88a8294d90d3f20f1c72ef16ac40c2b09d72c6fa Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 22 Jun 2024 00:37:21 -0300 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(chat.py):=20remove=20unne?= =?UTF-8?q?cessary=20type=20hint=20for=20graph=20variable=20to=20improve?= =?UTF-8?q?=20code=20readability=20and=20consistency?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/api/v1/chat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/api/v1/chat.py b/src/backend/base/langflow/api/v1/chat.py index 32b59f151..c0feb621f 100644 --- a/src/backend/base/langflow/api/v1/chat.py +++ b/src/backend/base/langflow/api/v1/chat.py @@ -158,11 +158,11 @@ async def build_vertex( if not cache: # If there's no cache logger.warning(f"No cache found for {flow_id_str}. Building graph starting at {vertex_id}") - graph = await build_graph_from_db( + graph: "Graph" = await build_graph_from_db( flow_id=flow_id_str, session=next(get_session()), chat_service=chat_service ) else: - graph: "Graph" = cache.get("result") + graph = cache.get("result") await graph.initialize_run() vertex = graph.get_vertex(vertex_id)