Solution ready for publishing: add main.py
This commit is contained in:
@@ -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!")
|
||||||
Reference in New Issue
Block a user