scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-29 06:33:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 200

fix(transcription): type SDK stream duration metadata explicitly

Browse source
This commit is contained in:
Emerson Gomes 2026-09-15 12:40:44 -05:00
parent bd603fa44e
commit d9f21872e2
No known key found for this signature in database
GPG key ID: D3DF28AB5D1B5E17
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -8218,7 +8218,7 @@ async def atranscription(
if file is not None:
calculated_duration = calculate_request_duration(file)
if calculated_duration is not None:
setattr(response, "_litellm_audio_duration", calculated_duration)
response._litellm_audio_duration = calculated_duration # pyright: ignore[reportAttributeAccess] # SDK streams permit runtime metadata but do not type this field
return response
if not isinstance(response, TranscriptionResponse):
raise ValueError(
@ -8562,7 +8562,7 @@ def transcription(
calculated_duration: Final = calculate_request_duration(file)
if isinstance(response, (Stream, AsyncStream)):
if calculated_duration is not None:
setattr(response, "_litellm_audio_duration", calculated_duration)
response._litellm_audio_duration = calculated_duration # pyright: ignore[reportAttributeAccess] # SDK streams permit runtime metadata but do not type this field
elif getattr(response, "duration", None) is None and calculated_duration is not None:
response.set_audio_transcription_duration(calculated_duration)