rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cohere: Update embeddings to v2 

#2962
This commit is contained in:
Cohee 2024-10-09 00:52:07 +03:00
parent c2935ba4a7
commit 9db761cea4
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/vectors/cohere-vectors.js
View File

@ -17,7 +17,7 @@ async function getCohereBatchVector(texts, isQuery, directories, model) {
throw new Error('No API key found'); 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', method: 'POST',
headers: { headers: {
'Content-Type': 'application/json', 'Content-Type': 'application/json',
@ -26,6 +26,7 @@ async function getCohereBatchVector(texts, isQuery, directories, model) {
body: JSON.stringify({ body: JSON.stringify({
texts: texts, texts: texts,
model: model, model: model,
embedding_types: ['float'],
input_type: isQuery ? 'search_query' : 'search_document', input_type: isQuery ? 'search_query' : 'search_document',
truncate: 'END', truncate: 'END',
}), }),
@ -38,12 +39,12 @@ async function getCohereBatchVector(texts, isQuery, directories, model) {
} }
const data = await response.json(); const data = await response.json();
if (!Array.isArray(data?.embeddings)) { if (!Array.isArray(data?.embeddings?.float)) {
console.log('API response was not an array'); console.log('API response was not an array');
throw new Error('API response was not an array'); throw new Error('API response was not an array');
} }
return data.embeddings; return data.embeddings.float;
} }
/** /**