fix(timeout.py): import errors

This commit is contained in:
Krrish Dholakia 2023-11-04 16:05:14 -07:00
parent 5bd6d6ff4c
commit ab54262d37
2 changed files with 2 additions and 1 deletions

View file

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