From 4317ab35b75b6116b38be0408647307341ef6809 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 25 Apr 2024 16:06:14 -0300 Subject: [PATCH] Refactor build_and_cache_graph_from_data function in chat.py to use model_dump() method for graph data --- src/backend/base/langflow/api/v1/chat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/api/v1/chat.py b/src/backend/base/langflow/api/v1/chat.py index e839df1c6..5d8e4ace3 100644 --- a/src/backend/base/langflow/api/v1/chat.py +++ b/src/backend/base/langflow/api/v1/chat.py @@ -87,7 +87,9 @@ async def retrieve_vertices_order( flow_id=flow_id, session=session, chat_service=chat_service, graph=graph ) else: - graph = await build_and_cache_graph_from_data(flow_id=flow_id, data=data, chat_service=chat_service) + graph = await build_and_cache_graph_from_data( + flow_id=flow_id, graph_data=data.model_dump(), chat_service=chat_service + ) if stop_component_id or start_component_id: try: first_layer = graph.sort_vertices(stop_component_id, start_component_id)