fix(openai.py): fix linting errors

2023-10-13 22:24:50 -07:00 · 2023-10-13 22:24:50 -07:00 · d77eee34f0
commit d77eee34f0
parent 3210ebfc7a
1 changed files with 2 additions and 0 deletions
--- a/litellm/llms/openai.py
+++ b/litellm/llms/openai.py
@ -246,6 +246,8 @@ class OpenAIChatCompletion(BaseLLM):
               logger_fn=None):
        super().completion()
        headers = self.validate_environment(api_key=api_key)
+        if model is None or messages is None:
+            raise CustomOpenAIError(status_code=422, message=f"Missing model or messages")

        for _ in range(2): # if call fails due to alternating messages, retry with reformatted message
            data = {