Better error message when trying to use without having it installed

oskarhane · oskarhane · commit 54fe2ebc07f0 · 2025-04-04T16:46:29.000+02:00
diff --git a/src/neo4j_graphrag/experimental/pipeline/pipeline.py b/src/neo4j_graphrag/experimental/pipeline/pipeline.py
@@ -215,7 +215,7 @@ def _get_neo4j_viz_graph(
         """Generate a neo4j-viz visualization of the pipeline graph"""
         if not neo4j_viz_available:
             raise ImportError(
-                "Could not import neo4j-viz. Install it with 'pip install neo4j-viz'"
+                "Could not import neo4j-viz. Install it with 'pip install \"neo4j-graphrag[experimental]\"'"
             )
 
         self.validate_parameter_mapping()

Original file line number	Diff line number	Diff line change
`@@ -215,7 +215,7 @@ def _get_neo4j_viz_graph(`
`215`	`215`	`"""Generate a neo4j-viz visualization of the pipeline graph"""`
`216`	`216`	`if not neo4j_viz_available:`
`217`	`217`	`raise ImportError(`
`218`		`- "Could not import neo4j-viz. Install it with 'pip install neo4j-viz'"`
	`218`	`+ "Could not import neo4j-viz. Install it with 'pip install \"neo4j-graphrag[experimental]\"'"`
`219`	`219`	`)`
`220`	`220`
`221`	`221`	`self.validate_parameter_mapping()`