fix: cache and imports

r2e-project · Nov 26, 2024 · aba8089 · aba8089
1 parent 65f9431
commit aba8089
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/r2e/llms/__init__.py b/src/r2e/llms/__init__.py
@@ -0,0 +1,2 @@
+from r2e.llms.llm_args import LLMArgs
+from r2e.llms.completions import LLMCompletions
diff --git a/src/r2e/llms/cache_object.py b/src/r2e/llms/cache_object.py
@@ -1,14 +1,14 @@
 import os
 import json
-from diskcache import Cache
+from diskcache import Cache as DiskCache
 
 from r2e.paths import CACHE_PATH, CACHE_DIR
 
 
 class Cache:
     def __init__(self) -> None:
         os.makedirs(CACHE_DIR, exist_ok=True)
-        self.cache_dict = Cache(CACHE_PATH)
+        self.cache_dict = DiskCache(CACHE_DIR)
 
     @staticmethod
     def process_payload(payload):
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from r2e.llms.llm_args import LLMArgs
		from r2e.llms.completions import LLMCompletions