mirror of
https://github.com/googleapis/genai-toolbox.git
synced 2026-02-03 11:45:10 -05:00
add more test case + remove flaky test
This commit is contained in:
@@ -1,5 +1,4 @@
|
||||
Final Client Response:
|
||||
AI:
|
||||
Booking Confirmed
|
||||
Loyalty Points
|
||||
POLICY CHECK: Intercepting 'update-hotel'
|
||||
POLICY CHECK: Intercepting 'update-hotel'
|
||||
@@ -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())
|
||||
|
||||
Reference in New Issue
Block a user