Fix session retrieval in build_vertex function

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-20 16:01:23 -03:00
commit a4c74f099c

View file

@ -330,7 +330,7 @@ async def build_vertex(
f"No cache found for {flow_id}. Building graph starting at {vertex_id}"
)
graph = build_and_cache_graph(
flow_id=flow_id, session=get_session(), chat_service=chat_service
flow_id=flow_id, session=next(get_session()), chat_service=chat_service
)
else:
graph = cache.get("result")