@assistant-ui/core 0.3.18 → 0.3.19

package/dist/react/client/interactable-model-context.js.map

@@ -1 +1 @@
-{"version":3,"file":"interactable-model-context.js","names":["Tool","toJSONSchema","Unstable_InteractableDefinition","interactableToolName","shallowMergeInteractableState","Unstable_InteractableSnapshotEntry","generateId","isRecord","PartialJSONSchema","ReturnType","ID_PROPERTY","type","const","description","ITEM_ID_PROPERTY","hasIdProperty","schema","Record","properties","id","undefined","withRequiredItemId","required","withoutItemId","_omitted","Array","isArray","filter","key","toArrayUpdateSchema","field","itemSchema","items","idKeyed","add","remove","clear","update","additionalProperties","idKeyedArrayFieldNames","Set","names","value","Object","entries","withArrayUpdateSchemas","fromEntries","map","withRequiredId","partial","process","env","NODE_ENV","console","warn","_reserved","buildInteractableModelContext","definitions","partialSchemaCache","Map","setDefState","updater","prev","streamBaselines","targetId","state","tools","unstable_composerMetadata","values","length","byName","def","list","get","name","push","set","instances","toolName","first","partialSchema","idKeyedFields","resolveTarget","parameters","streamCall","reader","toolCallId","partialArgs","args","streamValues","keys","idIndex","indexOf","target","baseline","arrayBaseline","execute","validIds","d","success","error","JSON","stringify","join","delete","addedItemIds","idFactory","itemId","result","interactables"],"sources":["../../../src/react/client/interactable-model-context.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\nimport { toJSONSchema } from \"assistant-stream\";\nimport type { Unstable_InteractableDefinition } from \"../types/scopes/interactables\";\nimport {\n  interactableToolName,\n  shallowMergeInteractableState,\n  type Unstable_InteractableSnapshotEntry,\n} from \"../../model-context/interactable-composer-metadata\";\nimport { generateId } from \"../../utils/id\";\nimport { isRecord } from \"../../utils/json/is-json\";\n\nexport type PartialJSONSchema = ReturnType<typeof toJSONSchema>;\n\nconst ID_PROPERTY = {\n  type: \"string\" as const,\n  description:\n    \"The id of the instance to update, as shown in its state snapshot in the conversation.\",\n};\n\nconst ITEM_ID_PROPERTY = {\n  type: \"string\" as const,\n  description: \"The id of an item currently in this list.\",\n};\n\nconst hasIdProperty = (schema: Record<string, unknown>) => {\n  const properties = schema.properties;\n  return isRecord(properties) && properties.id !== undefined;\n};\n\nconst withRequiredItemId = (schema: Record<string, unknown>) => ({\n  ...schema,\n  required: [\"id\"],\n});\n\n// The framework assigns ids to new items, so the model never sees the `id`\n// field when adding (it can't address an item that doesn't exist yet).\nconst withoutItemId = (schema: Record<string, unknown>) => {\n  const { id: _omitted, ...properties } = isRecord(schema.properties)\n    ? schema.properties\n    : {};\n  const required = Array.isArray(schema.required)\n    ? schema.required.filter((key) => key !== \"id\")\n    : schema.required;\n  return { ...schema, properties, required };\n};\n\nconst toArrayUpdateSchema = (schema: unknown, field: string) => {\n  if (!isRecord(schema) || schema.type !== \"array\") return schema;\n  const itemSchema = schema.items;\n  if (Array.isArray(itemSchema) || !isRecord(itemSchema)) return schema;\n\n  const idKeyed = hasIdProperty(itemSchema);\n  const properties: Record<string, unknown> = {\n    add: {\n      type: \"array\",\n      items: idKeyed ? withoutItemId(itemSchema) : itemSchema,\n    },\n    remove: {\n      type: \"array\",\n      items: idKeyed ? ITEM_ID_PROPERTY : itemSchema,\n    },\n    clear: { type: \"boolean\" },\n  };\n\n  if (idKeyed) {\n    properties.update = {\n      type: \"array\",\n      items: withRequiredItemId(itemSchema),\n    };\n  }\n\n  return {\n    type: \"object\" as const,\n    description:\n      `Operations for array field \"${field}\". To change one item use update ` +\n      `with its id; add new items (their ids are assigned for you), remove items ` +\n      `by id, or clear to empty. Change only the items you mean to — never resend ` +\n      `the whole array to edit one item.`,\n    properties,\n    additionalProperties: false,\n  };\n};\n\n// Top-level array fields whose items carry an `id` — the fields the framework\n// mints ids for on add.\nconst idKeyedArrayFieldNames = (\n  properties: Record<string, unknown>,\n): Set<string> => {\n  const names = new Set<string>();\n  for (const [key, value] of Object.entries(properties)) {\n    if (\n      isRecord(value) &&\n      value.type === \"array\" &&\n      isRecord(value.items) &&\n      hasIdProperty(value.items)\n    ) {\n      names.add(key);\n    }\n  }\n  return names;\n};\n\nconst withArrayUpdateSchemas = (properties: Record<string, unknown>) =>\n  Object.fromEntries(\n    Object.entries(properties).map(([key, value]) => [\n      key,\n      toArrayUpdateSchema(value, key),\n    ]),\n  );\n\n/**\n * Wraps an interactable's partial state schema with the required `id`\n * parameter. Falls back to a permissive schema when the partial conversion\n * failed at registration time.\n */\nfunction withRequiredId(partial: PartialJSONSchema | undefined) {\n  if (!partial || typeof partial !== \"object\" || partial.type !== \"object\") {\n    return {\n      type: \"object\" as const,\n      properties: { id: ID_PROPERTY },\n      required: [\"id\"],\n      additionalProperties: true,\n    };\n  }\n  if (process.env.NODE_ENV !== \"production\" && partial.properties?.id) {\n    console.warn(\n      `[Interactables] a top-level \"id\" field in an interactable's stateSchema is ` +\n        `reserved for instance addressing by the update tool and cannot be updated ` +\n        `by the model. Rename the field to make it model-writable.`,\n    );\n  }\n  const { id: _reserved, ...properties } = partial.properties ?? {};\n  return {\n    ...partial,\n    properties: { id: ID_PROPERTY, ...withArrayUpdateSchemas(properties) },\n    required: [\"id\"],\n  };\n}\n\nexport function buildInteractableModelContext(\n  definitions: Record<string, Unstable_InteractableDefinition>,\n  partialSchemaCache: Map<string, PartialJSONSchema>,\n  setDefState: (id: string, updater: (prev: unknown) => unknown) => void,\n  streamBaselines = new Map<string, { targetId: string; state: unknown }>(),\n):\n  | {\n      tools: Record<string, Tool<any, any>>;\n      unstable_composerMetadata?: Record<string, unknown>;\n    }\n  | undefined {\n  const entries = Object.values(definitions);\n  if (entries.length === 0) return undefined;\n\n  const byName = new Map<string, Unstable_InteractableDefinition[]>();\n  for (const def of entries) {\n    const list = byName.get(def.name) ?? [];\n    list.push(def);\n    byName.set(def.name, list);\n  }\n\n  const tools: Record<string, Tool<any, any>> = {};\n\n  for (const [name, instances] of byName) {\n    const toolName = interactableToolName(name);\n    if (tools[toolName]) {\n      if (process.env.NODE_ENV !== \"production\") {\n        console.warn(\n          `[Interactables] interactable names \"${name}\" and another registered name ` +\n            `both sanitize to the tool name \"${toolName}\". Rename one of them.`,\n        );\n      }\n      continue;\n    }\n\n    const first = instances[0]!;\n    const partialSchema = partialSchemaCache.get(first.id);\n    const idKeyedFields =\n      partialSchema && isRecord(partialSchema.properties)\n        ? idKeyedArrayFieldNames(partialSchema.properties)\n        : new Set<string>();\n\n    // `id` resolves to a definition of this name; an id-less call is accepted\n    // only while exactly one instance exists.\n    const resolveTarget = (\n      id: unknown,\n    ): Unstable_InteractableDefinition | undefined => {\n      if (typeof id === \"string\") {\n        const def = definitions[id];\n        return def?.name === name ? def : undefined;\n      }\n      return instances.length === 1 ? first : undefined;\n    };\n\n    tools[toolName] = {\n      type: \"frontend\",\n      description:\n        `Update the state of interactable component \"${name}\". ${first.description} ` +\n        `Pass the id of the instance to update — instance ids and current state ` +\n        `appear in the conversation as state snapshots. Only include the fields ` +\n        `you want to change; omitted fields keep their current values.`,\n      parameters: withRequiredId(partialSchema),\n      streamCall: async (reader, { toolCallId }) => {\n        try {\n          for await (const partialArgs of reader.args.streamValues()) {\n            if (!partialArgs || typeof partialArgs !== \"object\") continue;\n            const args = partialArgs as Record<string, unknown>;\n            const keys = Object.keys(args);\n            const idIndex = keys.indexOf(\"id\");\n            if (idIndex === keys.length - 1) continue;\n\n            const { id, ...partial } = args;\n            if (Object.keys(partial).length === 0) continue;\n            const target = resolveTarget(id);\n            if (!target) continue;\n\n            const baseline = streamBaselines.get(toolCallId);\n            const arrayBaseline =\n              baseline?.targetId === target.id ? baseline.state : target.state;\n            if (!baseline || baseline.targetId !== target.id) {\n              streamBaselines.set(toolCallId, {\n                targetId: target.id,\n                state: target.state,\n              });\n            }\n\n            setDefState(target.id, (prev) =>\n              shallowMergeInteractableState(prev, partial, { arrayBaseline }),\n            );\n          }\n        } catch {\n          // Non-fatal: execute handles the final state\n        }\n      },\n      execute: async (args: unknown, { toolCallId }) => {\n        const { id, ...partial } = (args ?? {}) as Record<string, unknown>;\n        const target = resolveTarget(id);\n        if (!target) {\n          const validIds = instances.map((d) => d.id);\n          return {\n            success: false,\n            error: `Unknown id ${JSON.stringify(id)} for interactable \"${name}\". Valid ids: ${validIds.join(\", \")}`,\n          };\n        }\n        const baseline = streamBaselines.get(toolCallId);\n        streamBaselines.delete(toolCallId);\n        const addedItemIds: Record<string, string[]> = {};\n        setDefState(target.id, (prev) =>\n          shallowMergeInteractableState(prev, partial, {\n            arrayBaseline:\n              baseline?.targetId === target.id ? baseline.state : undefined,\n            idFactory: (field) => {\n              const itemId = generateId();\n              (addedItemIds[field] ??= []).push(itemId);\n              return itemId;\n            },\n            idKeyedFields,\n          }),\n        );\n        // The resolved id lets an id-less call's UI (and the model) address\n        // the instance that was actually updated.\n        const result: {\n          success: true;\n          id: string;\n          addedItemIds?: Record<string, string[]>;\n        } = { success: true, id: target.id };\n        if (Object.keys(addedItemIds).length > 0) {\n          result.addedItemIds = addedItemIds;\n        }\n        return result;\n      },\n    };\n  }\n\n  const interactables: Unstable_InteractableSnapshotEntry[] = entries.map(\n    (def) => ({\n      id: def.id,\n      name: def.name,\n      state: def.state,\n    }),\n  );\n  return { tools, unstable_composerMetadata: { interactables } };\n}\n"],"mappings":";;;;;AAaA,MAAMU,cAAc;CAClBC,MAAM;CACNE,aACE;AACJ;AAEA,MAAMC,mBAAmB;CACvBH,MAAM;CACNE,aAAa;AACf;AAEA,MAAME,iBAAiBC,WAAoC;CACzD,MAAME,aAAaF,OAAOE;CAC1B,OAAOX,SAASW,UAAU,KAAKA,WAAWC,OAAOC,KAAAA;AACnD;AAEA,MAAMC,sBAAsBL,YAAqC;CAC/D,GAAGA;CACHM,UAAU,CAAC,IAAI;AACjB;AAIA,MAAMC,iBAAiBP,WAAoC;CACzD,MAAM,EAAEG,IAAIK,UAAU,GAAGN,eAAeX,SAASS,OAAOE,UAAU,IAC9DF,OAAOE,aACP,CAAC;CACL,MAAMI,WAAWG,MAAMC,QAAQV,OAAOM,QAAQ,IAC1CN,OAAOM,SAASK,QAAQC,QAAQA,QAAQ,IAAI,IAC5CZ,OAAOM;CACX,OAAO;EAAE,GAAGN;EAAQE;EAAYI;CAAS;AAC3C;AAEA,MAAMO,uBAAuBb,QAAiBc,UAAkB;CAC9D,IAAI,CAACvB,SAASS,MAAM,KAAKA,OAAOL,SAAS,SAAS,OAAOK;CACzD,MAAMe,aAAaf,OAAOgB;CAC1B,IAAIP,MAAMC,QAAQK,UAAU,KAAK,CAACxB,SAASwB,UAAU,GAAG,OAAOf;CAE/D,MAAMiB,UAAUlB,cAAcgB,UAAU;CACxC,MAAMb,aAAsC;EAC1CgB,KAAK;GACHvB,MAAM;GACNqB,OAAOC,UAAUV,cAAcQ,UAAU,IAAIA;EAC/C;EACAI,QAAQ;GACNxB,MAAM;GACNqB,OAAOC,UAAUnB,mBAAmBiB;EACtC;EACAK,OAAO,EAAEzB,MAAM,UAAU;CAC3B;CAEA,IAAIsB,SACFf,WAAWmB,SAAS;EAClB1B,MAAM;EACNqB,OAAOX,mBAAmBU,UAAU;CACtC;CAGF,OAAO;EACLpB,MAAM;EACNE,aACE,+BAA+BiB,MAAK;EAItCZ;EACAoB,sBAAsB;CACxB;AACF;AAIA,MAAMC,0BACJrB,eACgB;CAChB,MAAMuB,wBAAQ,IAAID,IAAY;CAC9B,KAAK,MAAM,CAACZ,KAAKc,UAAUC,OAAOC,QAAQ1B,UAAU,GAClD,IACEX,SAASmC,KAAK,KACdA,MAAM/B,SAAS,WACfJ,SAASmC,MAAMV,KAAK,KACpBjB,cAAc2B,MAAMV,KAAK,GAEzBS,MAAMP,IAAIN,GAAG;CAGjB,OAAOa;AACT;AAEA,MAAMI,0BAA0B3B,eAC9ByB,OAAOG,YACLH,OAAOC,QAAQ1B,UAAU,CAAC,CAAC6B,KAAK,CAACnB,KAAKc,WAAW,CAC/Cd,KACAC,oBAAoBa,OAAOd,GAAG,CAAC,CAChC,CACH;;;;;;AAOF,SAASoB,eAAeC,SAAwC;CAC9D,IAAI,CAACA,WAAW,OAAOA,YAAY,YAAYA,QAAQtC,SAAS,UAC9D,OAAO;EACLA,MAAM;EACNO,YAAY,EAAEC,IAAIT,YAAY;EAC9BY,UAAU,CAAC,IAAI;EACfgB,sBAAsB;CACxB;CAEF,IAAIY,QAAQC,IAAIC,aAAa,gBAAgBH,QAAQ/B,YAAYC,IAC/DkC,QAAQC,KACN,kNAGF;CAEF,MAAM,EAAEnC,IAAIoC,WAAW,GAAGrC,eAAe+B,QAAQ/B,cAAc,CAAC;CAChE,OAAO;EACL,GAAG+B;EACH/B,YAAY;GAAEC,IAAIT;GAAa,GAAGmC,uBAAuB3B,UAAU;EAAE;EACrEI,UAAU,CAAC,IAAI;CACjB;AACF;AAEA,SAAgBkC,8BACdC,aACAC,oBACAE,aACAG,kCAAkB,IAAIJ,IAAkD,GAM5D;CACZ,MAAMf,UAAUD,OAAOyB,OAAOX,WAAW;CACzC,IAAIb,QAAQyB,WAAW,GAAG,OAAOjD,KAAAA;CAEjC,MAAMkD,yBAAS,IAAIX,IAA+C;CAClE,KAAK,MAAMY,OAAO3B,SAAS;EACzB,MAAM4B,OAAOF,OAAOG,IAAIF,IAAIG,IAAI,KAAK,CAAA;EACrCF,KAAKG,KAAKJ,GAAG;EACbD,OAAOM,IAAIL,IAAIG,MAAMF,IAAI;CAC3B;CAEA,MAAMN,QAAwC,CAAC;CAE/C,KAAK,MAAM,CAACQ,MAAMG,cAAcP,QAAQ;EACtC,MAAMQ,WAAW3E,qBAAqBuE,IAAI;EAC1C,IAAIR,MAAMY,WAAW;GACnB,IAAI5B,QAAQC,IAAIC,aAAa,cAC3BC,QAAQC,KACN,uCAAuCoB,KAAI,gEACNI,SAAQ,uBAC/C;GAEF;EACF;EAEA,MAAMC,QAAQF,UAAU;EACxB,MAAMG,gBAAgBtB,mBAAmBe,IAAIM,MAAM5D,EAAE;EACrD,MAAM8D,gBACJD,iBAAiBzE,SAASyE,cAAc9D,UAAU,IAC9CqB,uBAAuByC,cAAc9D,UAAU,oBAC/C,IAAIsB,IAAY;EAItB,MAAM0C,iBACJ/D,OACgD;GAChD,IAAI,OAAOA,OAAO,UAAU;IAC1B,MAAMoD,MAAMd,YAAYtC;IACxB,OAAOoD,KAAKG,SAASA,OAAOH,MAAMnD,KAAAA;GACpC;GACA,OAAOyD,UAAUR,WAAW,IAAIU,QAAQ3D,KAAAA;EAC1C;EAEA8C,MAAMY,YAAY;GAChBnE,MAAM;GACNE,aACE,+CAA+C6D,KAAI,KAAMK,MAAMlE,YAAW;GAI5EsE,YAAYnC,eAAegC,aAAa;GACxCI,YAAY,OAAOC,QAAQ,EAAEC,iBAAiB;IAC5C,IAAI;KACF,WAAW,MAAMC,eAAeF,OAAOG,KAAKC,aAAa,GAAG;MAC1D,IAAI,CAACF,eAAe,OAAOA,gBAAgB,UAAU;MACrD,MAAMC,OAAOD;MACb,MAAMG,OAAO/C,OAAO+C,KAAKF,IAAI;MAE7B,IADgBE,KAAKE,QAAQ,IACzBD,MAAYD,KAAKrB,SAAS,GAAG;MAEjC,MAAM,EAAElD,IAAI,GAAG8B,YAAYuC;MAC3B,IAAI7C,OAAO+C,KAAKzC,OAAO,CAAC,CAACoB,WAAW,GAAG;MACvC,MAAMwB,SAASX,cAAc/D,EAAE;MAC/B,IAAI,CAAC0E,QAAQ;MAEb,MAAMC,WAAW/B,gBAAgBU,IAAIa,UAAU;MAC/C,MAAMS,gBACJD,UAAU9B,aAAa6B,OAAO1E,KAAK2E,SAAS7B,QAAQ4B,OAAO5B;MAC7D,IAAI,CAAC6B,YAAYA,SAAS9B,aAAa6B,OAAO1E,IAC5C4C,gBAAgBa,IAAIU,YAAY;OAC9BtB,UAAU6B,OAAO1E;OACjB8C,OAAO4B,OAAO5B;MAChB,CAAC;MAGHL,YAAYiC,OAAO1E,KAAK2C,SACtB1D,8BAA8B0D,MAAMb,SAAS,EAAE8C,cAAc,CAAC,CAChE;KACF;IACF,QAAQ,CACN;GAEJ;GACAC,SAAS,OAAOR,MAAe,EAAEF,iBAAiB;IAChD,MAAM,EAAEnE,IAAI,GAAG8B,YAAauC,QAAQ,CAAC;IACrC,MAAMK,SAASX,cAAc/D,EAAE;IAC/B,IAAI,CAAC0E,QAAQ;KACX,MAAMI,WAAWpB,UAAU9B,KAAKmD,MAAMA,EAAE/E,EAAE;KAC1C,OAAO;MACLgF,SAAS;MACTC,OAAO,cAAcC,KAAKC,UAAUnF,EAAE,EAAC,qBAAsBuD,KAAI,gBAAiBuB,SAASM,KAAK,IAAI;KACtG;IACF;IACA,MAAMT,WAAW/B,gBAAgBU,IAAIa,UAAU;IAC/CvB,gBAAgByC,OAAOlB,UAAU;IACjC,MAAMmB,eAAyC,CAAC;IAChD7C,YAAYiC,OAAO1E,KAAK2C,SACtB1D,8BAA8B0D,MAAMb,SAAS;KAC3C8C,eACED,UAAU9B,aAAa6B,OAAO1E,KAAK2E,SAAS7B,QAAQ7C,KAAAA;KACtDsF,YAAY5E,UAAU;MACpB,MAAM6E,SAASrG,WAAW;MAC1B,CAACmG,aAAa3E,WAAW,CAAA,EAAA,CAAI6C,KAAKgC,MAAM;MACxC,OAAOA;KACT;KACA1B;IACF,CAAC,CACH;IAGA,MAAM2B,SAIF;KAAET,SAAS;KAAMhF,IAAI0E,OAAO1E;IAAG;IACnC,IAAIwB,OAAO+C,KAAKe,YAAY,CAAC,CAACpC,SAAS,GACrCuC,OAAOH,eAAeA;IAExB,OAAOG;GACT;EACF;CACF;CASA,OAAO;EAAE1C;EAAOC,2BAA2B,EAAE0C,eAPejE,QAAQG,KACjEwB,SAAS;GACRpD,IAAIoD,IAAIpD;GACRuD,MAAMH,IAAIG;GACVT,OAAOM,IAAIN;EACb,EAE2C4C,EAAc;CAAE;AAC/D"}
+{"version":3,"file":"interactable-model-context.js","names":["Tool","toJSONSchema","Unstable_InteractableDefinition","interactableToolName","shallowMergeInteractableState","Unstable_InteractableSnapshotEntry","generateId","isRecord","nullProtoRecord","PartialJSONSchema","ReturnType","ID_PROPERTY","type","const","description","ITEM_ID_PROPERTY","hasIdProperty","schema","Record","properties","id","undefined","withRequiredItemId","required","withoutItemId","_omitted","Array","isArray","filter","key","toArrayUpdateSchema","field","itemSchema","items","idKeyed","add","remove","clear","update","additionalProperties","idKeyedArrayFieldNames","Set","names","value","Object","entries","withArrayUpdateSchemas","fromEntries","map","withRequiredId","partial","process","env","NODE_ENV","console","warn","_reserved","buildInteractableModelContext","definitions","partialSchemaCache","Map","setDefState","updater","prev","streamBaselines","targetId","state","tools","unstable_composerMetadata","values","length","byName","def","list","get","name","push","set","instances","toolName","first","partialSchema","idKeyedFields","resolveTarget","parameters","streamCall","reader","toolCallId","partialArgs","args","streamValues","keys","idIndex","indexOf","target","baseline","arrayBaseline","execute","validIds","d","success","error","JSON","stringify","join","delete","addedItemIds","idFactory","itemId","result","interactables"],"sources":["../../../src/react/client/interactable-model-context.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\nimport { toJSONSchema } from \"assistant-stream\";\nimport type { Unstable_InteractableDefinition } from \"../types/scopes/interactables\";\nimport {\n  interactableToolName,\n  shallowMergeInteractableState,\n  type Unstable_InteractableSnapshotEntry,\n} from \"../../model-context/interactable-composer-metadata\";\nimport { generateId } from \"../../utils/id\";\nimport { isRecord } from \"../../utils/json/is-json\";\nimport { nullProtoRecord } from \"../../utils/record\";\n\nexport type PartialJSONSchema = ReturnType<typeof toJSONSchema>;\n\nconst ID_PROPERTY = {\n  type: \"string\" as const,\n  description:\n    \"The id of the instance to update, as shown in its state snapshot in the conversation.\",\n};\n\nconst ITEM_ID_PROPERTY = {\n  type: \"string\" as const,\n  description: \"The id of an item currently in this list.\",\n};\n\nconst hasIdProperty = (schema: Record<string, unknown>) => {\n  const properties = schema.properties;\n  return isRecord(properties) && properties.id !== undefined;\n};\n\nconst withRequiredItemId = (schema: Record<string, unknown>) => ({\n  ...schema,\n  required: [\"id\"],\n});\n\n// The framework assigns ids to new items, so the model never sees the `id`\n// field when adding (it can't address an item that doesn't exist yet).\nconst withoutItemId = (schema: Record<string, unknown>) => {\n  const { id: _omitted, ...properties } = isRecord(schema.properties)\n    ? schema.properties\n    : {};\n  const required = Array.isArray(schema.required)\n    ? schema.required.filter((key) => key !== \"id\")\n    : schema.required;\n  return { ...schema, properties, required };\n};\n\nconst toArrayUpdateSchema = (schema: unknown, field: string) => {\n  if (!isRecord(schema) || schema.type !== \"array\") return schema;\n  const itemSchema = schema.items;\n  if (Array.isArray(itemSchema) || !isRecord(itemSchema)) return schema;\n\n  const idKeyed = hasIdProperty(itemSchema);\n  const properties: Record<string, unknown> = {\n    add: {\n      type: \"array\",\n      items: idKeyed ? withoutItemId(itemSchema) : itemSchema,\n    },\n    remove: {\n      type: \"array\",\n      items: idKeyed ? ITEM_ID_PROPERTY : itemSchema,\n    },\n    clear: { type: \"boolean\" },\n  };\n\n  if (idKeyed) {\n    properties.update = {\n      type: \"array\",\n      items: withRequiredItemId(itemSchema),\n    };\n  }\n\n  return {\n    type: \"object\" as const,\n    description:\n      `Operations for array field \"${field}\". To change one item use update ` +\n      `with its id; add new items (their ids are assigned for you), remove items ` +\n      `by id, or clear to empty. Change only the items you mean to — never resend ` +\n      `the whole array to edit one item.`,\n    properties,\n    additionalProperties: false,\n  };\n};\n\n// Top-level array fields whose items carry an `id` — the fields the framework\n// mints ids for on add.\nconst idKeyedArrayFieldNames = (\n  properties: Record<string, unknown>,\n): Set<string> => {\n  const names = new Set<string>();\n  for (const [key, value] of Object.entries(properties)) {\n    if (\n      isRecord(value) &&\n      value.type === \"array\" &&\n      isRecord(value.items) &&\n      hasIdProperty(value.items)\n    ) {\n      names.add(key);\n    }\n  }\n  return names;\n};\n\nconst withArrayUpdateSchemas = (properties: Record<string, unknown>) =>\n  Object.fromEntries(\n    Object.entries(properties).map(([key, value]) => [\n      key,\n      toArrayUpdateSchema(value, key),\n    ]),\n  );\n\n/**\n * Wraps an interactable's partial state schema with the required `id`\n * parameter. Falls back to a permissive schema when the partial conversion\n * failed at registration time.\n */\nfunction withRequiredId(partial: PartialJSONSchema | undefined) {\n  if (!partial || typeof partial !== \"object\" || partial.type !== \"object\") {\n    return {\n      type: \"object\" as const,\n      properties: { id: ID_PROPERTY },\n      required: [\"id\"],\n      additionalProperties: true,\n    };\n  }\n  if (process.env.NODE_ENV !== \"production\" && partial.properties?.id) {\n    console.warn(\n      `[Interactables] a top-level \"id\" field in an interactable's stateSchema is ` +\n        `reserved for instance addressing by the update tool and cannot be updated ` +\n        `by the model. Rename the field to make it model-writable.`,\n    );\n  }\n  const { id: _reserved, ...properties } = partial.properties ?? {};\n  return {\n    ...partial,\n    properties: { id: ID_PROPERTY, ...withArrayUpdateSchemas(properties) },\n    required: [\"id\"],\n  };\n}\n\nexport function buildInteractableModelContext(\n  definitions: Record<string, Unstable_InteractableDefinition>,\n  partialSchemaCache: Map<string, PartialJSONSchema>,\n  setDefState: (id: string, updater: (prev: unknown) => unknown) => void,\n  streamBaselines = new Map<string, { targetId: string; state: unknown }>(),\n):\n  | {\n      tools: Record<string, Tool<any, any>>;\n      unstable_composerMetadata?: Record<string, unknown>;\n    }\n  | undefined {\n  const entries = Object.values(definitions);\n  if (entries.length === 0) return undefined;\n\n  const byName = new Map<string, Unstable_InteractableDefinition[]>();\n  for (const def of entries) {\n    const list = byName.get(def.name) ?? [];\n    list.push(def);\n    byName.set(def.name, list);\n  }\n\n  const tools: Record<string, Tool<any, any>> = {};\n\n  for (const [name, instances] of byName) {\n    const toolName = interactableToolName(name);\n    if (tools[toolName]) {\n      if (process.env.NODE_ENV !== \"production\") {\n        console.warn(\n          `[Interactables] interactable names \"${name}\" and another registered name ` +\n            `both sanitize to the tool name \"${toolName}\". Rename one of them.`,\n        );\n      }\n      continue;\n    }\n\n    const first = instances[0]!;\n    const partialSchema = partialSchemaCache.get(first.id);\n    const idKeyedFields =\n      partialSchema && isRecord(partialSchema.properties)\n        ? idKeyedArrayFieldNames(partialSchema.properties)\n        : new Set<string>();\n\n    // `id` resolves to a definition of this name; an id-less call is accepted\n    // only while exactly one instance exists.\n    const resolveTarget = (\n      id: unknown,\n    ): Unstable_InteractableDefinition | undefined => {\n      if (typeof id === \"string\") {\n        const def = definitions[id];\n        return def?.name === name ? def : undefined;\n      }\n      return instances.length === 1 ? first : undefined;\n    };\n\n    tools[toolName] = {\n      type: \"frontend\",\n      description:\n        `Update the state of interactable component \"${name}\". ${first.description} ` +\n        `Pass the id of the instance to update — instance ids and current state ` +\n        `appear in the conversation as state snapshots. Only include the fields ` +\n        `you want to change; omitted fields keep their current values.`,\n      parameters: withRequiredId(partialSchema),\n      streamCall: async (reader, { toolCallId }) => {\n        try {\n          for await (const partialArgs of reader.args.streamValues()) {\n            if (!partialArgs || typeof partialArgs !== \"object\") continue;\n            const args = partialArgs as Record<string, unknown>;\n            const keys = Object.keys(args);\n            const idIndex = keys.indexOf(\"id\");\n            if (idIndex === keys.length - 1) continue;\n\n            const { id, ...partial } = args;\n            if (Object.keys(partial).length === 0) continue;\n            const target = resolveTarget(id);\n            if (!target) continue;\n\n            const baseline = streamBaselines.get(toolCallId);\n            const arrayBaseline =\n              baseline?.targetId === target.id ? baseline.state : target.state;\n            if (!baseline || baseline.targetId !== target.id) {\n              streamBaselines.set(toolCallId, {\n                targetId: target.id,\n                state: target.state,\n              });\n            }\n\n            setDefState(target.id, (prev) =>\n              shallowMergeInteractableState(prev, partial, { arrayBaseline }),\n            );\n          }\n        } catch {\n          // Non-fatal: execute handles the final state\n        }\n      },\n      execute: async (args: unknown, { toolCallId }) => {\n        const { id, ...partial } = (args ?? {}) as Record<string, unknown>;\n        const target = resolveTarget(id);\n        if (!target) {\n          const validIds = instances.map((d) => d.id);\n          return {\n            success: false,\n            error: `Unknown id ${JSON.stringify(id)} for interactable \"${name}\". Valid ids: ${validIds.join(\", \")}`,\n          };\n        }\n        const baseline = streamBaselines.get(toolCallId);\n        streamBaselines.delete(toolCallId);\n        const addedItemIds = nullProtoRecord<string[]>();\n        setDefState(target.id, (prev) =>\n          shallowMergeInteractableState(prev, partial, {\n            arrayBaseline:\n              baseline?.targetId === target.id ? baseline.state : undefined,\n            idFactory: (field) => {\n              const itemId = generateId();\n              (addedItemIds[field] ??= []).push(itemId);\n              return itemId;\n            },\n            idKeyedFields,\n          }),\n        );\n        // The resolved id lets an id-less call's UI (and the model) address\n        // the instance that was actually updated.\n        const result: {\n          success: true;\n          id: string;\n          addedItemIds?: Record<string, string[]>;\n        } = { success: true, id: target.id };\n        if (Object.keys(addedItemIds).length > 0) {\n          result.addedItemIds = { ...addedItemIds };\n        }\n        return result;\n      },\n    };\n  }\n\n  const interactables: Unstable_InteractableSnapshotEntry[] = entries.map(\n    (def) => ({\n      id: def.id,\n      name: def.name,\n      state: def.state,\n    }),\n  );\n  return { tools, unstable_composerMetadata: { interactables } };\n}\n"],"mappings":";;;;;;AAcA,MAAMW,cAAc;CAClBC,MAAM;CACNE,aACE;AACJ;AAEA,MAAMC,mBAAmB;CACvBH,MAAM;CACNE,aAAa;AACf;AAEA,MAAME,iBAAiBC,WAAoC;CACzD,MAAME,aAAaF,OAAOE;CAC1B,OAAOZ,SAASY,UAAU,KAAKA,WAAWC,OAAOC,KAAAA;AACnD;AAEA,MAAMC,sBAAsBL,YAAqC;CAC/D,GAAGA;CACHM,UAAU,CAAC,IAAI;AACjB;AAIA,MAAMC,iBAAiBP,WAAoC;CACzD,MAAM,EAAEG,IAAIK,UAAU,GAAGN,eAAeZ,SAASU,OAAOE,UAAU,IAC9DF,OAAOE,aACP,CAAC;CACL,MAAMI,WAAWG,MAAMC,QAAQV,OAAOM,QAAQ,IAC1CN,OAAOM,SAASK,QAAQC,QAAQA,QAAQ,IAAI,IAC5CZ,OAAOM;CACX,OAAO;EAAE,GAAGN;EAAQE;EAAYI;CAAS;AAC3C;AAEA,MAAMO,uBAAuBb,QAAiBc,UAAkB;CAC9D,IAAI,CAACxB,SAASU,MAAM,KAAKA,OAAOL,SAAS,SAAS,OAAOK;CACzD,MAAMe,aAAaf,OAAOgB;CAC1B,IAAIP,MAAMC,QAAQK,UAAU,KAAK,CAACzB,SAASyB,UAAU,GAAG,OAAOf;CAE/D,MAAMiB,UAAUlB,cAAcgB,UAAU;CACxC,MAAMb,aAAsC;EAC1CgB,KAAK;GACHvB,MAAM;GACNqB,OAAOC,UAAUV,cAAcQ,UAAU,IAAIA;EAC/C;EACAI,QAAQ;GACNxB,MAAM;GACNqB,OAAOC,UAAUnB,mBAAmBiB;EACtC;EACAK,OAAO,EAAEzB,MAAM,UAAU;CAC3B;CAEA,IAAIsB,SACFf,WAAWmB,SAAS;EAClB1B,MAAM;EACNqB,OAAOX,mBAAmBU,UAAU;CACtC;CAGF,OAAO;EACLpB,MAAM;EACNE,aACE,+BAA+BiB,MAAK;EAItCZ;EACAoB,sBAAsB;CACxB;AACF;AAIA,MAAMC,0BACJrB,eACgB;CAChB,MAAMuB,wBAAQ,IAAID,IAAY;CAC9B,KAAK,MAAM,CAACZ,KAAKc,UAAUC,OAAOC,QAAQ1B,UAAU,GAClD,IACEZ,SAASoC,KAAK,KACdA,MAAM/B,SAAS,WACfL,SAASoC,MAAMV,KAAK,KACpBjB,cAAc2B,MAAMV,KAAK,GAEzBS,MAAMP,IAAIN,GAAG;CAGjB,OAAOa;AACT;AAEA,MAAMI,0BAA0B3B,eAC9ByB,OAAOG,YACLH,OAAOC,QAAQ1B,UAAU,CAAC,CAAC6B,KAAK,CAACnB,KAAKc,WAAW,CAC/Cd,KACAC,oBAAoBa,OAAOd,GAAG,CAAC,CAChC,CACH;;;;;;AAOF,SAASoB,eAAeC,SAAwC;CAC9D,IAAI,CAACA,WAAW,OAAOA,YAAY,YAAYA,QAAQtC,SAAS,UAC9D,OAAO;EACLA,MAAM;EACNO,YAAY,EAAEC,IAAIT,YAAY;EAC9BY,UAAU,CAAC,IAAI;EACfgB,sBAAsB;CACxB;CAEF,IAAIY,QAAQC,IAAIC,aAAa,gBAAgBH,QAAQ/B,YAAYC,IAC/DkC,QAAQC,KACN,kNAGF;CAEF,MAAM,EAAEnC,IAAIoC,WAAW,GAAGrC,eAAe+B,QAAQ/B,cAAc,CAAC;CAChE,OAAO;EACL,GAAG+B;EACH/B,YAAY;GAAEC,IAAIT;GAAa,GAAGmC,uBAAuB3B,UAAU;EAAE;EACrEI,UAAU,CAAC,IAAI;CACjB;AACF;AAEA,SAAgBkC,8BACdC,aACAC,oBACAE,aACAG,kCAAkB,IAAIJ,IAAkD,GAM5D;CACZ,MAAMf,UAAUD,OAAOyB,OAAOX,WAAW;CACzC,IAAIb,QAAQyB,WAAW,GAAG,OAAOjD,KAAAA;CAEjC,MAAMkD,yBAAS,IAAIX,IAA+C;CAClE,KAAK,MAAMY,OAAO3B,SAAS;EACzB,MAAM4B,OAAOF,OAAOG,IAAIF,IAAIG,IAAI,KAAK,CAAA;EACrCF,KAAKG,KAAKJ,GAAG;EACbD,OAAOM,IAAIL,IAAIG,MAAMF,IAAI;CAC3B;CAEA,MAAMN,QAAwC,CAAC;CAE/C,KAAK,MAAM,CAACQ,MAAMG,cAAcP,QAAQ;EACtC,MAAMQ,WAAW5E,qBAAqBwE,IAAI;EAC1C,IAAIR,MAAMY,WAAW;GACnB,IAAI5B,QAAQC,IAAIC,aAAa,cAC3BC,QAAQC,KACN,uCAAuCoB,KAAI,gEACNI,SAAQ,uBAC/C;GAEF;EACF;EAEA,MAAMC,QAAQF,UAAU;EACxB,MAAMG,gBAAgBtB,mBAAmBe,IAAIM,MAAM5D,EAAE;EACrD,MAAM8D,gBACJD,iBAAiB1E,SAAS0E,cAAc9D,UAAU,IAC9CqB,uBAAuByC,cAAc9D,UAAU,oBAC/C,IAAIsB,IAAY;EAItB,MAAM0C,iBACJ/D,OACgD;GAChD,IAAI,OAAOA,OAAO,UAAU;IAC1B,MAAMoD,MAAMd,YAAYtC;IACxB,OAAOoD,KAAKG,SAASA,OAAOH,MAAMnD,KAAAA;GACpC;GACA,OAAOyD,UAAUR,WAAW,IAAIU,QAAQ3D,KAAAA;EAC1C;EAEA8C,MAAMY,YAAY;GAChBnE,MAAM;GACNE,aACE,+CAA+C6D,KAAI,KAAMK,MAAMlE,YAAW;GAI5EsE,YAAYnC,eAAegC,aAAa;GACxCI,YAAY,OAAOC,QAAQ,EAAEC,iBAAiB;IAC5C,IAAI;KACF,WAAW,MAAMC,eAAeF,OAAOG,KAAKC,aAAa,GAAG;MAC1D,IAAI,CAACF,eAAe,OAAOA,gBAAgB,UAAU;MACrD,MAAMC,OAAOD;MACb,MAAMG,OAAO/C,OAAO+C,KAAKF,IAAI;MAE7B,IADgBE,KAAKE,QAAQ,IACzBD,MAAYD,KAAKrB,SAAS,GAAG;MAEjC,MAAM,EAAElD,IAAI,GAAG8B,YAAYuC;MAC3B,IAAI7C,OAAO+C,KAAKzC,OAAO,CAAC,CAACoB,WAAW,GAAG;MACvC,MAAMwB,SAASX,cAAc/D,EAAE;MAC/B,IAAI,CAAC0E,QAAQ;MAEb,MAAMC,WAAW/B,gBAAgBU,IAAIa,UAAU;MAC/C,MAAMS,gBACJD,UAAU9B,aAAa6B,OAAO1E,KAAK2E,SAAS7B,QAAQ4B,OAAO5B;MAC7D,IAAI,CAAC6B,YAAYA,SAAS9B,aAAa6B,OAAO1E,IAC5C4C,gBAAgBa,IAAIU,YAAY;OAC9BtB,UAAU6B,OAAO1E;OACjB8C,OAAO4B,OAAO5B;MAChB,CAAC;MAGHL,YAAYiC,OAAO1E,KAAK2C,SACtB3D,8BAA8B2D,MAAMb,SAAS,EAAE8C,cAAc,CAAC,CAChE;KACF;IACF,QAAQ,CACN;GAEJ;GACAC,SAAS,OAAOR,MAAe,EAAEF,iBAAiB;IAChD,MAAM,EAAEnE,IAAI,GAAG8B,YAAauC,QAAQ,CAAC;IACrC,MAAMK,SAASX,cAAc/D,EAAE;IAC/B,IAAI,CAAC0E,QAAQ;KACX,MAAMI,WAAWpB,UAAU9B,KAAKmD,MAAMA,EAAE/E,EAAE;KAC1C,OAAO;MACLgF,SAAS;MACTC,OAAO,cAAcC,KAAKC,UAAUnF,EAAE,EAAC,qBAAsBuD,KAAI,gBAAiBuB,SAASM,KAAK,IAAI;KACtG;IACF;IACA,MAAMT,WAAW/B,gBAAgBU,IAAIa,UAAU;IAC/CvB,gBAAgByC,OAAOlB,UAAU;IACjC,MAAMmB,eAAelG,gBAA0B;IAC/CqD,YAAYiC,OAAO1E,KAAK2C,SACtB3D,8BAA8B2D,MAAMb,SAAS;KAC3C8C,eACED,UAAU9B,aAAa6B,OAAO1E,KAAK2E,SAAS7B,QAAQ7C,KAAAA;KACtDsF,YAAY5E,UAAU;MACpB,MAAM6E,SAAStG,WAAW;MAC1B,CAACoG,aAAa3E,WAAW,CAAA,EAAA,CAAI6C,KAAKgC,MAAM;MACxC,OAAOA;KACT;KACA1B;IACF,CAAC,CACH;IAGA,MAAM2B,SAIF;KAAET,SAAS;KAAMhF,IAAI0E,OAAO1E;IAAG;IACnC,IAAIwB,OAAO+C,KAAKe,YAAY,CAAC,CAACpC,SAAS,GACrCuC,OAAOH,eAAe,EAAE,GAAGA,aAAa;IAE1C,OAAOG;GACT;EACF;CACF;CASA,OAAO;EAAE1C;EAAOC,2BAA2B,EAAE0C,eAPejE,QAAQG,KACjEwB,SAAS;GACRpD,IAAIoD,IAAIpD;GACRuD,MAAMH,IAAIG;GACVT,OAAOM,IAAIN;EACb,EAE2C4C,EAAc;CAAE;AAC/D"}

