Upgrade OpenAI library to version 4

package.json

@@ -34,6 +34,6 @@
     "@octokit/core": "^4.1.0",
     "@octokit/plugin-paginate-rest": "^5.0.1",
     "@octokit/rest": "^19.0.5",
-    "openai": "^3.1.0"
+    "openai": "^4.86.2"
   }
 }

src/commitSummary.ts

@@ -94,23 +94,23 @@ async function getOpenAICompletion(
       throw new Error("OpenAI query too big");
     }
 
-    const response = await openai.createCompletion({
+    const response = await openai.chat.completions.create({
       model: MODEL_NAME,
-      prompt: openAIPrompt,
+      messages: [
+        { role: 'user', content: openAIPrompt }
+      ],
       max_tokens: MAX_TOKENS,
       temperature: TEMPERATURE,
     });
-
-    if (
-      response.data.choices !== undefined &&
-      response.data.choices.length > 0
-    ) {
+
+    if (response.choices !== undefined && response.choices.length > 0) {
       completion = postprocessSummary(
         diffResponse.data.files.map((file: any) => file.filename),
-        response.data.choices[0].text ?? "Error: couldn't generate summary",
+        response.choices[0].message?.content ?? "Error: couldn't generate summary",
         diffMetadata
       );
     }
+
   } catch (error) {
     console.error(error);
   }

src/filesSummary.ts

@@ -44,20 +44,19 @@ async function getOpenAISummaryForFile(
       throw new Error("OpenAI query too big");
     }
 
-    const response = await openai.createCompletion({
+    const response = await openai.chat.completions.create({
       model: MODEL_NAME,
-      prompt: openAIPrompt,
+      messages: [
+        { role: 'user', content: openAIPrompt }
+      ],
       max_tokens: MAX_TOKENS,
       temperature: TEMPERATURE,
     });
-    if (
-      response.data.choices !== undefined &&
-      response.data.choices.length > 0
-    ) {
-      return (
-        response.data.choices[0].text ?? "Error: couldn't generate summary"
-      );
+
+    if (response.choices !== undefined && response.choices.length > 0) {
+      return response.choices[0].message?.content ?? "Error: couldn't generate summary";
     }
+
   } catch (error) {
     console.error(error);
   }

src/openAi.ts

@@ -1,12 +1,11 @@
-import { Configuration, OpenAIApi } from "openai";
+import OpenAI from 'openai';
 
-const configuration = new Configuration({
-  apiKey: process.env.OPENAI_API_KEY,
+export const openai = new OpenAI({
+  apiKey: process.env.OPENAI_API_KEY, // This is the default and can be omitted
 });
 
 export const MAX_OPEN_AI_QUERY_LENGTH = 20000;
-export const MODEL_NAME = "text-davinci-003";
+export const MODEL_NAME = "gpt-4o-mini";
 export const TEMPERATURE = 0.5;
 export const MAX_TOKENS = 512;
 
-export const openai = new OpenAIApi(configuration);

src/summarizePr.ts

@@ -48,15 +48,19 @@ export async function summarizePr(
   }
 
   try {
-    const response = await openai.createCompletion({
+    const response = await openai.chat.completions.create({
       model: MODEL_NAME,
-      prompt: openAIPrompt,
+      messages: [
+        { role: 'user', content: openAIPrompt }
+      ],
       max_tokens: MAX_TOKENS,
       temperature: TEMPERATURE,
     });
-    return response.data.choices[0].text ?? "Error: couldn't generate summary";
+
+    return response.choices[0].message?.content ?? "Error: couldn't generate summary";
   } catch (error) {
     console.error(error);
     return "Error: couldn't generate summary";
   }
+
 }