39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
|
"""LangGraph 脚手架冒烟脚本（作业 1）。
|
||
|
|
|
||
|
|
同一 thread 输入两次、另一个 thread 输入一次，打印轮数和文本。
|
||
|
|
预期输出轮数依次为 1、2、1。
|
||
|
|
|
||
|
|
本脚本不导入 FastGPT、不读取其密钥；离线即可运行：
|
||
|
|
|
||
|
|
python examples/langgraph_smoke.py
|
||
|
|
"""
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from src.agent.graph import build_graph
|
||
|
|
|
||
|
|
|
||
|
|
async def main() -> None:
|
||
|
|
graph = build_graph()
|
||
|
|
|
||
|
|
async def ask(thread_id: str, text: str) -> None:
|
||
|
|
# 只传本轮变化的字段；会话字段（turn_count 等）由 checkpoint 保管。
|
||
|
|
# 注意：对已有 checkpoint 的 thread 传入完整初始状态，
|
||
|
|
# 其中的 turn_count=0 会覆盖已累计的值。
|
||
|
|
result = await graph.ainvoke(
|
||
|
|
{"session_id": thread_id, "input_text": text},
|
||
|
|
config={"configurable": {"thread_id": thread_id}},
|
||
|
|
)
|
||
|
|
print(
|
||
|
|
f"thread={thread_id} turn_count={result['turn_count']} "
|
||
|
|
f"stage_code={result['stage_code']} text={result['response_text']}"
|
||
|
|
)
|
||
|
|
|
||
|
|
await ask("thread-a", "你好")
|
||
|
|
await ask("thread-a", "发生了交通事故")
|
||
|
|
await ask("thread-b", "你好")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(main())
|