Solution ready for publishing: add main.py

This commit is contained in:
2026-06-02 16:08:13 +00:00
parent b6a91b0713
commit be0d4fe8ef
+17
View File
@@ -0,0 +1,17 @@
import sys
from agent import graph
if __name__ == "__main__":
print("LangGraph streammode 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!")