Merge pull request #2847 from ceruleandeep/staging Update makersuite-vectors.js to use Gemini API text-embedding-004 model

34a1b5586ae7aafded39a797119f8299073a87f1

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

Signed

src/vectors/makersuite-vectors.js+18 -12 Viewed

 const fetch = require('node-fetch').default;
 const { SECRET_KEYS, readSecret } = require('../endpoints/secrets');
 +const API_MAKERSUITE = 'https://generativelanguage.googleapis.com';
 /**
  * Gets the vector for the given text from gecko model

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

         throw new Error('No Google AI Studio key found');
+    }
 -    const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/embedding-gecko-001:embedText?key=${key}`, {
 +    const apiUrl = new URL(API_MAKERSUITE);
 +    const model = 'text-embedding-004';
 +    const url = `${apiUrl.origin}/v1beta/models/${model}:embedContent?key=${key}`;
 +    const body = {
 +        content: {
 +            parts: [
 +                { text: text },
 +            ],
 +        },
 +    };
++
 +    const response = await fetch(url, {
 +        body: JSON.stringify(body),
         method: 'POST',
         headers: {
             'Content-Type': 'application/json',
         },
 -        body: JSON.stringify({
 -            text: text,
 -        }),
     });
     if (!response.ok) {

+    }
     const data = await response.json();
+-
 +    // noinspection JSValidateTypes
 -    // Access the "value" dictionary
 +    return data['embedding']['values'];
 -    const vector = data.embedding.value;
+-
 -    return vector;
+}
 module.exports = {