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

Use cache when rerunning to allow pinned components

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-16 16:15:13 -03:00
commit 71c94f85fb
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/api/v1/chat.py
View file

@ -284,11 +284,16 @@ async def get_vertices(
):
"""Check the flow_id is in the flow_data_store."""
try:
flow: Flow = session.get(Flow, flow_id)
if not flow or not flow.data:
raise ValueError("Invalid flow ID")
graph = Graph.from_payload(flow.data)
chat_service.set_cache(flow_id, graph)
# First, we need to check if the flow_id is in the cache
if cache := chat_service.get_cache(flow_id):
graph = cache.get("result")
else:
flow: Flow = session.get(Flow, flow_id)
if not flow or not flow.data:
raise ValueError("Invalid flow ID")
graph = Graph.from_payload(flow.data)
chat_service.set_cache(flow_id, graph)
if component_id:
vertices = graph.sort_up_to_vertex(component_id)