From 73dd6a21a76d095ff8025a5c39ab857b52fbaf40 Mon Sep 17 00:00:00 2001
From: HoshinoSuzumi <master@uniiem.com>
Date: Wed, 3 Apr 2024 18:01:45 +0800
Subject: [PATCH] wip

---
 pages/aigc/chat/index.vue           | 144 +++++++++++++++++-----------
 typings/llm.ts                      |  15 +++
 typings/{schema.d.ts => types.d.ts} |   5 +
 3 files changed, 110 insertions(+), 54 deletions(-)
 rename typings/{schema.d.ts => types.d.ts} (96%)

diff --git a/pages/aigc/chat/index.vue b/pages/aigc/chat/index.vue
index e1be545..60c54be 100644
--- a/pages/aigc/chat/index.vue
+++ b/pages/aigc/chat/index.vue
@@ -1,28 +1,31 @@
 <script lang="ts" setup>
-import ChatItem from "~/components/aigc/chat/ChatItem.vue";
-import Message from "~/components/aigc/chat/Message.vue";
+import ChatItem from '~/components/aigc/chat/ChatItem.vue'
+import Message from '~/components/aigc/chat/Message.vue'
 import {
+  type Assistant,
   type ChatMessage,
   type ChatMessageId,
   type ChatSession,
   type ChatSessionId,
   llmModels,
-  type ModelTag
-} from "~/typings/llm";
-import {useHistory} from "~/composables/useHistory";
-import {uuidv4} from "@uniiem/uuid";
-import {useLLM} from "~/composables/useLLM";
-import {trimObject} from "@uniiem/object-trim";
-import ModalAuthentication from "~/components/ModalAuthentication.vue";
+  type ModelTag,
+} from '~/typings/llm'
+import {useHistory} from '~/composables/useHistory'
+import {uuidv4} from '@uniiem/uuid'
+import {useLLM} from '~/composables/useLLM'
+import {trimObject} from '@uniiem/object-trim'
+import ModalAuthentication from '~/components/ModalAuthentication.vue'
+import {useAsyncData} from '#app'
 
 useHead({
-  title: '聊天 | XSH AI'
+  title: '聊天 | XSH AI',
 })
 
 const dayjs = useDayjs()
 const toast = useToast()
 const modal = useModal()
 const loginState = useLoginState()
+const {token, user, is_logged_in} = storeToRefs(loginState)
 const historyStore = useHistory()
 const {chatSessions} = storeToRefs(historyStore)
 const {setChatSessions} = historyStore
@@ -35,14 +38,36 @@ const user_input = ref('')
 const responding = ref(false)
 const currentModel = ref<ModelTag>('spark3_5')
 const modals = reactive({
-  modelSelect: false
+  modelSelect: false,
 })
 
+loginState.$subscribe((mutation, state) => {
+  console.log(mutation, state)
+})
+
+const {
+  data: assistantTemplates
+} = await useAsyncData(
+    'App.Assistant_Template.GetList',
+    () => useFetchWrapped<
+        req.AssistantTemplateList & AuthedRequest, PagedData<Assistant>
+    >('App.Assistant_Template.GetList', {
+      user_id: loginState.user.id,
+      token: loginState.token as string,
+      page: 1,
+      perpage: 20,
+    }),
+    {
+      immediate: true,
+      watch: [is_logged_in],
+    },
+)
+
 /**
  * 获取指定 ID 的会话数据
  * @param chatSessionId
  */
-const getSessionCopyById = (chatSessionId: ChatSessionId): ChatSession | undefined => chatSessions.value.find(s => s.id === chatSessionId);
+const getSessionCopyById = (chatSessionId: ChatSessionId): ChatSession | undefined => chatSessions.value.find(s => s.id === chatSessionId)
 /**
  * 切换当前会话
  * @param chatSessionId 指定会话 ID，不传则切换到列表中第一个会话
@@ -95,7 +120,8 @@ const handleClickCreateSession = () => {
   ])
   // 切换到新的会话
   selectCurrentSessionId(sessionId)
-  modals.modelSelect = true
+  // TODO: Model or Assistant Selection
+  // modals.modelSelect = true
   nextTick(() => {
     insetMessage({
       id: uuidv4(),
@@ -112,12 +138,12 @@ const handleClickCreateSession = () => {
 const handleClickSend = (event: any) => {
   if (!loginState.is_logged_in) {
     modal.open(ModalAuthentication)
-    return;
+    return
   }
   if (event.ctrlKey) {
-    return;
+    return
   }
-  if (responding.value) return;
+  if (responding.value) return
   if (!user_input.value) return
   if (!currentSessionId.value) {
     toast.add({
@@ -126,7 +152,7 @@ const handleClickSend = (event: any) => {
       color: 'red',
       icon: 'i-tabler-circle-x',
     })
-    return;
+    return
   }
   // 插入用户消息
   insetMessage({
@@ -146,10 +172,10 @@ const handleClickSend = (event: any) => {
   })
   // 请求模型回复
   const trimmedMessages = trimObject(getMessages(), 2000, {
-    keys: ['content']
+    keys: ['content'],
   })
   useLLM(trimmedMessages, {
-    modelTag: currentModel.value
+    modelTag: currentModel.value,
   }).then(reply => {
     modifyMessageContent(assistantReplyId, reply)
   }).catch(err => {
@@ -173,8 +199,8 @@ const insetMessage = (message: ChatMessage): ChatMessageId => {
     ...s,
     messages: [
       ...s.messages,
-      message
-    ]
+      message,
+    ],
   } : s))
   scrollToMessageListBottom()
   return message.id
@@ -186,7 +212,7 @@ const modifyMessageContent = (
     messageId: ChatMessageId,
     content: string,
     interrupted: boolean = false,
-    updateTime: boolean = true
+    updateTime: boolean = true,
 ) => {
   setChatSessions(chatSessions.value.map(s => s.id === currentSessionId.value ? {
     ...s,
@@ -195,7 +221,7 @@ const modifyMessageContent = (
       content,
       interrupted,
       create_at: updateTime ? dayjs().unix() : m.create_at,
-    } : m)
+    } : m),
   } : s))
   scrollToMessageListBottom()
 }
@@ -248,7 +274,8 @@ onMounted(() => {
       </div>
     </div>
     <div class="h-[calc(100vh-4rem)] bg-white dark:bg-neutral-900 flex-1 flex flex-col">
-      <div class="w-full p-4 bg-neutral-50 dark:bg-neutral-800/50 border-b dark:border-neutral-700/50 flex items-center gap-2">
+      <div
+          class="w-full p-4 bg-neutral-50 dark:bg-neutral-800/50 border-b dark:border-neutral-700/50 flex items-center gap-2">
         <UButton
             class="md:hidden"
             color="black"
@@ -269,40 +296,49 @@ onMounted(() => {
             />
           </TransitionGroup>
         </div>
+        <pre>{{assistantTemplates}}</pre>
       </div>
-      <div
-          class="w-full p-4 pt-2 flex flex-col gap-2 bg-neutral-50 dark:bg-neutral-800/50 border-t dark:border-neutral-700/50">
-        <div class="flex items-center gap-2 overflow-auto overflow-y-hidden">
-          <button class="chat-option-btn" @click="modals.modelSelect = true">
-            <Icon name="tabler:robot-face"/>
-            <span class="text-xs">
+      <ClientOnly>
+        <div
+            class="w-full p-4 pt-2 flex flex-col gap-2 bg-neutral-50 dark:bg-neutral-800/50 border-t dark:border-neutral-700/50">
+          <div class="flex items-center gap-2 overflow-auto overflow-y-hidden">
+            <button class="chat-option-btn" @click="modals.modelSelect = true">
+              <Icon name="tabler:box"/>
+              <span class="text-xs">
               {{ llmModels.find(m => m.tag === currentModel)?.name.toUpperCase() || '模型' }}
             </span>
-          </button>
+            </button>
+            <button class="chat-option-btn" @click="modals.modelSelect = true">
+              <Icon name="tabler:robot-face"/>
+              <span class="text-xs">
+              助手
+            </span>
+            </button>
+          </div>
+          <div class="relative">
+            <UTextarea
+                v-model="user_input"
+                size="lg"
+                autoresize
+                :rows="5"
+                :maxrows="12"
+                class="font-sans"
+                placeholder="Enter 发送, Ctrl + Enter 换行"
+                @keydown.ctrl.enter="user_input += '\n'"
+                @keydown.enter.prevent="handleClickSend"
+            />
+            <UButton
+                color="black"
+                variant="solid"
+                icon="i-tabler-send-2"
+                class="absolute bottom-2.5 right-3"
+                @click.stop="handleClickSend"
+            >
+              发送
+            </UButton>
+          </div>
         </div>
-        <div class="relative">
-          <UTextarea
-              v-model="user_input"
-              size="lg"
-              autoresize
-              :rows="5"
-              :maxrows="12"
-              class="font-sans"
-              placeholder="Enter 发送, Ctrl + Enter 换行"
-              @keydown.ctrl.enter="user_input += '\n'"
-              @keydown.enter.prevent="handleClickSend"
-          />
-          <UButton
-              color="black"
-              variant="solid"
-              icon="i-tabler-send-2"
-              class="absolute bottom-2.5 right-3"
-              @click.stop="handleClickSend"
-          >
-            发送
-          </UButton>
-        </div>
-      </div>
+      </ClientOnly>
     </div>
 
     <!-- Modals -->
diff --git a/typings/llm.ts b/typings/llm.ts
index b313916..20e4af3 100644
--- a/typings/llm.ts
+++ b/typings/llm.ts
@@ -14,10 +14,25 @@ export interface LLMModal {
   endpoint: string
 }
 
+export interface Assistant {
+  id: number;
+  user_id: number;
+  create_time: number;
+  tpl_name: string;
+  types: string;
+  des: string;
+  input_tpl: string;
+  role: string;
+  target: string;
+  style: string;
+  demand: string;
+}
+
 export namespace LLMSpark {
   export interface request {
     prompt: string
   }
+
   export interface response {
     request_msg?: string
     request_fail?: {
diff --git a/typings/schema.d.ts b/typings/types.d.ts
similarity index 96%
rename from typings/schema.d.ts
rename to typings/types.d.ts
index 2937484..d055640 100644
--- a/typings/schema.d.ts
+++ b/typings/types.d.ts
@@ -55,6 +55,11 @@ namespace req {
       file_size: number
     }
   }
+
+  interface AssistantTemplateList {
+    page: number
+    perpage: number
+  }
 }
 
 namespace resp {