withPersistence stores the durability batch hint on a stopped or failed assistant message
I maintainer di solito rispondono entro 1 giorno
@tombeckenham ci sta già lavorando.
Dal 16/10/2026.
Valutazione
Questa issue non è ancora stata valutata.
Descrizione
TanStack AI version
@tanstack/ai 0.69.0, @tanstack/ai-persistence 0.9.0 (the same code is on main at 85c2956)
Framework/Library version
Node 26.3.1, server only (no UI framework involved)
Describe the bug and the steps to reproduce it
With snapshotStreaming: true, withPersistence keeps a metadata.tanstack.batch: true key on the stored assistant message when a run is stopped or fails after a middleware called ctx.emitCustomEvent(name, value, { batch: true }).
This is about the message store (stores.messages, which withPersistence writes and reconstructChat reads). It is not about the durability log, the delivery stream that toServerSentEventsResponse({ durability }) appends to. They are separate stores.
The batch hint is an in-process flag for the durability producer. #1355 says it "is stripped before the log and the wire", and stripDurabilityBatchHint does that in toServerSentEventsResponse and in the strip-to-spec middleware. That covers the durability log and the wire. flush() in stream-to-response.ts strips each chunk before durability.append and before forwarding it. The emitted CUSTOM chunk also passes through the middleware chain, though, before any of that. In withPersistence, onChunk merges chunk.metadata of every parent chunk into the streaming snapshot metadata (packages/ai-persistence/src/middleware.ts, the block under "Keep the call metadata (source, response id) for the snapshot rows."). The hint goes in with it. Every later snapshot row carries it. On Stop or on an error, updateStreamingRows(ctx, 'aborted' | 'error') writes the merged metadata to the final stored row, so the hint stays in storage and comes back from reconstructChat. A run that finishes normally is clean, because the final save replaces the snapshot rows.
Steps:
- Install
@tanstack/[email protected]and@tanstack/[email protected]. - Save the script below as
repro.mjsin an ESM package and runnode repro.mjs.
repro.mjs:
import { chat, defineChatMiddleware } from '@tanstack/ai'
import { memoryPersistence, withPersistence } from '@tanstack/ai-persistence'
const controller = new AbortController()
const adapter = {
kind: 'text',
name: 'fixed',
model: 'test-model',
'~types': {},
async *chatStream() {
const t = Date.now()
yield { type: 'RUN_STARTED', threadId: 'thread-1', runId: 'run-1', timestamp: t }
yield { type: 'TEXT_MESSAGE_START', messageId: 'msg-1', role: 'assistant', timestamp: t }
yield { type: 'TEXT_MESSAGE_CONTENT', messageId: 'msg-1', delta: 'partial', timestamp: t }
controller.abort() // the user presses Stop
},
}
const progress = defineChatMiddleware({
name: 'progress',
onChunk(ctx, chunk) {
if (chunk.type === 'TEXT_MESSAGE_CONTENT') {
ctx.emitCustomEvent('progress', { step: 1 }, { batch: true })
}
},
})
const persistence = memoryPersistence()
const stream = chat({
adapter,
messages: [{ role: 'user', content: 'hi' }],
threadId: 'thread-1',
runId: 'run-1',
abortController: controller,
middleware: [
withPersistence(persistence, { snapshotStreaming: true, snapshotIntervalMs: 0 }),
progress,
],
})
for await (const _ of stream) void _
const stored = await persistence.stores.messages.loadThread('thread-1')
console.log(JSON.stringify(stored.at(-1).metadata))
Expected
{"tanstack":{"source":{"provider":"fixed","api":"text","model":"test-model"},"stopReason":"aborted"}}
Actual
{"tanstack":{"source":{"provider":"fixed","api":"text","model":"test-model"},"batch":true,"stopReason":"aborted"}}
If the adapter throws instead of the abort, the row ends with "batch":true,"stopReason":"error". A snapshot row read in the middle of the run (after a second TEXT_MESSAGE_CONTENT) is {"tanstack":{"source":{...},"batch":true}}.
I found nothing that reads tanstack.batch on a stored message, so the visible effect is a stray internal key in the message store and in the hydrated messages. Any store that validates message metadata will see an unknown key. The hint is meant to stay in process, as the durability log and the wire already show, but the message store keeps it.
Your Minimal, Reproducible Example - (Sandbox Highly Recommended)
Inline above. Plain Node script, no network, no model.
Screenshots or Videos (Optional)
None.
Do you intend to try to help solve this bug with your own PR?
Yes, I am also opening a PR that solves the problem along side this issue
Terms & Code of Conduct
- I agree to follow this project's Code of Conduct
- I understand that if my bug cannot be reliable reproduced in a debuggable environment, it will probably not be fixed and this issue may even be closed.
- Lingua principale
- TypeScript
- Stelle
- 3.2k
- Fork
- 365
- Merge medio
- 1g 9h
- PR unite (30g)
- 242
Preparare l'ambiente
- Nessun Dockerfile né file Docker Compose
- Ha un modello di pull request
- Leggi la guida per i contributori
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di TanStack/ai
-
withPersistence does not save activity rows when a run is stopped or failsForse già presa @tombeckenham l’ha presa 4 giorni fa. Apertawaiting-on: maintainer
TanStack/ai#1804 · 1 assegnatario ·
I maintainer di solito rispondono entro 1 giorno
-
Activity events that a chat middleware emits are streamed but never recorded or persistedForse già presa @AlemTuzlak l’ha presa 4 giorni fa. Apertawaiting-on: maintainer
TanStack/ai#1803 · 1 assegnatario ·
I maintainer di solito rispondono entro 1 giorno
-
withPersistence stores the durability batch hint on a stopped or failed assistant messageForse già presa @tombeckenham l’ha presa 4 giorni fa. Apertahas-pr waiting-on: maintainer
TanStack/ai#1802 · 1 assegnatario ·
I maintainer di solito rispondono entro 1 giorno
-
Discriminate status-dependent record and option types (RunRecord, SessionIndexEntry, InputState, durable-only options)Forse già presa @jherr l’ha presa 4 giorni fa. Apertaenhancement waiting-on: maintainer
TanStack/ai#1799 · 1 assegnatario ·
I maintainer di solito rispondono entro 1 giorno
-
ai-harness: an input with no principal runs with the owner session's principalForse già presa @tombeckenham l’ha presa 5 giorni fa. Apertawaiting-on: maintainer
TanStack/ai#1797 · 1 assegnatario ·
I maintainer di solito rispondono entro 1 giorno
Issue simili
-
area: repo priority: P2 type: bug
Difficoltà 2/5 1-3 ore Idoneità per principianti 76/100
appandflow/stim#4107 · 1 commento ·
I maintainer di solito rispondono entro 1 giorno
-
community first-timers-only good first issue hacktoberfest help wanted low hanging fruit up-for-grabs
Difficoltà 1/5 1-3 ore Idoneità per principianti 60/100
lingdojo/kana-dojo#32859 · 1 commento · 5 reazioni ·
I maintainer di solito rispondono entro 1 giorno
-
code-quality refactoring
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
github/gh-aw-firewall#10364 ·
I maintainer di solito rispondono entro 1 giorno
-
Merge the duplicate kaffir_corn record into sorghumForse già presa @Crypto-Shroom l’ha presa oggi. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 65/100
Crypto-Shroom/Bird-Feed-Calculator#322 ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
drizzle-team/drizzle-orm#6552 ·
I maintainer di solito rispondono entro 2 giorni