Update agent.py

This commit is contained in:
Twisha Bansal
2026-02-09 11:23:58 +05:30
committed by GitHub
parent 7e175c62a0
commit acbb9c15ed

View File

@@ -33,7 +33,7 @@ async def before_tool_callback(
tool_name = tool.name
print(f"POLICY CHECK: Intercepting '{tool_name}'")
if "checkin_date" in args and "checkout_date" in args:
if tool_name == "update-hotel" and "checkin_date" in args and "checkout_date" in args:
start = datetime.fromisoformat(args["checkin_date"])
end = datetime.fromisoformat(args["checkout_date"])
duration = (end - start).days