package/dist/react/index.d.ts

@@ -5,7 +5,7 @@ import { PartState } from "../store/scopes/part.js";
 import { RuntimeAdapter } from "./RuntimeAdapter.js";
 import { RuntimeAdapters } from "../runtimes/remote-thread-list/types.js";
 import { JoinStrategy } from "../runtime/utils/external-message-conversion.js";
-import { convertExternalMessages, useExternalMessageConverter } from "./runtimes/external-message-converter.js";
+import { ExternalMessageConversionCache, convertExternalMessages, createExternalMessageConversionCache, useExternalMessageConverter } from "./runtimes/external-message-converter.js";
 import { StreamingTimingAccessors, StreamingTimingOptions, StreamingTimingState } from "../runtime/utils/streaming-timing.js";
 import { useStreamingTiming } from "./runtimes/useStreamingTiming.js";
 import { DataMessagePartComponent, DataMessagePartProps, EmptyMessagePartComponent, EmptyMessagePartProps, FileMessagePartComponent, FileMessagePartProps, GenerativeUIComponentRegistry, GenerativeUIMessagePartComponent, GenerativeUIMessagePartProps, GenerativeUIRenderProps, ImageMessagePartComponent, ImageMessagePartProps, QuoteMessagePartComponent, QuoteMessagePartProps, ReasoningGroupComponent, ReasoningGroupProps, ReasoningMessagePartComponent, ReasoningMessagePartProps, SourceMessagePartComponent, SourceMessagePartProps, TextMessagePartComponent, TextMessagePartProps, ToolCallMessagePartComponent, ToolCallMessagePartProps, Unstable_AudioMessagePartComponent, Unstable_AudioMessagePartProps } from "./types/MessagePartComponentTypes.js";
