Skip to content

fix(plugin-chatbot): accept the AI SDK's UIMessage in the exported mappers - #8341

Merged
os-justin merged 1 commit into
mainfrom
claude/issue-8214-chatbot-anypart-state-widen
Sep 7, 2026
Merged

fix(plugin-chatbot): accept the AI SDK's UIMessage in the exported mappers#8341
os-justin merged 1 commit into
mainfrom
claude/issue-8214-chatbot-anypart-state-widen

Conversation

@os-justin

Copy link
Copy Markdown
Collaborator

Fixes #8214

Spelling note, same reason the card gives: generic type arguments are written as WORDS below (UIMessage of unknown, UIDataTypes, UITools), because GitHub's body sanitizer eats tag-shaped fragments — backticks and fences included.

The defect

packages/plugin-chatbot/src/mapMessages.ts declares a deliberately permissive local AnyPart — every member is string / unknown — but typed one member against the output contract:

state?: ChatToolInvocation['state'];   // the TOOL-invocation lifecycle

@ai-sdk/react's text and reasoning parts carry state: 'streaming' | 'done', which is not in that union. So the interface written to absorb the SDK's part union was, on that one property, stricter than the union it exists to absorb, and the whole UIMessage[] assignment failed — at the exact call the docblock and the README export the mappers for.

The fix (PM ruling: option A)

AnyPart.state is now string, and the file's single read site narrows through a guard:

// a Record keyed by (NonNullable of ChatToolInvocation's state), valued true
const TOOL_STATES = { 'input-streaming': true, /* … the other 6 rows … */ };

// a type predicate narrowing to (NonNullable of ChatToolInvocation's state)
function isToolState(state) { return state !== undefined && state in TOOL_STATES; }

// the one read site, was: const persistedState = p.state;
const persistedState = isToolState(p.state) ? p.state : undefined;

Why string and not the union of both known state sets. Both widens cost the same at the read site — neither one lets p.state flow into ChatToolInvocation['state'] without an explicit narrowing, because AnyPart is a flat interface and the p.type filter above does not narrow a sibling property. Given equal cost, string is the one consistent with the ruling: it is a purely local widening. Enumerating one dependency's current literals inside a published parameter type keeps the type coupled to @ai-sdk/react@4.0.68 without the SDK's type as the source of truth — the coupling of option B with none of its authority — and re-creates this identical defect on the next SDK bump. string matches every sibling member of AnyPart and ends the class.

Cost at the read sites. mapMessages.ts has exactly one read of p.state (line 634 on the base; the other four state occurrences in extractToolInvocations are locals). It pays one guard call. The Record over the union is what keeps the guard honest in both directions: a typo in the table is a compile error, and a state added to ChatToolInvocation['state'] fails here until a row is added.

