update openai

vldc-hq · Dec 28, 2023 · 8e61b89 · 8e61b89
1 parent 71eb8d7
commit 8e61b89
Show file tree

Hide file tree

Showing 3 changed files with 191 additions and 8 deletions.
diff --git a/Pipfile b/Pipfile
@@ -40,7 +40,7 @@ Pillow = "==10.0.1"
 toml = "==0.10.2"
 install = "==1.3.5"
 aiohttp = "==3.8.5"
-openai = "==0.28.0"
+openai = "==1.6.1"
 typing-extensions = "==4.8.0"
 google-generativeai = "==0.3.2"
 

diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/bot/skills/buktopuha.py b/bot/skills/buktopuha.py
@@ -330,16 +330,16 @@ def check_for_answer(update: Update, context: CallbackContext):
 def generate_question(prompt, word) -> str:
     model = random.choice(["gpt-3.5-turbo", "gpt-4-1106-preview", "gemini-pro"])
     if model.startswith("gpt"):
-        response = openai.ChatCompletion.create(
+        response = openai.chat.completions.create(
             model=model,
-            prompt=prompt,
+            messages=[{"role": "system", "content": prompt}],
             temperature=0.9,
             max_tokens=150,
             top_p=1,
             frequency_penalty=0,
             presence_penalty=0.6,
         )
-        rs = response["choices"][0]["text"]
+        rs = response.choices[0].message.content
         return f"{model}: " + re.sub(
             word, "***", rs, flags=re.IGNORECASE
         ).strip().strip('"')