From 2ea4d9a2f846401fd41caafefc0a61a1c644493b Mon Sep 17 00:00:00 2001 From: Yuta Saito Date: Fri, 2 Jan 2026 16:25:37 +0900 Subject: [PATCH] fix: test error --- litellm/proxy/spend_tracking/spend_tracking_utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/spend_tracking/spend_tracking_utils.py b/litellm/proxy/spend_tracking/spend_tracking_utils.py index dedb9b6659..1c457d7bf4 100644 --- a/litellm/proxy/spend_tracking/spend_tracking_utils.py +++ b/litellm/proxy/spend_tracking/spend_tracking_utils.py @@ -397,9 +397,8 @@ def get_logging_payload( # noqa: PLR0915 # Extract agent_id for A2A requests (set directly on model_call_details) agent_id: Optional[str] = kwargs.get("agent_id") custom_llm_provider = kwargs.get("custom_llm_provider") - model_name = reconstruct_model_name( - kwargs.get("model", ""), custom_llm_provider, metadata - ) + raw_model = cast(str, kwargs.get("model") or "") + model_name = reconstruct_model_name(raw_model, custom_llm_provider, metadata or {}) try: payload: SpendLogsPayload = SpendLogsPayload(