From aaa0ec96993f829270856c043da91168c0bbf8d9 Mon Sep 17 00:00:00 2001 From: Samarth KaPatel Date: Mon, 6 Apr 2026 22:24:30 -0400 Subject: [PATCH] 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> --- .../test_litellm/proxy/test_litellm_pre_call_utils.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/proxy/test_litellm_pre_call_utils.py b/tests/test_litellm/proxy/test_litellm_pre_call_utils.py index 314227d1765..e78da936ea4 100644 --- a/tests/test_litellm/proxy/test_litellm_pre_call_utils.py +++ b/tests/test_litellm/proxy/test_litellm_pre_call_utils.py @@ -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("../../..")