diff --git a/litellm/proxy/hooks/proxy_track_cost_callback.py b/litellm/proxy/hooks/proxy_track_cost_callback.py
index 088f1af2d5c..8c6d5ab69ab 100644
--- a/litellm/proxy/hooks/proxy_track_cost_callback.py
+++ b/litellm/proxy/hooks/proxy_track_cost_callback.py
@@ -58,6 +58,8 @@ from litellm.proxy.spend_tracking.spend_tracking_utils import (  # noqa: F401  #
     should_store_prompts_and_responses_in_spend_logs,
 )
 from litellm.proxy.utils import BilledFailedAttempt, ProxyUpdateSpend
+from litellm.types.integrations.slack_alerting import CallInfo, Litellm_EntityType
+from litellm.types.proxy.model_access_group_budget import ModelAccessGroupBudget
 from litellm.types.utils import (
     CallTypes,
     LiteLLMBatch,