From fcc86cddf7a51787ff9e77870daecdbb37847b1b Mon Sep 17 00:00:00 2001
From: Luke Hollenback <8144622+lukehollenback@users.noreply.github.com>
Date: Mon, 1 Apr 2024 18:51:54 -0600
Subject: [PATCH] Forced JSON mode on newer GPT-3 Turbo models as well.

As supported according to https://platform.openai.com/docs/guides/text-generation/json-mode.
---
 src/main.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.ts b/src/main.ts
index 106470c..ed701c4 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -127,7 +127,7 @@ async function getAIResponse(prompt: string): Promise<Array<{
     const response = await openai.chat.completions.create({
       ...queryConfig,
       // return JSON if the model supports it:
-      ...(OPENAI_API_MODEL === "gpt-4-turbo-preview" || OPENAI_API_MODEL === "gpt-4-0125-preview" || OPENAI_API_MODEL === "gpt-4-1106-preview"
+      ...(OPENAI_API_MODEL === "gpt-4-turbo-preview" || OPENAI_API_MODEL === "gpt-3.5-turbo" || OPENAI_API_MODEL === "gpt-4-0125-preview" || OPENAI_API_MODEL === "gpt-4-1106-preview" || OPENAI_API_MODEL === "gpt-3.5-turbo-0125" || OPENAI_API_MODEL === "gpt-3.5-turbo-1106"
         ? { response_format: { type: "json_object" } }
         : {}),
       messages: [