chore: drop pydantic.v1 API usage after LlamaIndex 0.11 upgrade

Signed-off-by: Panos Vagenas <[email protected]>
DS4SD · Aug 28, 2024 · 58e57a4 · 58e57a4
1 parent e22b90f
commit 58e57a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quackling/llama_index/node_parsers/hier_node_parser.py b/quackling/llama_index/node_parsers/hier_node_parser.py
@@ -59,7 +59,7 @@ def _parse_nodes(
         rd.seed(seed)
 
         for input_node in nodes_with_progress:
-            li_doc = LIDocument.parse_obj(input_node)
+            li_doc = LIDocument.model_validate(input_node)
             dl_doc: DLDocument = DLDocument.model_validate_json(li_doc.get_content())
             chunk_iter = chunker.chunk(dl_doc=dl_doc)
             for chunk in chunk_iter: