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!")