mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-10 22:41:41 +00:00
39 lines
1,019 B
Python
39 lines
1,019 B
Python
from __future__ import annotations
|
|
|
|
from typing import Final
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
from .profiler import FunctionTraceEvent
|
|
|
|
|
|
class _TraceEventPayload(BaseModel):
|
|
model_config = ConfigDict(strict=True, extra="forbid")
|
|
id: int
|
|
parent_id: int | None
|
|
function: str
|
|
module_path: str | None = None
|
|
file: str | None = None
|
|
line: int | None = None
|
|
|
|
|
|
class TraceResponsePayload(BaseModel):
|
|
model_config = ConfigDict(strict=True, extra="forbid")
|
|
response: object = None
|
|
error: str | None = None
|
|
trace: tuple[_TraceEventPayload, ...] | list[_TraceEventPayload]
|
|
|
|
|
|
def native_trace_events(payload: object) -> tuple[FunctionTraceEvent, ...]:
|
|
response: Final = TraceResponsePayload.model_validate(payload)
|
|
return tuple(
|
|
FunctionTraceEvent(
|
|
event.id,
|
|
event.parent_id,
|
|
event.function,
|
|
event.module_path,
|
|
event.file,
|
|
event.line,
|
|
)
|
|
for event in response.trace
|
|
)
|