scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

fix: address PR review feedback on formatting and types

Browse source
This commit is contained in:
Het1819 2026-06-04 17:31:11 -04:00
parent 4cf9229dfb
commit b880e045fd
2 changed files with 4 additions and 43 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
litellm/litellm_core_utils/litellm_logging.py
View file

@ -406,43 +406,6 @@ class Logging(LiteLLMLoggingBaseClass):
self._defer_async_logging: bool = False
self._enqueue_deferred_logging: Optional[Callable[[], None]] = None
# def _calculate_model_transparency(self, response_obj: Any = None) -> dict:
# """
# Calculates the requested vs resolved vs response model mismatch metadata.
# """
# requested_model = self.kwargs.get("model", "")
# # The router usually populates the final resolved model here
# resolved_model = self.kwargs.get("litellm_params", {}).get("model", requested_model)
# response_model = ""
# if response_obj:
# if hasattr(response_obj, "model"):
# response_model = getattr(response_obj, "model", "")
# elif isinstance(response_obj, dict):
# response_model = response_obj.get("model", "")
# model_mismatch = False
# if requested_model != resolved_model:
# model_mismatch = "requested_vs_resolved_mismatch"
# elif response_model and resolved_model:
# # Note: providers sometimes append dates to models (e.g., gpt-4-0613)
# if resolved_model not in response_model and response_model not in resolved_model:
# model_mismatch = "resolved_vs_response_mismatch"
# has_usage = False
# if response_obj:
# if hasattr(response_obj, "usage") and getattr(response_obj, "usage"):
# has_usage = True
# elif isinstance(response_obj, dict) and response_obj.get("usage"):
# has_usage = True
# return {
# "requested_model": requested_model,
# "resolved_model": resolved_model,
# "response_model": response_model,
# "model_mismatch": model_mismatch,
# "usage_source": "upstream" if has_usage else "missing"
# }
def _calculate_model_transparency(self, response_obj: Any = None) -> dict:
"""
Calculates the requested vs resolved vs response model mismatch metadata.
@ -5293,18 +5256,15 @@ def get_standard_logging_object_payload(
standard_built_in_tools_params=standard_built_in_tools_params,
)
# emit_standard_logging_payload(payload) - Moved to success_handler to prevent double emitting
# --- 🚀 ADD STEP 2 INJECTION HERE 🚀 ---
if logging_obj and hasattr(logging_obj, "_calculate_model_transparency"):
# emit_standard_logging_payload(payload) - Moved to success_handler to prevent double emitting
if logging_obj is not None:
transparency_data = logging_obj._calculate_model_transparency(response_obj=init_response_obj)
payload["requested_model"] = transparency_data.get("requested_model", "")
payload["resolved_model"] = transparency_data.get("resolved_model", "")
payload["response_model"] = transparency_data.get("response_model", "")
payload["model_mismatch"] = transparency_data.get("model_mismatch", False)
payload["usage_source"] = transparency_data.get("usage_source", "missing")
# ---------------------------------------
# emit_standard_logging_payload(payload) - Moved to success_handler to prevent double emitting
return payload

1
litellm/types/utils.py
View file

@ -2971,6 +2971,7 @@ class StandardLoggingPayload(TypedDict):
model_mismatch: NotRequired[Union[str, bool]]
usage_source: NotRequired[str]
from typing import AsyncIterator, Iterator