diff --git a/python/pathway/xpacks/llm/vector_store.py b/python/pathway/xpacks/llm/vector_store.py index f5395960..e2839af3 100644 --- a/python/pathway/xpacks/llm/vector_store.py +++ b/python/pathway/xpacks/llm/vector_store.py @@ -231,7 +231,10 @@ def _build_graph(self) -> dict: @pw.udf def parse_doc(data: bytes, metadata) -> list[pw.Json]: rets = self.parser(data) - metadata = metadata.value + if metadata is None: + metadata = {} + else: + metadata = metadata.value return [dict(text=ret[0], metadata={**metadata, **ret[1]}) for ret in rets] # type: ignore parsed_docs = docs.select(data=parse_doc(docs.data, docs._metadata)).flatten(