You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.1 KiB
TypeScript
46 lines
1.1 KiB
TypeScript
import * as dotenv from 'dotenv'
|
|
import type { SendMessageOptions } from 'chatgpt'
|
|
import { ChatGPTAPI } from 'chatgpt'
|
|
import { sendResponse } from './utils'
|
|
|
|
export interface ChatContext {
|
|
conversationId?: string
|
|
parentMessageId?: string
|
|
}
|
|
|
|
dotenv.config()
|
|
|
|
const apiKey = process.env.OPENAI_API_KEY
|
|
|
|
if (apiKey === undefined)
|
|
throw new Error('OPENAI_API_KEY is not defined')
|
|
|
|
/**
|
|
* More Info: https://github.com/transitive-bullshit/chatgpt-api
|
|
*/
|
|
const api = new ChatGPTAPI({ apiKey, debug: false })
|
|
|
|
async function chatReply(
|
|
message: string,
|
|
lastContext?: { conversationId?: string; parentMessageId?: string },
|
|
) {
|
|
if (!message)
|
|
return sendResponse({ type: 'Fail', message: 'Message is empty' })
|
|
|
|
try {
|
|
let options: SendMessageOptions = {}
|
|
|
|
if (lastContext)
|
|
options = { ...lastContext }
|
|
|
|
const response = await api.sendMessage(message, { ...options })
|
|
|
|
return sendResponse({ type: 'Success', data: response })
|
|
}
|
|
catch (error: any) {
|
|
return sendResponse({ type: 'Fail', message: error.message })
|
|
}
|
|
}
|
|
|
|
export { chatReply }
|