fix(ai-opencode): bound memory while buffering stream parts

.changeset/calm-dolphins-buffer.md

@@ -0,0 +1,5 @@
+---
+'@tanstack/ai-opencode': patch
+---
+
+fix(ai-opencode): bound memory while buffering parts before their message role is known

packages/ai-opencode/src/stream/translate.ts

@@ -375,7 +375,13 @@ export async function* translateOpencodeStream(
       const role = messageRoles.get(part.messageID)
       if (role === undefined) {
         const parts = pendingParts.get(part.messageID) ?? []
-        parts.push({ part, delta })
+        // A text or reasoning part carries its full text, so keep only its
+        // latest update. Keeping each update holds every snapshot in memory.
+        const index = isToolPart(part)
+          ? -1
+          : parts.findIndex((pending) => pending.part.id === part.id)
+        if (index === -1) parts.push({ part, delta })
+        else parts[index] = { part }
         pendingParts.set(part.messageID, parts)
       } else if (role === 'assistant') {
         yield* handlePart(part, delta)

packages/ai-opencode/tests/translate.test.ts

@@ -146,6 +146,7 @@ describe('translateOpencodeStream', () => {
   it('accumulates incremental text deltas', async () => {
     const chunks = await collect([
       session,
+      messageUpdated('msg-1', 'assistant'),
       textPart('part-1', 'Hel', 'Hel'),
       textPart('part-1', 'Hello', 'lo'),
       done(),
@@ -164,6 +165,7 @@ describe('translateOpencodeStream', () => {
   it('derives the delta from full-text snapshots when no delta is given', async () => {
     const chunks = await collect([
       session,
+      messageUpdated('msg-1', 'assistant'),
       textPart('part-1', 'Hel'),
       textPart('part-1', 'Hello'),
       done(),
@@ -173,6 +175,23 @@ describe('translateOpencodeStream', () => {
     expect(contents[1]).toMatchObject({ delta: 'lo', content: 'Hello' })
   })
 
+  it('keeps only the latest update of a part while its role is unknown', async () => {
+    const chunks = await collect([
+      session,
+      textPart('part-1', 'The', 'The'),
+      textPart('part-1', 'The answer', ' answer'),
+      textPart('part-1', 'The answer is 42', ' is 42'),
+      done(),
+    ])
+    const contents = chunks.filter((c) => c.type === 'TEXT_MESSAGE_CONTENT')
+
+    expect(contents).toHaveLength(1)
+    expect(contents[0]).toMatchObject({
+      delta: 'The answer is 42',
+      content: 'The answer is 42',
+    })
+  })
+
   it('reports usage with cache and reasoning details', async () => {
     const chunks = await collect([
       session,