From 22f588ccf3fa74adddefdac0088643c0f9bd5222 Mon Sep 17 00:00:00 2001
From: Jicheng Lu <103353@smsassist.com>
Date: Wed, 4 Dec 2024 16:11:28 -0600
Subject: [PATCH] refine utility name

---
 src/lib/helpers/types/agentTypes.js           |  1 +
 src/lib/helpers/utils/common.js               | 17 +++++
 .../[conversationId]/chat-box.svelte          |  4 +-
 .../page/agent/[agentId]/agent-utility.svelte | 63 ++++++++++++++-----
 4 files changed, 67 insertions(+), 18 deletions(-)

diff --git a/src/lib/helpers/types/agentTypes.js b/src/lib/helpers/types/agentTypes.js
index b68a9b0d..1059e53a 100644
--- a/src/lib/helpers/types/agentTypes.js
+++ b/src/lib/helpers/types/agentTypes.js
@@ -128,6 +128,7 @@
 /**
  * @typedef {Object} UtilityBase
  * @property {string} name
+ * @property {string?} [displayName]
  */
 
 export default {};
\ No newline at end of file
diff --git a/src/lib/helpers/utils/common.js b/src/lib/helpers/utils/common.js
index 84b5d2ef..d3ae5566 100644
--- a/src/lib/helpers/utils/common.js
+++ b/src/lib/helpers/utils/common.js
@@ -50,4 +50,21 @@ export function formatObject(object) {
     } catch {
         return object;
     }
+}
+
+
+/**
+ * @param {string?} str
+ * @param {string?} prefix
+ */
+export function truncateByPrefix(str, prefix) {
+    if (!str || !prefix) {
+        return str;
+    }
+
+    if (!str.startsWith(prefix)) {
+        return str;
+    }
+
+    return str.replace(prefix, '');
 }
