diff --git a/main.py b/main.py new file mode 100644 index 0000000..cd512db --- /dev/null +++ b/main.py @@ -0,0 +1,17 @@ +import sys +from agent import graph + +if __name__ == "__main__": + print("LangGraph stream‑mode demo. Type 'exit' to quit.") + state = {"messages": []} + for line in sys.stdin: + user_input = line.rstrip('\n') + if not user_input: + continue + if user_input.lower() in {"exit", "quit"}: + break + for partial in graph.stream({"input": user_input, "messages": state["messages"]}): + print(partial.get("partial", ""), end="") + print() + state["messages"] = graph.invoke({"input": user_input, "messages": state["messages"]})["messages"] + print("Goodbye!")