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

fix: fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-20 05:17:27 +05:30
parent 5be647dd76
commit aed1cd3283
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

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

6
tests/local_testing/test_whisper.py
View file

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