Skip to content

Commit

Permalink
Feat: Now saves a png of graph in docker files each time you run
Browse files Browse the repository at this point in the history
  • Loading branch information
EldarAlvik committed Oct 22, 2024
1 parent 2353f70 commit 43a3ffb
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions core/graphAgent.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 43a3ffb

Please sign in to comment.