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

Merge branch 'feat/progressbar' of https://github.com/logspace-ai/langflow into feat/progressbar

Browse source
This commit is contained in:
Cristhian Zanforlin Lousa 2023-06-21 14:05:26 -03:00
commit 3a9315ed77
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -75,7 +75,7 @@ async def stream_build(flow_id: str):
try:
if flow_id not in flow_data_store:
error_message = "Invalid session ID"
yield f"data: {json.dumps({'error': error_message})}\n\n"
yield f"error: {json.dumps({'error': error_message})}\n\n"
return
graph_data = flow_data_store[flow_id].get("data")
@ -86,7 +86,15 @@ async def stream_build(flow_id: str):
return
logger.debug("Building langchain object")
graph = Graph.from_payload(graph_data)
try:
# Some error could happen when building the graph
graph = Graph.from_payload(graph_data)
except Exception as exc:
logger.error(exc)
error_message = str(exc)
yield f"error: {json.dumps({'error': error_message})}\n\n"
return
number_of_nodes = len(graph.nodes)
for i, vertex in enumerate(graph.generator_build(), 1):
try: