From 317f1408227f161b7ac2a10c0bd6830ba49c897d Mon Sep 17 00:00:00 2001
From: Emanuel Ferreira <contatoferreirads@gmail.com>
Date: Tue, 6 Feb 2024 12:01:48 -0300
Subject: [PATCH] fix: revert embed batch temporarily (#520)
---
.../core/src/indices/vectorStore/VectorStoreIndex.ts | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/packages/core/src/indices/vectorStore/VectorStoreIndex.ts b/packages/core/src/indices/vectorStore/VectorStoreIndex.ts
index 76f67f56d..bc1b6c230 100644
--- a/packages/core/src/indices/vectorStore/VectorStoreIndex.ts
+++ b/packages/core/src/indices/vectorStore/VectorStoreIndex.ts
@@ -168,14 +168,14 @@ export class VectorStoreIndex extends BaseIndex<IndexDict> {
): Promise<BaseNode[]> {
const nodesWithEmbeddings: BaseNode[] = [];
- const embeddingResults = await this.embedModel.getTextEmbeddingsBatch(
- nodes.map((node) => node.getContent(MetadataMode.EMBED)),
- options,
- );
-
for (let i = 0; i < nodes.length; ++i) {
const node = nodes[i];
- node.embedding = embeddingResults[i];
+ if (options?.logProgress) {
+ console.log(`Getting embedding for node ${i + 1}/${nodes.length}`);
+ }
+ node.embedding = await this.embedModel.getTextEmbedding(
+ node.getContent(MetadataMode.EMBED),
+ );
nodesWithEmbeddings.push(node);
}
--
GitLab