Change max retries to 1

This commit is contained in:
Jason Liu
2023-10-12 20:02:05 -04:00
parent 899fca354e
commit 41c4636dd2
+2 -2
View File
@@ -103,7 +103,7 @@ def wrap_chatcompletion(func: Callable) -> Callable:
@wraps(func)
async def new_chatcompletion_async(
response_model=None, validation_context=None, *args, max_retries=0, **kwargs
response_model=None, validation_context=None, *args, max_retries=1, **kwargs
):
response_model, new_kwargs = handle_response_model(response_model, kwargs) # type: ignore
response, error = await retry_async(
@@ -120,7 +120,7 @@ def wrap_chatcompletion(func: Callable) -> Callable:
@wraps(func)
def new_chatcompletion_sync(
response_model=None, validation_context=None, *args, max_retries=0, **kwargs
response_model=None, validation_context=None, *args, max_retries=1, **kwargs
):
response_model, new_kwargs = handle_response_model(response_model, kwargs) # type: ignore
response, error = retry_sync(