Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 13:40:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Implement internal cache path management for improved resource handling

Browse Source
This commit is contained in:
JB
2025-10-06 16:34:01 -07:00
parent e76701ab32
commit fd8ede318f
2 changed files with 21 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
abogen/utils.py
+18
View File
@@ -183,6 +183,8 @@ def get_user_cache_root():
for env_var in ("HUGGINGFACE_HUB_CACHE", "TRANSFORMERS_CACHE"):
os.environ.setdefault(env_var, hf_cache)
os.environ.setdefault("ABOGEN_INTERNAL_CACHE_ROOT", cache_base)
cache_root = None
override = os.environ.get("ABOGEN_TEMP_DIR")
@@ -217,6 +219,22 @@ def get_user_cache_root():
return cache_root
def get_internal_cache_root():
root = os.environ.get("ABOGEN_INTERNAL_CACHE_ROOT") or os.environ.get("XDG_CACHE_HOME")
if root:
return ensure_directory(root)
home_dir = os.environ.get("HOME") or os.path.join("/tmp", "abogen-home")
home_dir = ensure_directory(home_dir)
return ensure_directory(os.path.join(home_dir, ".cache"))
def get_internal_cache_path(folder=None):
base = get_internal_cache_root()
if folder:
return ensure_directory(os.path.join(base, folder))
return base
def get_user_cache_path(folder=None):
base = get_user_cache_root()
if folder: