It looks good. Can you try it with a mock response that is a bit shorter? And also tell the assistant to wait for the tool's response