From 0d9d4856fb0cb589161333b70d2274b771abcdb1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 15:35:14 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(celery=5Fapp.py):=20ignore?= =?UTF-8?q?=20type=20error=20for=20Celery=20import=20to=20avoid=20linting?= =?UTF-8?q?=20issues?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🐛 fix(base.py): ignore type error for AsyncResult import to avoid linting issues 🐛 fix(worker.py): ignore type error for SoftTimeLimitExceeded import to avoid linting issues ✨ feat(worker.py): add conditional clearing of session cache based on clear_cache flag to improve performance --- src/backend/langflow/core/celery_app.py | 2 +- src/backend/langflow/graph/vertex/base.py | 2 +- src/backend/langflow/worker.py | 6 +++--- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/core/celery_app.py b/src/backend/langflow/core/celery_app.py index 85223491e..699fc3e14 100644 --- a/src/backend/langflow/core/celery_app.py +++ b/src/backend/langflow/core/celery_app.py @@ -1,4 +1,4 @@ -from celery import Celery +from celery import Celery # type: ignore def make_celery(app_name: str): diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index ccab84179..d8de30ca9 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -76,7 +76,7 @@ class Vertex: def get_task(self): # using the task_id, get the task from celery # and return it - from celery.result import AsyncResult + from celery.result import AsyncResult # type: ignore return AsyncResult(self.task_id) diff --git a/src/backend/langflow/worker.py b/src/backend/langflow/worker.py index 2b66b160c..2077bc3ef 100644 --- a/src/backend/langflow/worker.py +++ b/src/backend/langflow/worker.py @@ -1,9 +1,8 @@ from langflow.core.celery_app import celery_app from typing import Any, Dict, Optional, Tuple from typing import TYPE_CHECKING -from celery.exceptions import SoftTimeLimitExceeded +from celery.exceptions import SoftTimeLimitExceeded # type: ignore from langflow.processing.process import ( - clear_caches_if_needed, generate_result, process_inputs, ) @@ -42,8 +41,9 @@ def process_graph_cached_task( session_id=None, ) -> Tuple[Any, str]: initialize_session_manager() - clear_caches_if_needed(clear_cache) session_manager = get_session_manager() + if clear_cache: + session_manager.clear_session(session_id) # Load the graph using SessionManager langchain_object, artifacts = session_manager.load_session(session_id, data_graph) processed_inputs = process_inputs(inputs, artifacts)