feat: 添加上下文关联

electron/main/chat.ts

@@ -2,9 +2,12 @@ import { ipcMain } from 'electron'
 import { Configuration, OpenAIApi } from 'openai'
 import HttpsProxyAgent from 'https-proxy-agent'
 import HttpProxyAgent from 'http-proxy-agent'
-import { createFailMessage, createSuccessMessage } from '../utils/message'
+import { Context, createFailMessage, createSuccessMessage } from '../utils/message'
 import { md2Html } from '../utils/markdown'
 
+// 上下文
+const context = new Context()
+
 const httpsAgent = new HttpsProxyAgent('http://127.0.0.1:4780')
 const httpAgent = new HttpProxyAgent('http://127.0.0.1:4780')
 export async function chat(win: Electron.BrowserWindow) {
@@ -16,19 +19,22 @@ export async function chat(win: Electron.BrowserWindow) {
 
   ipcMain.handle('send-chat', async (e, { content }: { content: string }) => {
     try {
+      const preContext = context.context
       const res = await openai.createChatCompletion({
         model: 'gpt-3.5-turbo',
         messages: [
           {
             role: 'system',
             content: '麻烦请用 markdown 回复我。',
           },
+          ...preContext,
           { role: 'user', content },
         ],
       }, {
         httpAgent,
         httpsAgent,
       })
+      context.add(content, res.data.choices[0].message.content)
       return createSuccessMessage({
         message: md2Html(res.data.choices[0].message.content),
       })

electron/utils/message.ts

@@ -1,3 +1,35 @@
+import type { ChatCompletionRequestMessage } from 'openai'
+
+// 最大上下文数量
+const MAX_MESSAGE = 3
+
+export class Context {
+  private readonly list: ChatCompletionRequestMessage[]
+
+  constructor() {
+    this.list = []
+  }
+
+  public add(userMessage: string, assistantMessage: string) {
+    if (!userMessage || !assistantMessage)
+      return
+    if (this.list.length === MAX_MESSAGE * 2)
+      this.list.splice(0, 2)
+
+    this.list.push({
+      role: 'user',
+      content: userMessage,
+    }, {
+      role: 'assistant',
+      content: assistantMessage,
+    })
+  }
+
+  public get context() {
+    return [...this.list]
+  }
+}
+
 const errorCodeMap: Record<string, string> = {
   401: 'error 401 可能是你的 apiKey 不对劲！',
   403: 'error 403 拒绝访问，哒咩！',

tsconfig.node.json

@@ -1,5 +1,6 @@
 {
   "compilerOptions": {
+    "target": "ESNext",
     "composite": true,
     "module": "ESNext",
     "moduleResolution": "Node",