Spaces:
Runtime error
Runtime error
| import sys | |
| import os | |
| # 确保路径能够导入 backend | |
| sys.path.insert(0, os.path.abspath(os.path.dirname(__file__))) | |
| from backend.agent.graph import app | |
| def main(): | |
| print("==================================================") | |
| print("🚀 开始模拟 VTP A2A 协商流程: Planner vs Budget") | |
| print("==================================================\n") | |
| initial_state = { | |
| "user_query": "I want a 7-day luxurious trip to Paris.", | |
| "messages": ["User: I want a 7-day luxurious trip to Paris."], | |
| "current_plan": None, | |
| "negotiation_rounds": 0, | |
| "feedback": None, | |
| "is_approved": False, | |
| "final_output": None | |
| } | |
| # 使用 LangGraph 的 stream 执行图结构，监控每次节点运转 | |
| for step_data in app.stream(initial_state): | |
| for node_name, state_update in step_data.items(): | |
| print(f"\n✅ [LangGraph Node Finished]: -> {node_name.upper()}") | |
| print("-" * 50) | |
| # 打印当前节点的输出变动 | |
| if "messages" in state_update: | |
| for msg in state_update["messages"]: | |
| print(f"💬 Message: {msg}") | |
| if "feedback" in state_update and state_update["feedback"]: | |
| print(f"⚠️ Explicit Constraint (Feedback): {state_update['feedback']}") | |
| if "negotiation_rounds" in state_update: | |
| print(f"🔄 Negotiation Rounds Updated To: {state_update['negotiation_rounds']}") | |
| if "final_output" in state_update: | |
| print(f"🎯 Final Output Generated: {state_update['final_output']}") | |
| print("-" * 50) | |
| print("\n==================================================") | |
| print("🎉 协商流程演示结束！") | |
| print("==================================================") | |
| if __name__ == "__main__": | |
| main() | |