fix: use Array.prototype.flat (#760)

Co-authored-by: Alex Yang <himself65@outlook.com>
run-llama · Apr 24, 2024 · 13d8d7c · 13d8d7c
1 parent 9c34e44
commit 13d8d7c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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(