phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: add diskcache as extra dependency

Browse source
This commit is contained in:
Antonio Loison 2024-05-10 11:19:14 +02:00
parent 7ee07cd961
commit c1ba4ec078
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/caching.py
View file

@ -15,7 +15,6 @@ from openai._models import BaseModel as OpenAIObject
from litellm._logging import verbose_logger
from litellm.types.services import ServiceLoggerPayload, ServiceTypes
import traceback
import diskcache as dc
def print_verbose(print_statement):
@ -1919,6 +1918,7 @@ class Cache:
class DiskCache(BaseCache):
def __init__(self, cache_dir: str = ".litellm_cache"):
import diskcache as dc
# if users don't provider one, use the default litellm cache
self.disk_cache = dc.Cache(cache_dir)