mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
fix(otel): read cache payload fields defensively without casts
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
0a8ae7e624
commit
1051c9c100
2 changed files with 3 additions and 3 deletions
|
|
@ -246,7 +246,7 @@ class LLMCallEvent:
|
|||
auth_metadata=auth_metadata(payload, kwargs),
|
||||
is_no_upstream_call=bool(kwargs.get(LITELLM_LOGGING_NO_UPSTREAM_LLM_CALL)),
|
||||
upstream_started=kwargs.get("api_call_start_time") is not None,
|
||||
cache_hit=bool(cast("Mapping[str, object]", payload).get("cache_hit")) if payload else False,
|
||||
cache_hit=bool(payload.get("cache_hit")) if payload else False, # pyright: ignore[reportUnknownMemberType] # payload dicts may omit this key
|
||||
provisional_span_name=f"{operation.value} {model}".strip(),
|
||||
time_to_first_chunk_seconds=time_to_first_chunk_seconds(kwargs),
|
||||
trace=trace,
|
||||
|
|
|
|||
|
|
@ -471,8 +471,8 @@ class LLMCallSpanData:
|
|||
trace=trace or TraceControls(),
|
||||
session_id=session_id or None,
|
||||
embedding_output=embedding_output if capture_content else None,
|
||||
cache_hit=as_bool(cast("Mapping[str, object]", payload).get("cache_hit")),
|
||||
saved_cache_cost=as_float(cast("Mapping[str, object]", payload).get("saved_cache_cost")),
|
||||
cache_hit=as_bool(payload.get("cache_hit")), # pyright: ignore[reportUnknownMemberType] # payload dicts may omit this key
|
||||
saved_cache_cost=as_float(payload.get("saved_cache_cost")), # pyright: ignore[reportUnknownMemberType] # payload dicts may omit this key
|
||||
)
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue