feat: set attributes in _set_doc_by_id for weaviate

docarray · alaeddine-13 · May 25, 2022 · May 9, 2022 · May 9, 2022 · May 10, 2022
commit c1a3fdb88dedb6357bfa883a19f5115e03deefa3
diff --git a/docarray/array/storage/weaviate/backend.py b/docarray/array/storage/weaviate/backend.py
@@ -315,8 +315,13 @@ def _doc2weaviate_create_payload(self, value: 'Document'):
         :param value: document to create a payload for
         :return: the payload dictionary
         """
+        extra_columns = {col: value.tags.get(col) for col, _ in self._config.columns}
+
         return dict(
-            data_object={'_serialized': value.to_base64(**self._serialize_config)},
+            data_object={
+                '_serialized': value.to_base64(**self._serialize_config),
+                **extra_columns,
+            },
             class_name=self._class_name,
             uuid=self._map_id(value.id),
             vector=self._map_embedding(value.embedding),