Ensure cache folder is created

HishamAlyahya · Feb 2, 2024 · 965f86d · 965f86d
1 parent 56fbc19
commit 965f86d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -20,7 +20,7 @@ from pyllm import CodeLLM
 
 llm = CodeLLM()
 swap_numbers = llm.def_function(
-    "Swap two input numbers", unit_tests=[((1, 2), (2, 1))], use_cached=False
+    "Swap two input numbers", unit_tests=[((1, 2), (2, 1))]
 )
 print(swap_numbers(20, 40))
 # Output: (40, 20)

diff --git a/pyllm/llm.py b/pyllm/llm.py
@@ -15,6 +15,9 @@
 from pyllm.exceptions import TooManyRetries
 
 
+os.makedirs(user_cache_dir("PyLLM"), exist_ok=True)
+
+
 class CacheHandler:
     CACHE_FILE = os.path.join(user_cache_dir("PyLLM"), "cached_functions.json")