litellm/tests/local_testing/test_disk_cache_unit_tests.py
Krrish Dholakia 16bbed72d4 test(caching_unit_tests.py): add unit tests for llm caching
ensures coverage for common caching scenarios across different implementations
2024-11-12 13:21:22 +05:30

11 lines
307 B
Python

from cache_unit_tests import LLMCachingUnitTests
from litellm.caching import LiteLLMCacheType
class TestDiskCacheUnitTests(LLMCachingUnitTests):
def get_cache_type(self) -> LiteLLMCacheType:
return LiteLLMCacheType.DISK
# if __name__ == "__main__":
# pytest.main([__file__, "-v", "-s"])