Fix: issues with endpoints

Author: Mcastres

admin/src/api/completions/index.js

@@ -1,10 +1,10 @@
 import { request } from '@strapi/helper-plugin';
 
 const completions = {
-  create: async ({ model, prompt, temperature, maxTokens }) => {
+  create: async ({ model, messages, temperature, maxTokens }) => {
     const data = await request(`/open-ai/completions`, {
       method: 'POST',
-      body: { model, prompt, temperature, maxTokens },
+      body: { model, messages, temperature, maxTokens },
     });
     return data;
   },

admin/src/components/CMEditView/RightLinksCompo/index.js

@@ -126,13 +126,15 @@ const RightLinksCompo = () => {
 
   const handlePromptSubmit = () => {
     if (model && prompt && temperature && maxTokens) {
+      const messages = [{ role: 'user', content: prompt }];
       setGenerateCompletionText('Generating completion...');
       completionAPI
-        .create({ model, prompt, temperature, maxTokens })
+        .create({ model, messages, temperature, maxTokens })
         .then((data) => {
-          console.log(data);
           setCompletion(data?.choices[0]?.text.trim());
           setFinishReason(data?.choices[0]?.finish_reason);
+        })
+        .finally(() => {
           setGenerateCompletionText('Generate');
         });
     }
@@ -242,10 +244,7 @@ const RightLinksCompo = () => {
                           paddingLeft={4}
                           background="neutral0"
                         >
-                          <Button
-                            paddingTop={4}
-                            onClick={() => handlePromptSubmit()}
-                          >
+                          <Button onClick={() => handlePromptSubmit()}>
                             {generateCompletionText}
                           </Button>
                         </Box>

server/controllers/completions.js

@@ -4,13 +4,13 @@ module.exports = ({ strapi }) => {
   const completionService = strapi.plugins['open-ai'].services.completions;
 
   const createCompletion = async (ctx) => {
-    const { model, prompt, temperature, maxTokens } = ctx.request.body;
+    const { model, messages, temperature, maxTokens } = ctx.request.body;
 
-    if (model && prompt && temperature && maxTokens) {
+    if (model && messages && temperature && maxTokens) {
       try {
         return completionService.createCompletion({
           model,
-          prompt,
+          messages,
           temperature,
           maxTokens,
         });
@@ -19,7 +19,7 @@ module.exports = ({ strapi }) => {
         ctx.throw(500, err);
       }
     }
-    return ctx.throw(400, 'Either the prompt, temperature, model or maxToken parameter is missing');
+    return ctx.throw(400, 'Either the messages, temperature, model or maxToken parameter is missing');
   };
 
   return {

server/services/completions.js

@@ -3,18 +3,31 @@
 const fetch = require('node-fetch');
 
 module.exports = ({ strapi }) => {
-  const createCompletion = async ({ model, prompt, temperature, maxTokens }) => {
+  const createCompletion = async ({
+    model,
+    messages,
+    temperature,
+    maxTokens,
+  }) => {
     try {
-      const response = await fetch(`https://api.openai.com/v1/completions`, {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-          Authorization: `Bearer ${strapi
-            .plugin('open-ai')
-            .config('API_TOKEN')}`,
-        },
-        body: JSON.stringify({ model, prompt, temperature, max_tokens: maxTokens }),
-      });
+      const response = await fetch(
+        `https://api.openai.com/v1/chat/completions`,
+        {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${strapi
+              .plugin('open-ai')
+              .config('API_TOKEN')}`,
+          },
+          body: JSON.stringify({
+            model,
+            messages,
+            temperature,
+            max_tokens: maxTokens,
+          }),
+        }
+      );
 
       const res = await response.json();
       return res;

server/services/settings.js

@@ -17,72 +17,27 @@ module.exports = ({ strapi }) => {
       temperature: 1,
       maxTokens: 16,
       models: [
-        'babbage',
-        'ada',
-        'davinci',
+        'dall-e-3',
+        'whisper-1',
+        'davinci-002',
+        'babbage-002',
+        'dall-e-2',
+        'gpt-3.5-turbo-16k',
+        'tts-1-hd-1106',
+        'tts-1-hd',
+        'gpt-3.5-turbo-1106',
+        'gpt-3.5-turbo-instruct-0914',
+        'gpt-3.5-turbo-instruct',
+        'tts-1',
+        'gpt-3.5-turbo-0301',
+        'tts-1-1106',
+        'gpt-3.5-turbo-0125',
+        'text-embedding-3-large',
+        'gpt-3.5-turbo',
+        'text-embedding-3-small',
+        'gpt-3.5-turbo-0613',
         'text-embedding-ada-002',
-        'babbage-code-search-code',
-        'text-similarity-babbage-001',
-        'text-davinci-001',
-        'curie-instruct-beta',
-        'babbage-code-search-text',
-        'babbage-similarity',
-        'curie-search-query',
-        'code-search-babbage-text-001',
-        'code-cushman-001',
-        'code-search-babbage-code-001',
-        'code-davinci-002',
-        'text-ada-001',
-        'text-similarity-ada-001',
-        'text-davinci-insert-002',
-        'ada-code-search-code',
-        'text-davinci-002',
-        'ada-similarity',
-        'text-davinci-003',
-        'code-search-ada-text-001',
-        'text-search-ada-query-001',
-        'text-curie-001',
-        'text-davinci-edit-001',
-        'davinci-search-document',
-        'ada-code-search-text',
-        'text-search-ada-doc-001',
-        'code-davinci-edit-001',
-        'davinci-instruct-beta',
-        'text-babbage-001',
-        'text-similarity-curie-001',
-        'code-search-ada-code-001',
-        'ada-search-query',
-        'text-search-davinci-query-001',
-        'curie-similarity',
-        'davinci-search-query',
-        'text-davinci-insert-001',
-        'babbage-search-document',
-        'ada-search-document',
-        'curie',
-        'text-search-babbage-doc-001',
-        'text-search-curie-doc-001',
-        'text-search-curie-query-001',
-        'babbage-search-query',
-        'text-search-davinci-doc-001',
-        'text-search-babbage-query-001',
-        'curie-search-document',
-        'text-similarity-davinci-001',
-        'audio-transcribe-001',
-        'davinci-similarity',
-        'cushman:2020-05-03',
-        'ada:2020-05-03',
-        'babbage:2020-05-03',
-        'curie:2020-05-03',
-        'davinci:2020-05-03',
-        'if-davinci-v2',
-        'if-curie-v2',
-        'if-davinci:3.0.0',
-        'davinci-if:3.0.0',
-        'davinci-instruct-beta:2.0.0',
-        'text-ada:001',
-        'text-davinci:001',
-        'text-curie:001',
-        'text-babbage:001',
+        'gpt-3.5-turbo-16k-0613',
       ],
     };
     await pluginStore.set({ key: 'settings', value });