Update myagent_langraph.py

myagent_langraph.py

@@ -5,7 +5,7 @@ from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
 from langgraph.graph.message import add_messages
 from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import ToolNode, tools_condition
-from IPython.display import Image, display
+#from IPython.display import Image, display
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.tools.tavily_search import TavilySearchResults
 
@@ -215,10 +215,10 @@ if __name__ == "__main__":
     react_graph = build_graph()
     # Show the butler's thought process
     # Save the graph visualization to disk
-    graph_image = react_graph.get_graph(xray=True).draw_mermaid_png()
-    with open("graph_visualization.png", "wb") as f:
-        f.write(graph_image)
-    print("Graph visualization saved to 'graph_visualization.png'")
+    #graph_image = react_graph.get_graph(xray=True).draw_mermaid_png()
+    #with open("graph_visualization.png", "wb") as f:
+    #    f.write(graph_image)
+    #print("Graph visualization saved to 'graph_visualization.png'")
 
     # Also display it if in an environment that supports display
     #display(Image(graph_image))