fix(main.py): fix linting issue

2023-11-17 15:45:00 -08:00 · 2023-11-17 15:45:00 -08:00 · 17bb1184bd
commit 17bb1184bd
parent 0ab6b2451d
2 changed files with 3 additions and 2 deletions
--- 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,
--- 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(