@@ -119,4 +119,4 @@ import { useEditComposerCancel } from "./primitive-hooks/useEditComposerCancel.j
 import { useEditComposerSend } from "./primitive-hooks/useEditComposerSend.js";
 import { useMessageError } from "./primitive-hooks/useMessageError.js";
 import { LocalRuntimeOptions, splitLocalRuntimeOptions, useLocalRuntime } from "./runtimes/useLocalRuntime.js";
-export { type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantInteractableProps, AssistantProviderBase, type AssistantProviderBaseProps, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportCommand, type AssistantTransportConnectionMetadata, type AssistantTransportOptions, type AssistantTransportProtocol, type AsyncStorageLike, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, type CloudThreadListAdapterOptions, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, type DataRenderersClientSchema, type DataRenderersMethods, type DataRenderersState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, type FileMessagePartComponent, type FileMessagePartProps, type GenerativeUIComponentRegistry, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GroupByContext, type ImageMessagePartComponent, type ImageMessagePartProps, type InteractableDefinition, type InteractablePersistedState, type InteractablePersistenceAdapter, type InteractablePersistenceStatus, type InteractableRegistration, type InteractableStateSchema, Interactables, type InteractablesClientSchema, type InteractablesMethods, type InteractablesState, type JoinStrategy, type LocalRuntimeOptions, type McpAppResourceOutput, type McpToolkitDefinition, type McpToolkitEntry, type McpToolkitToolConfig, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, type PartState, type ProviderToolConfig, QueueItemByIndexProvider, type QueueItemByIndexProviderProps, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, type RuntimeAdapters, type RuntimeExtras, type SendCommandsRequestBody, type SourceMessagePartComponent, type SourceMessagePartProps, type StreamingTimingAccessors, type StreamingTimingOptions, type StreamingTimingState, SuggestionByIndexProvider, type SuggestionByIndexProviderProps, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, ThreadPrimitiveUnstable_MessageById, type TitleGenerationAdapter, type ToolArgsStatus, type ToolCallMessagePartComponent, type ToolCallMessagePartProps, type ToolCallText, type ToolDefinition, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type ToolsClientSchema, type ToolsMethods, type ToolsState, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type Unstable_InferInteractableState, type Unstable_InteractableConfig, type Unstable_InteractableDefinition, type Unstable_InteractablePersistedState, type Unstable_InteractablePersistenceAdapter, type Unstable_InteractablePersistenceStatus, type Unstable_InteractableRegistration, type Unstable_InteractableStateSchema, type Unstable_InteractableToolConfig, type Unstable_InteractableToolRenderProps, type Unstable_InteractableVersionInfo, type Unstable_InteractablesClientSchema, type Unstable_InteractablesConfig, type Unstable_InteractablesMethods, type Unstable_InteractablesState, type UseActionBarCopyOptions, type UseComposerIfProps, type UseSuggestionTriggerOptions, convertExternalMessages, createCloudThreadListAdapter, createLocalStorageAdapter, createMessageConverter, createRuntimeExtras, createSimpleTitleAdapter, defineMcpToolkit, defineToolkit, externalTool, getMessageQuote, getRenderComponent, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, providerTool, splitLocalRuntimeOptions, stubTool, unstable_Interactables, unstable_createRuntimeExtrasFromBrand, unstable_interactableTool, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useThreadMessageIds, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAuiToolOverrides, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useCloudThreadListRuntime, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useStreamingTiming, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };
+export { type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantInteractableProps, AssistantProviderBase, type AssistantProviderBaseProps, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportCommand, type AssistantTransportConnectionMetadata, type AssistantTransportOptions, type AssistantTransportProtocol, type AsyncStorageLike, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, type CloudThreadListAdapterOptions, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, type DataRenderersClientSchema, type DataRenderersMethods, type DataRenderersState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, type ExternalMessageConversionCache, type FileMessagePartComponent, type FileMessagePartProps, type GenerativeUIComponentRegistry, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GroupByContext, type ImageMessagePartComponent, type ImageMessagePartProps, type InteractableDefinition, type InteractablePersistedState, type InteractablePersistenceAdapter, type InteractablePersistenceStatus, type InteractableRegistration, type InteractableStateSchema, Interactables, type InteractablesClientSchema, type InteractablesMethods, type InteractablesState, type JoinStrategy, type LocalRuntimeOptions, type McpAppResourceOutput, type McpToolkitDefinition, type McpToolkitEntry, type McpToolkitToolConfig, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, type PartState, type ProviderToolConfig, QueueItemByIndexProvider, type QueueItemByIndexProviderProps, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, type RuntimeAdapters, type RuntimeExtras, type SendCommandsRequestBody, type SourceMessagePartComponent, type SourceMessagePartProps, type StreamingTimingAccessors, type StreamingTimingOptions, type StreamingTimingState, SuggestionByIndexProvider, type SuggestionByIndexProviderProps, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, ThreadPrimitiveUnstable_MessageById, type TitleGenerationAdapter, type ToolArgsStatus, type ToolCallMessagePartComponent, type ToolCallMessagePartProps, type ToolCallText, type ToolDefinition, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type ToolsClientSchema, type ToolsMethods, type ToolsState, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type Unstable_InferInteractableState, type Unstable_InteractableConfig, type Unstable_InteractableDefinition, type Unstable_InteractablePersistedState, type Unstable_InteractablePersistenceAdapter, type Unstable_InteractablePersistenceStatus, type Unstable_InteractableRegistration, type Unstable_InteractableStateSchema, type Unstable_InteractableToolConfig, type Unstable_InteractableToolRenderProps, type Unstable_InteractableVersionInfo, type Unstable_InteractablesClientSchema, type Unstable_InteractablesConfig, type Unstable_InteractablesMethods, type Unstable_InteractablesState, type UseActionBarCopyOptions, type UseComposerIfProps, type UseSuggestionTriggerOptions, convertExternalMessages, createCloudThreadListAdapter, createExternalMessageConversionCache, createLocalStorageAdapter, createMessageConverter, createRuntimeExtras, createSimpleTitleAdapter, defineMcpToolkit, defineToolkit, externalTool, getMessageQuote, getRenderComponent, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, providerTool, splitLocalRuntimeOptions, stubTool, unstable_Interactables, unstable_createRuntimeExtrasFromBrand, unstable_interactableTool, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useThreadMessageIds, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAuiToolOverrides, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useCloudThreadListRuntime, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useStreamingTiming, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/react/index.js

@@ -27,7 +27,7 @@ import { unstable_useInteractableVersions } from "./model-context/useInteractabl
 import { unstable_interactableTool } from "./model-context/interactableTool.js";
 import { useToolArgsStatus } from "./model-context/useToolArgsStatus.js";
 import { useRuntimeAdapters } from "./runtimes/useRuntimeAdapters.js";
-import { convertExternalMessages, useExternalMessageConverter } from "./runtimes/external-message-converter.js";
+import { convertExternalMessages, createExternalMessageConversionCache, useExternalMessageConverter } from "./runtimes/external-message-converter.js";
 import { useStreamingTiming } from "./runtimes/useStreamingTiming.js";
 import { Interactables } from "./interactables-legacy/Interactables.js";
 import { unstable_Interactables } from "./client/Interactables.js";
@@ -105,4 +105,4 @@ import { useEditComposerCancel } from "./primitive-hooks/useEditComposerCancel.j
 import { useEditComposerSend } from "./primitive-hooks/useEditComposerSend.js";
 import { useMessageError } from "./primitive-hooks/useMessageError.js";
 import { splitLocalRuntimeOptions, useLocalRuntime } from "./runtimes/useLocalRuntime.js";
-export { AssistantProviderBase, AssistantRuntimeProvider, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, QueueItemByIndexProvider, ReadonlyThreadProvider, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, SuggestionByIndexProvider, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, ThreadPrimitiveUnstable_MessageById, Tools, convertExternalMessages, createCloudThreadListAdapter, createLocalStorageAdapter, createMessageConverter, createRuntimeExtras, createSimpleTitleAdapter, defineMcpToolkit, defineToolkit, externalTool, getMessageQuote, getRenderComponent, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, providerTool, splitLocalRuntimeOptions, stubTool, unstable_Interactables, unstable_createRuntimeExtrasFromBrand, unstable_interactableTool, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useThreadMessageIds, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAuiToolOverrides, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useCloudThreadListRuntime, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useStreamingTiming, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };
+export { AssistantProviderBase, AssistantRuntimeProvider, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, QueueItemByIndexProvider, ReadonlyThreadProvider, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, SuggestionByIndexProvider, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, ThreadPrimitiveUnstable_MessageById, Tools, convertExternalMessages, createCloudThreadListAdapter, createExternalMessageConversionCache, createLocalStorageAdapter, createMessageConverter, createRuntimeExtras, createSimpleTitleAdapter, defineMcpToolkit, defineToolkit, externalTool, getMessageQuote, getRenderComponent, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, providerTool, splitLocalRuntimeOptions, stubTool, unstable_Interactables, unstable_createRuntimeExtrasFromBrand, unstable_interactableTool, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useThreadMessageIds, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAuiToolOverrides, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useCloudThreadListRuntime, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useStreamingTiming, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/react/interactables-legacy/useAssistantInteractable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantInteractable.d.ts","names":[],"sources":["../../../src/react/interactables-legacy/useAssistantInteractable.ts"],"mappings":";;;;;;;KASY;EACV;EACA,aAAa;EACb;EACA;EACA;;;;;;;;;;;;;;cAeW,2BACX,cACA,QAAQ"}
+{"version":3,"file":"useAssistantInteractable.d.ts","names":[],"sources":["../../../src/react/interactables-legacy/useAssistantInteractable.ts"],"mappings":";;;;;;;KAUY;EACV;EACA,aAAa;EACb;EACA;EACA;;;;;;;;;;;;;;cAeW,2BACX,cACA,QAAQ"}

package/dist/react/interactables-legacy/useAssistantInteractable.js

@@ -1,3 +1,4 @@
+import { useJSONSchemaDependency } from "../utils/useJSONSchemaDependency.js";
 import { useEffect, useId, useRef } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 //#region src/react/interactables-legacy/useAssistantInteractable.ts
@@ -19,6 +20,7 @@ const useAssistantInteractable = (name, config) => {
 	const id = config.id ?? autoId;
 	const stateSchemaRef = useRef(config.stateSchema);
 	stateSchemaRef.current = config.stateSchema;
+	const normalizedStateSchema = useJSONSchemaDependency(config.stateSchema);
 	const initialStateRef = useRef(config.initialState);
 	initialStateRef.current = config.initialState;
 	useEffect(() => {
@@ -35,6 +37,7 @@ const useAssistantInteractable = (name, config) => {
 		id,
 		name,
 		config.description,
+		normalizedStateSchema,
 		config.selected
 	]);
 	return id;

package/dist/react/interactables-legacy/useAssistantInteractable.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantInteractable.js","names":["useEffect","useId","useRef","useAui","InteractableStateSchema","AssistantInteractableProps","description","stateSchema","initialState","id","selected","useAssistantInteractable","name","config","aui","autoId","replace","stateSchemaRef","current","initialStateRef","interactables","register"],"sources":["../../../src/react/interactables-legacy/useAssistantInteractable.ts"],"sourcesContent":["import { useEffect, useId, useRef } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type { InteractableStateSchema } from \"./scopes\";\n\n/**\n * @deprecated Since 2026-06-14 — migrate to the Unstable / Experimental API.\n * Scheduled for removal on/after 2026-09-14. See\n * {@link https://www.assistant-ui.com/docs/tools/interactables#migrating-from-the-previous-api | Interactables migration guide}.\n */\nexport type AssistantInteractableProps = {\n  description: string;\n  stateSchema: InteractableStateSchema;\n  initialState: unknown;\n  id?: string;\n  selected?: boolean;\n};\n\n/**\n * @deprecated Since 2026-06-14 — migrate to the Unstable / Experimental API.\n * Scheduled for removal on/after 2026-09-14. See\n * {@link https://www.assistant-ui.com/docs/tools/interactables#migrating-from-the-previous-api | Interactables migration guide}.\n *\n * Registers an interactable with the AI assistant.\n *\n * This hook handles registration only. To read and write the interactable's\n * state, use {@link useInteractableState} with the returned id.\n *\n * @returns The interactable instance id.\n */\nexport const useAssistantInteractable = (\n  name: string,\n  config: AssistantInteractableProps,\n): string => {\n  const aui = useAui();\n\n  const autoId = useId().replace(/[^a-zA-Z0-9]/g, \"\");\n  const id = config.id ?? autoId;\n\n  const stateSchemaRef = useRef(config.stateSchema);\n  stateSchemaRef.current = config.stateSchema;\n  const initialStateRef = useRef(config.initialState);\n  initialStateRef.current = config.initialState;\n\n  useEffect(() => {\n    return aui.interactables.register({\n      id,\n      name,\n      description: config.description,\n      stateSchema: stateSchemaRef.current,\n      initialState: initialStateRef.current,\n      selected: config.selected,\n    });\n  }, [aui, id, name, config.description, config.selected]);\n\n  return id;\n};\n"],"mappings":";;;;;;;;;;;;;;;AA6BA,MAAaW,4BACXC,MACAC,WACW;CACX,MAAMC,MAAMX,OAAO;CAEnB,MAAMY,SAASd,MAAM,CAAC,CAACe,QAAQ,iBAAiB,EAAE;CAClD,MAAMP,KAAKI,OAAOJ,MAAMM;CAExB,MAAME,iBAAiBf,OAAOW,OAAON,WAAW;CAChDU,eAAeC,UAAUL,OAAON;CAChC,MAAMY,kBAAkBjB,OAAOW,OAAOL,YAAY;CAClDW,gBAAgBD,UAAUL,OAAOL;CAEjCR,gBAAgB;EACd,OAAOc,IAAIM,cAAcC,SAAS;GAChCZ;GACAG;GACAN,aAAaO,OAAOP;GACpBC,aAAaU,eAAeC;GAC5BV,cAAcW,gBAAgBD;GAC9BR,UAAUG,OAAOH;EACnB,CAAC;CACH,GAAG;EAACI;EAAKL;EAAIG;EAAMC,OAAOP;EAAaO,OAAOH;CAAQ,CAAC;CAEvD,OAAOD;AACT"}
+{"version":3,"file":"useAssistantInteractable.js","names":["useEffect","useId","useRef","useAui","useJSONSchemaDependency","InteractableStateSchema","AssistantInteractableProps","description","stateSchema","initialState","id","selected","useAssistantInteractable","name","config","aui","autoId","replace","stateSchemaRef","current","normalizedStateSchema","initialStateRef","interactables","register"],"sources":["../../../src/react/interactables-legacy/useAssistantInteractable.ts"],"sourcesContent":["import { useEffect, useId, useRef } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { useJSONSchemaDependency } from \"../utils/useJSONSchemaDependency\";\nimport type { InteractableStateSchema } from \"./scopes\";\n\n/**\n * @deprecated Since 2026-06-14 — migrate to the Unstable / Experimental API.\n * Scheduled for removal on/after 2026-09-14. See\n * {@link https://www.assistant-ui.com/docs/tools/interactables#migrating-from-the-previous-api | Interactables migration guide}.\n */\nexport type AssistantInteractableProps = {\n  description: string;\n  stateSchema: InteractableStateSchema;\n  initialState: unknown;\n  id?: string;\n  selected?: boolean;\n};\n\n/**\n * @deprecated Since 2026-06-14 — migrate to the Unstable / Experimental API.\n * Scheduled for removal on/after 2026-09-14. See\n * {@link https://www.assistant-ui.com/docs/tools/interactables#migrating-from-the-previous-api | Interactables migration guide}.\n *\n * Registers an interactable with the AI assistant.\n *\n * This hook handles registration only. To read and write the interactable's\n * state, use {@link useInteractableState} with the returned id.\n *\n * @returns The interactable instance id.\n */\nexport const useAssistantInteractable = (\n  name: string,\n  config: AssistantInteractableProps,\n): string => {\n  const aui = useAui();\n\n  const autoId = useId().replace(/[^a-zA-Z0-9]/g, \"\");\n  const id = config.id ?? autoId;\n\n  const stateSchemaRef = useRef(config.stateSchema);\n  stateSchemaRef.current = config.stateSchema;\n  const normalizedStateSchema = useJSONSchemaDependency(config.stateSchema);\n  const initialStateRef = useRef(config.initialState);\n  initialStateRef.current = config.initialState;\n\n  useEffect(() => {\n    return aui.interactables.register({\n      id,\n      name,\n      description: config.description,\n      stateSchema: stateSchemaRef.current,\n      initialState: initialStateRef.current,\n      selected: config.selected,\n    });\n  }, [\n    aui,\n    id,\n    name,\n    config.description,\n    normalizedStateSchema,\n    config.selected,\n  ]);\n\n  return id;\n};\n"],"mappings":";;;;;;;;;;;;;;;;AA8BA,MAAaY,4BACXC,MACAC,WACW;CACX,MAAMC,MAAMZ,OAAO;CAEnB,MAAMa,SAASf,MAAM,CAAC,CAACgB,QAAQ,iBAAiB,EAAE;CAClD,MAAMP,KAAKI,OAAOJ,MAAMM;CAExB,MAAME,iBAAiBhB,OAAOY,OAAON,WAAW;CAChDU,eAAeC,UAAUL,OAAON;CAChC,MAAMY,wBAAwBhB,wBAAwBU,OAAON,WAAW;CACxE,MAAMa,kBAAkBnB,OAAOY,OAAOL,YAAY;CAClDY,gBAAgBF,UAAUL,OAAOL;CAEjCT,gBAAgB;EACd,OAAOe,IAAIO,cAAcC,SAAS;GAChCb;GACAG;GACAN,aAAaO,OAAOP;GACpBC,aAAaU,eAAeC;GAC5BV,cAAcY,gBAAgBF;GAC9BR,UAAUG,OAAOH;EACnB,CAAC;CACH,GAAG;EACDI;EACAL;EACAG;EACAC,OAAOP;EACPa;EACAN,OAAOH;CAAQ,CAChB;CAED,OAAOD;AACT"}

package/dist/react/model-context/useInteractable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useInteractable.d.ts","names":[],"sources":["../../../src/react/model-context/useInteractable.ts"],"mappings":";;;;;;;;KAkBY,gCAAgC,WAAW;EACrD;IAAe;MAAU,cAAc;;;IAErC;;;;;;KAQQ,iCAAiC;;EAE3C,OAAO;;EAEP;;EAEA;;;;;;KAOU,4BACV,gBAAgB;EAEhB;EACA,aAAa;EACb,cAAc,gCAAgC;;EAE9C;;;;;;;;EAQA,eAAe;;;;;;;;;;;;;;;;;;;;cAkJJ,2BACX,gBAAgB,kCAEhB,cACA,QAAQ,4BAA4B,uBAEpC,gCAAgC;EAE9B;EACA,SACI,iCACE,gCAAgC;EAGtC,WACE,SACI,gCAAgC,aAE9B,MAAM,gCAAgC,aACnC,gCAAgC;EAE3C;EACA;EACA,aAAa"}
+{"version":3,"file":"useInteractable.d.ts","names":[],"sources":["../../../src/react/model-context/useInteractable.ts"],"mappings":";;;;;;;;KAmBY,gCAAgC,WAAW;EACrD;IAAe;MAAU,cAAc;;;IAErC;;;;;;KAQQ,iCAAiC;;EAE3C,OAAO;;EAEP;;EAEA;;;;;;KAOU,4BACV,gBAAgB;EAEhB;EACA,aAAa;EACb,cAAc,gCAAgC;;EAE9C;;;;;;;;EAQA,eAAe;;;;;;;;;;;;;;;;;;;;cAoJJ,2BACX,gBAAgB,kCAEhB,cACA,QAAQ,4BAA4B,uBAEpC,gCAAgC;EAE9B;EACA,SACI,iCACE,gCAAgC;EAGtC,WACE,SACI,gCAAgC,aAE9B,MAAM,gCAAgC,aACnC,gCAAgC;EAE3C;EACA;EACA,aAAa"}

package/dist/react/model-context/useInteractable.js

@@ -1,7 +1,8 @@
 "use client";
 import { interactableToolName, unstable_getInteractableVersions } from "../../model-context/interactable-composer-metadata.js";
-import { unstable_useInteractableState } from "./useInteractableState.js";
 import { useJSONEqual } from "../utils/useJSONEqual.js";
+import { useJSONSchemaDependency } from "../utils/useJSONSchemaDependency.js";
+import { unstable_useInteractableState } from "./useInteractableState.js";
 import { useEffect, useId, useMemo, useRef } from "@assistant-ui/tap/react-shim";
 import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/react/model-context/useInteractable.ts
@@ -26,6 +27,7 @@ const useInteractable = (name, config) => {
 	const internalScope = part?.toolName === name || part?.toolName === updateToolName ? "thread" : void 0;
 	const stateSchemaRef = useRef(config.stateSchema);
 	stateSchemaRef.current = config.stateSchema;
+	const normalizedStateSchema = useJSONSchemaDependency(config.stateSchema);
 	const initialStateRef = useRef(config.initialState);
 	initialStateRef.current = config.initialState;
 	const interactables = useAuiState(() => aui.unstable_interactables);
@@ -44,6 +46,7 @@ const useInteractable = (name, config) => {
 		id,
 		name,
 		config.description,
+		normalizedStateSchema,
 		internalScope,
 		config.updateRender
 	]);

package/dist/react/model-context/useInteractable.js.map

@@ -1 +1 @@
-{"version":3,"file":"useInteractable.js","names":["useEffect","useId","useMemo","useRef","useAui","useAuiState","Unstable_InteractableStateSchema","ToolCallMessagePartComponent","unstable_getInteractableVersions","getInteractableVersions","interactableToolName","unstable_useInteractableState","useInteractableState","useJSONEqual","Unstable_InferInteractableState","TSchema","types","output","TOutput","Unstable_InteractableVersionInfo","state","TState","isLatest","restore","Unstable_InteractableConfig","description","stateSchema","initialState","id","updateRender","useInteractable","name","config","version","setState","updater","prev","isPending","error","flush","Promise","aui","autoId","replace","updateToolName","part","s","p","optional","type","undefined","inferredId","toolName","result","Record","args","success","internalScope","stateSchemaRef","current","initialStateRef","interactables","unstable_interactables","register","scope","Parameters","myToolCallId","toolCallId","registeredState","methods","versionValue","versions","thread","messages","some","v","origin","mine","find","latestToolCallId","findLast","const","unstable_useInteractable"],"sources":["../../../src/react/model-context/useInteractable.ts"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useId, useMemo, useRef } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { Unstable_InteractableStateSchema } from \"../types/scopes/interactables\";\nimport type { ToolCallMessagePartComponent } from \"../types/MessagePartComponentTypes\";\nimport {\n  unstable_getInteractableVersions as getInteractableVersions,\n  interactableToolName,\n} from \"../../model-context/interactable-composer-metadata\";\nimport { unstable_useInteractableState as useInteractableState } from \"./useInteractableState\";\nimport { useJSONEqual } from \"../utils/useJSONEqual\";\n\n/**\n * The state type described by an interactable's `stateSchema`. Resolves the\n * output type of a StandardSchemaV1 schema (e.g. Zod); plain JSON Schema\n * falls back to `unknown`.\n */\nexport type Unstable_InferInteractableState<TSchema> = TSchema extends {\n  \"~standard\": { types?: { output: infer TOutput } | undefined };\n}\n  ? TOutput\n  : unknown;\n\n/**\n * One message's version of an interactable, for components rendered inside\n * tool-call parts. Show `state` to display history; call `restore()` to set\n * the live instance back to it.\n */\nexport type Unstable_InteractableVersionInfo<TState> = {\n  /** The interactable's state as it was at this message. */\n  state: TState;\n  /** Whether this message holds the instance's most recent tool-driven version. */\n  isLatest: boolean;\n  /** Sets the live state back to this version. */\n  restore: () => void;\n};\n\n/**\n * Unstable / Experimental — the interactables API is still evolving and may change in any release.\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport type Unstable_InteractableConfig<\n  TSchema extends Unstable_InteractableStateSchema,\n> = {\n  description: string;\n  stateSchema: TSchema;\n  initialState: Unstable_InferInteractableState<TSchema>;\n  /** Unique instance ID; required to address this instance when multiple interactables share a name. Auto-generated if omitted. */\n  id?: string | undefined;\n  /**\n   * Component installed as the tool UI for this interactable's `update_{name}`\n   * tool calls, so a model edit re-renders the interactable at the message\n   * that made it instead of only mutating an earlier one. Prefer\n   * `unstable_interactableTool`, which wires this up. Pass a stable component reference;\n   * changing identity re-registers the tool UI.\n   */\n  updateRender?: ToolCallMessagePartComponent | undefined;\n};\n\nconst useInteractable = <TSchema extends Unstable_InteractableStateSchema>(\n  name: string,\n  config: Unstable_InteractableConfig<TSchema>,\n): readonly [\n  Unstable_InferInteractableState<TSchema>,\n  {\n    id: string;\n    /**\n     * This message's version of the instance, when rendered inside a\n     * tool-created interactable message part; `undefined` elsewhere.\n     */\n    version:\n      | Unstable_InteractableVersionInfo<\n          Unstable_InferInteractableState<TSchema>\n        >\n      | undefined;\n    setState: (\n      updater:\n        | Unstable_InferInteractableState<TSchema>\n        | ((\n            prev: Unstable_InferInteractableState<TSchema>,\n          ) => Unstable_InferInteractableState<TSchema>),\n    ) => void;\n    isPending: boolean;\n    error: unknown;\n    flush: () => Promise<void>;\n  },\n] => {\n  const aui = useAui();\n\n  const autoId = useId().replace(/[^a-zA-Z0-9]/g, \"\");\n\n  const updateToolName = interactableToolName(name);\n  const part = useAuiState((s) => {\n    const p = s.optional.part;\n    return p?.type === \"tool-call\" ? p : undefined;\n  });\n\n  // Inside an update_{name} part, the instance id comes from the call itself:\n  // the result's resolved id (covers id-less calls), or the id argument.\n  let inferredId: string | undefined;\n  if (part?.toolName === updateToolName) {\n    const result = part.result as Record<string, unknown> | undefined;\n    const args = part.args as Record<string, unknown> | undefined;\n    if (result?.success !== false) {\n      if (typeof result?.id === \"string\") inferredId = result.id;\n      else if (typeof args?.id === \"string\") inferredId = args.id;\n    }\n  }\n\n  const id = config.id ?? inferredId ?? autoId;\n  const internalScope =\n    part?.toolName === name || part?.toolName === updateToolName\n      ? \"thread\"\n      : undefined;\n\n  const stateSchemaRef = useRef(config.stateSchema);\n  stateSchemaRef.current = config.stateSchema;\n  const initialStateRef = useRef(config.initialState);\n  initialStateRef.current = config.initialState;\n\n  const interactables = useAuiState(() => aui.unstable_interactables);\n\n  useEffect(() => {\n    return interactables.register({\n      id,\n      name,\n      description: config.description,\n      stateSchema: stateSchemaRef.current,\n      initialState: initialStateRef.current,\n      updateRender: config.updateRender,\n      ...(internalScope ? { scope: internalScope } : {}),\n    } as Parameters<typeof interactables.register>[0] & {\n      scope?: \"thread\" | undefined;\n    });\n  }, [\n    interactables,\n    id,\n    name,\n    config.description,\n    internalScope,\n    config.updateRender,\n  ]);\n\n  const myToolCallId = part?.toolCallId;\n\n  const [registeredState, methods] =\n    useInteractableState<Unstable_InferInteractableState<TSchema>>(id);\n  const { setState } = methods;\n\n  const versionValue = useAuiState(\n    useJSONEqual((s) => {\n      if (!internalScope || !myToolCallId) return undefined;\n      const versions = getInteractableVersions(s.thread.messages, id, name);\n      if (!versions.some((v) => v.origin === \"create\" && v.toolCallId === id)) {\n        return undefined;\n      }\n      const mine = versions.find((v) => v.toolCallId === myToolCallId);\n      if (!mine) return undefined;\n      const latestToolCallId = versions.findLast(\n        (v) => v.toolCallId !== undefined,\n      )?.toolCallId;\n      return { state: mine.state, isLatest: latestToolCallId === myToolCallId };\n    }),\n  );\n\n  const version = useMemo(\n    () =>\n      versionValue && {\n        state: versionValue.state as Unstable_InferInteractableState<TSchema>,\n        isLatest: versionValue.isLatest,\n        restore: () =>\n          setState(\n            versionValue.state as Unstable_InferInteractableState<TSchema>,\n          ),\n      },\n    [versionValue, setState],\n  );\n\n  const state =\n    registeredState === undefined ? config.initialState : registeredState;\n\n  return [state, { id, version, ...methods }] as const;\n};\n\n/**\n * Registers an interactable with the AI assistant and returns its live state,\n * like `useState` that the model can also read and update.\n *\n * Call this once per place that shows the interactable. Other components can\n * read and write the same instance by passing its `id` to\n * `unstable_useInteractableState`.\n *\n * For tool-created interactables rendered inside tool-call message parts,\n * `version` carries this message's version of the instance — its state as of\n * that point in the conversation, whether it is the most recent tool-driven\n * version, and a `restore()` back to it. Whether older messages render frozen history or stay\n * live-editable is the component's choice. Inside an `update_{name}` part the\n * instance `id` is inferred from the call, so the same component works at the\n * creating call and at update calls.\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport const unstable_useInteractable: <\n  TSchema extends Unstable_InteractableStateSchema,\n>(\n  name: string,\n  config: Unstable_InteractableConfig<TSchema>,\n) => readonly [\n  Unstable_InferInteractableState<TSchema>,\n  {\n    id: string;\n    version:\n      | Unstable_InteractableVersionInfo<\n          Unstable_InferInteractableState<TSchema>\n        >\n      | undefined;\n    setState: (\n      updater:\n        | Unstable_InferInteractableState<TSchema>\n        | ((\n            prev: Unstable_InferInteractableState<TSchema>,\n          ) => Unstable_InferInteractableState<TSchema>),\n    ) => void;\n    isPending: boolean;\n    error: unknown;\n    flush: () => Promise<void>;\n  },\n] = useInteractable;\n"],"mappings":";;;;;;;AA4DA,MAAM8B,mBACJC,MACAC,WAyBG;CACH,MAAMS,MAAMrC,OAAO;CAEnB,MAAMsC,SAASzC,MAAM,CAAC,CAAC0C,QAAQ,iBAAiB,EAAE;CAElD,MAAMC,iBAAiBlC,qBAAqBqB,IAAI;CAChD,MAAMc,OAAOxC,aAAayC,MAAM;EAC9B,MAAMC,IAAID,EAAEE,SAASH;EACrB,OAAOE,GAAGE,SAAS,cAAcF,IAAIG,KAAAA;CACvC,CAAC;CAID,IAAIC;CACJ,IAAIN,MAAMO,aAAaR,gBAAgB;EACrC,MAAMS,SAASR,KAAKQ;EACpB,MAAME,OAAOV,KAAKU;EAClB,IAAIF,QAAQG,YAAY,OAAO;GAC7B,IAAI,OAAOH,QAAQzB,OAAO,UAAUuB,aAAaE,OAAOzB;QACnD,IAAI,OAAO2B,MAAM3B,OAAO,UAAUuB,aAAaI,KAAK3B;EAC3D;CACF;CAEA,MAAMA,KAAKI,OAAOJ,MAAMuB,cAAcT;CACtC,MAAMe,gBACJZ,MAAMO,aAAarB,QAAQc,MAAMO,aAAaR,iBAC1C,WACAM,KAAAA;CAEN,MAAMQ,iBAAiBvD,OAAO6B,OAAON,WAAW;CAChDgC,eAAeC,UAAU3B,OAAON;CAChC,MAAMkC,kBAAkBzD,OAAO6B,OAAOL,YAAY;CAClDiC,gBAAgBD,UAAU3B,OAAOL;CAEjC,MAAMkC,gBAAgBxD,kBAAkBoC,IAAIqB,sBAAsB;CAElE9D,gBAAgB;EACd,OAAO6D,cAAcE,SAAS;GAC5BnC;GACAG;GACAN,aAAaO,OAAOP;GACpBC,aAAagC,eAAeC;GAC5BhC,cAAciC,gBAAgBD;GAC9B9B,cAAcG,OAAOH;GACrB,GAAI4B,gBAAgB,EAAEO,OAAOP,cAAc,IAAI,CAAC;EAClD,CAEC;CACH,GAAG;EACDI;EACAjC;EACAG;EACAC,OAAOP;EACPgC;EACAzB,OAAOH;CAAY,CACpB;CAED,MAAMqC,eAAerB,MAAMsB;CAE3B,MAAM,CAACC,iBAAiBC,WACtBzD,8BAA+DgB,EAAE;CACnE,MAAM,EAAEM,aAAamC;CAErB,MAAMC,eAAejE,YACnBQ,cAAciC,QAAM;EAClB,IAAI,CAACW,iBAAiB,CAACS,cAAc,OAAOhB,KAAAA;EAC5C,MAAMqB,WAAW9D,iCAAwBqC,IAAE0B,OAAOC,UAAU7C,IAAIG,IAAI;EACpE,IAAI,CAACwC,SAASG,MAAMC,MAAMA,EAAEC,WAAW,YAAYD,EAAER,eAAevC,EAAE,GACpE;EAEF,MAAMiD,OAAON,SAASO,MAAMH,QAAMA,IAAER,eAAeD,YAAY;EAC/D,IAAI,CAACW,MAAM,OAAO3B,KAAAA;EAClB,MAAM6B,mBAAmBR,SAASS,UAC/BL,QAAMA,IAAER,eAAejB,KAAAA,CAC1B,CAAC,EAAEiB;EACH,OAAO;GAAE/C,OAAOyD,KAAKzD;GAAOE,UAAUyD,qBAAqBb;EAAa;CAC1E,CAAC,CACH;CAEA,MAAMjC,UAAU/B,cAEZoE,gBAAgB;EACdlD,OAAOkD,aAAalD;EACpBE,UAAUgD,aAAahD;EACvBC,eACEW,SACEoC,aAAalD,KACf;CACJ,GACF,CAACkD,cAAcpC,QAAQ,CACzB;CAKA,OAAO,CAFLkC,oBAAoBlB,KAAAA,IAAYlB,OAAOL,eAAeyC,iBAEzC;EAAExC;EAAIK;EAAS,GAAGoC;CAAQ,CAAC;AAC5C;;;;;;;;;;;;;;;;;;;AAoBA,MAAaa,2BAyBTpD"}
+{"version":3,"file":"useInteractable.js","names":["useEffect","useId","useMemo","useRef","useAui","useAuiState","Unstable_InteractableStateSchema","ToolCallMessagePartComponent","unstable_getInteractableVersions","getInteractableVersions","interactableToolName","unstable_useInteractableState","useInteractableState","useJSONEqual","useJSONSchemaDependency","Unstable_InferInteractableState","TSchema","types","output","TOutput","Unstable_InteractableVersionInfo","state","TState","isLatest","restore","Unstable_InteractableConfig","description","stateSchema","initialState","id","updateRender","useInteractable","name","config","version","setState","updater","prev","isPending","error","flush","Promise","aui","autoId","replace","updateToolName","part","s","p","optional","type","undefined","inferredId","toolName","result","Record","args","success","internalScope","stateSchemaRef","current","normalizedStateSchema","initialStateRef","interactables","unstable_interactables","register","scope","Parameters","myToolCallId","toolCallId","registeredState","methods","versionValue","versions","thread","messages","some","v","origin","mine","find","latestToolCallId","findLast","const","unstable_useInteractable"],"sources":["../../../src/react/model-context/useInteractable.ts"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useId, useMemo, useRef } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { Unstable_InteractableStateSchema } from \"../types/scopes/interactables\";\nimport type { ToolCallMessagePartComponent } from \"../types/MessagePartComponentTypes\";\nimport {\n  unstable_getInteractableVersions as getInteractableVersions,\n  interactableToolName,\n} from \"../../model-context/interactable-composer-metadata\";\nimport { unstable_useInteractableState as useInteractableState } from \"./useInteractableState\";\nimport { useJSONEqual } from \"../utils/useJSONEqual\";\nimport { useJSONSchemaDependency } from \"../utils/useJSONSchemaDependency\";\n\n/**\n * The state type described by an interactable's `stateSchema`. Resolves the\n * output type of a StandardSchemaV1 schema (e.g. Zod); plain JSON Schema\n * falls back to `unknown`.\n */\nexport type Unstable_InferInteractableState<TSchema> = TSchema extends {\n  \"~standard\": { types?: { output: infer TOutput } | undefined };\n}\n  ? TOutput\n  : unknown;\n\n/**\n * One message's version of an interactable, for components rendered inside\n * tool-call parts. Show `state` to display history; call `restore()` to set\n * the live instance back to it.\n */\nexport type Unstable_InteractableVersionInfo<TState> = {\n  /** The interactable's state as it was at this message. */\n  state: TState;\n  /** Whether this message holds the instance's most recent tool-driven version. */\n  isLatest: boolean;\n  /** Sets the live state back to this version. */\n  restore: () => void;\n};\n\n/**\n * Unstable / Experimental — the interactables API is still evolving and may change in any release.\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport type Unstable_InteractableConfig<\n  TSchema extends Unstable_InteractableStateSchema,\n> = {\n  description: string;\n  stateSchema: TSchema;\n  initialState: Unstable_InferInteractableState<TSchema>;\n  /** Unique instance ID; required to address this instance when multiple interactables share a name. Auto-generated if omitted. */\n  id?: string | undefined;\n  /**\n   * Component installed as the tool UI for this interactable's `update_{name}`\n   * tool calls, so a model edit re-renders the interactable at the message\n   * that made it instead of only mutating an earlier one. Prefer\n   * `unstable_interactableTool`, which wires this up. Pass a stable component reference;\n   * changing identity re-registers the tool UI.\n   */\n  updateRender?: ToolCallMessagePartComponent | undefined;\n};\n\nconst useInteractable = <TSchema extends Unstable_InteractableStateSchema>(\n  name: string,\n  config: Unstable_InteractableConfig<TSchema>,\n): readonly [\n  Unstable_InferInteractableState<TSchema>,\n  {\n    id: string;\n    /**\n     * This message's version of the instance, when rendered inside a\n     * tool-created interactable message part; `undefined` elsewhere.\n     */\n    version:\n      | Unstable_InteractableVersionInfo<\n          Unstable_InferInteractableState<TSchema>\n        >\n      | undefined;\n    setState: (\n      updater:\n        | Unstable_InferInteractableState<TSchema>\n        | ((\n            prev: Unstable_InferInteractableState<TSchema>,\n          ) => Unstable_InferInteractableState<TSchema>),\n    ) => void;\n    isPending: boolean;\n    error: unknown;\n    flush: () => Promise<void>;\n  },\n] => {\n  const aui = useAui();\n\n  const autoId = useId().replace(/[^a-zA-Z0-9]/g, \"\");\n\n  const updateToolName = interactableToolName(name);\n  const part = useAuiState((s) => {\n    const p = s.optional.part;\n    return p?.type === \"tool-call\" ? p : undefined;\n  });\n\n  // Inside an update_{name} part, the instance id comes from the call itself:\n  // the result's resolved id (covers id-less calls), or the id argument.\n  let inferredId: string | undefined;\n  if (part?.toolName === updateToolName) {\n    const result = part.result as Record<string, unknown> | undefined;\n    const args = part.args as Record<string, unknown> | undefined;\n    if (result?.success !== false) {\n      if (typeof result?.id === \"string\") inferredId = result.id;\n      else if (typeof args?.id === \"string\") inferredId = args.id;\n    }\n  }\n\n  const id = config.id ?? inferredId ?? autoId;\n  const internalScope =\n    part?.toolName === name || part?.toolName === updateToolName\n      ? \"thread\"\n      : undefined;\n\n  const stateSchemaRef = useRef(config.stateSchema);\n  stateSchemaRef.current = config.stateSchema;\n  const normalizedStateSchema = useJSONSchemaDependency(config.stateSchema);\n  const initialStateRef = useRef(config.initialState);\n  initialStateRef.current = config.initialState;\n\n  const interactables = useAuiState(() => aui.unstable_interactables);\n\n  useEffect(() => {\n    return interactables.register({\n      id,\n      name,\n      description: config.description,\n      stateSchema: stateSchemaRef.current,\n      initialState: initialStateRef.current,\n      updateRender: config.updateRender,\n      ...(internalScope ? { scope: internalScope } : {}),\n    } as Parameters<typeof interactables.register>[0] & {\n      scope?: \"thread\" | undefined;\n    });\n  }, [\n    interactables,\n    id,\n    name,\n    config.description,\n    normalizedStateSchema,\n    internalScope,\n    config.updateRender,\n  ]);\n\n  const myToolCallId = part?.toolCallId;\n\n  const [registeredState, methods] =\n    useInteractableState<Unstable_InferInteractableState<TSchema>>(id);\n  const { setState } = methods;\n\n  const versionValue = useAuiState(\n    useJSONEqual((s) => {\n      if (!internalScope || !myToolCallId) return undefined;\n      const versions = getInteractableVersions(s.thread.messages, id, name);\n      if (!versions.some((v) => v.origin === \"create\" && v.toolCallId === id)) {\n        return undefined;\n      }\n      const mine = versions.find((v) => v.toolCallId === myToolCallId);\n      if (!mine) return undefined;\n      const latestToolCallId = versions.findLast(\n        (v) => v.toolCallId !== undefined,\n      )?.toolCallId;\n      return { state: mine.state, isLatest: latestToolCallId === myToolCallId };\n    }),\n  );\n\n  const version = useMemo(\n    () =>\n      versionValue && {\n        state: versionValue.state as Unstable_InferInteractableState<TSchema>,\n        isLatest: versionValue.isLatest,\n        restore: () =>\n          setState(\n            versionValue.state as Unstable_InferInteractableState<TSchema>,\n          ),\n      },\n    [versionValue, setState],\n  );\n\n  const state =\n    registeredState === undefined ? config.initialState : registeredState;\n\n  return [state, { id, version, ...methods }] as const;\n};\n\n/**\n * Registers an interactable with the AI assistant and returns its live state,\n * like `useState` that the model can also read and update.\n *\n * Call this once per place that shows the interactable. Other components can\n * read and write the same instance by passing its `id` to\n * `unstable_useInteractableState`.\n *\n * For tool-created interactables rendered inside tool-call message parts,\n * `version` carries this message's version of the instance — its state as of\n * that point in the conversation, whether it is the most recent tool-driven\n * version, and a `restore()` back to it. Whether older messages render frozen history or stay\n * live-editable is the component's choice. Inside an `update_{name}` part the\n * instance `id` is inferred from the call, so the same component works at the\n * creating call and at update calls.\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport const unstable_useInteractable: <\n  TSchema extends Unstable_InteractableStateSchema,\n>(\n  name: string,\n  config: Unstable_InteractableConfig<TSchema>,\n) => readonly [\n  Unstable_InferInteractableState<TSchema>,\n  {\n    id: string;\n    version:\n      | Unstable_InteractableVersionInfo<\n          Unstable_InferInteractableState<TSchema>\n        >\n      | undefined;\n    setState: (\n      updater:\n        | Unstable_InferInteractableState<TSchema>\n        | ((\n            prev: Unstable_InferInteractableState<TSchema>,\n          ) => Unstable_InferInteractableState<TSchema>),\n    ) => void;\n    isPending: boolean;\n    error: unknown;\n    flush: () => Promise<void>;\n  },\n] = useInteractable;\n"],"mappings":";;;;;;;;AA6DA,MAAM+B,mBACJC,MACAC,WAyBG;CACH,MAAMS,MAAMtC,OAAO;CAEnB,MAAMuC,SAAS1C,MAAM,CAAC,CAAC2C,QAAQ,iBAAiB,EAAE;CAElD,MAAMC,iBAAiBnC,qBAAqBsB,IAAI;CAChD,MAAMc,OAAOzC,aAAa0C,MAAM;EAC9B,MAAMC,IAAID,EAAEE,SAASH;EACrB,OAAOE,GAAGE,SAAS,cAAcF,IAAIG,KAAAA;CACvC,CAAC;CAID,IAAIC;CACJ,IAAIN,MAAMO,aAAaR,gBAAgB;EACrC,MAAMS,SAASR,KAAKQ;EACpB,MAAME,OAAOV,KAAKU;EAClB,IAAIF,QAAQG,YAAY,OAAO;GAC7B,IAAI,OAAOH,QAAQzB,OAAO,UAAUuB,aAAaE,OAAOzB;QACnD,IAAI,OAAO2B,MAAM3B,OAAO,UAAUuB,aAAaI,KAAK3B;EAC3D;CACF;CAEA,MAAMA,KAAKI,OAAOJ,MAAMuB,cAAcT;CACtC,MAAMe,gBACJZ,MAAMO,aAAarB,QAAQc,MAAMO,aAAaR,iBAC1C,WACAM,KAAAA;CAEN,MAAMQ,iBAAiBxD,OAAO8B,OAAON,WAAW;CAChDgC,eAAeC,UAAU3B,OAAON;CAChC,MAAMkC,wBAAwB/C,wBAAwBmB,OAAON,WAAW;CACxE,MAAMmC,kBAAkB3D,OAAO8B,OAAOL,YAAY;CAClDkC,gBAAgBF,UAAU3B,OAAOL;CAEjC,MAAMmC,gBAAgB1D,kBAAkBqC,IAAIsB,sBAAsB;CAElEhE,gBAAgB;EACd,OAAO+D,cAAcE,SAAS;GAC5BpC;GACAG;GACAN,aAAaO,OAAOP;GACpBC,aAAagC,eAAeC;GAC5BhC,cAAckC,gBAAgBF;GAC9B9B,cAAcG,OAAOH;GACrB,GAAI4B,gBAAgB,EAAEQ,OAAOR,cAAc,IAAI,CAAC;EAClD,CAEC;CACH,GAAG;EACDK;EACAlC;EACAG;EACAC,OAAOP;EACPmC;EACAH;EACAzB,OAAOH;CAAY,CACpB;CAED,MAAMsC,eAAetB,MAAMuB;CAE3B,MAAM,CAACC,iBAAiBC,WACtB3D,8BAA+DiB,EAAE;CACnE,MAAM,EAAEM,aAAaoC;CAErB,MAAMC,eAAenE,YACnBQ,cAAckC,QAAM;EAClB,IAAI,CAACW,iBAAiB,CAACU,cAAc,OAAOjB,KAAAA;EAC5C,MAAMsB,WAAWhE,iCAAwBsC,IAAE2B,OAAOC,UAAU9C,IAAIG,IAAI;EACpE,IAAI,CAACyC,SAASG,MAAMC,MAAMA,EAAEC,WAAW,YAAYD,EAAER,eAAexC,EAAE,GACpE;EAEF,MAAMkD,OAAON,SAASO,MAAMH,QAAMA,IAAER,eAAeD,YAAY;EAC/D,IAAI,CAACW,MAAM,OAAO5B,KAAAA;EAClB,MAAM8B,mBAAmBR,SAASS,UAC/BL,QAAMA,IAAER,eAAelB,KAAAA,CAC1B,CAAC,EAAEkB;EACH,OAAO;GAAEhD,OAAO0D,KAAK1D;GAAOE,UAAU0D,qBAAqBb;EAAa;CAC1E,CAAC,CACH;CAEA,MAAMlC,UAAUhC,cAEZsE,gBAAgB;EACdnD,OAAOmD,aAAanD;EACpBE,UAAUiD,aAAajD;EACvBC,eACEW,SACEqC,aAAanD,KACf;CACJ,GACF,CAACmD,cAAcrC,QAAQ,CACzB;CAKA,OAAO,CAFLmC,oBAAoBnB,KAAAA,IAAYlB,OAAOL,eAAe0C,iBAEzC;EAAEzC;EAAIK;EAAS,GAAGqC;CAAQ,CAAC;AAC5C;;;;;;;;;;;;;;;;;;;AAoBA,MAAaa,2BAyBTrD"}

