add more test case + remove flaky test

This commit is contained in:
Twisha Bansal
2026-02-03 16:38:38 +05:30
parent 911069ae8d
commit 336743f747
2 changed files with 9 additions and 2 deletions

View File

@@ -1,5 +1,4 @@
Final Client Response:
AI:
Booking Confirmed
Loyalty Points
POLICY CHECK: Intercepting 'update-hotel'
POLICY CHECK: Intercepting 'update-hotel'

View File

@@ -106,6 +106,14 @@ async def main():
last_ai_msg = response["messages"][-1].content
print(f"AI: {last_ai_msg}")
# Test Pre-processing
print("-" * 50)
user_input = "Update booking for hotel 3 with checkin 2025-01-01 and checkout 2025-01-20"
response = await agent.ainvoke(
{"messages": [{"role": "user", "content": user_input}]}
)
print(f"AI: {response['messages'][-1].content}")
if __name__ == "__main__":
asyncio.run(main())