From 86fcbbff0bcba7da231c50992f8528fff7e70f34 Mon Sep 17 00:00:00 2001
From: ChenZhaoYu <790348264@qq.com>
Date: Tue, 14 Feb 2023 10:04:32 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=B8=8A=E9=99=90=E6=96=87=E8=81=94?=
 =?UTF-8?q?=E6=83=B3=E6=94=B9=E4=B8=BA=E5=89=8D=E7=AB=AF=E8=BF=9B=E8=A1=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 package.json                           |  5 +-
 service/src/chatgpt.ts                 | 48 ++-------------
 service/src/index.ts                   | 24 ++------
 src/api/index.ts                       |  7 ++-
 src/components/business/Chat/index.vue | 83 +++++++++++++-------------
 src/components/business/Chat/types.ts  | 12 ++++
 6 files changed, 69 insertions(+), 110 deletions(-)
 create mode 100644 src/components/business/Chat/types.ts

diff --git a/package.json b/package.json
index 0976459..c55f308 100644
--- a/package.json
+++ b/package.json
@@ -47,9 +47,8 @@
     "vue-tsc": "^1.0.24"
   },
   "lint-staged": {
-    "*.{js,ts,jsx,tsx,vue}": [
-      "pnpm lint:fix",
-      "pnpm type-check"
+    "*.{ts,tsx,vue}": [
+      "pnpm lint:fix"
     ]
   }
 }
diff --git a/service/src/chatgpt.ts b/service/src/chatgpt.ts
index 9e310b1..f50d1af 100644
--- a/service/src/chatgpt.ts
+++ b/service/src/chatgpt.ts
@@ -15,67 +15,31 @@ 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 })
+const api = new ChatGPTAPI({ apiKey, debug: false })
 
