mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +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("../../../..")
|
0, os.path.abspath("../../../..")
|
||||||
) # Adds the parent directory to the system path
|
) # Adds the parent directory to the system path
|
||||||
|
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
from litellm.caching import DualCache
|
from litellm.caching import DualCache
|
||||||
from litellm.proxy.hooks.managed_files import _PROXY_LiteLLMManagedFiles
|
from litellm.proxy.hooks.managed_files import _PROXY_LiteLLMManagedFiles
|
||||||
from litellm.types.utils import SpecialEnums
|
from litellm.types.utils import SpecialEnums
|
||||||
|
|
||||||
|
|
||||||
def test_get_file_ids_and_decode_b64_to_unified_uid_from_messages():
|
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 = [
|
messages = [
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "user",
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue