From 263f4d098c32cff7cfeb594c9ce17b3acc6949f5 Mon Sep 17 00:00:00 2001
From: MY <715711877@qq.com>
Date: Fri, 6 Sep 2024 14:48:50 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=9B=A0=E5=AE=8C=E5=96=84=20LLM=20&=20LLM?=
 =?UTF-8?q?=20=E6=B7=BB=E5=8A=A0=E8=87=B3=E6=89=8B=E6=9C=BA=E7=AB=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/pages/doc/MobileToolBox.vue        | 10 +++----
 src/pages/doc/knowledge/LLM.vue        | 40 ++++++++++++++++++++++++--
 src/pages/doc/mobile/MobileDocPage.vue |  9 ++++++
 3 files changed, 51 insertions(+), 8 deletions(-)

diff --git a/src/pages/doc/MobileToolBox.vue b/src/pages/doc/MobileToolBox.vue
index ddf20cb384..457c3cb86d 100644
--- a/src/pages/doc/MobileToolBox.vue
+++ b/src/pages/doc/MobileToolBox.vue
@@ -52,9 +52,8 @@ type ActionType =
     'goToDoc' | 
     'goToPpt' |
     'showKnowledgeSystem' | 
-    'showKwFinder'
-
-type LocalActionType = ActionType & 'showMoreSetting'
+    'showKwFinder' |
+    'showLlm'
 
 interface Action {
   name: string
@@ -90,7 +89,8 @@ export default defineComponent({
     'showKnowledgeReviewer',
     'showKnowledgeTrend',
     'showKnowledgeRedundancy',
-    'showKwFinder'
+    'showKwFinder',
+    'showLlm'
   ],
   data(){
     return {
@@ -103,7 +103,7 @@ export default defineComponent({
         {name: '知识体系', type: 'success', action: 'showKnowledgeSystem' },
         {name: '知识回顾', type: 'primary', action: 'showKnowledgeReviewer'},
         {name: '知识趋势', type: 'danger', action: 'showKnowledgeTrend'},
-        {name: '知识冗余', type: 'success', action: 'showKnowledgeRedundancy',},
+        {name: '知识助手', type: 'warning', action: 'showLlm'},
         {name: '阅读历史', type: 'primary', action: 'showReadingHistory', divided: true},
         {name: '文本搜索', type: 'warning', action: 'showKwFinder'},
         // {name: '添加书签', type: 'danger', action: 'showBookMarkAdder', divided: true},
diff --git a/src/pages/doc/knowledge/LLM.vue b/src/pages/doc/knowledge/LLM.vue
index b5b0a00719..a9c1688bd1 100644
--- a/src/pages/doc/knowledge/LLM.vue
+++ b/src/pages/doc/knowledge/LLM.vue
@@ -1,5 +1,5 @@
 <template>
-  <el-drawer v-model="showDrawer" :size="$isMobile() ? '50%' : '44%'" :direction="$isMobile() ? 'btt' : 'rtl'"
+  <el-drawer v-model="showDrawer" :size="$isMobile() ? '75%' : '44%'" :direction="$isMobile() ? 'btt' : 'rtl'"
     title="LLM" :modal="false" @close="$emit('close')" :lock-scroll="false" modal-class="operational-drawer-modal"
     class="operational-drawer">
     <div class="llm-container" v-loading="loading">
@@ -9,7 +9,14 @@
           }}</el-option>
       </el-select>
       <el-input type="textarea" rows="5" v-model="query" @keyup="handleQueryKeyUp"/>
-      <el-button type="primary" @click="handleSend">发送</el-button>
+      <el-row>
+        <el-col :span="12">
+          <el-button type="primary" @click="handleSend" style="width:100%">发送</el-button>
+        </el-col>
+        <el-col :span="12">
+          <el-button type="success" @click="handleCopy" style="width:100%">复制</el-button>
+        </el-col>
+      </el-row>
     </div>
 
   </el-drawer>
@@ -17,12 +24,13 @@
 
 <script lang="ts">
 import { defineComponent } from "vue";
-import api from "@/api";
 import { ElMessage } from "element-plus";
 import DocService from "@/service/DocService";
 import Content from '@/dto/Content'
 import axios from 'axios'
 import { marked } from 'marked'
+import CategoryService from "@/service/CategoryService";
+import Category from "@/dto/Category";
 
 function contents2String(contents: Content[]): string {
   if (!contents || contents.length == 0) {
@@ -31,6 +39,13 @@ function contents2String(contents: Content[]): string {
   return contents.map(v => tab(v.level) + "- " + v.name + "\n" + contents2String(v.chidren)).join('\n')
 }
 
+function categorys2String(categorys: Category[], level: number): string {
+  if (!categorys || categorys.length == 0) {
+    return ''
+  }
+  return categorys.map(v => tab(level) + "- " + v.name + "\n" + categorys2String(v.chidren, level + 1)).join('\n')
+}
+
 function tab(t: number) {
   let str = ""
   for (let i = 0; i < t; i++) {
@@ -54,6 +69,7 @@ export default defineComponent({
       query: '',
       loading: false,
       presets: [
+        { name: '根据整个目录回答问题', value: 'answerByAllCategory', template: () => this.answerByAllCategoryTemplate() },
         { name: '概括总结', value: 'summary', template: () => this.summaryTemplate() },
         { name: '根据内容回答问题', value: 'ansuwerByContent', template: () => this.answerByContentTempalte() },
         { name: '根据目录回答问题', value: 'ansuwerByCategory', template: () => this.answerByCategoryTempalte() },
@@ -86,6 +102,20 @@ export default defineComponent({
         ---
       `
     },
+    async answerByAllCategoryTemplate() {
+      const categorys = await CategoryService.getCategoryList()
+      return `
+        以下是个人知识体系的目录，请根据目录回答问题：
+
+        问题：
+
+
+        目录：
+        ---
+        ${categorys2String(categorys, 0)}
+        ---
+      `
+    },
     async answerByCategoryTempalte() {
       const contents = await DocService.getContentByDocId(this.doc)
       const text = contents2String(contents)
@@ -135,6 +165,10 @@ export default defineComponent({
         this.handleSend()
       }
     },
+    handleCopy() {
+      navigator.clipboard.writeText(this.query);
+      ElMessage.success("复制成功")
+    },
     async handleSend() {
       let stream = null
       try {
diff --git a/src/pages/doc/mobile/MobileDocPage.vue b/src/pages/doc/mobile/MobileDocPage.vue
index 8b7178caaa..299969cb9f 100644
--- a/src/pages/doc/mobile/MobileDocPage.vue
+++ b/src/pages/doc/mobile/MobileDocPage.vue
@@ -25,7 +25,9 @@
     @showKnowledgeNetwork="showKnowledgeNetwork"
     @showKnowledgeTrend="showKnowledgeTrend"
     @showKwFinder="showKwFinder"
+    @showLlm="showLLM"
   />
+  <LLM :doc="doc" ref="llm"/>
   <key-word-finder ref="kwFinder" :kw="kw" @kwChanged="handleKwChanged"/>
 </template>
 
@@ -50,6 +52,7 @@ import KeyWordFinder from '../KeyWordFinder.vue'
 import TouchUtils from '@/util/TouchUtils'
 import SelectionPopover from '../tool/SelectionPopover.vue'
 import MermaidShower from '../mermaid-shower/MermaidShower.vue';
+import LLM from '../knowledge/LLM.vue';
 
 export default defineComponent({
   components: {
@@ -66,6 +69,7 @@ export default defineComponent({
     KeyWordFinder,
     SelectionPopover,
     MermaidShower,
+    LLM,
 },
   setup() {
     const knowledgeReviewer = ref<InstanceType<typeof KnowledgeReviewer>>()
@@ -73,6 +77,7 @@ export default defineComponent({
     const readingHistory = ref<InstanceType<typeof ReadingHistory>>()
     const mindGraph = ref<InstanceType<typeof MindGraph>>()
     const kwFinder = ref<InstanceType<typeof KeyWordFinder>>()
+    const llm = ref<InstanceType<typeof LLM>>()
     const showReadingHistory = () => {
       readingHistory.value?.show()
     }
@@ -85,11 +90,15 @@ export default defineComponent({
     const showKnowledgeNetwork = () => {
       knowledgeNetwork.value?.show()
     }
+    const showLLM = () => {
+      llm.value?.show();
+    }
     return {
       readingHistory,showReadingHistory,
       knowledgeReviewer, showKnowledgeReviewer,
       knowledgeNetwork, showKnowledgeNetwork,
       mindGraph, showMindGraph,
+      llm, showLLM,
       kwFinder
     }
   },