kamp/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2024-11-14 02:20:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improve VoyageAI error responses and textChunk handler

Browse Source 
resolves #1491
This commit is contained in:
timothycarambat 2024-05-22 13:42:48 -05:00
parent 3ef009de73
commit 7bace20769
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
server/utils/EmbeddingEngines/voyageAi/index.js
View File

@ -38,7 +38,10 @@ class VoyageAiEmbedder {
Array.isArray(textInput) ? textInput : [textInput], Array.isArray(textInput) ? textInput : [textInput],
{ modelName: this.model } { modelName: this.model }
); );
return result || [];
// If given an array return the native Array[Array] format since that should be the outcome.
// But if given a single string, we need to flatten it so that we have a 1D array.
return (Array.isArray(textInput) ? result : result.flat()) || [];
} }
async embedChunks(textChunks = []) { async embedChunks(textChunks = []) {
@ -50,6 +53,12 @@ class VoyageAiEmbedder {
return embeddings; return embeddings;
} catch (error) { } catch (error) {
console.error("Voyage AI Failed to embed:", error); console.error("Voyage AI Failed to embed:", error);
if (
error.message.includes(
"Cannot read properties of undefined (reading '0')"
)
)
throw new Error("Voyage AI failed to embed: Rate limit reached");
throw error; throw error;
} }
} }