From bc627808136274ebd064742ad5aa3a34dd076fd0 Mon Sep 17 00:00:00 2001
From: Luke Hollenback <8144622+lukehollenback@users.noreply.github.com>
Date: Mon, 1 Apr 2024 18:48:27 -0600
Subject: [PATCH] Forced JSON mode on all GPT-4 Turbo models newer than the
 1106 preview.

In response to https://github.com/freeedcom/ai-codereviewer/issues/56.
---
 src/main.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.ts b/src/main.ts
index 13677ae4..106470cb 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-1106-preview"
+      ...(OPENAI_API_MODEL === "gpt-4-turbo-preview" || OPENAI_API_MODEL === "gpt-4-0125-preview" || OPENAI_API_MODEL === "gpt-4-1106-preview"
         ? { response_format: { type: "json_object" } }
         : {}),
       messages: [