From e8e534eabd5f4d6636e5606141b90db7529d4303 Mon Sep 17 00:00:00 2001
From: Twilight <im@ivampiresp.com>
Date: Mon, 16 Sep 2024 00:43:12 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9B=20=E5=8A=A9=E7=90=86?=
 =?UTF-8?q?=E5=B7=A5=E5=85=B7=E9=80=89=E6=8B=A9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/settings/AssistantSettings.vue | 53 ++++++++++++++++---
 1 file changed, 47 insertions(+), 6 deletions(-)

diff --git a/src/components/settings/AssistantSettings.vue b/src/components/settings/AssistantSettings.vue
index b586031..30269ee 100644
--- a/src/components/settings/AssistantSettings.vue
+++ b/src/components/settings/AssistantSettings.vue
@@ -50,7 +50,7 @@
       v-model:show="showSettingsDialog"
       :width="drawerWidth"
     >
-      <n-drawer-content closable title="编辑助理">
+      <n-drawer-content closable title="编辑助理" :native-scrollbar="false">
         <div v-if="currentAssistant">
           <n-form>
             <n-form-item label="助理名称">
@@ -122,16 +122,27 @@
 
           <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 v-for="c in userTools" :key="c.id">
+              <!-- 如果是第 2 或不是最后一个，则添加 hr -->
+              <n-divider v-if="userTools.indexOf(c) !== 0 " />
+              <div class="flex justify-between items-center mt-3">
+                <div>
+                  {{ c.name }}
+                  <div v-if="c.description">{{ c.description }}</div>
+                </div>
+                <div>
+                  <n-switch
+                    :value="findTool(c.id ?? 0)"
+                    @click="bindOrUnbind(c.id ?? 0)"
+                  />
+                </div>
               </div>
             </div>
           </div>
         </div>
 
-        <n-divider />
+        <n-divider v-if="!userTools.length" />
+        <div v-else class="mt-10"></div>
         <div>
           <div class="flex justify-between align-middle items-center">
             <n-h3>助理 API</n-h3>
@@ -364,6 +375,36 @@ const deleteAssistantKey = async (id: number) => {
   await getAssistantsKeys();
 };
 
+const findTool = (id: number) => {
+  let binded = false;
+  for (let i = 0; i < currentAssistantTools.value.length; i++) {
+    if (currentAssistantTools.value[i].tool_id === id) {
+      binded = true;
+      break;
+    }
+  }
+
+  return binded;
+};
+
+const bindOrUnbind = async (id: number) => {
+  let binded = findTool(id);
+
+  if (binded) {
+    await getApi().Assistant.apiV1AssistantsIdToolsToolIdDelete(
+      currentAssistantId.value,
+      id
+    );
+  } else {
+    await getApi().Assistant.apiV1AssistantsIdToolsToolIdPost(
+      currentAssistantId.value,
+      id
+    );
+  }
+
+  getTools();
+};
+
 getChats();
 getLibraries();
 getAssistants();