invoke.py

from graph import graph
from states.state import State


def main():
    result = graph.invoke(State(graph_state="Initial State"))

    print("Final state:")
    print(result["graph_state"])
    print()
    print("Logs:")
    for log in result["logs"]:
        print(f"- {log}")

if __name__ == "__main__":
    main()