From 621777e4c689a78267b4fc858c7a05a395015567 Mon Sep 17 00:00:00 2001
From: Twilight <im@ivampiresp.com>
Date: Sat, 14 Sep 2024 10:06:34 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9B=20=E5=8A=A9=E7=90=86?=
 =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E9=A1=B5=E9=9D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/AssistantSettings.vue | 119 +++++++++++++++++++++++++--
 1 file changed, 114 insertions(+), 5 deletions(-)

diff --git a/src/components/AssistantSettings.vue b/src/components/AssistantSettings.vue
index fb54065..9b7ed20 100644
--- a/src/components/AssistantSettings.vue
+++ b/src/components/AssistantSettings.vue
@@ -1,13 +1,25 @@
 <template>
+  <div class="mb-3">
+    <n-button tertiary @click="showCreateDialog = true"> 新建助理 </n-button>
+  </div>
   <n-list hoverable clickable v-if="assistants.length">
-    <n-list-item v-for="c in assistants" :key="c.id" @click="showEditAssistant(c.id ?? 0)">
+    <n-list-item
+      v-for="c in assistants"
+      :key="c.id"
+      @click="showEditAssistant(c.id ?? 0)"
+    >
       <n-thing>
         <div class="flex justify-between items-center">
           <div>
             {{ c.name }}
           </div>
           <div>
-            <n-button quaternary circle type="info" @click.stop="showEditAssistant(c.id ?? 0)">
+            <n-button
+              quaternary
+              circle
+              type="info"
+              @click.stop="showEditAssistant(c.id ?? 0)"
+            >
               <template #icon>
                 <n-icon size="16" class="cursor-pointer">
                   <SettingsOutline />
@@ -19,8 +31,15 @@
       </n-thing>
     </n-list-item>
   </n-list>
-  <div v-else>
-    <n-result status="404" title="还没有助理" description="助理可以整合一系列工具，也可以指定人设和自定义提示词">
+  <div v-else class="text-center">
+    <n-result
+      status="404"
+      title="还没有助理"
+      description="助理可以整合一系列工具，也可以指定人设和自定义提示词"
+    >
+      <n-button type="primary" @click="showCreateDialog = true">
+        新建助理
+      </n-button>
     </n-result>
   </div>
   <div>
@@ -35,6 +54,21 @@
               />
             </n-form-item>
 
+            <n-form-item label="描述">
+              <n-input
+                v-model:value="currentAssistant.description"
+                @keydown.enter.prevent
+              />
+            </n-form-item>
+
+            <n-form-item label="提示词">
+              <n-input
+                type="textarea"
+                v-model:value="currentAssistant.prompt"
+                @keydown.enter.prevent
+              />
+            </n-form-item>
+
             <n-form-item label="资料库选择">
               <n-select
                 :style="{ width: '33%' }"
@@ -45,6 +79,50 @@
 
             <n-button type="primary" @click="editAssistant"> 更新 </n-button>
           </n-form>
+
+          <div class="mt-3" v-if="userTools.length">
+            <n-h3>工具</n-h3>
+            <div class="flex justify-between items-center" >
+              <div>
+                工具 1
+              </div>
+              <div>
+                <n-switch />
+              </div>
+            </div>
+          </div>
+        </div>
+      </n-drawer-content>
+    </n-drawer>
+
+    <n-drawer placement="left" v-model:show="showCreateDialog" :width="600">
+      <n-drawer-content closable title="新建助理">
+        <div v-if="currentAssistant">
+          <n-form>
+            <n-form-item label="助理名称">
+              <n-input
+                v-model:value="currentAssistant.name"
+                @keydown.enter.prevent
+              />
+            </n-form-item>
+
+            <n-form-item label="描述">
+              <n-input
+                v-model:value="currentAssistant.description"
+                @keydown.enter.prevent
+              />
+            </n-form-item>
+
+            <n-form-item label="提示词">
+              <n-input
+                type="textarea"
+                v-model:value="currentAssistant.prompt"
+                @keydown.enter.prevent
+              />
+            </n-form-item>
+
+            <n-button type="primary" @click="createAssistant"> 新建 </n-button>
+          </n-form>
         </div>
       </n-drawer-content>
     </n-drawer>
@@ -58,11 +136,19 @@ import getApi from "../plugins/api";
 import { useChatStore } from "../stores/chat";
 import router from "@/router";
 import { ref } from "vue";
-import { EntityAssistant, EntityLibrary } from "@/api";
+import {
+  EntityAssistant,
+  EntityAssistantTool,
+  EntityLibrary,
+  EntityTool,
+} from "@/api";
 
 const dialog = useDialog();
 const chatStore = useChatStore();
 const showSettingsDialog = ref(false);
+const showCreateDialog = ref(false);
+const currentAssistantTools: Ref<EntityAssistantTool[]> = ref([]);
+const userTools: Ref<EntityTool[]> = ref([]);
 const currentAssistantId = ref();
 const currentAssistant: Ref<EntityAssistant> = ref({});
 const assistants: Ref<EntityAssistant[]> = ref([]);
@@ -90,6 +176,8 @@ const showEditAssistant = async (id: number) => {
   currentAssistantId.value = id;
   currentAssistant.value =
     (await getApi().Assistant.apiV1AssistantsIdGet(id)).data.data ?? {};
+
+  getTools();
 };
 
 const editAssistant = async () => {
@@ -100,6 +188,27 @@ const editAssistant = async () => {
   await getAssistants();
 };
 
+const getTools = async () => {
+  currentAssistantTools.value =
+    (
+      await getApi().Assistant.apiV1AssistantsIdToolsGet(
+        currentAssistantId.value
+      )
+    ).data.data ?? [];
+
+  userTools.value = (await getApi().Tool.apiV1ToolsGet()).data.data ?? [];
+};
+
+const createAssistant = async () => {
+  await getApi().Assistant.apiV1AssistantsPost({
+    name: currentAssistant.value.name ?? "",
+    prompt: currentAssistant.value.prompt ?? "",
+    description:
+      currentAssistant.value.description ?? currentAssistant.value.name ?? "",
+  });
+  await getAssistants();
+};
+
 const getLibraries = async () => {
   libraries.value =
     (await getApi().Library.apiV1LibrariesGet()).data.data ?? [];