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

refac: change cache folder

Browse source
This commit is contained in:
Ibis Prevedello 2023-04-11 23:31:08 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit 296ecafdcf
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/cache/utils.py vendored
View file

@ -40,8 +40,8 @@ PREFIX = "langflow_cache"
def clear_old_cache_files(max_cache_size: int = 3): def clear_old_cache_files(max_cache_size: int = 3):
cache_dir = Path(tempfile.gettempdir()) cache_dir = Path(tempfile.gettempdir()) / PREFIX
cache_files = list(cache_dir.glob(f"{PREFIX}_*.dill")) cache_files = list(cache_dir.glob("*.dill"))
if len(cache_files) > max_cache_size: if len(cache_files) > max_cache_size:
cache_files_sorted_by_mtime = sorted( cache_files_sorted_by_mtime = sorted(
@ -85,7 +85,7 @@ def filter_json(json_data):
def save_cache(hash_val: str, chat_data, clean_old_cache_files: bool): def save_cache(hash_val: str, chat_data, clean_old_cache_files: bool):
cache_path = Path(tempfile.gettempdir()) / f"{PREFIX}_{hash_val}.dill" cache_path = Path(tempfile.gettempdir()) / PREFIX / f"{hash_val}.dill"
with cache_path.open("wb") as cache_file: with cache_path.open("wb") as cache_file:
dill.dump(chat_data, cache_file) dill.dump(chat_data, cache_file)
@ -94,7 +94,7 @@ def save_cache(hash_val: str, chat_data, clean_old_cache_files: bool):
def load_cache(hash_val): def load_cache(hash_val):
cache_path = Path(tempfile.gettempdir()) / f"{PREFIX}_{hash_val}.dill" cache_path = Path(tempfile.gettempdir()) / PREFIX / f"{hash_val}.dill"
if cache_path.exists(): if cache_path.exists():
with cache_path.open("rb") as cache_file: with cache_path.open("rb") as cache_file:
return dill.load(cache_file) return dill.load(cache_file)