fix: include node relationships when converting jsonToDoc (run-llama#968

)
himself65 · Jun 25, 2024 · 0730140 · 0730140
1 parent 810711d
commit 0730140
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.changeset/pink-keys-kiss.md b/.changeset/pink-keys-kiss.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+include node relationships when converting jsonToDoc
diff --git a/packages/llamaindex/src/storage/docStore/utils.ts b/packages/llamaindex/src/storage/docStore/utils.ts
@@ -44,6 +44,7 @@ export function jsonToDoc(docDict: DocJson): BaseNode {
       id_: dataDict.id_,
       hash: dataDict.hash,
       metadata: dataDict.metadata,
+      relationships: dataDict.relationships,
     });
   } else {
     throw new Error(`Unknown doc type: ${docType}`);