Cohere: Update embeddings to v2 #2962

9db761cea42dede55747beb4f9edabbc2bbb2c41

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

src/vectors/cohere-vectors.js+4 -3 Viewed

         throw new Error('No API key found');
+    }
     const response = await fetch('https://api.cohere.ai/v1v2/embed', {
         method: 'POST',
         headers: {
             'Content-Type': 'application/json',

         body: JSON.stringify({
             texts: texts,
             model: model,
 +            embedding_types: ['float'],
             input_type: isQuery ? 'search_query' : 'search_document',
             truncate: 'END',
         }),

@@ -38,12 +39,12 @@ async function getCohereBatchVector(texts, isQuery, directories, model) {
38	39	}
39	40
40	41	const data = await response.json();
41	42	if (!Array.isArray(data?.embeddings?.float)) {
42	43	console.log('API response was not an array');
43	44	throw new Error('API response was not an array');
44	45	}
45	46
46	47	return data.embeddings.float;
47	48	}
48	49
49	50	/**