fix: add metadata in add and add_local (#343)

This commit is contained in:
juaneloDev
2023-07-20 21:40:01 -05:00
committed by GitHub
parent fa63a16591
commit aed894246f

View File

@@ -121,7 +121,7 @@ class EmbedChain:
chunks_before_addition = self.count()
# Add metadata to each document
metadatas_with_metadata = [meta or metadata for meta in metadatas]
metadatas_with_metadata = [{**meta, **metadata} for meta in metadatas]
self.collection.add(documents=documents, metadatas=list(metadatas_with_metadata), ids=ids)
print((f"Successfully saved {src}. New chunks count: " f"{self.count() - chunks_before_addition}"))