Vectors: Fix passing model id to AI Studio

5a6d93fc0311f1a891ee2b453b30566f79225262

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

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

         case 'extras':
             return getExtrasVector(text, sourceSettings.extrasUrl, sourceSettings.extrasKey);
         case 'palm':
-            return getMakerSuiteVector(text, directories);
+            return getMakerSuiteVector(text, directories, sourceSettings.model);
         case 'cohere':
             return getCohereVector(text, isQuery, directories, sourceSettings.model);
         case 'llamacpp':

                 results.push(...await getExtrasBatchVector(batch, sourceSettings.extrasUrl, sourceSettings.extrasKey));
                 break;
             case 'palm':
-                results.push(...await getMakerSuiteBatchVector(batch, directories));
+                results.push(...await getMakerSuiteBatchVector(batch, directories, sourceSettings.model));
                 break;
             case 'cohere':
                 results.push(...await getCohereBatchVector(batch, isQuery, directories, sourceSettings.model));

src/vectors/makersuite-vectors.js+5 -4 Viewed

  * Gets the vector for the given text from gecko model
  * @param {string[]} texts - The array of texts to get the vector for
  * @param {import('../users.js').UserDirectoryList} directories - The directories object for the user
+ * @param {string} model - The model to use for embedding
  * @returns {Promise<number[][]>} - The array of vectors for the texts
  */
-export async function getMakerSuiteBatchVector(texts, directories) {
+export async function getMakerSuiteBatchVector(texts, directories, model) {
-    const promises = texts.map(text => getMakerSuiteVector(text, directories));
+    const promises = texts.map(text => getMakerSuiteVector(text, directories, model));
     return await Promise.all(promises);
+}

  * Gets the vector for the given text from Gemini API text-embedding-004 model
  * @param {string} text - The text to get the vector for
  * @param {import('../users.js').UserDirectoryList} directories - The directories object for the user
+ * @param {string} model - The model to use for embedding (default is 'text-embedding-004')
  * @returns {Promise<number[]>} - The vector for the text
  */
-export async function getMakerSuiteVector(text, directories) {
+export async function getMakerSuiteVector(text, directories, model) {
     const key = readSecret(directories, SECRET_KEYS.MAKERSUITE);
     if (!key) {

+    }
     const apiUrl = trimTrailingSlash(API_MAKERSUITE);
-    const model = 'text-embedding-004';
     const url = `${apiUrl}/v1beta/models/${model}:embedContent?key=${key}`;
     const body = {
         content: {