Ignore function arguments when caching

dottxt-ai · Nov 20, 2023 · acbdc4f · acbdc4f
1 parent 0409e15
commit acbdc4f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/outlines/caching.py b/outlines/caching.py
@@ -1,5 +1,5 @@
 import os
-from typing import Callable
+from typing import Callable, Optional
 
 from perscache import Cache, NoCache
 from perscache.serializers import JSONSerializer
@@ -10,8 +10,11 @@
 memory = Cache(serializer=JSONSerializer(), storage=LocalFileStorage(cache_dir))
 
 
-def cache(fn: Callable):
-    return memory.cache()(fn)
+def cache(ignore: Optional[str]):
+    def cache_fn(fn: Callable):
+        return memory.cache(ignore=ignore)(fn)
+
+    return cache_fn
 
 
 def get_cache():

diff --git a/outlines/models/openai.py b/outlines/models/openai.py
@@ -9,6 +9,7 @@
 import numpy as np
 
 import outlines
+from outlines.caching import cache
 
 __all__ = ["OpenAI", "openai"]
 
@@ -287,6 +288,7 @@ def __repr__(self):
         return str(self.config)
 
 
+@cache(ignore="client")
 @functools.partial(outlines.vectorize, signature="(),(),()->(s)")
 async def generate_chat(
     prompt: str, client: "AsyncOpenAI", config: OpenAIConfig