Behaviour fix that falls out of it. An unrecognized state string (an AI SDK v4 snapshot's 'result', say) used to pass through verbatim into ChatToolInvocation.state, then fall past every branch in getToolState and render a finished call as "Running" forever. It now normalizes to undefined, which is that function's documented "infer from errorText / result" case. The output type is now true at runtime instead of merely asserted.

Evidence

Instrument is the package's own type-check (tsc --noEmit && tsc -p tsconfig.test.json), not vitest — vitest transpiles types away. tsconfig.test.json was confirmed to actually read the test file (--listFiles, 1876 files, src/__tests__/mapMessages.test.ts present; a made-up filename control returns 0 hits).

leg verdict
base, cast in place type-check exit 0
base, cast removed (lit control) exit 2TS2345 at mapMessages.test.ts(136,42), the card's exact chain down to Type '"done"' is not assignable to …
after the widen, cast removed type-check exit 0
ablation: narrow state restored on the committed implementation exit 2, same TS2345 by name
dependents (--filter '...@object-ui/plugin-chatbot') exit 0, Scope: 7 of 47 workspace projects, all 7 Done, 0 errors
pnpm --filter @object-ui/plugin-chatbot test 39 files / 461 tests passed
eslint . (package-scoped) 0 errors, 90 warnings (all pre-existing, none in a touched file)

Ablation discipline: mutation proved on disk by git hash-object differing from git rev-parse HEAD:PATH, restore proved by state (git diff HEAD empty and the hashes equal again), trap … EXIT INT TERM with absolute paths, from a committed implementation. A separate positive control confirms the vitest case is really executed: an injected failing expectation turns that exact file and test name red, then is restored by state.

The two load-bearing casts

  • mapMessages.test.ts:135 msgs as never — removed. The fixture is now declared UIMessage[] imported from @ai-sdk/react, so the pin is about the dependency's real shape rather than a hand-rolled literal. Note the cast was doing more than the card's diagnostic: the untyped literal also had role: string, so the cast was hiding two things. The SDK type fixes both, and nothing else in the fixture had to change.
  • useObjectChat.ts:683 chatResult as any — measured, left in place. With the widen applied, removing it leaves exactly one diagnostic, and it has nothing to do with this card: TS2322 at useObjectChat.ts:803 — the hook's own returned setMessages is declared (messages: unknown[]) => void while the SDK's is (messages: UIMessage[] | updater) => void, which is contravariantly incompatible. Out of scope here; reported rather than chased.

That measurement also substantiates the card's central claim rather than repeating it. With the narrow state restored and the as any removed, the mapper call in production source goes red too: TS2345 at useObjectChat.ts(701,69). So the widen fixes the package's own call site, not just the test — the as any had been masking this defect as well as the unrelated one.

README

The doc-snippet: fragment marker does exist on this base (packages/plugin-chatbot/README.md, line 361), so batch 30's PR #8228 has landed. It is removed here, and the doc-snippet gate is the instrument that proves the block now compiles:

run verdict line
marker present (before) Covered blocks: 790 — 631 to compile, 159 declared fragment(s). / Semantic phase: 631 of 631 block(s) judged, 0 failed.exit 0
marker removed (after) Covered blocks: 790 — 632 to compile, 158 declared fragment(s). / Semantic phase: 632 of 632 block(s) judged, 0 failed.exit 0
control: marker removed, dist rebuilt from the ablated source exit 1[semantic] packages/plugin-chatbot/README.md:371:42 TS2345 …

Exit 1 is "I ran and found errors", not the exit 2 "PRECONDITION NOT MET" a fresh worktree gives before its closure is built; that closure (34 filters, from the gate's own --build-filter) was built for every one of these runs.

One addition beyond the brief: the seven-line prose blockquote directly above the marker — "Today this call needs a cast on the reader's side…" — is removed with it. It documents the defect as a live constraint on the reader; leaving it would keep the page telling readers to write a cast the compiler no longer wants, which is the same over-claim in prose that the marker was in metadata.

Changeset

minor on @object-ui/plugin-chatbot. Not breaking: the parameter type only got looser, so every argument that compiled before still compiles (verified — dist/mapMessages.d.ts now emits state?: string;). minor rather than patch because a published signature accepts input it previously refused, which is a capability a consumer can newly depend on. The repo forbids major; pnpm gate check-changeset-no-major is green, as are check-changeset-presence / -fixed / -overwrite.

Other gates run

check:control-bytes, check:readme-exports, check:doc-fences, check:doc-types, check:phantom-deps, check:unused-deps — all exit 0. check-governed-queue-guard --test on the four changed paths: NOT GOVERNED — 4 path(s) checked against 5 governed surface(s); none matched.


🤖 Generated with Claude Code

https://claude.ai/code/session_01YBWFb5YgMU5dw8p2VKj16S


Generated by Claude Code

…ppers

`mapMessages.ts`'s local `AnyPart` is written to absorb whatever a producer
hands the mapper — every other member is `string` / `unknown` — but `state` was
typed against the OUTPUT contract (`ChatToolInvocation['state']`, the
tool-invocation lifecycle). `@ai-sdk/react`'s text and reasoning parts carry
`state: 'streaming' | 'done'`, which is not in that union, so the
deliberately-permissive input interface was on that one property STRICTER than
the union it exists to absorb and the whole `UIMessage[]` assignment was refused
with TS2345 — at the exact call the docblock and the README export it for.

Widen `AnyPart.state` to `string` and narrow at the single read site through an
`isToolState` guard whose table is a `Record` over `ChatToolInvocation['state']`,
so the output stays exactly as checked as before and the table cannot drift from
the union it guards. `string` rather than the union of both known state sets:
every other `AnyPart` member is already maximally loose, and enumerating one
dependency's current literals in a published parameter type re-creates this same
defect on the next SDK bump.

The test's `msgs as never` is gone: the fixture is now typed `UIMessage[]` from
`@ai-sdk/react`, so the assertion is about the dependency's real shape rather
than a hand-rolled literal, and the compiler — not a cast — judges the seam.
`useObjectChat.ts:683`'s `chatResult as any` is left in place deliberately;
measured separately, it hides one unrelated diagnostic (see the PR body).

The README's mapper block loses its `doc-snippet: fragment` marker and the prose
warning above it: the block compiles as written now, and `check:doc-snippets`
moves it from the declared-fragment tier into the compiled tier (631 -> 632
compiled, 159 -> 158 fragments, 0 failed).

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01YBWFb5YgMU5dw8p2VKj16S
@github-actions

github-actions Bot commented Sep 7, 2026

Copy link
Copy Markdown
Contributor

✅ Console Performance Budget

Metric Value Budget
Eager closure (gzip, 50 chunks) 3472.2 KB 3512.7 KB
Main entry chunk (gzip) 143.9 KB 350 KB
Entry file index-CUwgXj8u.js
Status PASS

The eager closure is every chunk the entry reaches through static imports — what the browser fetches and parses before the app renders. The entry chunk on its own is a small fraction of it.


📦 Bundle Size Report

Package Size Gzipped
app-shell (consoleActionDispatch.js) 0.20KB 0.19KB
app-shell (index.js) 15.67KB 5.75KB
app-shell (runtime-config.js) 20.68KB 7.36KB
app-shell (types.js) 0.01KB 0.04KB
app-shell (urlParams.js) 10.06KB 3.86KB
auth (ActiveOrganizationStorage.js) 25.05KB 9.16KB
auth (AuthContext.js) 0.31KB 0.24KB
auth (AuthGuard.js) 2.07KB 1.00KB
auth (AuthProvider.js) 40.18KB 10.59KB
auth (AuthShell.js) 3.49KB 1.40KB
auth (ForgotPasswordForm.js) 12.21KB 3.45KB
auth (LoginForm.js) 18.15KB 5.39KB
auth (PreviewBanner.js) 0.90KB 0.50KB
auth (RegisterForm.js) 6.65KB 2.22KB
auth (SocialSignInButtons.js) 9.61KB 3.89KB
auth (UserMenu.js) 3.41KB 1.23KB
auth (auth-gate-events.js) 1.29KB 0.66KB
auth (authStyles.js) 5.04KB 1.72KB
auth (createAuthClient.js) 40.21KB 10.80KB
auth (createAuthenticatedFetch.js) 8.46KB 3.43KB
auth (index.js) 3.19KB 1.44KB
auth (invitation-status.js) 1.22KB 0.70KB
auth (org-roles.js) 6.66KB 2.78KB
auth (phone-identifier.js) 1.11KB 0.66KB
auth (types.js) 0.59KB 0.35KB
auth (useAuth.js) 5.30KB 1.02KB
auth (useWorkspaceAdminStatus.js) 5.13KB 2.35KB
collaboration (CommentThread.js) 26.08KB 7.56KB
collaboration (LiveCursors.js) 3.17KB 1.27KB
collaboration (PresenceAvatars.js) 6.49KB 2.64KB
collaboration (PresenceProvider.js) 2.79KB 1.13KB
collaboration (index.js) 1.68KB 0.73KB
collaboration (useCollaborationTranslation.js) 6.05KB 2.52KB
collaboration (useCommentSearch.js) 1.98KB 0.88KB
collaboration (useConflictResolution.js) 7.75KB 1.86KB
collaboration (useMentionNotifications.js) 1.81KB 0.68KB
collaboration (usePresence.js) 6.33KB 1.84KB
collaboration (useRealtimeSubscription.js) 7.91KB 2.01KB
components (index.js) 498.43KB 114.00KB
core (index.js) 7.48KB 2.96KB
create-plugin (index.js) 10.08KB 3.26KB
data-objectstack (index.js) 189.15KB 52.56KB
fields (index.js) 243.15KB 61.40KB
i18n (LocalizationContext.js) 1.76KB 0.96KB
i18n (builtinAggregateLabels.js) 0.86KB 0.49KB
i18n (currency.js) 1.22KB 0.64KB
i18n (fallbackInterpolation.js) 6.25KB 2.77KB
i18n (i18n.js) 6.57KB 2.76KB
i18n (index.js) 3.65KB 1.47KB
i18n (pickLocalized.js) 7.62KB 3.26KB
i18n (provider.js) 26.89KB 9.04KB
i18n (useDisplayLocale.js) 2.85KB 1.45KB
i18n (useObjectLabel.js) 34.34KB 9.17KB
i18n (useSafeTranslation.js) 5.60KB 2.33KB
layout (index.js) 38.84KB 10.94KB
mobile (MobileProvider.js) 0.92KB 0.49KB
mobile (ResponsiveContainer.js) 0.94KB 0.38KB
mobile (breakpoints.js) 1.51KB 0.70KB
mobile (createOfflineDataSource.js) 5.61KB 1.75KB
mobile (index.js) 1.99KB 0.87KB
mobile (offlineQueue.js) 3.91KB 1.35KB
mobile (pwa.js) 0.97KB 0.49KB
mobile (serviceWorker.js) 1.48KB 0.62KB
mobile (serviceWorkerSource.js) 3.41KB 1.48KB
mobile (useBreakpoint.js) 1.54KB 0.65KB
mobile (useGesture.js) 6.96KB 1.98KB
mobile (useOfflineSync.js) 1.99KB 0.72KB
mobile (usePullToRefresh.js) 2.53KB 0.85KB
mobile (useResponsive.js) 0.72KB 0.42KB
mobile (useSpecGesture.js) 4.39KB 1.66KB
mobile (useTouchTarget.js) 1.01KB 0.54KB
permissions (MePermissionsProvider.js) 11.71KB 4.29KB
permissions (PermissionContext.js) 0.31KB 0.25KB
permissions (PermissionGuard.js) 0.89KB 0.45KB
permissions (PermissionProvider.js) 6.24KB 2.16KB
permissions (discardProofCache.js) 1.04KB 0.55KB
permissions (evaluator.js) 5.12KB 1.74KB
permissions (index.js) 0.93KB 0.41KB
permissions (store.js) 0.91KB 0.42KB
permissions (useFieldPermissions.js) 1.28KB 0.53KB
permissions (usePermissions.js) 4.83KB 2.27KB
plugin-ai (index.js) 15.16KB 3.68KB
plugin-calendar (index.js) 47.98KB 13.51KB
plugin-charts (index.js) 71.39KB 19.92KB
plugin-chatbot (index.js) 193.79KB 46.09KB
plugin-dashboard (index.js) 131.48KB 34.45KB
plugin-designer (index.js) 213.21KB 43.63KB
plugin-detail (index.js) 248.56KB 63.89KB
plugin-editor (index.js) 2.23KB 1.05KB
plugin-form (index.js) 131.01KB 32.32KB
plugin-gantt (index.js) 167.16KB 40.99KB
plugin-grid (index.js) 208.58KB 56.63KB
plugin-kanban (index.js) 55.38KB 15.72KB
plugin-list (index.js) 113.38KB 27.73KB
plugin-map (index.js) 20.49KB 6.83KB
plugin-markdown (index.js) 13.88KB 4.80KB
plugin-report (index.js) 43.42KB 11.92KB
plugin-timeline (index.js) 30.10KB 8.74KB
plugin-tree (index.js) 9.33KB 3.25KB
plugin-view (index.js) 84.46KB 20.80KB
providers (DataSourceProvider.js) 0.75KB 0.39KB
providers (MetadataProvider.js) 1.37KB 0.59KB
providers (ThemeProvider.js) 1.90KB 0.85KB
providers (UploadProvider.js) 11.66KB 3.50KB
providers (index.js) 0.45KB 0.23KB
providers (types.js) 0.01KB 0.04KB
react-runtime (index.js) 5.62KB 2.34KB
react (LazyPluginLoader.js) 4.47KB 1.63KB
react (SchemaRenderer.js) 81.07KB 26.86KB
react (data-invalidation.js) 5.05KB 2.08KB
react (index.js) 4.63KB 2.18KB
react (schema-input.js) 2.32KB 1.24KB
react (spec-input.js) 0.20KB 0.18KB
sdui-parser (codegen.js) 6.58KB 2.74KB
sdui-parser (dashboard-widget-options.js) 3.08KB 1.30KB
sdui-parser (index.js) 5.55KB 2.45KB
sdui-parser (input-type.js) 2.84KB 1.40KB
sdui-parser (parse.js) 20.57KB 5.88KB
sdui-parser (provenance.js) 3.66KB 1.82KB
sdui-parser (types.js) 0.28KB 0.23KB
sdui-parser (validate.js) 13.64KB 4.59KB
types (ai.js) 0.20KB 0.17KB
types (api-types.js) 0.20KB 0.18KB
types (app.js) 2.87KB 1.00KB
types (base.js) 0.20KB 0.18KB
types (blocks.js) 0.20KB 0.18KB
types (complex.js) 2.93KB 1.49KB
types (crud.js) 0.20KB 0.18KB
types (dashboard-filter-alias.js) 6.23KB 2.74KB
types (data-display.js) 3.75KB 1.85KB
types (data-protocol.js) 0.20KB 0.19KB
types (data.js) 0.20KB 0.18KB
types (designer.js) 1.85KB 0.85KB
types (disclosure.js) 0.20KB 0.18KB
types (error-code.js) 1.54KB 0.88KB
types (expression.js) 0.20KB 0.18KB
types (feedback.js) 0.20KB 0.18KB
types (field-types.js) 0.20KB 0.18KB
types (form.js) 0.20KB 0.18KB
types (http-inflight.js) 8.87KB 3.73KB
types (http-retry.js) 4.32KB 2.02KB
types (icon-key-migration.js) 4.26KB 1.63KB
types (index.js) 4.74KB 2.25KB
types (layout.js) 0.20KB 0.18KB
types (managed-by.js) 0.19KB 0.18KB
types (mobile.js) 4.73KB 2.28KB
types (navigation.js) 0.20KB 0.18KB
types (objectql.js) 0.20KB 0.18KB
types (overlay.js) 0.20KB 0.18KB
types (permissions.js) 0.20KB 0.18KB
types (plugin-scope.js) 0.20KB 0.18KB
types (record-components.js) 0.20KB 0.19KB
types (record-semantics.js) 1.28KB 0.67KB
types (registry.js) 0.20KB 0.18KB
types (reports.js) 0.20KB 0.18KB
types (select-option.js) 0.20KB 0.19KB
types (spec-report.js) 5.05KB 1.93KB
types (spec-ui-namespace.js) 0.20KB 0.19KB
types (system-fields.js) 3.33KB 1.54KB
types (theme.js) 6.28KB 2.87KB
types (ui-action.js) 8.11KB 3.32KB
types (views.js) 0.20KB 0.18KB
types (widget.js) 0.20KB 0.18KB

Size Limits

  • ✅ Core packages should be < 50KB gzipped
  • ✅ Component packages should be < 100KB gzipped
  • ⚠️ Plugin packages should be < 150KB gzipped

@os-justin
os-justin marked this pull request as ready for review September 7, 2026 13:43
@os-justin
os-justin enabled auto-merge September 7, 2026 13:43
@os-justin
os-justin added this pull request to the merge queue Sep 7, 2026
Merged via the queue into main with commit 411a132 Sep 7, 2026
34 checks passed
@os-justin
os-justin deleted the claude/issue-8214-chatbot-anypart-state-widen branch September 7, 2026 14:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation plugin tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

finding(plugin-chatbot): uiMessagesToChatMessages' parameter type refuses @ai-sdk/react's UIMessage — the exact input it is exported for

1 participant