withPersistence stores the durability batch hint on a stopped or failed assistant message
メンテナーはふだん 1 日以内に返信
評価
この issue はまだ評価されていません。
説明
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.
- 主要言語
- TypeScript
- スター
- 3.2k
- フォーク
- 365
- 平均マージ
- 1日 8時間
- マージ済み PR(30日)
- 261
環境構築
- Dockerfile・Docker Compose ファイルなし
- プルリクエストのテンプレートあり
- コントリビューションガイドを読む
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
TanStack/ai のほかの issue
-
withPersistence does not save activity rows when a run is stopped or fails対応中かも @tombeckenham が 1 日前に担当しました。 オープンwaiting-on: maintainer
メンテナーはふだん 1 日以内に返信
-
Activity events that a chat middleware emits are streamed but never recorded or persisted対応中かも @AlemTuzlak が 1 日前に担当しました。 オープンwaiting-on: maintainer
メンテナーはふだん 1 日以内に返信
-
Activity events that a chat middleware emits are streamed but never recorded or persisted対応中かも @AlemTuzlak が 1 日前に担当しました。 オープンwaiting-on: maintainer
メンテナーはふだん 1 日以内に返信
-
withPersistence stores the durability batch hint on a stopped or failed assistant message対応中かも @tombeckenham が 1 日前に担当しました。 オープンwaiting-on: maintainer
メンテナーはふだん 1 日以内に返信
-
Discriminate status-dependent record and option types (RunRecord, SessionIndexEntry, InputState, durable-only options)対応中かも @jherr が 1 日前に担当しました。 オープンenhancement waiting-on: maintainer
メンテナーはふだん 1 日以内に返信
似ている issue
-
難易度 2/5 1〜3時間 初心者へのやさしさ 77/100
TomerAberbach/profiler-md#733 ·
メンテナーはふだん 1 日以内に返信
-
bug priority: medium
難易度 2/5 1〜3時間 初心者へのやさしさ 68/100
brentspine/brents-toasts#143 ·
-
🐛 bug 🗺️ DB Models & Ops
難易度 2/5 1〜3時間 初心者へのやさしさ 83/100
LibreChat-AI/LibreChat#17214 ·
メンテナーはふだん 1 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 67/100
Wavvon/Wavvon-clients#136 ·
メンテナーはふだん 1 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 62/100
umami-software/umami#4637 ·
メンテナーはふだん 3 日以内に返信