fix: Register Neo4j upload stager (#323)

ds-filipknefel · web-flow · commit 2ac6c818dd49 · 2025-01-07T11:20:58.000+01:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,9 +1,12 @@
-## 0.3.12-dev3
+## 0.3.12-dev4
 
 ### Enhancements
 
 * **Migrate Vectara Destination Connector to v2**
 
+### Fixes
+* **Register Neo4j Upload Stager**
+
 ## 0.3.12-dev2
 
 ### Enhancements
diff --git a/unstructured_ingest/__version__.py b/unstructured_ingest/__version__.py
@@ -1 +1 @@
-__version__ = "0.3.12-dev3"  # pragma: no cover
+__version__ = "0.3.12-dev4"  # pragma: no cover
diff --git a/unstructured_ingest/v2/processes/connectors/neo4j.py b/unstructured_ingest/v2/processes/connectors/neo4j.py
@@ -378,6 +378,8 @@ def _create_edges_query(edges: list[_Edge], relationship: Relationship) -> tuple
 
 neo4j_destination_entry = DestinationRegistryEntry(
     connection_config=Neo4jConnectionConfig,
+    upload_stager=Neo4jUploadStager,
+    upload_stager_config=Neo4jUploadStagerConfig,
     uploader=Neo4jUploader,
     uploader_config=Neo4jUploaderConfig,
 )

Original file line number	Diff line number	Diff line change
`@@ -378,6 +378,8 @@ def _create_edges_query(edges: list[_Edge], relationship: Relationship) -> tuple`
`378`	`378`
`379`	`379`	`neo4j_destination_entry = DestinationRegistryEntry(`
`380`	`380`	`connection_config=Neo4jConnectionConfig,`
	`381`	`+ upload_stager=Neo4jUploadStager,`
	`382`	`+ upload_stager_config=Neo4jUploadStagerConfig,`
`381`	`383`	`uploader=Neo4jUploader,`
`382`	`384`	`uploader_config=Neo4jUploaderConfig,`
`383`	`385`	`)`