fix global_metadata set

LazyAGI · Dec 16, 2024 · 3df22d0 · 3df22d0
1 parent 93866e0
commit 3df22d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/lazyllm/tools/rag/doc_node.py b/lazyllm/tools/rag/doc_node.py
@@ -89,9 +89,7 @@ def global_metadata(self) -> Dict[str, Any]:
 
     @global_metadata.setter
     def global_metadata(self, global_metadata: Dict) -> None:
-        if self.parent:
-            raise ValueError("only root node can set global metadata.")
-        self._global_metadata = global_metadata
+        self.root_node._global_metadata = global_metadata
 
     @property
     def metadata(self) -> Dict:

diff --git a/lazyllm/tools/rag/milvus_store.py b/lazyllm/tools/rag/milvus_store.py
@@ -299,7 +299,6 @@ def _deserialize_node_partial(self, result: Dict) -> DocNode:
             if k.startswith(self._embedding_key_prefix):
                 doc.embedding[k[len(self._embedding_key_prefix):]] = v
             elif k.startswith(self._global_metadata_key_prefix):
-                if doc.parent:
-                    doc._global_metadata[k[len(self._global_metadata_key_prefix):]] = v
+                doc.global_metadata[k[len(self._global_metadata_key_prefix):]] = v
 
         return doc