limit openai dependency to <1.0.0 (#2027)

zenml-io · Nov 7, 2023 · 999e96f · 999e96f
1 parent 40fe6c7
commit 999e96f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/zenml/integrations/openai/__init__.py b/src/zenml/integrations/openai/__init__.py
@@ -21,7 +21,7 @@ class OpenAIIntegration(Integration):
     """Definition of OpenAI integration for ZenML."""
 
     NAME = OPEN_AI
-    REQUIREMENTS = ["openai>=0.27.0"]
+    REQUIREMENTS = ["openai>=0.27.0,<1.0.0"]
 
 
 OpenAIIntegration.check_installation()
diff --git a/src/zenml/integrations/openai/hooks/open_ai_failure_hook.py b/src/zenml/integrations/openai/hooks/open_ai_failure_hook.py
@@ -62,7 +62,7 @@ def openai_alerter_failure_hook_helper(
         sys.stdout = original_stdout
         rich_traceback = output_captured.getvalue()
 
-        response = openai.ChatCompletion.create(  # type: ignore[no-untyped-call]
+        response = openai.ChatCompletion.create(  # type: ignore
             model=model_name,
             messages=[
                 {