Merge pull request #306 from C-Pro/modelname

use chat completions api
vldc-hq · Dec 28, 2023 · af80412 · af80412
2 parents 7a0e4c2 + 71eb8d7
commit af80412
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,