phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: fix tests

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-13 19:26:10 -07:00
parent 5a3aaabf05
commit 2e94df30bf
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_add_update_models.py
View file

@ -15,8 +15,12 @@ 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
import pytest, logging, asyncio import pytest, logging, asyncio
import litellm, asyncio import litellm
from litellm.proxy.proxy_server import add_new_model, update_model, LitellmUserRoles from litellm.proxy.management_endpoints.model_management_endpoints import (
add_new_model,
update_model,
)
from litellm.proxy._types import LitellmUserRoles
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
from litellm.proxy.utils import PrismaClient, ProxyLogging from litellm.proxy.utils import PrismaClient, ProxyLogging
from litellm.proxy.management_endpoints.team_endpoints import new_team from litellm.proxy.management_endpoints.team_endpoints import new_team