sij/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2025-04-17 18:18:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions1

Adjust how text is split depending on input type ()

Browse source 
resolves
This commit is contained in:
Timothy Carambat 2024-04-30 10:11:56 -07:00 • committed by GitHub
parent ca63012c0f
commit bf435b2861
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 18 additions and 6 deletions
server/utils/EmbeddingEngines
azureOpenAi
index.js
lmstudio
index.js
localAi
index.js
native
index.js
ollama
index.js
openAi
index.js

4
server/utils/EmbeddingEngines/azureOpenAi/index.js
View file

@ -23,7 +23,9 @@ class AzureOpenAiEmbedder {
}
async embedTextInput(textInput) {
const result = await this.embedChunks(textInput);
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}

4
server/utils/EmbeddingEngines/lmstudio/index.js
View file

@ -31,7 +31,9 @@ class LMStudioEmbedder {
}
async embedTextInput(textInput) {
const result = await this.embedChunks(textInput);
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}

4
server/utils/EmbeddingEngines/localAi/index.js
View file

@ -23,7 +23,9 @@ class LocalAiEmbedder {
}
async embedTextInput(textInput) {
const result = await this.embedChunks(textInput);
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}

4
server/utils/EmbeddingEngines/native/index.js
View file

@ -119,7 +119,9 @@ class NativeEmbedder {
}
async embedTextInput(textInput) {
const result = await this.embedChunks(textInput);
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}

4
server/utils/EmbeddingEngines/ollama/index.js
View file

@ -30,7 +30,9 @@ class OllamaEmbedder {
}
async embedTextInput(textInput) {
const result = await this.embedChunks([textInput]);
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}

4
server/utils/EmbeddingEngines/openAi/index.js
View file

@ -19,7 +19,9 @@ class OpenAiEmbedder {
}
async embedTextInput(textInput) {
const result = await this.embedChunks(textInput);
const result = await this.embedChunks(
Array.isArray(textInput) ? textInput : [textInput]
);
return result?.[0] || [];
}