From 33742cb29e098e9f4e577fe5655ceb04ebc04698 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 11 Apr 2025 19:58:13 -0700 Subject: [PATCH] test: fix test --- tests/litellm/proxy/hooks/test_managed_files.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/litellm/proxy/hooks/test_managed_files.py b/tests/litellm/proxy/hooks/test_managed_files.py index 006a99d324..032de65a3f 100644 --- a/tests/litellm/proxy/hooks/test_managed_files.py +++ b/tests/litellm/proxy/hooks/test_managed_files.py @@ -9,13 +9,17 @@ sys.path.insert( 0, os.path.abspath("../../../..") ) # Adds the parent directory to the system path +from unittest.mock import MagicMock + from litellm.caching import DualCache from litellm.proxy.hooks.managed_files import _PROXY_LiteLLMManagedFiles from litellm.types.utils import SpecialEnums def test_get_file_ids_and_decode_b64_to_unified_uid_from_messages(): - proxy_managed_files = _PROXY_LiteLLMManagedFiles(DualCache()) + proxy_managed_files = _PROXY_LiteLLMManagedFiles( + DualCache(), prisma_client=MagicMock() + ) messages = [ { "role": "user",