fix(mermaid): added correct output image path

EURAC-EEBgroup · Oct 2, 2024 · ff42d5a · ff42d5a
1 parent 2e3ca0f
commit ff42d5a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/brickllm/graphs/brickschema_graph.py b/brickllm/graphs/brickschema_graph.py
@@ -56,7 +56,7 @@ def display(self, filename="graph.png"):
             raise ValueError("Graph is not compiled yet. Please compile the graph first.")
 
         # Save the image to the specified file
-        self.graph.get_graph().draw_mermaid_png(filename)
+        self.graph.get_graph().draw_mermaid_png(output_file_path=filename)
 
         # Open the image using PIL (Pillow)
         if os.path.exists(filename):

diff --git a/examples/brickschema_ttl.py b/examples/brickschema_ttl.py
@@ -15,7 +15,7 @@
 """
 
 # Display the graph
-# brick_graph.display()
+brick_graph.display()
 
 # Run the graph without streaming
 brick_graph.run(