chore: 调整文件结构 service 和 web 分离

service/src/chatgpt.ts

@@ -0,0 +1,54 @@
+import * as 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 }

service/src/index.ts

@@ -0,0 +1,26 @@
+import express from 'express'
+import { chatReply, clearChatContext } from './chatgpt'
+
+const app = express()
+
+app.use(express.json())
+
+app.all('*', (req, res, next) => {
+  res.header('Access-Control-Allow-Origin', '*')
+  res.header('Access-Control-Allow-Headers', 'Content-Type')
+  res.header('Access-Control-Allow-Methods', '*')
+  next()
+})
+
+app.listen(3002, () => globalThis.console.log('Server is running on port 3002'))
+
+app.post('/chat', async (req, res) => {
+  const { message } = req.body
+  const response = await chatReply(message)
+  res.send(response)
+})
+
+app.post('/clear', async (req, res) => {
+  const response = await clearChatContext()
+  res.send(response)
+})