aboutsummaryrefslogtreecommitdiff
path: root/src/client/apis/gpt/customization.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/apis/gpt/customization.ts')
-rw-r--r--src/client/apis/gpt/customization.ts13
1 files changed, 7 insertions, 6 deletions
diff --git a/src/client/apis/gpt/customization.ts b/src/client/apis/gpt/customization.ts
index 135b83353..db91e45a7 100644
--- a/src/client/apis/gpt/customization.ts
+++ b/src/client/apis/gpt/customization.ts
@@ -1,5 +1,6 @@
-import { ChatCompletionRequestMessage } from 'openai';
+import { ChatCompletionMessageParam } from 'openai/resources';
import { openai } from './setup';
+import { ClientOptions, OpenAI } from 'openai';
export enum CustomizationType {
PRES_TRAIL_SLIDE = 'trails',
@@ -67,7 +68,7 @@ export const gptTrailSlideCustomization = async (inputText: string) => {
prompt += 'If the input does not contain info a specific key, please set their value to null. Please only return the json with these keys and their values.';
try {
- const response = await openai.createChatCompletion({
+ const response = await openai.chat.completions.create({
model: 'gpt-3.5-turbo',
messages: [
{ role: 'system', content: prompt },
@@ -76,7 +77,7 @@ export const gptTrailSlideCustomization = async (inputText: string) => {
temperature: 0.1,
max_tokens: 1000,
});
- return response.data.choices[0].message?.content;
+ return response.choices[0].message?.content;
} catch (err) {
console.log(err);
return 'Error connecting with API.';
@@ -115,7 +116,7 @@ export const generatePalette = async (inputData: StyleInput, useImageData: boole
// iteration
- let messages: ChatCompletionRequestMessage[] = [
+ let messages: ChatCompletionMessageParam[] = [
{ role: 'system', content: prompt },
{ role: 'user', content: JSON.stringify(inputData) },
];
@@ -131,13 +132,13 @@ export const generatePalette = async (inputData: StyleInput, useImageData: boole
console.log('Messages: ', messages);
try {
- const response = await openai.createChatCompletion({
+ const response = await openai.chat.completions.create({
model: 'gpt-4',
messages: messages,
temperature: 0.1,
max_tokens: 2000,
});
- const content = response.data.choices[0].message?.content;
+ const content = response.choices[0].message?.content;
console.log(content);
if (content) {
return content;