From 1734e75f25b239bfcee550baf2851d1208a99be2 Mon Sep 17 00:00:00 2001
From: Vesna Tanko <tankovesna@hotmail.com>
Date: Thu, 7 Dec 2023 08:48:02 +0100
Subject: [PATCH] GPT: Migrate to newest API

---
 orangecontrib/prototypes/widgets/owchatgptbase.py | 4 ++--
 setup.py                                          | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/orangecontrib/prototypes/widgets/owchatgptbase.py b/orangecontrib/prototypes/widgets/owchatgptbase.py
index 2e51819f..6086be86 100644
--- a/orangecontrib/prototypes/widgets/owchatgptbase.py
+++ b/orangecontrib/prototypes/widgets/owchatgptbase.py
@@ -26,12 +26,12 @@ def run_gpt(
         prompt_start: str,
         prompt_end: str
 ) -> str:
-    openai.api_key = api_key
+    client = openai.OpenAI(api_key=api_key)
     enc = tiktoken.encoding_for_model(model)
 
     text = enc.decode(enc.encode(text)[:3500])
     content = f"{prompt_start}\n{text}.\n{prompt_end}"
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model=model,
         messages=[
             {"role": "system", "content": "You are a helpful assistant."},
diff --git a/setup.py b/setup.py
index 5f85978c..8bfcd3b5 100644
--- a/setup.py
+++ b/setup.py
@@ -70,7 +70,7 @@ def include_documentation(local_dir, install_dir):
             'pyqtgraph',
             'AnyQt>=0.1.0',
             'pandas>=1.3.0',
-            'openai',
+            'openai>=1',
             'tiktoken',
         ],
         extras_require={