From 4b165609583c6f482c028cc4d732e074a0cd9105 Mon Sep 17 00:00:00 2001
From: ChenZhaoYu <790348264@qq.com>
Date: Sat, 11 Feb 2023 15:20:58 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81=E4=B8=8A=E4=B8=8B?=
 =?UTF-8?q?=E6=96=87=E8=81=94=E6=83=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/chatgpt.ts        | 43 +++++++++++++++++++++++++++++++++++----
 service/index.ts          |  7 ++++++-
 src/views/Chat/index.vue  | 17 +++++++++++-----
 src/views/Chat/request.ts | 22 ++++++++++++++++----
 4 files changed, 75 insertions(+), 14 deletions(-)

diff --git a/service/chatgpt.ts b/service/chatgpt.ts
index 874d6d7..e2cc28c 100644
--- a/service/chatgpt.ts
+++ b/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 }
diff --git a/service/index.ts b/service/index.ts
index ce7df19..d6317b5 100644
--- a/service/index.ts
+++ b/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)
+})
diff --git a/src/views/Chat/index.vue b/src/views/Chat/index.vue
index e2cd6ec..75dd1dc 100644
--- a/src/views/Chat/index.vue
+++ b/src/views/Chat/index.vue
@@ -2,7 +2,7 @@
 import { nextTick, onMounted, ref } from 'vue'
 import { NButton, NInput, NPopover, useMessage } from 'naive-ui'
 import { Message } from './components'
-import { fetchChatAPI } from './request'
+import { clearChatContext, fetchChatAPI } from './request'
 import { Icon } from '@/components'
 
 interface ListProps {
@@ -27,9 +27,16 @@ function initChat() {
   addMessage('Hi, I am ChatGPT, a chatbot based on GPT-3.', false)
 }
 
-function handleClear() {
-  list.value = []
-  setTimeout(initChat, 100)
+async function handleClear() {
+  try {
+    const { message } = await clearChatContext()
+    ms.success(message)
+    list.value = []
+    setTimeout(initChat, 100)
+  }
+  catch (error) {
+    ms.error('Clear failed, please try again later.')
+  }
 }
 
 function handleEnter(event: KeyboardEvent) {
@@ -81,7 +88,7 @@ function addMessage(message: string, reversal = false) {
               <Icon icon="ri:delete-bin-6-line" />
             </button>
           </template>
-          <span>Clear</span>
+          <span>Clear Context</span>
         </NPopover>
       </div>
     </header>
diff --git a/src/views/Chat/request.ts b/src/views/Chat/request.ts
index b05cb36..cc32f56 100644
--- a/src/views/Chat/request.ts
+++ b/src/views/Chat/request.ts
@@ -1,13 +1,13 @@
 import axios from 'axios'
 
-async function fetchChatAPI(message: string) {
-  const url = `${import.meta.env.VITE_GLOB_API_URL}/chat`
+const BASE_URL = import.meta.env.VITE_GLOB_API_URL
 
+async function fetchChatAPI(message: string) {
   if (!message || message.trim() === '')
     return
 
   try {
-    const { status, data } = await axios.post(url, { message })
+    const { status, data } = await axios.post(`${BASE_URL}/chat`, { message })
 
     if (status === 200) {
       if (data.text)
@@ -24,4 +24,18 @@ async function fetchChatAPI(message: string) {
   }
 }
 
-export { fetchChatAPI }
+async function clearChatContext() {
+  try {
+    const { status, data } = await axios.post(`${BASE_URL}/clear`)
+
+    if (status === 200)
+      return Promise.resolve(data)
+
+    return Promise.reject(new Error('Request failed'))
+  }
+  catch (error) {
+    return Promise.reject(error)
+  }
+}
+
+export { fetchChatAPI, clearChatContext }