From dbf223db8776609d46c5abffe99a967a9b05c997 Mon Sep 17 00:00:00 2001
From: Mervin van Brakel <63094424+mervinvb@users.noreply.github.com>
Date: Thu, 19 Dec 2024 14:12:34 +0100
Subject: [PATCH] Change to gpt-4o-mini

---
 src/openai_interfacing.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/openai_interfacing.py b/src/openai_interfacing.py
index c397d23..927d689 100644
--- a/src/openai_interfacing.py
+++ b/src/openai_interfacing.py
@@ -167,13 +167,9 @@ async def send_ai_text(prompt_type: str, prompt: str, job_id: str, websocket) ->
 
     prompt_length = await get_token_size(prompt)
 
-    if 4000 < prompt_length < 16000:
-        model = "gpt-3.5-turbo-16k-0613"
-    elif prompt_length > 16000:
-        model = "gpt-3.5-turbo"
+    model = "gpt-4o-mini"
+    if prompt_length > 128000:
         prompt = prompts["log_too_log"]
-    else:
-        model = "gpt-3.5-turbo"
 
     try:
         response = await openai.ChatCompletion.acreate(