[Bug]: `No module named 'diskcache'`

[jamesbraza]

What happened?

With Python 3.11 and after pip install litellm==1.72.1:

from litellm.caching import Cache

Cache(type="disk")

Can litellm make an extra for diskcache, to avoid the ModuleNotFoundError

Relevant log output

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/path/to/.venv/lib/python3.11/site-packages/litellm/caching/caching.py", line 205, in __init__
    self.cache = DiskCache(disk_cache_dir=disk_cache_dir)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/path/to/.venv/lib/python3.11/site-packages/litellm/caching/disk_cache.py", line 16, in __init__
    import diskcache as dc
ModuleNotFoundError: No module named 'diskcache'

Are you a ML Ops Team?

No

What LiteLLM version are you on ?

1.72.1

Twitter / LinkedIn details

No response

[ishaan-jaff]

It's not a required dep, how would you suggest we handle this. We don't want to add bloat to the core SDK

[jamesbraza]

Thanks for asking, yeah the way to do it is:

1. Add an optional extra such as caching (see how we do it here)
2. Add instructions to pip install litellm[caching]
3. Lazily import diskcache with try-except that prints a nice message such as:

Please install litellm with litellm[caching] to use disk caching.

[ishaan-jaff]

@jamesbraza is this what you wanted ?

#11600

[jamesbraza]

Yeah correct, looks good