\ No newline at end of file
diff --git a/src/routes/chat/[agentId]/[conversationId]/chat-box.svelte b/src/routes/chat/[agentId]/[conversationId]/chat-box.svelte
index 8220910d..94378811 100644
--- a/src/routes/chat/[agentId]/[conversationId]/chat-box.svelte
+++ b/src/routes/chat/[agentId]/[conversationId]/chat-box.svelte
@@ -1759,7 +1759,7 @@
 							{#if !!lastBotMsg && !isSendingMsg && !isThinking}
 								<RichContent
 									message={lastBotMsg}
-									disabled={isSendingMsg || isThinking}
+									disabled={isSendingMsg || isThinking || disableAction}
 									onConfirm={(title, payload) => confirmSelectedOption(title, payload)}
 								/>
 							{/if}
@@ -1843,7 +1843,7 @@
 									<div class="chat-util-links">
 										{#if !isLite}
 											<ChatBigMessage
-												disabled={disableAction}
+												disabled={isSendingMsg || isThinking || disableAction}
 												on:click={() => toggleBigMessageModal()}
 											/>
 										{/if}
diff --git a/src/routes/page/agent/[agentId]/agent-utility.svelte b/src/routes/page/agent/[agentId]/agent-utility.svelte
index 20c7f3c1..69afa03c 100644
--- a/src/routes/page/agent/[agentId]/agent-utility.svelte
+++ b/src/routes/page/agent/[agentId]/agent-utility.svelte
@@ -2,9 +2,10 @@
     import { onMount } from 'svelte';
     import { Card, CardBody, Input, Button } from '@sveltestrap/sveltestrap';
     import { getAgentUtilityOptions } from '$lib/services/agent-service';
+	import { truncateByPrefix } from '$lib/helpers/utils/common';
 
     const limit = 5;
-    const maxLength = 30;
+    const prefix = "util-";
 
     /** @type {import('$agentTypes').AgentModel} */
     export let agent;
@@ -60,13 +61,29 @@
         getAgentUtilityOptions().then(data => {
             const list = data || [];
             list.forEach(item => {
-                const fns = item.functions.map(f => f.name);
-                const tps = item.templates.map(t => t.name);
+                const fns = item.functions.map(f => {
+                    return {
+                        name: f.name,
+                        displayName: truncateByPrefix(f.name, prefix)
+                    };
+                });
+                const tps = item.templates.map(t => {
+                    return {
+                        name: t.name,
+                        displayName: truncateByPrefix(t.name, prefix)
+                    };
+                });
 
                 if (!utilityMapper[item.name]) {
                     utilityMapper[item.name] = {
-                        functions: ["", ...fns],
-                        templates: ["", ...tps]
+                        functions: [{
+                            name: "",
+                            displayName: ""
+                        }, ...fns],
+                        templates: [{
+                            name: "",
+                            displayName: ""
+                        }, ...tps]
                     };
                 } else {
                     const prevFns = utilityMapper[item.name].functions;
@@ -82,7 +99,20 @@
     });
 
     function init() {
-        refresh(agent.utilities);
+        const list = agent.utilities?.map(x => {
+            return {
+                ...x,
+                functions: x.functions?.map(f => ({
+                    ...f,
+                    displayName: truncateByPrefix(f.name, prefix)
+                })) || [],
+                templates: x.templates?.map(t => ({
+                    ...t,
+                    displayName: truncateByPrefix(t.name, prefix)
+                })) || []
+            };
+        }) || [];
+        refresh(list);
     }
 
     /**
@@ -97,11 +127,11 @@
         found.name = name;
         found.functions = [
             // @ts-ignore
-            ...utilityMapper[name].functions?.filter(x => !!x)?.map(x => ({ name: x })) || []
+            ...utilityMapper[name].functions?.filter(x => !!x.name) || []
         ];
         found.templates = [
             // @ts-ignore
-            ...utilityMapper[name].templates?.filter(x => !!x)?.map(x => ({ name: x })) || []
+            ...utilityMapper[name].templates?.filter(x => !!x.name) || []
         ];
         refresh(innerUtilities);
     }
@@ -132,9 +162,9 @@
         if (!found || found.disabled) return;
 
         if (type === 'function') {
-            found.functions.push({ name: '' });
+            found.functions.push({ name: '', displayName: '' });
         } else if (type === 'template') {
-            found.templates.push({ name: '' });
+            found.templates.push({ name: '', displayName: '' });
         }
 
         refresh(innerUtilities);
@@ -171,15 +201,18 @@
         const found = innerUtilities.find((_, index) => index === uid);
         if (!found) return;
 
+        const vals = e.target.value.split("#");
         if (type === 'function') {
             const fn = found.functions?.find((_, index) => index === idx);
             if (fn) {
-                fn.name = e.target.value;
+                fn.name = vals[0];
+                fn.displayName = vals[1];
             }
         } else if (type === 'template') {
             const tp = found.templates?.find((_, index) => index === idx);
             if (tp) {
-                tp.name = e.target.value;
+                tp.name = vals[0];
+                tp.displayName = vals[1];
             }
         }
         refresh(innerUtilities);
@@ -305,12 +338,11 @@
                                         <div class="utility-input line-align-center">
                                             <Input
                                                 type="select"
-                                                value={fn.name}
                                                 disabled={utility.disabled}
                                                 on:change={e => selectContent(e, uid, fid, 'function')}
                                             >
                                                 {#each (utilityMapper[utility.name]?.functions || []) as option}
-                                                    <option value={option} selected={option == fn.name}>{option}</option>
+                                                    <option value={`${option.name}#${option.dsiplayName}`} selected={option.name == fn.name}>{option.displayName}</option>
                                                 {/each}
                                             </Input>
                                         </div>
@@ -355,12 +387,11 @@
                                         <div class="utility-input line-align-center">
                                             <Input
                                                 type="select"
-                                                value={tp.name}
                                                 disabled={utility.disabled}
                                                 on:change={e => selectContent(e, uid, tid, 'template')}
                                             >
                                                 {#each (utilityMapper[utility.name]?.templates || []) as option}
-                                                    <option value={option} selected={option == tp.name}>{option}</option>
+                                                    <option value={`${option.name}#${option.dsiplayName}`} selected={option.name == tp.name}>{option.displayName}</option>
                                                 {/each}
                                             </Input>
                                         </div>