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

Add Gemini embedding model for vector storage

Browse Source
This commit is contained in:
Cohee
2025-04-01 21:42:26 +03:00
parent 80e821d12d
commit 70fe5b6e01
3 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -63,6 +63,7 @@ const settings = {
ollama_keep: false,
vllm_model: '',
webllm_model: '',
google_model: 'text-embedding-004',
summarize: false,
summarize_sent: false,
summary_source: 'main',
@@ -789,6 +790,9 @@ function getVectorsRequestBody(args = {}) {
case 'webllm':
body.model = extension_settings.vectors.webllm_model;
break;
case 'palm':
body.model = extension_settings.vectors.google_model;
break;
default:
break;
}
@@ -1082,6 +1086,7 @@ function toggleSettings() {
$('#nomicai_apiKey').toggle(settings.source === 'nomicai');
$('#webllm_vectorsModel').toggle(settings.source === 'webllm');
$('#koboldcpp_vectorsModel').toggle(settings.source === 'koboldcpp');
$('#google_vectorsModel').toggle(settings.source === 'palm');
if (settings.source === 'webllm') {
loadWebLlmModels();
}
@@ -1738,6 +1743,12 @@ jQuery(async () => {
toastr.success('WebLLM model loaded');
});
$('#vectors_google_model').val(settings.google_model).on('input', () => {
settings.google_model = String($('#vectors_google_model').val());
Object.assign(extension_settings.vectors, settings);
saveSettingsDebounced();
});
$('#api_key_nomicai').toggleClass('success', !!secret_state[SECRET_KEYS.NOMICAI]);
toggleSettings();