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/v1/embed', {
+    const response = await fetch('https://api.cohere.ai/v2/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	const data = await response.json();	41	const data = await response.json();
41	if (!Array.isArray(data?.embeddings)) {	42	if (!Array.isArray(data?.embeddings?.float)) {
42	console.log('API response was not an array');	43	console.log('API response was not an array');
43	throw new Error('API response was not an array');	44	throw new Error('API response was not an array');
44	}	45	}
45		46
46	return data.embeddings;	47	return data.embeddings.float;
47	}	48	}
48		49
49	/**	50	/**