Update tests/test_litellm/proxy/test_litellm_pre_call_utils.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Samarth KaPatel 2026-04-06 22:24:30 -04:00 committed by GitHub
parent 76867f3aa3
commit aaa0ec9699
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -10,8 +10,6 @@ from fastapi import Request
import litellm
from litellm.proxy._types import TeamCallbackMetadata, UserAPIKeyAuth
from unittest.mock import MagicMock
from litellm.proxy.litellm_pre_call_utils import LiteLLMProxyRequestSetup
from litellm.proxy.litellm_pre_call_utils import (
KeyAndTeamLoggingSettings,
LiteLLMProxyRequestSetup,
@ -23,6 +21,14 @@ from litellm.proxy.litellm_pre_call_utils import (
add_litellm_data_to_request,
check_if_token_is_service_account,
)
_get_dynamic_logging_metadata,
_get_enforced_params,
_get_metadata_variable_name,
_update_model_if_key_alias_exists,
add_guardrails_from_policy_engine,
add_litellm_data_to_request,
check_if_token_is_service_account,
)
sys.path.insert(
0, os.path.abspath("../../..")