michael/merlyn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(embedder): surface Mistral embedding failures (#5513)

Browse Source 
* fix(embedder): surface Mistral embedding failures

* fix pr

---------

Co-authored-by: Timothy Carambat <rambat1010@gmail.com>
This commit is contained in:
Chris Z 2026-04-25 03:55:51 +08:00 committed by GitHub
parent 680d38a3ce
commit eeb29948e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
server/utils/EmbeddingEngines/mistral/index.js
View File

@ -12,16 +12,10 @@ class MistralEmbedder {
}
async embedTextInput(textInput) {
try {
const response = await this.openai.embeddings.create({
model: this.model,
input: textInput,
});
return response?.data[0]?.embedding || [];
} catch (error) {
console.error("Failed to get embedding from Mistral.", error.message);
return [];
}
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}
async embedChunks(textChunks = []) {
@ -30,10 +24,13 @@ class MistralEmbedder {
model: this.model,
input: textChunks,
});
return response?.data?.map((emb) => emb.embedding) || [];
const embeddings = response?.data?.map((emb) => emb.embedding) || [];
if (embeddings.length === 0)
throw new Error("Mistral returned empty embeddings for batch");
return embeddings;
} catch (error) {
console.error("Failed to get embeddings from Mistral.", error.message);
return new Array(textChunks.length).fill([]);
throw new Error(`Mistral Failed to embed: ${error.message}`);
}
}
}