18 lines
629 B
Python
18 lines
629 B
Python
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!")
|