use chat completions api

vldc-hq · Dec 28, 2023 · 71eb8d7 · 71eb8d7
1 parent 2624978
commit 71eb8d7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/bot/skills/buktopuha.py b/bot/skills/buktopuha.py
@@ -328,11 +328,9 @@ def check_for_answer(update: Update, context: CallbackContext):
 
 
 def generate_question(prompt, word) -> str:
-    model = random.choice(
-        ["gpt-3.5-turbo-instruct", "gpt-4-1106-preview", "gemini-pro"]
-    )
+    model = random.choice(["gpt-3.5-turbo", "gpt-4-1106-preview", "gemini-pro"])
     if model.startswith("gpt"):
-        response = openai.Completion.create(
+        response = openai.ChatCompletion.create(
             model=model,
             prompt=prompt,
             temperature=0.9,