-async function chatReply(message: string) {
+async function chatReply(
+  message: string,
+  lastContext?: { conversationId?: string; parentMessageId?: string },
+) {
   if (!message)
     return sendResponse({ type: 'Fail', message: 'Message is empty' })
 
   try {
-    // Get the last context from the chat context
     let options: SendMessageOptions = {}
 
-    const lastContext = Array.from(chatContext).pop()
-
     if (lastContext)
       options = { ...lastContext }
 
     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 sendResponse({ type: 'Success', data: response })
   }
   catch (error: any) {
-    global.console.log(error)
     return sendResponse({ type: 'Fail', message: error.message })
   }
 }
 
-async function chatReplayOne(message: string, options?: ChatContext) {
-  if (!message)
-    return sendResponse({ type: 'Fail', message: 'Message is empty' })
-
-  try {
-    let messageOptions: SendMessageOptions = {}
-
-    if (options) {
-      const { conversationId, parentMessageId } = options
-      messageOptions = { conversationId, parentMessageId }
-
-      const response = await api.sendMessage(message, { ...messageOptions })
-
-      return sendResponse({ type: 'Success', data: response })
-    }
-  }
-  catch (error: any) {
-    return sendResponse({ type: 'Fail', message: error.message })
-  }
-}
-
-async function clearChatContext() {
-  // Clear the chat context
-  chatContext.clear()
-  return sendResponse({ type: 'Success', message: 'Chat context cleared' })
-}
-
-export { chatReply, chatReplayOne, clearChatContext }
+export { chatReply }
diff --git a/service/src/index.ts b/service/src/index.ts
index 91ba3e5..ad2bfa3 100644
--- a/service/src/index.ts
+++ b/service/src/index.ts
@@ -1,35 +1,22 @@
 import express from 'express'
 import type { ChatContext } from './chatgpt'
-import { chatReplayOne, chatReply, clearChatContext } from './chatgpt'
+import { chatReply } from './chatgpt'
 
 const app = express()
 
 app.use(express.json())
 
-app.all('*', (req, res, next) => {
+app.all('*', (_, 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) => {
-  try {
-    const { prompt } = req.body as { prompt: string }
-    const response = await chatReply(prompt)
-    res.send(response)
-  }
-  catch (error) {
-    res.send(error)
-  }
-})
-
-app.post('./chatOne', async (req, res) => {
   try {
     const { prompt, options = {} } = req.body as { prompt: string; options?: ChatContext }
-    const response = await chatReplayOne(prompt, options)
+    const response = await chatReply(prompt, options)
     res.send(response)
   }
   catch (error) {
@@ -37,7 +24,4 @@ app.post('./chatOne', async (req, res) => {
   }
 })
 
-app.post('/clear', async (req, res) => {
-  const response = await clearChatContext()
-  res.send(response)
-})
+app.listen(3002, () => globalThis.console.log('Server is running on port 3002'))
diff --git a/src/api/index.ts b/src/api/index.ts
index b49b52c..e0e87d6 100644
--- a/src/api/index.ts
+++ b/src/api/index.ts
@@ -1,9 +1,12 @@
 import { post } from '@/utils/request'
 
-export function fetchChatAPI<T = any>(prompt: string) {
+export function fetchChatAPI<T = any>(
+  prompt: string,
+  options?: { conversationId?: string; parentMessageId?: string },
+) {
   return post<T>({
     url: '/chat',
-    data: { prompt },
+    data: { prompt, options },
   })
 }
 
diff --git a/src/components/business/Chat/index.vue b/src/components/business/Chat/index.vue
index 0903a66..279ea44 100644
--- a/src/components/business/Chat/index.vue
+++ b/src/components/business/Chat/index.vue
@@ -1,51 +1,27 @@
 <script setup lang='ts'>
-import { nextTick, onMounted, ref } from 'vue'
+import { computed, nextTick, onMounted, ref } from 'vue'
 import { NButton, NInput, useMessage } from 'naive-ui'
+import type { ChatOptions, ChatProps } from './types'
 import { Message } from './components'
 import { Layout } from './layout'
-import { clearConversations, fetchChatAPI } from '@/api'
+import { fetchChatAPI } from '@/api'
 import { HoverButton, SvgIcon } from '@/components/common'
 
-interface ListProps {
-  dateTime: string
-  message: string
-  reversal?: boolean
-  error?: boolean
-}
-
 const scrollRef = ref<HTMLDivElement>()
 
 const ms = useMessage()
 
 const prompt = ref('')
-
 const loading = ref(false)
 
-const list = ref<ListProps[]>([])
-
-onMounted(initChat)
+const list = ref<ChatProps[]>([])
+const chatList = computed(() => list.value.filter(item => (!item.reversal && !item.error)))
 
 function initChat() {
-  addMessage('Hi, I am ChatGPT, a chatbot based on GPT-3.', false)
+  addMessage('Hi, I am ChatGPT, a chatbot based on GPT-3.')
 }
 
-async function handleClear() {
-  try {
-    await clearConversations()
-  }
-  catch (error) {
-    ms.error('Clear failed, please try again later.')
-  }
-  finally {
-    list.value = []
-    setTimeout(initChat, 100)
-  }
-}
-
-function handleEnter(event: KeyboardEvent) {
-  if (event.key === 'Enter')
-    handleSubmit()
-}
+onMounted(initChat)
 
 async function handleSubmit() {
   if (loading.value)
@@ -58,26 +34,51 @@ async function handleSubmit() {
     return
   }
 
-  addMessage(message, true)
+  addMessage(message, { reversal: true })
   prompt.value = ''
 
+  let options: ChatOptions = {}
+  const lastContext = chatList.value[chatList.value.length - 1]?.options
+
+  if (lastContext)
+    options = { ...lastContext }
+
   try {
     loading.value = true
-    const { data } = await fetchChatAPI(message)
-    addMessage(data?.text ?? '', false)
+    const { data } = await fetchChatAPI(message, options)
+    addMessage(data?.text ?? '', { options: { conversationId: data.conversationId, parentMessageId: data.id } })
   }
   catch (error: any) {
-    addMessage(`Error: ${error.message ?? 'Request failed, please try again later.'}`, false, true)
+    addMessage(`Error: ${error.message ?? 'Request failed, please try again later.'}`, { error: true })
   }
   finally {
     loading.value = false
   }
 }
 
-function addMessage(message: string, reversal = false, error = false) {
-  list.value.push({ dateTime: new Date().toLocaleString(), message, reversal, error })
+function handleEnter(event: KeyboardEvent) {
+  if (event.key === 'Enter')
+    handleSubmit()
+}
+
+function addMessage(
+  message: string,
+  args?: { reversal?: boolean; error?: boolean; options?: ChatOptions },
+) {
+  list.value.push({
+    dateTime: new Date().toLocaleString(),
+    message,
+    reversal: args?.reversal ?? false,
+    error: args?.error ?? false,
+    options: args?.options ?? undefined,
+  })
   nextTick(() => scrollRef.value && (scrollRef.value.scrollTop = scrollRef.value.scrollHeight))
 }
+
+function handleClear() {
+  list.value = []
+  setTimeout(initChat, 100)
+}
 </script>
 
 <template>
@@ -87,12 +88,8 @@ function addMessage(message: string, reversal = false, error = false) {
         <div ref="scrollRef" class="h-full p-4 overflow-hidden overflow-y-auto">
           <div>
             <Message
-              v-for="(item, index) of list"
-              :key="index"
-              :date-time="item.dateTime"
-              :message="item.message"
-              :reversal="item.reversal"
-              :error="item.error"
+              v-for="(item, index) of list" :key="index" :date-time="item.dateTime" :message="item.message"
+              :reversal="item.reversal" :error="item.error"
             />
           </div>
         </div>
diff --git a/src/components/business/Chat/types.ts b/src/components/business/Chat/types.ts
new file mode 100644
index 0000000..c43dd62
--- /dev/null
+++ b/src/components/business/Chat/types.ts
@@ -0,0 +1,12 @@
+export interface ChatOptions {
+  conversationId?: string
+  parentMessageId?: string
+}
+
+export interface ChatProps {
+  dateTime: string
+  message: string
+  reversal?: boolean
+  error?: boolean
+  options?: ChatOptions
+}