Add Gemini embedding model for vector storage

70fe5b6e01b17139c6d86cd18ad0458ec4b657d3

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/scripts/extensions/vectors/index.js+11 -0 Viewed

     ollama_keep: false,
     vllm_model: '',
     webllm_model: '',
 +    google_model: 'text-embedding-004',
     summarize: false,
     summarize_sent: false,
     summary_source: 'main',

         case 'webllm':
             body.model = extension_settings.vectors.webllm_model;
             break;
 +        case 'palm':
 +            body.model = extension_settings.vectors.google_model;
 +            break;
         default:
             break;
+    }

     $('#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();
+    }

         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();

public/scripts/extensions/vectors/settings.html+10 -1 Viewed

                     Hint: Set the URL in the API connection settings.
                 </i>
             </div>
+-
 +            <div class="flex-container flexFlowColumn" id="google_vectorsModel">
 +                <label for="vectors_google_model" data-i18n="Vectorization Model">
 +                    Vectorization Model
 +                </label>
 +                <select id="vectors_google_model" class="text_pole">
 +                    <option value="gemini-embedding-exp-03-07">gemini-embedding-exp-03-07</option>
 +                    <option value="text-embedding-004">text-embedding-004</option>
 +                    <option value="embedding-001">embedding-001</option>
 +                </select>
 +            </div>
             <div class="flex-container alignItemsCenter" id="nomicai_apiKey">
                 <label for="api_key_nomicai" class="flex1">
                     <span data-i18n="NomicAI API Key">NomicAI API Key</span>

src/endpoints/vectors.js+1 -2 Viewed

             };
         case 'palm':
             return {
 -                // TODO: Add support for multiple models
 +                model: String(request.body.model || 'text-embedding-004'),
 -                model: 'text-embedding-004',
             };
         case 'mistral':
             return {