fix(timeout.py): fix timeout issue

This commit is contained in:
Krrish Dholakia 2023-11-13 11:07:09 -08:00
parent 8ae511b178
commit b6eea3cf8e
3 changed files with 5 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 import Timeout
from litellm.exceptions import Timeout
def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout):