perf: Assistant chat ui

Author: fit2cloud-chenyw

backend/apps/chat/task/llm.py

@@ -296,7 +296,7 @@ def generate_recommend_questions_task(self):
 
         # get schema
         if self.ds and not self.chat_question.db_schema:
-            self.chat_question.db_schema = get_table_schema(session=self.session, ds=self.ds)
+            self.chat_question.db_schema = get_table_schema(session=self.session, current_user=self.current_user, ds=self.ds)
 
         guess_msg: List[Union[BaseMessage, dict[str, Any]]] = []
         guess_msg.append(SystemMessage(content=self.chat_question.guess_sys_question()))
@@ -726,7 +726,7 @@ def run_task(llm_service: LLMService, in_chat: bool = True):
                 yield orjson.dumps({'id': llm_service.ds.id, 'datasource_name': llm_service.ds.name,
                                     'engine_type': llm_service.ds.type_name, 'type': 'datasource'}).decode() + '\n\n'
 
-            llm_service.chat_question.db_schema = get_table_schema(session=llm_service.session, ds=llm_service.ds)
+            llm_service.chat_question.db_schema = get_table_schema(session=llm_service.session, current_user=llm_service.current_user, ds=llm_service.ds)
 
         # generate sql
         sql_res = llm_service.generate_sql()

frontend/src/assets/svg/icon_new_chat_outlined.svg

@@ -292,7 +292,7 @@ const createNewChat = async () => {
   if (isAssistant.value) {
     const assistantChat = await assistantStore.setChat()
     if (assistantChat) {
-      onChatCreated(assistantChat as any)
+      onChatCreatedQuick(assistantChat as any)
     }
     return
   }
@@ -317,6 +317,26 @@ function onClickHistory(chat: Chat) {
   scrollToBottom()
 }
 
+function toAssistantHistory(chat: Chat) {
+  currentChat.value = new ChatInfo(chat)
+  if (chat !== undefined && chat.id !== undefined && !loading.value) {
+    currentChatId.value = chat.id
+    loading.value = true
+    chatApi
+      .get(chat.id)
+      .then((res) => {
+        const info = chatApi.toChatInfo(res)
+        if (info) {
+          currentChat.value = info
+          onClickHistory(info)
+        }
+      })
+      .finally(() => {
+        loading.value = false
+      })
+  }
+}
+
 function onChatDeleted(id: number) {
   console.log('deleted', id)
 }
@@ -434,11 +454,6 @@ const sendMessage = async () => {
   loading.value = true
   isTyping.value = true
 
-  /* const assistantChat = await assistantStore.setChat()
-  if (assistantChat) {
-    onChatCreated(assistantChat as any)
-  } */
-
   const currentRecord = new ChatRecord()
   currentRecord.create_time = new Date()
   currentRecord.chat_id = currentChatId.value
@@ -833,10 +848,9 @@ const getCurrentChatId = () => {
 }
 defineExpose({
   getHistoryList,
-  onClickHistory,
-  onChatDeleted,
-  onChatRenamed,
+  toAssistantHistory,
   getCurrentChatId,
+  createNewChat,
 })
 </script>
 

frontend/src/views/chat/index.vue

@@ -11,6 +11,9 @@
     <chat-component v-if="!loading" ref="chatRef" class="sqlbot-chat-container" />
   </div>
   <div class="sqlbot-top-btn">
+    <el-icon style="cursor: pointer" @click="createChat">
+      <icon_new_chat_outlined />
+    </el-icon>
     <el-icon style="cursor: pointer" @click="openHistory">
       <history></history>
     </el-icon>
@@ -41,10 +44,10 @@
                 @click="onClickHistory(chat)"
               >
                 <span class="title">{{ chat.brief ?? 'Untitled' }}</span>
-                <div class="history-operate">
+                <!-- <div class="history-operate">
                   <el-icon @click="EditPen(chat)"><IconOpeEdit /></el-icon>
                   <el-icon @click="Delete(chat)"><IconOpeDelete /></el-icon>
-                </div>
+                </div> -->
               </div>
             </template>
           </el-scrollbar>
@@ -58,8 +61,9 @@ import { onBeforeMount, ref } from 'vue'
 import ChatComponent from '@/views/chat/index.vue'
 import AssistantGif from '@/assets/img/assistant.gif'
 import history from '@/assets/svg/chart/history.svg'
-import IconOpeEdit from '@/assets/svg/operate/ope-edit.svg'
-import IconOpeDelete from '@/assets/svg/operate/ope-delete.svg'
+import icon_new_chat_outlined from '@/assets/svg/icon_new_chat_outlined.svg'
+/* import IconOpeEdit from '@/assets/svg/operate/ope-edit.svg'
+import IconOpeDelete from '@/assets/svg/operate/ope-delete.svg' */
 import { useRoute } from 'vue-router'
 import { assistantApi } from '@/api/assistant'
 import { useAssistantStore } from '@/stores/assistant'
@@ -72,21 +76,24 @@ const chatList = ref<Array<any>>([])
 const drawer = ref(false)
 const currentId = ref()
 
+const createChat = () => {
+  chatRef.value?.createNewChat()
+}
 const openHistory = () => {
   chatList.value = chatRef.value?.getHistoryList()
   currentId.value = chatRef.value?.getCurrentChatId()
   drawer.value = true
 }
 const onClickHistory = (chat: any) => {
-  chatRef.value?.onClickHistory(chat)
+  chatRef.value?.toAssistantHistory(chat)
 }
 
-const EditPen = (chat: any) => {
+/* const EditPen = (chat: any) => {
   chatRef.value?.onChatRenamed(chat)
 }
 const Delete = (chat: any) => {
   chatRef.value?.onChatDeleted(chat.id)
-}
+} */
 const validator = ref({
   id: '',
   valid: false,
@@ -155,8 +162,13 @@ onBeforeMount(async () => {
   z-index: 2009;
   position: absolute;
   top: 16px;
+  display: flex;
+  column-gap: 16px;
   // right: 16px;
   font-size: 22px;
+  i:first-child {
+    color: var(--primary-color);
+  }
 }
 .sqlbot-history-container {
   width: 100%;