feat: 支持上下文联想

service/chatgpt.ts

@@ -1,19 +1,54 @@
 import dotenv from 'dotenv'
 import { ChatGPTAPI } from 'chatgpt'
 
+interface ChatContext {
+  conversationId?: string
+  parentMessageId?: string
+}
+
 dotenv.config()
 
-const apiKey = ''
+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: process.env.OPENAI_API_KEY || apiKey })
+const api = new ChatGPTAPI({ apiKey })
 
 async function chatReply(message: string) {
   if (!message)
     return
-  return await api.sendMessage(message)
+
+  // 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
 }
 
-export { chatReply }
+async function clearChatContext() {
+  // Clear the chat context
+  chatContext.clear()
+  return Promise.resolve({ message: 'Chat context cleared' })
+}
+
+export { chatReply, clearChatContext }

service/index.ts

@@ -1,5 +1,5 @@
 import express from 'express'
-import { chatReply } from './chatgpt'
+import { chatReply, clearChatContext } from './chatgpt'
 
 const app = express()
 
@@ -19,3 +19,8 @@ app.post('/chat', async (req, res) => {
   const response = await chatReply(message)
   res.send(response)
 })
+
+app.post('/clear', async (req, res) => {
+  const response = await clearChatContext()
+  res.send(response)
+})