From 13d8d7cbbe6b404a152977ced8430d60f2bfe7eb Mon Sep 17 00:00:00 2001
From: ezirmusitua <jferroal@gmail.com>
Date: Thu, 25 Apr 2024 03:36:12 +0800
Subject: [PATCH] fix: use `Array.prototype.flat` (#760)
Co-authored-by: Alex Yang <himself65@outlook.com>
---
packages/core/src/ingestion/IngestionPipeline.ts | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/packages/core/src/ingestion/IngestionPipeline.ts b/packages/core/src/ingestion/IngestionPipeline.ts
index 775f17123..d93867ce2 100644
--- a/packages/core/src/ingestion/IngestionPipeline.ts
+++ b/packages/core/src/ingestion/IngestionPipeline.ts
@@ -94,20 +94,20 @@ export class IngestionPipeline {
documents?: Document[],
nodes?: BaseNode[],
): Promise<BaseNode[]> {
- const inputNodes: BaseNode[] = [];
+ const inputNodes: BaseNode[][] = [];
if (documents) {
- inputNodes.push(...documents);
+ inputNodes.push(documents);
}
if (nodes) {
- inputNodes.push(...nodes);
+ inputNodes.push(nodes);
}
if (this.documents) {
- inputNodes.push(...this.documents);
+ inputNodes.push(this.documents);
}
if (this.reader) {
- inputNodes.push(...(await this.reader.loadData()));
+ inputNodes.push(await this.reader.loadData());
}
- return inputNodes;
+ return inputNodes.flat();
}
async run(
--
GitLab