From 2bd51c61041a98e6a697afd9f0f4bfbd4a9f822c Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 16 Jun 2023 14:31:16 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=20refactor(utils.py):=20remove=20u?= =?UTF-8?q?nused=20functions=20save=5Fcache=20and=20load=5Fcache=20The=20f?= =?UTF-8?q?unctions=20save=5Fcache=20and=20load=5Fcache=20are=20not=20used?= =?UTF-8?q?=20in=20the=20codebase=20and=20can=20be=20safely=20removed=20to?= =?UTF-8?q?=20improve=20code=20readability=20and=20maintainability.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/cache/utils.py | 20 -------------------- src/backend/langflow/interface/run.py | 16 +--------------- 2 files changed, 1 insertion(+), 35 deletions(-) diff --git a/src/backend/langflow/cache/utils.py b/src/backend/langflow/cache/utils.py index 0f1ff5d92..6058f6923 100644 --- a/src/backend/langflow/cache/utils.py +++ b/src/backend/langflow/cache/utils.py @@ -9,7 +9,6 @@ from collections import OrderedDict from pathlib import Path from typing import Any, Dict -import dill # type: ignore CACHE: Dict[str, Any] = {} @@ -133,22 +132,3 @@ def save_binary_file(content: str, file_name: str, accepted_types: list[str]) -> file.write(decoded_bytes) return file_path - - -@create_cache_folder -def save_cache(hash_val: str, chat_data, clean_old_cache_files: bool): - cache_path = Path(tempfile.gettempdir()) / PREFIX / f"{hash_val}.dill" - with cache_path.open("wb") as cache_file: - dill.dump(chat_data, cache_file) - - if clean_old_cache_files: - clear_old_cache_files() - - -@create_cache_folder -def load_cache(hash_val): - cache_path = Path(tempfile.gettempdir()) / PREFIX / f"{hash_val}.dill" - if cache_path.exists(): - with cache_path.open("rb") as cache_file: - return dill.load(cache_file) - return None diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py index 6fe4c089b..a3efe2b0c 100644 --- a/src/backend/langflow/interface/run.py +++ b/src/backend/langflow/interface/run.py @@ -1,22 +1,8 @@ -from langflow.cache.utils import compute_dict_hash, load_cache, memoize_dict +from langflow.cache.utils import memoize_dict from langflow.graph import Graph from langflow.utils.logger import logger -def load_langchain_object(data_graph, is_first_message=False): - """ - Load langchain object from cache if it exists, otherwise build it. - """ - computed_hash = compute_dict_hash(data_graph) - if is_first_message: - langchain_object = build_langchain_object(data_graph) - else: - logger.debug("Loading langchain object from cache") - langchain_object = load_cache(computed_hash) - - return computed_hash, langchain_object - - @memoize_dict(maxsize=10) def build_langchain_object_with_caching(data_graph): """