Gorantla Krishna commited on
Commit
306f0be
·
1 Parent(s): de93f01
src/langgraphagenticai/graph/graph_builder.py CHANGED
@@ -65,8 +65,8 @@ class GraphBuilder:
65
  elif usecase == "Chatbot with Tool":
66
  self.chatbot_with_tools_build_graph()
67
 
68
- graph = self.graph_builder.compile()
69
- display(Image(graph.get_graph().draw_mermaid_png(output_file_path=f"src\langgraphagenticai\graph\graph_images\graph-{usecase}.png")))
70
- return graph
71
 
72
 
 
65
  elif usecase == "Chatbot with Tool":
66
  self.chatbot_with_tools_build_graph()
67
 
68
+
69
+
70
+ return self.graph_builder.compile()
71
 
72