diff options
Diffstat (limited to 'src/client/apis/gpt/customization.ts')
-rw-r--r-- | src/client/apis/gpt/customization.ts | 13 |
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; |