fix(main.py): support max retries for transcription calls

This commit is contained in:
Krrish Dholakia 2024-04-01 18:37:53 -07:00
parent c3e4af76cf
commit ceabf726b0
3 changed files with 7 additions and 3 deletions

View file

@ -785,10 +785,10 @@ class OpenAIChatCompletion(BaseLLM):
optional_params: dict,
model_response: TranscriptionResponse,
timeout: float,
max_retries: int,
api_key: Optional[str] = None,
api_base: Optional[str] = None,
client=None,
max_retries=None,
logging_obj=None,
atranscription: bool = False,
):