From f3c7c40f1eb8d5f0920ad06259c50ce157b79ce6 Mon Sep 17 00:00:00 2001
From: ivamp <im@ivampiresp.com>
Date: Mon, 14 Oct 2024 23:56:39 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9B=20Header=20=E7=8A=B6?=
 =?UTF-8?q?=E6=80=81=E6=98=BE=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/chat/Chat.vue | 35 ++++++++++++++++++++++++++++++-----
 src/layouts/Header.vue       |  2 +-
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index e1f7cb3..8b0fa02 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -60,7 +60,7 @@
       </n-scrollbar>
     </div>
 
-    <div class="min-w-full pt-2 relative " style="max-height: 120px">
+    <div class="min-w-full pt-2 relative" style="max-height: 120px">
       <div class="relative bottom-0 pr-2 pl-2 min-w-full" ref="textContainer">
         <div
           class="mx-auto w-2xl max-w-2xl text-center mb-3 animate__animated animate__pulse text-lg"
@@ -674,11 +674,36 @@ function streamChat(streamId: String, redirect = false) {
         });
         break;
       case "tool_calling":
+        if (data.tool_call_message.function_name.startsWith("builtin_")) {
+          if (data.tool_call_message.function_name === "builtin_browser") {
+            let url = data.tool_call_message.args.query_or_url;
+            // 如果是 URL
+            if (url.startsWith("http")) {
+              chatStore.toolName = "正在浏览 " + url;
+            } else {
+              chatStore.toolName = "正在搜索 " + url;
+            }
+          } else if (data.tool_call_message.function_name === "calculator") {
+            chatStore.toolName = "正在计算";
+          } else if (
+            data.tool_call_message.function_name === "generate_image"
+          ) {
+            chatStore.toolName = "生成图片中";
+          } else if (
+            data.tool_call_message.function_name === "describe_image"
+          ) {
+            chatStore.toolName = "正在理解图片";
+          }
+        } else {
+          chatStore.toolName =
+            "正在执行 " + 
+            data.tool_call_message.tool_name +
+            " 中的 " +
+            data.tool_call_message.function_name;
+        }
+
         // toolCalling.value = true;
-        chatStore.toolName =
-          data.tool_call_message.tool_name +
-          " 中的 " +
-          data.tool_call_message.function_name;
+
         // toolName.value =
         //   data.tool_call_message.tool_name +
         //   " 中的 " +
diff --git a/src/layouts/Header.vue b/src/layouts/Header.vue
index 2f60772..6572dd5 100644
--- a/src/layouts/Header.vue
+++ b/src/layouts/Header.vue
@@ -69,7 +69,7 @@
 
             <div v-else-if="chatStore.toolName != ''">
               <n-gradient-text type="info">
-                正在执行 {{ chatStore.toolName }}
+                {{ chatStore.toolName }}
               </n-gradient-text>
             </div>
             <div v-else>