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