mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 02:34:29 +00:00
test: fix test
This commit is contained in:
parent
8b09a2721f
commit
1d878b7ea0
1 changed files with 5 additions and 1 deletions
|
@ -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",
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue