Fixed neo4j writer docstring

alexthomas93 · Sep 2, 2024 · e5fcf5a · e5fcf5a
1 parent 6e3c8fa
commit e5fcf5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/neo4j_genai/experimental/components/kg_writer.py b/src/neo4j_genai/experimental/components/kg_writer.py
@@ -85,7 +85,7 @@ class Neo4jWriter(KGWriter):
         DATABASE = "neo4j"
 
         driver = AsyncGraphDatabase.driver(URI, auth=AUTH, database=DATABASE)
-        writer = AsyncNeo4jWriter(driver=driver, neo4j_database=DATABASE)
+        writer = Neo4jWriter(driver=driver, neo4j_database=DATABASE)
 
         pipeline = Pipeline()
         pipeline.add_component("writer", writer)