Merge branch 'master' into sophie-ai-images

1 files changed, 35 insertions, 5 deletions

diff --git a/src/client/apis/gpt/GPT.ts b/src/client/apis/gpt/GPT.ts
index 9403e4d3a..cea862330 100644
--- a/src/client/apis/gpt/GPT.ts
+++ b/src/client/apis/gpt/GPT.ts
@@ -1,4 +1,23 @@
-import { GPTCallOpts, GPTCallType, callTypeMap, openai } from './setup';
+import { ClientOptions, OpenAI } from 'openai';
+
+enum GPTCallType {
+    SUMMARY = 'summary',
+    COMPLETION = 'completion',
+    EDIT = 'edit',
+}
+
+type GPTCallOpts = {
+    model: string;
+    maxTokens: number;
+    temp: number;
+    prompt: string;
+};
+
+const callTypeMap: { [type: string]: GPTCallOpts } = {
+    summary: { model: 'text-davinci-003', maxTokens: 256, temp: 0.5, prompt: 'Summarize this text in simpler terms: ' },
+    edit: { model: 'text-davinci-003', maxTokens: 256, temp: 0.5, prompt: 'Reword this: ' },
+    completion: { model: 'text-davinci-003', maxTokens: 256, temp: 0.5, prompt: '' },
+};
 
 /**
  * Calls the OpenAI API.
@@ -10,13 +29,18 @@ const gptAPICall = async (inputText: string, callType: GPTCallType) => {
     if (callType === GPTCallType.SUMMARY) inputText += '.';
     const opts: GPTCallOpts = callTypeMap[callType];
     try {
-        const response = await openai.createCompletion({
+        const configuration: ClientOptions = {
+            apiKey: process.env.OPENAI_KEY,
+            dangerouslyAllowBrowser: true,
+        };
+        const openai = new OpenAI(configuration);
+        const response = await openai.completions.create({
             model: opts.model,
             max_tokens: opts.maxTokens,
             temperature: opts.temp,
             prompt: `${opts.prompt}${inputText}`,
         });
-        return response.data.choices[0].text;
+        return response.choices[0].text;
     } catch (err) {
         console.log(err);
         return 'Error connecting with API.';
@@ -25,12 +49,18 @@ const gptAPICall = async (inputText: string, callType: GPTCallType) => {
 
 const gptImageCall = async (prompt: string, n?: number) => {
     try {
-        const response = await openai.createImage({
+        const configuration: ClientOptions = {
+            apiKey: process.env.OPENAI_KEY,
+            dangerouslyAllowBrowser: true,
+        };
+
+        const openai = new OpenAI(configuration);
+        const response = await openai.images.generate({
             prompt: prompt,
             n: n ?? 1,
             size: '1024x1024',
         });
-        return response.data.data.map(data => data.url);
+        return response.data.map((data: any) => data.url);
         // return response.data.data[0].url;
     } catch (err) {
         console.error(err);