Pushed a fix here:
https://github.com/discourse/discourse-ai/pull/708
Let me know if you still have problems, please.