test(batches): use the public hook class names main now exports

tests/unit/proxy/hooks/test_model_max_budget_limiter.py

@@ -202,7 +202,7 @@ async def test_a_batch_polled_within_every_budget_window_is_never_charged_again(
 async def test_batch_line_item_events_do_not_charge_the_model_budget():
     """Line events carry batch_parent_id; the aggregate aretrieve_batch event is
     the one that already bills the batch, so children must not double-charge."""
-    limiter: Final = _PROXY_VirtualKeyModelMaxBudgetLimiter(dual_cache=DualCache())
+    limiter: Final = PROXY_VirtualKeyModelMaxBudgetLimiter(dual_cache=DualCache())
 
     line_event = _event("acompletion", CHAT_COST)
     line_event["litellm_params"]["batch_parent_id"] = "batch_first"

tests/unit/proxy/hooks/test_proxy_track_cost_callback.py

@@ -2774,7 +2774,7 @@ async def test_async_post_call_failure_hook_persists_no_raw_model_on_an_unknown_
 
 @pytest.mark.asyncio
 async def test_batch_line_item_event_never_updates_spend():  # test-quality-ok: the observable contract is exactly that the spend path is never invoked
-    logger: Final = _ProxyDBLogger()
+    logger: Final = ProxyDBLogger()
     kwargs: Final = {
         "litellm_params": {"batch_parent_id": "batch_1", "metadata": {}},
         "model": "gpt-4o",