Separate UI message for each text part in a message[]

[alexduffdesign]

I have a tool that looks up user orders in Shopify.

In the tool we have ShippingInfoStatus :
READY_TO_FETCH: "READY_TO_FETCH",
NEEDS_ORDER_SELECTION: "NEEDS_ORDER_SELECTION",
NEEDS_ORDER_NUMBER: "NEEDS_ORDER_NUMBER",
INSUFFICIENT_INFO: "INSUFFICIENT_INFO",

And based of these we render different UI components in the frontend.

So if a user says 'where's my order' and they haven't included an order number it will output : 'needs_order_selection' and finds the orders, and we will render the OrderList component.

I've got this as a human in the loop step, so when they select the order(s) it sets the result of that tool as the order number(s).

We then in the routes run tool again with the order numbers - which finds the shipping information of those orders.

This sets the result of the tool again with the updated info - the assistant adds another text part which is added to the same UI message as the previous and the UI element is added showing the shipping info.

• The issue being that the text part gets added to the existing UI message
  So we get :

"It looks like you have an order waiting for your attention! 🎉 Here are the details:

Order Number: (ordernumber)
Order Date: (date)
Total: (price)

Items:
Beeswax Birthday Candle Set of 10 🐝 🎂
Blue Goose Cake Topper
Would you like to know more about this order? 🌈✨

Your order has been delivered! 🎉✨ Here are the details:
Order Number: (number)
Status: (status)
Estimated Delivery Date: (date)
Items:(items)

If you need anything else, just let me know! 🌈💖"

You see it has the two messages in the same UI message. I am guessing because its part of the same message[] in useChat.

---

Perhaps I am doing the wrong thing in having this as a human in the loop step?

Is this expected behaviour - is my setup wrong perhaps for this functionality. Any help would be awesome.

Thanks.

The other way I guess is - sending a user message when they select the orders 'I want to track (order number) then our agent would call the tool.