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

Refactor build_and_cache_graph_from_data function in chat.py to use model_dump() method for graph data

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-25 16:06:14 -03:00
commit 4317ab35b7
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/api/v1/chat.py
View file

@ -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)