feat: model tools assignment

This commit is contained in:
Timothy J. Baek
2024-06-11 15:29:46 -07:00
parent c3ce4d7f6a
commit 5cd28c04b8
5 changed files with 116 additions and 16 deletions

View File

@@ -74,6 +74,9 @@
let selectedModels = [''];
let atSelectedModel: Model | undefined;
let selectedModelIds = [];
$: selectedModelIds = atSelectedModel !== undefined ? [atSelectedModel.id] : selectedModels;
let selectedToolIds = [];
let webSearchEnabled = false;
@@ -1281,17 +1284,13 @@
bind:selectedToolIds
bind:webSearchEnabled
bind:atSelectedModel
availableTools={$user.role === 'admin'
? $tools.reduce((a, e, i, arr) => {
a[e.id] = {
name: e.name,
description: e.meta.description,
enabled: false
};
return a;
}, {})
: {}}
availableToolIds={selectedModelIds.reduce((a, e, i, arr) => {
const model = $models.find((m) => m.id === e);
if (model?.info?.meta?.toolIds ?? false) {
return [...new Set([...a, ...model.info.meta.toolIds])];
}
return a;
}, [])}
{selectedModels}
{messages}
{submitPrompt}

View File

@@ -9,7 +9,8 @@
models,
config,
showCallOverlay,
tools
tools,
user as _user
} from '$lib/stores';
import { blobToFile, calculateSHA256, findWordIndices } from '$lib/utils';
@@ -59,7 +60,7 @@
export let files = [];
export let availableTools = {};
export let availableToolIds = [];
export let selectedToolIds = [];
export let webSearchEnabled = false;
@@ -657,7 +658,16 @@
<InputMenu
bind:webSearchEnabled
bind:selectedToolIds
tools={availableTools}
tools={$tools.reduce((a, e, i, arr) => {
if (availableToolIds.includes(e.id) || ($_user?.role ?? 'user') === 'admin') {
a[e.id] = {
name: e.name,
description: e.meta.description,
enabled: false
};
}
return a;
}, {})}
uploadFilesHandler={() => {
filesInputElement.click();
}}