From 43a3ffbc2526b6a9ced08d529269e6a302d1be18 Mon Sep 17 00:00:00 2001 From: EldarAlvik Date: Tue, 22 Oct 2024 19:00:21 +0200 Subject: [PATCH] Feat: Now saves a png of graph in docker files each time you run --- core/graphAgent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/graphAgent.py b/core/graphAgent.py index cb3e9ab..9437f68 100644 --- a/core/graphAgent.py +++ b/core/graphAgent.py @@ -38,9 +38,9 @@ def __init__(self): ) self.graph = self.workflow.compile() - #with open("core/graph_node_network.png", 'wb') as f: - #f.write(self.graph.get_graph().draw_mermaid_png()) - + with open("graph_node_network.png", 'wb') as f: + f.write(self.graph.get_graph().draw_mermaid_png()) + def chatbot(self, state: GraphState): """ Simple bot that invokes the list of previous messages