rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix KoboldCpp saved vectors retrieval

Browse Source
This commit is contained in:
Cohee
2025-04-27 23:18:39 +03:00
parent ed895b7c3e
commit 05daddb60c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
public/scripts/extensions/vectors/index.js
View File

@ -829,11 +829,12 @@ async function getAdditionalArgs(items) {
* @returns {Promise<number[]>} Saved hashes * @returns {Promise<number[]>} Saved hashes
*/ */
async function getSavedHashes(collectionId) { async function getSavedHashes(collectionId) {
const args = await getAdditionalArgs([]);
const response = await fetch('/api/vector/list', { const response = await fetch('/api/vector/list', {
method: 'POST', method: 'POST',
headers: getRequestHeaders(), headers: getRequestHeaders(),
body: JSON.stringify({ body: JSON.stringify({
...getVectorsRequestBody(), ...getVectorsRequestBody(args),
collectionId: collectionId, collectionId: collectionId,
source: settings.source, source: settings.source,
}), }),
@ -1155,6 +1156,9 @@ function loadWebLlmModels() {
* @returns {Promise<Record<string, number[]>>} Calculated embeddings * @returns {Promise<Record<string, number[]>>} Calculated embeddings
*/ */
async function createWebLlmEmbeddings(items) { async function createWebLlmEmbeddings(items) {
if (items.length === 0) {
return /** @type {Record<string, number[]>} */ ({});
}
return executeWithWebLlmErrorHandling(async () => { return executeWithWebLlmErrorHandling(async () => {
const embeddings = await webllmProvider.embedTexts(items, settings.webllm_model); const embeddings = await webllmProvider.embedTexts(items, settings.webllm_model);
const result = /** @type {Record<string, number[]>} */ ({}); const result = /** @type {Record<string, number[]>} */ ({});