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

🔧 fix(endpoints.py): remove unused import 'process_graph_cached' and replace it with 'process_graph_cached_worker' for processing graph data asynchronously

Browse source 
🔧 fix(endpoints.py): add error handling for processing tweaks and log the error message
✨ feat(endpoints.py): use 'process_graph_cached_worker' to process graph data asynchronously and return the result in 'ProcessResponse'
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-13 19:26:48 -03:00
commit 5954e6daa0
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/api/v1/endpoints.py
View file

@ -3,10 +3,10 @@ from typing import Annotated, Optional
from langflow.cache.utils import save_uploaded_file from langflow.cache.utils import save_uploaded_file
from langflow.database.models.flow import Flow from langflow.database.models.flow import Flow
from langflow.processing.process import process_graph_cached, process_tweaks from langflow.processing.process import process_tweaks
from langflow.utils.logger import logger from langflow.utils.logger import logger
from langflow.settings import settings from langflow.settings import settings
from langflow.worker import process_graph_cached as process_graph_cached_worker
from fastapi import APIRouter, Depends, HTTPException, UploadFile, Body from fastapi import APIRouter, Depends, HTTPException, UploadFile, Body
from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.custom.custom_component import CustomComponent
@ -90,7 +90,12 @@ async def process_flow(
graph_data = process_tweaks(graph_data, tweaks) graph_data = process_tweaks(graph_data, tweaks)
except Exception as exc: except Exception as exc:
logger.error(f"Error processing tweaks: {exc}") logger.error(f"Error processing tweaks: {exc}")
response = process_graph_cached(graph_data, inputs, clear_cache) # ! This was added just for testing purposes
response = process_graph_cached_worker.delay(
graph_data=graph_data,
inputs=inputs,
clear_cache=clear_cache,
).get()
return ProcessResponse( return ProcessResponse(
result=response, result=response,
) )