From d13f0e77dfe94ae015d47cae5256f529b3cb9b5a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 15:34:45 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(utils.py):=20remove=20unus?= =?UTF-8?q?ed=20imports=20and=20classes=20=F0=9F=94=A5=20refactor(utils.py?= =?UTF-8?q?):=20remove=20unused=20Memoize=20class=20and=20its=20methods?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/cache/utils.py | 45 +------------------- 1 file changed, 2 insertions(+), 43 deletions(-) diff --git a/src/backend/langflow/services/cache/utils.py b/src/backend/langflow/services/cache/utils.py index 04fa4bbdf..5791e65c5 100644 --- a/src/backend/langflow/services/cache/utils.py +++ b/src/backend/langflow/services/cache/utils.py @@ -6,12 +6,12 @@ import os import tempfile from collections import OrderedDict from pathlib import Path -from typing import TYPE_CHECKING, Any, Callable, Dict +from typing import TYPE_CHECKING, Any, Dict from appdirs import user_cache_dir from langflow.services.database.models.base import orjson_dumps if TYPE_CHECKING: - from langflow.services.cache.base import BaseCacheManager + pass CACHE: Dict[str, Any] = {} @@ -195,44 +195,3 @@ def save_uploaded_file(file, folder_name): new_file.write(chunk) return file_path - - -class Memoize: - def __init__( - self, - get_cache_manager: Callable[[], "BaseCacheManager"], - ): - self.get_cache_manager = get_cache_manager - self.hash_func = compute_dict_hash - - def clear_cache(self, session_id): - cache_manager = self.get_cache_manager() - cache_manager.delete(session_id) - - def get_result_by_session_id(self, session_id): - cache_manager = self.get_cache_manager() - return cache_manager.get(session_id) - - def update_cache(self, session_id, value): - cache_manager = self.get_cache_manager() - cache_manager.set(session_id, value) - - def __call__(self, func: Callable[..., Any]): - @functools.wraps(func) - def wrapper(*args, **kwargs): - cache_manager = self.get_cache_manager() - session_id = self.hash_func(args[0]) - - result = cache_manager.get(session_id) - - if result is None: - result = func(*args, **kwargs) - cache_manager.set(session_id, result) - - wrapper.session_id = session_id - return result - - wrapper.clear_cache = self.clear_cache - wrapper.update_cache = self.update_cache - wrapper.get_result_by_session_id = self.get_result_by_session_id - return wrapper