test(test_completion.py): num retries 3

This commit is contained in:
Krrish Dholakia 2023-11-04 15:55:56 -07:00
parent 5b3978eff4
commit 5bd6d6ff4c
3 changed files with 6 additions and 4 deletions

View file

@ -17,7 +17,7 @@ from concurrent import futures
from inspect import iscoroutinefunction
from functools import wraps
from threading import Thread
from openai.error import Timeout
from litellm.error import Timeout
def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout):