phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat): Add timestamp_granularities parameter to transcription API (#6457)

Browse source 
* Add timestamp_granularities parameter to transcription API

* add param to the local test
This commit is contained in:
Show 2024-11-20 00:04:33 +01:00 • committed by GitHub
parent f4ec93fbc3
commit e89dcccdd9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -4729,6 +4729,7 @@ def transcription(
response_format: Optional[ response_format: Optional[
Literal["json", "text", "srt", "verbose_json", "vtt"] Literal["json", "text", "srt", "verbose_json", "vtt"]
] = None, ] = None,
timestamp_granularities: List[Literal["word", "segment"]] = None,
temperature: Optional[int] = None, # openai defaults this to 0 temperature: Optional[int] = None, # openai defaults this to 0
## LITELLM PARAMS ## ## LITELLM PARAMS ##
user: Optional[str] = None, user: Optional[str] = None,
@ -4778,6 +4779,7 @@ def transcription(
language=language, language=language,
prompt=prompt, prompt=prompt,
response_format=response_format, response_format=response_format,
timestamp_granularities=timestamp_granularities,
temperature=temperature, temperature=temperature,
custom_llm_provider=custom_llm_provider, custom_llm_provider=custom_llm_provider,
drop_params=drop_params, drop_params=drop_params,

1
litellm/utils.py
View file

@ -2125,6 +2125,7 @@ def get_optional_params_transcription(
prompt: Optional[str] = None, prompt: Optional[str] = None,
response_format: Optional[str] = None, response_format: Optional[str] = None,
temperature: Optional[int] = None, temperature: Optional[int] = None,
timestamp_granularities: Optional[List[Literal["word", "segment"]]] = None,
custom_llm_provider: Optional[str] = None, custom_llm_provider: Optional[str] = None,
drop_params: Optional[bool] = None, drop_params: Optional[bool] = None,
**kwargs, **kwargs,

4
tests/local_testing/test_whisper.py
View file

@ -53,8 +53,9 @@ from litellm import Router
) )
@pytest.mark.parametrize("response_format", ["json", "vtt"]) @pytest.mark.parametrize("response_format", ["json", "vtt"])
@pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.parametrize("timestamp_granularities", [["word"], ["segment"]])
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_transcription(model, api_key, api_base, response_format, sync_mode): async def test_transcription(model, api_key, api_base, response_format, sync_mode, timestamp_granularities):
if sync_mode: if sync_mode:
transcript = litellm.transcription( transcript = litellm.transcription(
model=model, model=model,
@ -62,6 +63,7 @@ async def test_transcription(model, api_key, api_base, response_format, sync_mod
api_key=api_key, api_key=api_key,
api_base=api_base, api_base=api_base,
response_format=response_format, response_format=response_format,
timestamp_granularities=timestamp_granularities,
drop_params=True, drop_params=True,
) )
else: else: