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

Update langflow worker.py with async_to_sync for generate_result

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-20 14:19:04 -03:00
commit 4751e91e90
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/worker.py
View file

@ -2,12 +2,13 @@ from typing import TYPE_CHECKING, Any, Dict, Optional
from asgiref.sync import async_to_sync from asgiref.sync import async_to_sync
from celery.exceptions import SoftTimeLimitExceeded # type: ignore from celery.exceptions import SoftTimeLimitExceeded # type: ignore
from loguru import logger
from rich import print
from langflow.core.celery_app import celery_app from langflow.core.celery_app import celery_app
from langflow.processing.process import Result, generate_result, process_inputs from langflow.processing.process import Result, generate_result, process_inputs
from langflow.services.deps import get_session_service from langflow.services.deps import get_session_service
from langflow.services.manager import initialize_session_service from langflow.services.manager import initialize_session_service
from loguru import logger
from rich import print
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.graph.vertex.base import Vertex from langflow.graph.vertex.base import Vertex
@ -62,7 +63,7 @@ def process_graph_cached_task(
logger.debug(f"Built object: {built_object}") logger.debug(f"Built object: {built_object}")
processed_inputs = process_inputs(inputs, artifacts or {}) processed_inputs = process_inputs(inputs, artifacts or {})
result = generate_result(built_object, processed_inputs) result = async_to_sync(generate_result, force_new_loop=True)(built_object, processed_inputs)
# Update the session with the new data # Update the session with the new data
session_service.update_session(session_id, (graph, artifacts)) session_service.update_session(session_id, (graph, artifacts))