Any tips on AIState? Seem to be doing a lot of fighting trying to get it to show what I'm expecting.

[shaded-blue]

To be clear, I'm not exactly new to this, but I feel like I have completely lost it since trying to integrate this AI SDK into my codebase / project, and part of my brain is just no longer functioning.

I seem to have absolutely no understanding of how this AIState is being updated.

For example, in my sendMessage server action, I am setting the initial message with update:

  aiState.update({
    ...aiState.get(),
    messages: [
      ...aiState.get().messages,
      {
        id: nanoid(),
        role: 'user',
        content
      }
    ]
  })

And I am using the same render function they are for the response, which uses done to add the response:

textStream.done()
      aiState.done({
        ...aiState.get(),
        messages: [
          ...aiState.get().messages,
          {
            id: nanoid(),
            role: 'assistant',
            content
          }
        ]
      })

I log aiState.get() through the mutation steps and see exactly what I expect, which is the new message after the update, and both at the done.

Similarly, if I log getAIState() in messageSave, which gets run after from the unstable_onSetAIState, I can see both messages.

****** aiState.done in response function, here's the updated state ****** :  {
  chatId: 'oshBrXcYVc0s',
  messages: [
    {
      id: 'RBjYZsmghB',
      role: 'user',
      content: 'Test 1',
      relatives: [Object],
      active: true
    },
    {
      id: 'i2xhhR5zQC',
      role: 'assistant',
      content: 'Hello! How can I assist you today?',
      relatives: [Object],
      active: true
    }
  ],
  terminalNode: 'i2xhhR5zQC'
}
Entered unstable_onSetAIState
Here's the state:  {
  chatId: 'oshBrXcYVc0s',
  messages: [
    {
      id: 'RBjYZsmghB',
      role: 'user',
      content: 'Test 1',
      relatives: [Object],
      active: true
    },
    {
      id: 'i2xhhR5zQC',
      role: 'assistant',
      content: 'Hello! How can I assist you today?',
      relatives: [Object],
      active: true
    }
  ],
  terminalNode: 'i2xhhR5zQC'
}
Here's the state.messages:  [
  {
    id: 'RBjYZsmghB',
    role: 'user',
    content: 'Test 1',
    relatives: { parentId: undefined, childId: 'i2xhhR5zQC' },
    active: true
  },
  {
    id: 'i2xhhR5zQC',
    role: 'assistant',
    content: 'Hello! How can I assist you today?',
    relatives: { parentId: 'RBjYZsmghB' },
    active: true
  }
]
****** ENTER messageSend ******
We're sending the following content:  Test 2
****** AIState in messageSend ****** :  {
  chatId: 'oshBrXcYVc0s',
  messages: [
    {
      id: 'i2xhhR5zQC',
      role: 'assistant',
      content: 'Hello! How can I assist you today?',
      relatives: [Object],
      active: true
    }
  ],
  terminalNode: 'i2xhhR5zQC'
}

When I send a follow-up message, the very first message I sent is gone from AIState, seemingly in between the last step of unstable_onSetAIState, and the absolute very start of the sendMessage action.

All follow up messages are successfully stored in the AIState.

Another oddity, is that when logged from the client, useAIState's state seems to look completely different from that on the server.

I've read through the rather terse docs many times, but again, I feel I've just found myself in some kind of rut here and my brain absolutely cannot seem to reconcile whatever is missing in my understanding, but I must be missing something, because for a tool that claims to streamline interaction with LLM's, I must say I've found this significantly more difficult to understand and implement than rolling an entire solution myself, even before getting to the UI part, which is the entire reason I started this attempt at a migration.

All advice is greatly appreciated, and thank you for your time. Sorry for the bit of a rant, this has really gotten to me. Lol.

[wolfpixels]

+1 having this exact problem

[wolfpixels]

I've mentioned it here: #1509 - this is the 2nd weird error I've encountered with their ai chat template. You are not going crazy, my guess is they are moving fast, perhaps too fast, to try and be the dominate ai sdk platform and stuff is breaking

[wolfpixels]

taking a quick look at Reddit, this seems to be a common feeling amongst dev community so dw