feat: move graph compilation inside class

brickllm/compiled_graphs.py

@@ -1,3 +1,3 @@
 from .graphs import BrickSchemaGraph
 
-brickschema_graph = BrickSchemaGraph().compile()
+brickschema_graph = BrickSchemaGraph()._compiled_graph()

brickllm/graphs/brickschema_graph.py

@@ -36,15 +36,20 @@ def __init__(self):
         self.workflow.add_edge("get_relationships", "get_sensors")
         self.workflow.add_edge("get_sensors", END)
 
-        # Initialize the compiled graph
-        self.graph = None
+        # Compile graph
+        try:
+            self.graph = self.workflow.compile()
+        except Exception as e:
+            raise ValueError(f"Failed to compile the graph: {e}")
 
         # Hardcoding the thread_id for now
         self.config = {"configurable": {"thread_id": "1"}}
 
-    def compile(self):
-        """Compile the workflow and update the compiled graph."""
-        self.graph = self.workflow.compile()
+    def _compiled_graph(self):
+        """Check if the graph is compiled and return the compiled graph."""
+        if self.graph is None:
+            raise ValueError("Graph is not compiled yet. Please compile the graph first.")
+        return self.graph
 
     def display(self, filename="graph.png"):
         """Display the compiled graph as an image.

examples/brickschema_ttl.py

@@ -2,7 +2,6 @@
 
 # Create an instance of BrickSchemaGraph
 brick_graph = BrickSchemaGraph()
-brick_graph.compile()
 
 # Specify the user prompt
 building_description = """