diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py index 27f0c3d92f..726c2b3b97 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -161,7 +161,7 @@ class OpenAIChatCompletion(BaseLLM): def completion(self, model_response: ModelResponse, - timeout: Any, + timeout: float, model: Optional[str]=None, messages: Optional[list]=None, print_verbose: Optional[Callable]=None, diff --git a/litellm/main.py b/litellm/main.py index b6056fda8c..2bb3671979 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -976,7 +976,8 @@ def completion( litellm_params=litellm_params, logger_fn=logger_fn, logging_obj=logging, - acompletion=acompletion + acompletion=acompletion, + timeout=timeout ) ## LOGGING logging.post_call(