scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 87

refactor(langfuse): read the response id through a typed protocol

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-15 08:09:55 +00:00
parent 79ec58ad33
commit 4cfed77d9a
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/integrations/langfuse/langfuse.py
View file

@ -8,7 +8,7 @@ from datetime import datetime
from functools import lru_cache
from importlib.metadata import version
from types import MappingProxyType
from typing import TYPE_CHECKING, Any, Final, Literal, Protocol, cast
from typing import TYPE_CHECKING, Any, Final, Literal, Protocol, cast, runtime_checkable
from packaging.version import Version
@ -143,13 +143,16 @@ def _logging_id(start_time: datetime | None, response_obj: object) -> str | None
return litellm.utils.get_logging_id(start_time, response_obj)
@runtime_checkable
class _ResponseWithId(Protocol):
"""Response payloads (ModelResponse and friends, or a plain dict) expose their provider id via ``get``."""
def get(self, key: Literal["id"], default: None = None, /) -> object: ...
def _lookup_ids(litellm_call_id: str | None, response_obj: object) -> Mapping[str, str]:
"""v2 carried the response id inside the generation id; v4 hashes ids to 16 hex chars, so they ride in metadata."""
response_id: Final[object] = (
cast(Mapping[str, object], response_obj).get("id")
if isinstance(response_obj, Mapping)
else getattr(response_obj, "id", None)
)
response_id: Final[object] = response_obj.get("id") if isinstance(response_obj, _ResponseWithId) else None
ids: Final[tuple[tuple[str, object], ...]] = (("litellm_call_id", litellm_call_id), ("response_id", response_id))
return MappingProxyType({key: str(value) for key, value in ids if value is not None})