package/dist/react/model-context/useInteractableVersions.js

@@ -1,6 +1,6 @@
 import { unstable_getInteractableVersions } from "../../model-context/interactable-composer-metadata.js";
-import { unstable_useInteractableState } from "./useInteractableState.js";
 import { useJSONEqual } from "../utils/useJSONEqual.js";
+import { unstable_useInteractableState } from "./useInteractableState.js";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import "@assistant-ui/tap/react-shim";
 import { useAuiState } from "@assistant-ui/store";

package/dist/react/primitives/message/MessageGroupedParts.d.ts

@@ -5,6 +5,13 @@ import { GroupByContext } from "../../utils/groupParts.js";
 import { ReactNode } from "react";
 //#region src/react/primitives/message/MessageGroupedParts.d.ts
 declare namespace MessagePrimitiveGroupedParts {
+  /** Per status tallies over the group's `indices`; they sum to `indices.length`. */
+  type GroupCounts = {
+    readonly running: number;
+    readonly complete: number;
+    readonly incomplete: number;
+    readonly requiresAction: number;
+  };
   /**
    * A coalesced group of adjacent parts. Surfaced through the same
    * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers
@@ -15,6 +22,8 @@ declare namespace MessagePrimitiveGroupedParts {
   type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {
     readonly type: TKey;
     readonly status: MessagePartStatus | ToolCallMessagePartStatus;
+    /** Per status tallies over `indices`. */
+    readonly counts: GroupCounts;
     readonly indices: readonly number[];
   };
   /**
@@ -68,10 +77,10 @@ declare namespace MessagePrimitiveGroupedParts {
      * `switch (part.type)` can tell groups apart from real part types.
      *
      * **Prefer {@link groupPartByType}** for the common case of mapping by
-     * `part.type` — it ships a stable memo fingerprint so the tree
+     * `part.type` or tool name — it ships a stable memo fingerprint so the tree
      * survives unrelated re-renders. Use an inline function only when
      * the helper isn't expressive enough (e.g. branching on
-     * `part.toolName` or part metadata).
+     * `part.parentId` or part metadata).
      *
      * The second argument is a {@link GroupByContext} carrying the tool-UI
      * registry, for grouping that depends on it (e.g. standalone tool calls).

package/dist/react/primitives/message/MessageGroupedParts.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageGroupedParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"mappings":";;;;;;kBAmBiB;;;;;;;;OAQH,UAAU;aACX,MAAM;aACN,QAAQ,oBAAoB;aAC5B;;;;;;;;;;;;OAaC;aACD;;;;;;;;;;;;;OAcC;OAEA,WAAW;;;;;;;aAOZ,MAAM,UAAU,QAAQ,oBAAoB;;;;;;;aAO5C,UAAU;;OAGT,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aA8BP,UACP,MAAM,WACN,SAAS,4BACG;;;;;;;;;aAUL,YAAY;;;;;;;;;;;;aAaZ,WAAW,MAAM,WAAW,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAmHtC;GAAgC,kCAA8B,SAAA,WAAA,YAIxE,6BAA6B,MAAM,QAAQ"}
+{"version":3,"file":"MessageGroupedParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"mappings":";;;;;;kBAmBiB;;OAEH;aACD;aACA;aACA;aACA;;;;;;;;;OAUC,UAAU;aACX,MAAM;aACN,QAAQ,oBAAoB;;aAE5B,QAAQ;aACR;;;;;;;;;;;;OAaC;aACD;;;;;;;;;;;;;OAcC;OAEA,WAAW;;;;;;;aAOZ,MAAM,UAAU,QAAQ,oBAAoB;;;;;;;aAO5C,UAAU;;OAGT,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aA8BP,UACP,MAAM,WACN,SAAS,4BACG;;;;;;;;;aAUL,YAAY;;;;;;;;;;;;aAaZ,WAAW,MAAM,WAAW,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAoHtC;GAAgC,kCAA8B,SAAA,WAAA,YAIxE,6BAA6B,MAAM,QAAQ"}

package/dist/react/primitives/message/MessageGroupedParts.js

@@ -1,5 +1,5 @@
 "use client";
-import { getGroupStatus } from "../../../utils/getGroupStatus.js";
+import { getGroupSummary } from "../../../utils/getGroupStatus.js";
 import { MessagePartChildren } from "./MessageParts.js";
 import { GROUPBY_MEMO_KEY, buildGroupTree } from "../../utils/groupParts.js";
 import { Fragment, useMemo } from "@assistant-ui/tap/react-shim";
@@ -36,10 +36,11 @@ const renderNode = (node, parts, render) => {
 			children: /* @__PURE__ */ jsx(PartChildrenSentinel, {})
 		})
 	}, node.idKey ? `part-${node.idKey}` : `part-${node.index}`);
