rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-02-13 18:50:39 +01:00
Code Issues Packages Projects Releases Wiki Activity

Group tool calls in the result

Browse Source
This commit is contained in:
Cohee 2024-10-06 20:19:58 +03:00
parent 2a25f75fa5
commit 927c2418e0
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
public/scripts/tool-calling.js
View File

@ -158,7 +158,7 @@ class ToolDefinition {
description: this.#description, description: this.#description,
parameters: this.#parameters, parameters: this.#parameters,
}, },
toString: function() { toString: function () {
return `<div><b>${this.function.name}</b></div><div><small>${this.function.description}</small></div><pre class="justifyLeft wordBreakAll"><code class="flex padding5">${JSON.stringify(this.function.parameters, null, 2)}</code></pre><hr>`; return `<div><b>${this.function.name}</b></div><div><small>${this.function.description}</small></div><pre class="justifyLeft wordBreakAll"><code class="flex padding5">${JSON.stringify(this.function.parameters, null, 2)}</code></pre><hr>`;
}, },
}; };
@ -580,6 +580,19 @@ export class ToolManager {
return result; return result;
} }
/**
* Groups tool names by count.
* @param {string[]} toolNames Tool names
* @returns {string} Grouped tool names
*/
static #groupToolNames(toolNames) {
const toolCounts = toolNames.reduce((acc, name) => {
acc[name] = (acc[name] || 0) + 1;
return acc;
}, {});
return Object.entries(toolCounts).map(([name, count]) => count > 1 ? `${name} (${count})` : name).join(', ');
}
/** /**
* Formats a message with tool invocations. * Formats a message with tool invocations.
* @param {ToolInvocation[]} invocations Tool invocations. * @param {ToolInvocation[]} invocations Tool invocations.
@ -597,8 +610,8 @@ export class ToolManager {
i.result = tryParse(i.result); i.result = tryParse(i.result);
}); });
codeElement.textContent = JSON.stringify(data, null, 2); codeElement.textContent = JSON.stringify(data, null, 2);
const toolNames = data.map(i => i.displayName || i.name).join(', '); const toolNames = data.map(i => i.displayName || i.name);
summaryElement.textContent = `Tool calls: ${toolNames}`; summaryElement.textContent = `Tool calls: ${this.#groupToolNames(toolNames)}`;
preElement.append(codeElement); preElement.append(codeElement);
detailsElement.append(summaryElement, preElement); detailsElement.append(summaryElement, preElement);
return detailsElement.outerHTML; return detailsElement.outerHTML;