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.
55 lines
1.3 KiB
TypeScript
55 lines
1.3 KiB
TypeScript
import dotenv from 'dotenv'
|
|
import { ChatGPTAPI } from 'chatgpt'
|
|
|
|
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')
|
|
|
|
const chatContext = new Set<ChatContext>()
|
|
|
|
/**
|
|
* More Info: https://github.com/transitive-bullshit/chatgpt-api
|
|
*/
|
|
const api = new ChatGPTAPI({ apiKey })
|
|
|
|
async function chatReply(message: string) {
|
|
if (!message)
|
|
return
|
|
|
|
// Get the last context from the chat context
|
|
// If there is a last context, add it to the options
|
|
let options = {}
|
|
const lastContext = Array.from(chatContext).pop()
|
|
if (lastContext) {
|
|
const { conversationId, parentMessageId } = lastContext
|
|
options = { conversationId, parentMessageId }
|
|
}
|
|
|
|
// Send the message to the API
|
|
const response = await api.sendMessage(message, { ...options })
|
|
|
|
const { conversationId, id } = response
|
|
|
|
// Add the new context to the chat context
|
|
if (conversationId && id)
|
|
chatContext.add({ conversationId, parentMessageId: id })
|
|
|
|
return response
|
|
}
|
|
|
|
async function clearChatContext() {
|
|
// Clear the chat context
|
|
chatContext.clear()
|
|
return Promise.resolve({ message: 'Chat context cleared' })
|
|
}
|
|
|
|
export { chatReply, clearChatContext }
|