-	const status = getGroupStatus(parts, node.indices);
+	const { status, counts } = getGroupSummary(parts, node.indices);
 	const groupPart = {
 		type: node.key,
 		status,
+		counts,
 		indices: node.indices
 	};
 	return /* @__PURE__ */ jsx(Fragment, { children: render({

package/dist/react/primitives/message/MessageGroupedParts.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessageGroupedParts.js","names":["Fragment","FC","ReactNode","useMemo","useAuiState","useShallowSelector","PartState","MessagePartStatus","ToolCallMessagePartStatus","getGroupStatus","buildGroupTree","GROUPBY_MEMO_KEY","GroupByContext","GroupNode","MessagePartChildren","EnrichedPartState","MessagePrimitiveGroupedParts","GroupPart","type","TKey","status","indices","IndicatorPart","IndicatorMode","RenderInfo","part","children","Props","groupBy","context","indicator","info","shouldShowIndicator","mode","parts","isRunning","length","last","undefined","PartChildrenSentinel","Error","renderNode","node","render","idKey","index","groupPart","key","nodeKey","map","child","s","message","toolUIs","tools","memoKey","memoDep","tree","toolCallId","displayName"],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallowSelector } from \"@assistant-ui/store/internal\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport { getGroupStatus } from \"../../../utils/getGroupStatus\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupByContext,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` is running when any contained part runs,\n   * otherwise it mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    readonly indices: readonly number[];\n  };\n\n  /**\n   * Synthetic trailing slot for a streaming/loading affordance (a\n   * \"thinking…\" dot, etc.). Surfaced through the same `{ part }` channel\n   * as groups and leaf parts so a single `switch (part.type)` renders it\n   * via `case \"indicator\"`.\n   *\n   * It is only ever emitted while the message is running, so its presence\n   * alone means \"render your loading UI here\" — there's no `status` to\n   * branch on.\n   */\n  export type IndicatorPart = {\n    readonly type: \"indicator\";\n  };\n\n  /**\n   * When to emit the synthetic {@link IndicatorPart}. It is **only** emitted\n   * while the message is running (streaming); the mode further restricts\n   * which running states qualify:\n   * - `\"never\"` — never.\n   * - `\"empty\"` — only when the message has no parts yet.\n   * - `\"no-text\"` (default) — when the message has no parts yet or the last\n   *   part isn't `text`/`reasoning` (e.g. it ended on a tool call, so the\n   *   assistant likely isn't done).\n   * - `\"always\"` — whenever the message is running, regardless of parts.\n   */\n  export type IndicatorMode = \"never\" | \"empty\" | \"no-text\" | \"always\";\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`), a single enriched leaf part, or the synthetic\n     * {@link IndicatorPart} (`type: \"indicator\"`). Use one switch over\n     * `part.type` to handle all three.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState | IndicatorPart;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.toolName` or part metadata).\n     *\n     * The second argument is a {@link GroupByContext} carrying the tool-UI\n     * registry, for grouping that depends on it (e.g. standalone tool calls).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      context: GroupByContext,\n    ) => readonly TKey[] | null;\n\n    /**\n     * Controls emission of the synthetic {@link IndicatorPart} — a\n     * trailing `{ part: { type: \"indicator\", status } }` render call you\n     * handle with `case \"indicator\"` to show loading/status UI.\n     *\n     * @default \"no-text\"\n     * @see IndicatorMode\n     */\n    readonly indicator?: IndicatorMode;\n\n    /**\n     * Render function called once per group node, once per leaf part, and\n     * (when the `indicator` condition is met) once for the trailing\n     * {@link IndicatorPart}. Switch on `part.type`: `\"group-…\"` cases wrap\n     * `children`; real part types (`\"text\"`, `\"tool-call\"`, …) render the\n     * part directly; `\"indicator\"` renders status/loading UI.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst shouldShowIndicator = (\n  mode: MessagePrimitiveGroupedParts.IndicatorMode,\n  parts: readonly PartState[],\n  isRunning: boolean,\n): boolean => {\n  // The indicator is a streaming affordance — never show it on a settled\n  // message, whatever the mode.\n  if (!isRunning) return false;\n\n  switch (mode) {\n    case \"never\":\n      return false;\n    case \"always\":\n      return true;\n    case \"empty\":\n      return parts.length === 0;\n    case \"no-text\": {\n      const last = parts[parts.length - 1];\n      return (\n        last === undefined ||\n        (last.type !== \"text\" && last.type !== \"reasoning\")\n      );\n    }\n  }\n};\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by part identity when available, else absolute part index — never\n    // the structural nodeKey, which leaves zombie fiber subscriptions when\n    // parts reshape (#4051).\n    return (\n      <MessagePartChildren\n        key={node.idKey ? `part-${node.idKey}` : `part-${node.index}`}\n        index={node.index}\n      >\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const status = getGroupStatus(parts, node.indices);\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.idKey ?? node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       case \"indicator\":       return <LoadingDots />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  indicator = \"no-text\",\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallowSelector((s) => s.message.parts));\n  // Handed to `groupBy` as its `context` argument (see GroupByContext).\n  const toolUIs = useAuiState((s) => s.tools.toolUIs);\n  // Subscribe to a boolean, not the status object: the tree only needs to\n  // re-render when running-ness flips, and `\"never\"` opts out entirely.\n  const isRunning = useAuiState((s) =>\n    indicator === \"never\" ? false : s.message.status?.type === \"running\",\n  );\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(() => {\n    const context: GroupByContext = { toolUIs };\n    return buildGroupTree(\n      parts.map((part) => groupBy(part, context) ?? []),\n      parts.map((part) =>\n        part.type === \"tool-call\" ? part.toolCallId : undefined,\n      ),\n    );\n    // oxlint-disable-next-line react/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n  }, [parts, memoDep, toolUIs]);\n\n  return (\n    <>\n      {tree.map((node) => renderNode(node, parts, children))}\n      {shouldShowIndicator(indicator, parts, isRunning) &&\n        children({\n          part: { type: \"indicator\" },\n          children: <PartChildrenSentinel />,\n        })}\n    </>\n  );\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;;AAyIA,MAAMgC,uBACJC,MACAC,OACAC,cACY;CAGZ,IAAI,CAACA,WAAW,OAAO;CAEvB,QAAQF,MAAR;EACE,KAAK,SACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,SACH,OAAOC,MAAME,WAAW;EAC1B,KAAK,WAAW;GACd,MAAMC,OAAOH,MAAMA,MAAME,SAAS;GAClC,OACEC,SAASC,KAAAA,KACRD,KAAKnB,SAAS,UAAUmB,KAAKnB,SAAS;EAE3C;CACF;AACF;;;;;;;AAQA,MAAMqB,6BAAiC;CACrC,MAAM,IAAIC,MACR,8LAGF;AACF;AAEA,MAAMC,cACJC,MACAR,OACAS,WACc;CACd,IAAID,KAAKxB,SAAS,QAIhB,OACE,oBAAC,qBAAD;EAEE,OAAOwB,KAAKG;EAEV,WAAA,EAAEpB,WAAWkB,OAAO;GAAElB;GAAMC,UAAU,oBAAC,sBAAD,CAAqB,CAAA;EAAI,CAAC;CAC/C,GAJdgB,KAAKE,QAAQ,QAAQF,KAAKE,UAAU,QAAQF,KAAKG,OAInC;CAIzB,MAAMzB,SAASX,eAAeyB,OAAOQ,KAAKrB,OAAO;CACjD,MAAMyB,YAA0D;EAC9D5B,MAAMwB,KAAKK;EACX3B;EACAC,SAASqB,KAAKrB;CAChB;CAEA,OACE,oBAAC,UAAD,EAAA,UACGsB,OAAO;EACNlB,MAAMqB;EACNpB,UACE,oBAAA,YAAA,EAAA,UAAGgB,KAAKhB,SAASuB,KAAKC,UAAUT,WAAWS,OAAOhB,OAAOS,MAAM,CAAC,EAAE,CAAA;CAEtE,CAAC,EACO,GAPKD,KAAKE,SAASF,KAAKM,OAOxB;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAahC,gCAAgE,EAC3EY,SACAE,YAAY,WACZJ,eACyD;CACzD,MAAMQ,QAAQ9B,YAAYC,oBAAoB8C,MAAMA,EAAEC,QAAQlB,KAAK,CAAC;CAEpE,MAAMmB,UAAUjD,aAAa+C,QAAMA,IAAEG,MAAMD,OAAO;CAGlD,MAAMlB,YAAY/B,aAAa+C,QAC7BrB,cAAc,UAAU,QAAQqB,IAAEC,QAAQhC,QAAQF,SAAS,SAC7D;CAUA,MAAMsC,UAHW5B,QACfjB,qBAEyBiB;CAC3B,MAAM6B,OAAOtD,cAAc;EACzB,MAAM0B,UAA0B,EAAEwB,QAAQ;EAC1C,OAAO3C,eACLwB,MAAMe,KAAKxB,SAASG,QAAQH,MAAMI,OAAO,KAAK,CAAA,CAAE,GAChDK,MAAMe,KAAKxB,WACTA,OAAKP,SAAS,cAAcO,OAAKiC,aAAapB,KAAAA,CAChD,CACF;CAEF,GAAG;EAACJ;EAAOsB;EAASH;CAAO,CAAC;CAE5B,OACE,qBAAA,YAAA,EAAA,UAAA,CACGI,KAAKR,KAAKP,SAASD,WAAWC,MAAMR,OAAOR,QAAQ,CAAC,GACpDM,oBAAoBF,WAAWI,OAAOC,SAAS,KAC9CT,SAAS;EACPD,MAAM,EAAEP,MAAM,YAAY;EAC1BQ,UAAU,oBAAC,sBAAD,CAAqB,CAAA;CACjC,CAAC,CACL,EAAA,CAAA;AAEJ;AAEAV,6BAA6B2C,cAAc"}
+{"version":3,"file":"MessageGroupedParts.js","names":["Fragment","FC","ReactNode","useMemo","useAuiState","useShallowSelector","PartState","MessagePartStatus","ToolCallMessagePartStatus","getGroupSummary","buildGroupTree","GROUPBY_MEMO_KEY","GroupByContext","GroupNode","MessagePartChildren","EnrichedPartState","MessagePrimitiveGroupedParts","GroupCounts","running","complete","incomplete","requiresAction","GroupPart","type","TKey","status","counts","indices","IndicatorPart","IndicatorMode","RenderInfo","part","children","Props","groupBy","context","indicator","info","shouldShowIndicator","mode","parts","isRunning","length","last","undefined","PartChildrenSentinel","Error","renderNode","node","render","idKey","index","groupPart","key","nodeKey","map","child","s","message","toolUIs","tools","memoKey","memoDep","tree","toolCallId","displayName"],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallowSelector } from \"@assistant-ui/store/internal\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport { getGroupSummary } from \"../../../utils/getGroupStatus\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupByContext,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /** Per status tallies over the group's `indices`; they sum to `indices.length`. */\n  export type GroupCounts = {\n    readonly running: number;\n    readonly complete: number;\n    readonly incomplete: number;\n    readonly requiresAction: number;\n  };\n\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` is running when any contained part runs,\n   * otherwise it mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    /** Per status tallies over `indices`. */\n    readonly counts: GroupCounts;\n    readonly indices: readonly number[];\n  };\n\n  /**\n   * Synthetic trailing slot for a streaming/loading affordance (a\n   * \"thinking…\" dot, etc.). Surfaced through the same `{ part }` channel\n   * as groups and leaf parts so a single `switch (part.type)` renders it\n   * via `case \"indicator\"`.\n   *\n   * It is only ever emitted while the message is running, so its presence\n   * alone means \"render your loading UI here\" — there's no `status` to\n   * branch on.\n   */\n  export type IndicatorPart = {\n    readonly type: \"indicator\";\n  };\n\n  /**\n   * When to emit the synthetic {@link IndicatorPart}. It is **only** emitted\n   * while the message is running (streaming); the mode further restricts\n   * which running states qualify:\n   * - `\"never\"` — never.\n   * - `\"empty\"` — only when the message has no parts yet.\n   * - `\"no-text\"` (default) — when the message has no parts yet or the last\n   *   part isn't `text`/`reasoning` (e.g. it ended on a tool call, so the\n   *   assistant likely isn't done).\n   * - `\"always\"` — whenever the message is running, regardless of parts.\n   */\n  export type IndicatorMode = \"never\" | \"empty\" | \"no-text\" | \"always\";\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`), a single enriched leaf part, or the synthetic\n     * {@link IndicatorPart} (`type: \"indicator\"`). Use one switch over\n     * `part.type` to handle all three.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState | IndicatorPart;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` or tool name — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.parentId` or part metadata).\n     *\n     * The second argument is a {@link GroupByContext} carrying the tool-UI\n     * registry, for grouping that depends on it (e.g. standalone tool calls).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      context: GroupByContext,\n    ) => readonly TKey[] | null;\n\n    /**\n     * Controls emission of the synthetic {@link IndicatorPart} — a\n     * trailing `{ part: { type: \"indicator\", status } }` render call you\n     * handle with `case \"indicator\"` to show loading/status UI.\n     *\n     * @default \"no-text\"\n     * @see IndicatorMode\n     */\n    readonly indicator?: IndicatorMode;\n\n    /**\n     * Render function called once per group node, once per leaf part, and\n     * (when the `indicator` condition is met) once for the trailing\n     * {@link IndicatorPart}. Switch on `part.type`: `\"group-…\"` cases wrap\n     * `children`; real part types (`\"text\"`, `\"tool-call\"`, …) render the\n     * part directly; `\"indicator\"` renders status/loading UI.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst shouldShowIndicator = (\n  mode: MessagePrimitiveGroupedParts.IndicatorMode,\n  parts: readonly PartState[],\n  isRunning: boolean,\n): boolean => {\n  // The indicator is a streaming affordance — never show it on a settled\n  // message, whatever the mode.\n  if (!isRunning) return false;\n\n  switch (mode) {\n    case \"never\":\n      return false;\n    case \"always\":\n      return true;\n    case \"empty\":\n      return parts.length === 0;\n    case \"no-text\": {\n      const last = parts[parts.length - 1];\n      return (\n        last === undefined ||\n        (last.type !== \"text\" && last.type !== \"reasoning\")\n      );\n    }\n  }\n};\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by part identity when available, else absolute part index — never\n    // the structural nodeKey, which leaves zombie fiber subscriptions when\n    // parts reshape (#4051).\n    return (\n      <MessagePartChildren\n        key={node.idKey ? `part-${node.idKey}` : `part-${node.index}`}\n        index={node.index}\n      >\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const { status, counts } = getGroupSummary(parts, node.indices);\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    counts,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.idKey ?? node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       case \"indicator\":       return <LoadingDots />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  indicator = \"no-text\",\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallowSelector((s) => s.message.parts));\n  // Handed to `groupBy` as its `context` argument (see GroupByContext).\n  const toolUIs = useAuiState((s) => s.tools.toolUIs);\n  // Subscribe to a boolean, not the status object: the tree only needs to\n  // re-render when running-ness flips, and `\"never\"` opts out entirely.\n  const isRunning = useAuiState((s) =>\n    indicator === \"never\" ? false : s.message.status?.type === \"running\",\n  );\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(() => {\n    const context: GroupByContext = { toolUIs };\n    return buildGroupTree(\n      parts.map((part) => groupBy(part, context) ?? []),\n      parts.map((part) =>\n        part.type === \"tool-call\" ? part.toolCallId : undefined,\n      ),\n    );\n    // oxlint-disable-next-line react/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n  }, [parts, memoDep, toolUIs]);\n\n  return (\n    <>\n      {tree.map((node) => renderNode(node, parts, children))}\n      {shouldShowIndicator(indicator, parts, isRunning) &&\n        children({\n          part: { type: \"indicator\" },\n          children: <PartChildrenSentinel />,\n        })}\n    </>\n  );\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;;AAmJA,MAAMsC,uBACJC,MACAC,OACAC,cACY;CAGZ,IAAI,CAACA,WAAW,OAAO;CAEvB,QAAQF,MAAR;EACE,KAAK,SACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,SACH,OAAOC,MAAME,WAAW;EAC1B,KAAK,WAAW;GACd,MAAMC,OAAOH,MAAMA,MAAME,SAAS;GAClC,OACEC,SAASC,KAAAA,KACRD,KAAKpB,SAAS,UAAUoB,KAAKpB,SAAS;EAE3C;CACF;AACF;;;;;;;AAQA,MAAMsB,6BAAiC;CACrC,MAAM,IAAIC,MACR,8LAGF;AACF;AAEA,MAAMC,cACJC,MACAR,OACAS,WACc;CACd,IAAID,KAAKzB,SAAS,QAIhB,OACE,oBAAC,qBAAD;EAEE,OAAOyB,KAAKG;EAEV,WAAA,EAAEpB,WAAWkB,OAAO;GAAElB;GAAMC,UAAU,oBAAC,sBAAD,CAAqB,CAAA;EAAI,CAAC;CAC/C,GAJdgB,KAAKE,QAAQ,QAAQF,KAAKE,UAAU,QAAQF,KAAKG,OAInC;CAIzB,MAAM,EAAE1B,QAAQC,WAAWjB,gBAAgB+B,OAAOQ,KAAKrB,OAAO;CAC9D,MAAMyB,YAA0D;EAC9D7B,MAAMyB,KAAKK;EACX5B;EACAC;EACAC,SAASqB,KAAKrB;CAChB;CAEA,OACE,oBAAC,UAAD,EAAA,UACGsB,OAAO;EACNlB,MAAMqB;EACNpB,UACE,oBAAA,YAAA,EAAA,UAAGgB,KAAKhB,SAASuB,KAAKC,UAAUT,WAAWS,OAAOhB,OAAOS,MAAM,CAAC,EAAE,CAAA;CAEtE,CAAC,EACO,GAPKD,KAAKE,SAASF,KAAKM,OAOxB;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAatC,gCAAgE,EAC3EkB,SACAE,YAAY,WACZJ,eACyD;CACzD,MAAMQ,QAAQpC,YAAYC,oBAAoBoD,MAAMA,EAAEC,QAAQlB,KAAK,CAAC;CAEpE,MAAMmB,UAAUvD,aAAaqD,QAAMA,IAAEG,MAAMD,OAAO;CAGlD,MAAMlB,YAAYrC,aAAaqD,QAC7BrB,cAAc,UAAU,QAAQqB,IAAEC,QAAQjC,QAAQF,SAAS,SAC7D;CAUA,MAAMuC,UAHW5B,QACfvB,qBAEyBuB;CAC3B,MAAM6B,OAAO5D,cAAc;EACzB,MAAMgC,UAA0B,EAAEwB,QAAQ;EAC1C,OAAOjD,eACL8B,MAAMe,KAAKxB,SAASG,QAAQH,MAAMI,OAAO,KAAK,CAAA,CAAE,GAChDK,MAAMe,KAAKxB,WACTA,OAAKR,SAAS,cAAcQ,OAAKiC,aAAapB,KAAAA,CAChD,CACF;CAEF,GAAG;EAACJ;EAAOsB;EAASH;CAAO,CAAC;CAE5B,OACE,qBAAA,YAAA,EAAA,UAAA,CACGI,KAAKR,KAAKP,SAASD,WAAWC,MAAMR,OAAOR,QAAQ,CAAC,GACpDM,oBAAoBF,WAAWI,OAAOC,SAAS,KAC9CT,SAAS;EACPD,MAAM,EAAER,MAAM,YAAY;EAC1BS,UAAU,oBAAC,sBAAD,CAAqB,CAAA;CACjC,CAAC,CACL,EAAA,CAAA;AAEJ;AAEAhB,6BAA6BiD,cAAc"}

package/dist/react/primitives/part/PartMessages.d.ts

@@ -20,6 +20,7 @@ declare namespace PartPrimitiveMessages {
  * This primitive reads `messages` from the current tool call part in the PartScope
  * and renders them using a readonly thread context. All existing message and part
  * primitives work inside, and parent tool UI registrations are inherited.
+ * The nested thread reports `isRunning` while its trailing assistant message is running.
  *
  * @example
  * ```tsx

package/dist/react/primitives/part/PartMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"PartMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"mappings":";;;;kBASiB;OACH;IAEN,YAAY,YAAY,wBAAwB;IAChD;;;IAIA,WAAW;MAAS,SAAS;UAAoB;IACjD;;;;;;;;;;;;;;;;;;;;;;;;;;;cAsCK,2BAA2B,GAAG,sBAAsB;cAyBpD,uCAAqB,qBAAA,sBAAA"}
+{"version":3,"file":"PartMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"mappings":";;;;kBASiB;OACH;IAEN,YAAY,YAAY,wBAAwB;IAChD;;;IAIA,WAAW;MAAS,SAAS;UAAoB;IACjD;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCK,2BAA2B,GAAG,sBAAsB;cAyBpD,uCAAqB,qBAAA,sBAAA"}

package/dist/react/primitives/part/PartMessages.js

@@ -14,6 +14,7 @@ const usePartMessages = () => {
 * This primitive reads `messages` from the current tool call part in the PartScope
 * and renders them using a readonly thread context. All existing message and part
 * primitives work inside, and parent tool UI registrations are inherited.
+* The nested thread reports `isRunning` while its trailing assistant message is running.
 *
 * @example
 * ```tsx

package/dist/react/primitives/part/PartMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"PartMessages.js","names":["FC","ReactNode","memo","useAuiState","ThreadMessage","ReadonlyThreadProvider","ThreadPrimitiveMessages","ThreadPrimitiveMessagesImpl","PartPrimitiveMessages","Props","components","NonNullable","children","value","message","usePartMessages","_temp","PartPrimitiveMessagesImpl","t0","$","_c","messages","length","t1","t2","displayName","s","part","type","undefined"],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"sourcesContent":["import { type FC, type ReactNode, memo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport type { ThreadMessage } from \"../../../types/message\";\nimport { ReadonlyThreadProvider } from \"../../providers/ReadonlyThreadProvider\";\nimport {\n  type ThreadPrimitiveMessages,\n  ThreadPrimitiveMessagesImpl,\n} from \"../thread/ThreadMessages\";\n\nexport namespace PartPrimitiveMessages {\n  export type Props =\n    | {\n        components: NonNullable<ThreadPrimitiveMessages.Props[\"components\"]>;\n        children?: never;\n      }\n    | {\n        /** Render function called for each message. Receives the message. */\n        children: (value: { message: ThreadMessage }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst usePartMessages = (): readonly ThreadMessage[] | undefined => {\n  return useAuiState((s) => {\n    const part = s.part;\n    if (part.type !== \"tool-call\") return undefined;\n    return \"messages\" in part\n      ? (part.messages as readonly ThreadMessage[] | undefined)\n      : undefined;\n  });\n};\n\n/**\n * Renders the nested messages of a tool call part (e.g. sub-agent conversation).\n *\n * This primitive reads `messages` from the current tool call part in the PartScope\n * and renders them using a readonly thread context. All existing message and part\n * primitives work inside, and parent tool UI registrations are inherited.\n *\n * @example\n * ```tsx\n * const toolkit = defineToolkit({\n *   invoke_sub_agent: {\n *     type: \"backend\",\n *     render: () => (\n *       <PartPrimitive.Messages>\n *         {({ message }) => {\n *           if (message.role === \"user\") return <MyUserMessage />;\n *           return <MyAssistantMessage />;\n *         }}\n *       </PartPrimitive.Messages>\n *     ),\n *   },\n * });\n * ```\n */\nexport const PartPrimitiveMessagesImpl: FC<PartPrimitiveMessages.Props> = ({\n  components,\n  children,\n}) => {\n  const messages = usePartMessages();\n\n  if (!messages?.length) return null;\n\n  if (components) {\n    return (\n      <ReadonlyThreadProvider messages={messages}>\n        <ThreadPrimitiveMessagesImpl components={components} />\n      </ReadonlyThreadProvider>\n    );\n  }\n\n  return (\n    <ReadonlyThreadProvider messages={messages}>\n      <ThreadPrimitiveMessagesImpl>{children}</ThreadPrimitiveMessagesImpl>\n    </ReadonlyThreadProvider>\n  );\n};\n\nPartPrimitiveMessagesImpl.displayName = \"PartPrimitive.Messages\";\n\nexport const PartPrimitiveMessages = memo(PartPrimitiveMessagesImpl);\n"],"mappings":";;;;;;;AAsBA,MAAMe,wBAAkB;CAAA,OACfZ,YAAYa,KAMlB;AAAC;;;;;;;;;;;;;;;;;;;;;;;;;AA2BJ,MAAaC,6BAA6DC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAV,YAAAE,aAAAM;CAIzE,MAAAG,WAAiBN,gBAAgB;CAEjC,IAAI,CAACM,UAAQC,QAAQ,OAAS;CAE9B,IAAIZ,YAAU;EAAA,IAAAa;EAAA,IAAAJ,EAAA,OAAAT,YAAA;GAGRa,KAAA,oBAAC,6BAAD,EAAyCb,WAAU,CAAA;GAAIS,EAAA,KAAAT;GAAAS,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAAA,IAAAK;EAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;GADzDC,KAAA,oBAAC,wBAAD;IAAkCH;IAChCE,UAAAA;GADqB,CAAA;GAEEJ,EAAA,KAAAE;GAAAF,EAAA,KAAAI;GAAAJ,EAAA,KAAAK;EAAA,OAAAA,KAAAL,EAAA;EAAA,OAFzBK;CAEyB;CAE5B,IAAAD;CAAA,IAAAJ,EAAA,OAAAP,UAAA;EAIGW,KAAA,oBAAC,6BAAD,EAA8BX,SAAF,CAAA;EAAyCO,EAAA,KAAAP;EAAAO,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAK;CAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;EADvEC,KAAA,oBAAC,wBAAD;GAAkCH;GAChCE,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAE;EAAAF,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAAA,OAFzBK;AAEyB;AAI7BP,0BAA0BQ,cAAc;AAExC,MAAajB,wBAAwBN,KAAKe,yBAAyB;AA3D3C,SAAAD,MAAAU,GAAA;CAEpB,MAAAC,OAAaD,EAACC;CACd,IAAIA,KAAIC,SAAU,aAAW;CAAmB,OACzC,cAAcD,OAChBA,KAAIN,WADFQ,KAAAA;AAEM"}
+{"version":3,"file":"PartMessages.js","names":["FC","ReactNode","memo","useAuiState","ThreadMessage","ReadonlyThreadProvider","ThreadPrimitiveMessages","ThreadPrimitiveMessagesImpl","PartPrimitiveMessages","Props","components","NonNullable","children","value","message","usePartMessages","_temp","PartPrimitiveMessagesImpl","t0","$","_c","messages","length","t1","t2","displayName","s","part","type","undefined"],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"sourcesContent":["import { type FC, type ReactNode, memo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport type { ThreadMessage } from \"../../../types/message\";\nimport { ReadonlyThreadProvider } from \"../../providers/ReadonlyThreadProvider\";\nimport {\n  type ThreadPrimitiveMessages,\n  ThreadPrimitiveMessagesImpl,\n} from \"../thread/ThreadMessages\";\n\nexport namespace PartPrimitiveMessages {\n  export type Props =\n    | {\n        components: NonNullable<ThreadPrimitiveMessages.Props[\"components\"]>;\n        children?: never;\n      }\n    | {\n        /** Render function called for each message. Receives the message. */\n        children: (value: { message: ThreadMessage }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst usePartMessages = (): readonly ThreadMessage[] | undefined => {\n  return useAuiState((s) => {\n    const part = s.part;\n    if (part.type !== \"tool-call\") return undefined;\n    return \"messages\" in part\n      ? (part.messages as readonly ThreadMessage[] | undefined)\n      : undefined;\n  });\n};\n\n/**\n * Renders the nested messages of a tool call part (e.g. sub-agent conversation).\n *\n * This primitive reads `messages` from the current tool call part in the PartScope\n * and renders them using a readonly thread context. All existing message and part\n * primitives work inside, and parent tool UI registrations are inherited.\n * The nested thread reports `isRunning` while its trailing assistant message is running.\n *\n * @example\n * ```tsx\n * const toolkit = defineToolkit({\n *   invoke_sub_agent: {\n *     type: \"backend\",\n *     render: () => (\n *       <PartPrimitive.Messages>\n *         {({ message }) => {\n *           if (message.role === \"user\") return <MyUserMessage />;\n *           return <MyAssistantMessage />;\n *         }}\n *       </PartPrimitive.Messages>\n *     ),\n *   },\n * });\n * ```\n */\nexport const PartPrimitiveMessagesImpl: FC<PartPrimitiveMessages.Props> = ({\n  components,\n  children,\n}) => {\n  const messages = usePartMessages();\n\n  if (!messages?.length) return null;\n\n  if (components) {\n    return (\n      <ReadonlyThreadProvider messages={messages}>\n        <ThreadPrimitiveMessagesImpl components={components} />\n      </ReadonlyThreadProvider>\n    );\n  }\n\n  return (\n    <ReadonlyThreadProvider messages={messages}>\n      <ThreadPrimitiveMessagesImpl>{children}</ThreadPrimitiveMessagesImpl>\n    </ReadonlyThreadProvider>\n  );\n};\n\nPartPrimitiveMessagesImpl.displayName = \"PartPrimitive.Messages\";\n\nexport const PartPrimitiveMessages = memo(PartPrimitiveMessagesImpl);\n"],"mappings":";;;;;;;AAsBA,MAAMe,wBAAkB;CAAA,OACfZ,YAAYa,KAMlB;AAAC;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BJ,MAAaC,6BAA6DC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAV,YAAAE,aAAAM;CAIzE,MAAAG,WAAiBN,gBAAgB;CAEjC,IAAI,CAACM,UAAQC,QAAQ,OAAS;CAE9B,IAAIZ,YAAU;EAAA,IAAAa;EAAA,IAAAJ,EAAA,OAAAT,YAAA;GAGRa,KAAA,oBAAC,6BAAD,EAAyCb,WAAU,CAAA;GAAIS,EAAA,KAAAT;GAAAS,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAAA,IAAAK;EAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;GADzDC,KAAA,oBAAC,wBAAD;IAAkCH;IAChCE,UAAAA;GADqB,CAAA;GAEEJ,EAAA,KAAAE;GAAAF,EAAA,KAAAI;GAAAJ,EAAA,KAAAK;EAAA,OAAAA,KAAAL,EAAA;EAAA,OAFzBK;CAEyB;CAE5B,IAAAD;CAAA,IAAAJ,EAAA,OAAAP,UAAA;EAIGW,KAAA,oBAAC,6BAAD,EAA8BX,SAAF,CAAA;EAAyCO,EAAA,KAAAP;EAAAO,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAK;CAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;EADvEC,KAAA,oBAAC,wBAAD;GAAkCH;GAChCE,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAE;EAAAF,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAAA,OAFzBK;AAEyB;AAI7BP,0BAA0BQ,cAAc;AAExC,MAAajB,wBAAwBN,KAAKe,yBAAyB;AA5D3C,SAAAD,MAAAU,GAAA;CAEpB,MAAAC,OAAaD,EAACC;CACd,IAAIA,KAAIC,SAAU,aAAW;CAAmB,OACzC,cAAcD,OAChBA,KAAIN,WADFQ,KAAAA;AAEM"}

package/dist/react/primitives/thread/ThreadMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/thread/ThreadMessages.tsx"],"mappings":";;;;KAYK;;EAGC,SAAS;;EAET,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,cAAc;;EAEd,mBAAmB;;EAEnB,gBAAgB;;;EAIhB,UAAU;;EAEV,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,aAAa;;EAEb,kBAAkB;;EAElB,gBAAgB;;kBAGL;OACH;;IAGN,YAAY;IACZ;;;IAIA,WAAW;MAAS,SAAS;UAAmB;IAChD;;;kBAsGS;OACH;IACV;IACA,YAAY;;;;;;cAOH,+BAA+B,GAAG,8BAA8B;kBAgB5D;OACH;IACV;IACA,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;cA2BH,qCAAqC,GAAG,oCAAoC;;;;;;;;;;;;;;cA4D5E,6BAA6B,GAAG,wBAAwB;cAkBxD,yCAAuB,qBAAA,wBAAA"}
+{"version":3,"file":"ThreadMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/thread/ThreadMessages.tsx"],"mappings":";;;;KAaK;;EAGC,SAAS;;EAET,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,cAAc;;EAEd,mBAAmB;;EAEnB,gBAAgB;;;EAIhB,UAAU;;EAEV,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,aAAa;;EAEb,kBAAkB;;EAElB,gBAAgB;;kBAGL;OACH;;IAGN,YAAY;IACZ;;;IAIA,WAAW;MAAS,SAAS;UAAmB;IAChD;;;kBAsGS;OACH;IACV;IACA,YAAY;;;;;;cAOH,+BAA+B,GAAG,8BAA8B;kBAgB5D;OACH;IACV;IACA,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;cA2BH,qCAAqC,GAAG,oCAAoC;;;;;;;;;;;;;;cA8D5E,6BAA6B,GAAG,wBAAwB;cAkBxD,yCAAuB,qBAAA,wBAAA"}

package/dist/react/primitives/thread/ThreadMessages.js

@@ -3,6 +3,7 @@ import { MessageByIdProvider } from "../../providers/MessageByIdProvider.js";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { memo, useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
+import { useShallowSelector } from "@assistant-ui/store/internal";
 import { jsx } from "react/jsx-runtime";
 //#region src/react/primitives/thread/ThreadMessages.tsx
 const isComponentsSame = (prev, next) => {
@@ -129,10 +130,10 @@ const ThreadPrimitiveUnstable_MessageById = memo((t0) => {
 }, (prev, next) => prev.messageId === next.messageId && isComponentsSame(prev.components, next.components));
 ThreadPrimitiveUnstable_MessageById.displayName = "ThreadPrimitive.Unstable_MessageById";
 const ThreadPrimitiveMessagesInner = ({ children }) => {
-	const messagesLength = useAuiState((s) => s.thread.messages.length);
+	const messageIds = useAuiState(useShallowSelector((s) => s.thread.messages.map((message) => message.id)));
 	return useMemo(() => {
-		if (messagesLength === 0) return null;
-		return Array.from({ length: messagesLength }, (_, index) => /* @__PURE__ */ jsx(MessageByIndexProvider, {
+		if (messageIds.length === 0) return null;
+		return messageIds.map((messageId, index) => /* @__PURE__ */ jsx(MessageByIndexProvider, {
 			index,
 			children: /* @__PURE__ */ jsx(RenderChildrenWithAccessor, {
 				getItemState: (aui) => aui.thread.message({ index }).getState(),
@@ -140,8 +141,8 @@ const ThreadPrimitiveMessagesInner = ({ children }) => {
 					return getItem();
 				} })
 			})
-		}, index));
-	}, [messagesLength, children]);
+		}, messageId));
+	}, [messageIds, children]);
 };
 /**
 * Renders all messages in the current thread.