@langchain/langgraph-sdk 1.8.10 → 1.9.1

package/dist/ui/messages.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"messages.cjs","names":["HumanMessageChunk","AIMessageChunk","SystemMessageChunk","ToolMessageChunk","RemoveMessage"],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  isBaseMessageChunk,\n  HumanMessageChunk,\n  SystemMessageChunk,\n  AIMessageChunk,\n  ToolMessageChunk,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends { messages: unknown }\n  ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (isBaseMessageChunk(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(message);\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToMessage(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>,\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    if (state.values == null) return state;\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;;AAiCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,OAAA,GAAA,yBAAA,oBAAuB,QAAQ,CAAE,QAAO;AACxC,UAAA,GAAA,yBAAA,gBAAsB,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAIA,yBAAAA,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAIC,yBAAAA,eAAe,QAAQ;AAGpC,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAIC,yBAAAA,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAIC,yBAAAA,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAIC,yBAAAA,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,SAAA,GAAA,yBAAA,4BAAkC,QAAQ;;AAG5C,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,8BAA8B,WAAW;EACzD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,UAAA,GAAA,yBAAA,oBACM,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAAI;QAE7D,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;AAC5B,MAAI,MAAM,UAAU,KAAM,QAAO;EACjC,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}
+{"version":3,"file":"messages.cjs","names":["BaseMessageChunk","HumanMessage","AIMessage","SystemMessage","ToolMessage","RemoveMessage","HumanMessageChunk","AIMessageChunk","SystemMessageChunk","ToolMessageChunk"],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  HumanMessageChunk,\n  HumanMessage,\n  SystemMessageChunk,\n  SystemMessage,\n  AIMessageChunk,\n  AIMessage,\n  ToolMessageChunk,\n  ToolMessage,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends { messages: unknown }\n  ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (BaseMessageChunk.isInstance(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessage(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessage(normalizeAIMessageToolCalls(message));\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessage(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessage({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nfunction tryCoerceMessageLikeToChunk(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(normalizeAIMessageToolCalls(message));\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  return tryCoerceMessageLikeToMessage(message);\n}\n\ntype ToolCallLike = {\n  id?: string;\n  name?: string;\n  args?: unknown;\n  input?: unknown;\n};\n\nfunction normalizeAIMessageToolCalls<\n  T extends Omit<Message, \"type\"> & { type: string },\n>(message: T): T {\n  const record = message as T & {\n    content?: unknown;\n    tool_calls?: unknown;\n  };\n  if (Array.isArray(record.tool_calls) && record.tool_calls.length > 0) {\n    return message;\n  }\n\n  const toolCalls = extractToolCallsFromContent(record.content);\n  if (toolCalls.length === 0) return message;\n  return {\n    ...message,\n    tool_calls: toolCalls,\n  };\n}\n\nfunction extractToolCallsFromContent(content: unknown) {\n  if (!Array.isArray(content)) return [];\n  return content.flatMap(\n    (\n      block\n    ): Array<{\n      id: string;\n      name: string;\n      args: Record<string, unknown>;\n      type: \"tool_call\";\n    }> => {\n      if (block == null || typeof block !== \"object\") return [];\n      const record = block as ToolCallLike & { type?: unknown };\n      if (record.type !== \"tool_call\" && record.type !== \"tool_use\") return [];\n      return [\n        {\n          id: record.id ?? \"\",\n          name: record.name ?? \"\",\n          args: normalizeToolCallArgs(record.args ?? record.input),\n          type: \"tool_call\",\n        },\n      ];\n    }\n  );\n}\n\nfunction normalizeToolCallArgs(value: unknown): Record<string, unknown> {\n  if (value != null && typeof value === \"object\" && !Array.isArray(value)) {\n    return value as Record<string, unknown>;\n  }\n  if (typeof value === \"string\" && value.length > 0) {\n    try {\n      const parsed = JSON.parse(value);\n      if (\n        parsed != null &&\n        typeof parsed === \"object\" &&\n        !Array.isArray(parsed)\n      ) {\n        return parsed as Record<string, unknown>;\n      }\n    } catch {\n      // Streaming input fragments are expected to be invalid until finalized.\n    }\n  }\n  return {};\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToChunk(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (BaseMessageChunk.isInstance(prev) ? prev : null)?.concat(chunk) ??\n        chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>,\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    if (state.values == null) return state;\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;;AAoCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,MAAIA,yBAAAA,iBAAiB,WAAW,QAAQ,CAAE,QAAO;AACjD,UAAA,GAAA,yBAAA,gBAAsB,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAIC,yBAAAA,aAAa,QAAQ;AAGlC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAIC,yBAAAA,UAAU,4BAA4B,QAAQ,CAAC;AAG5D,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAIC,yBAAAA,cAAc,QAAQ;AAGnC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAIC,yBAAAA,YAAY;EACrB,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAIC,yBAAAA,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,SAAA,GAAA,yBAAA,4BAAkC,QAAQ;;AAG5C,SAAS,4BACP,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAIC,yBAAAA,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAIC,yBAAAA,eAAe,4BAA4B,QAAQ,CAAC;AAGjE,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAIC,yBAAAA,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAIC,yBAAAA,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,QAAO,8BAA8B,QAAQ;;AAU/C,SAAS,4BAEP,SAAe;CACf,MAAM,SAAS;AAIf,KAAI,MAAM,QAAQ,OAAO,WAAW,IAAI,OAAO,WAAW,SAAS,EACjE,QAAO;CAGT,MAAM,YAAY,4BAA4B,OAAO,QAAQ;AAC7D,KAAI,UAAU,WAAW,EAAG,QAAO;AACnC,QAAO;EACL,GAAG;EACH,YAAY;EACb;;AAGH,SAAS,4BAA4B,SAAkB;AACrD,KAAI,CAAC,MAAM,QAAQ,QAAQ,CAAE,QAAO,EAAE;AACtC,QAAO,QAAQ,SAEX,UAMI;AACJ,MAAI,SAAS,QAAQ,OAAO,UAAU,SAAU,QAAO,EAAE;EACzD,MAAM,SAAS;AACf,MAAI,OAAO,SAAS,eAAe,OAAO,SAAS,WAAY,QAAO,EAAE;AACxE,SAAO,CACL;GACE,IAAI,OAAO,MAAM;GACjB,MAAM,OAAO,QAAQ;GACrB,MAAM,sBAAsB,OAAO,QAAQ,OAAO,MAAM;GACxD,MAAM;GACP,CACF;GAEJ;;AAGH,SAAS,sBAAsB,OAAyC;AACtE,KAAI,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACrE,QAAO;AAET,KAAI,OAAO,UAAU,YAAY,MAAM,SAAS,EAC9C,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,MAAM;AAChC,MACE,UAAU,QACV,OAAO,WAAW,YAClB,CAAC,MAAM,QAAQ,OAAO,CAEtB,QAAO;SAEH;AAIV,QAAO,EAAE;;AAGX,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,4BAA4B,WAAW;EACvD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,SACbT,yBAAAA,iBAAiB,WAAW,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAChE;QAEF,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;AAC5B,MAAI,MAAM,UAAU,KAAM,QAAO;EACjC,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}

package/dist/ui/messages.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"messages.d.cts","names":[],"sources":["../../src/ui/messages.ts"],"mappings":";;;;;;;;;AAqBA;;KAAY,qBAAA,MAA2B,CAAA;EAAY,QAAA;AAAA,IAC/C,IAAA,CAAK,CAAA;EAAmB,QAAA,EAAU,WAAA;AAAA,IAClC,CAAA;;;;;KAMQ,uBAAA,MAA6B,CAAA,SAAU,WAAA,cAC/C,WAAA,CAAY,qBAAA,CAAsB,CAAA,OAClC,CAAA;AAAA,cA0CS,mBAAA;EACX,MAAA,EAAQ,MAAA;IAGJ,KAAA,GAAQ,gBAAA,GAAmB,WAAA;IAC3B,QAAA,GAAW,MAAA;IACX,KAAA;EAAA;EAIJ,WAAA,CAAA;EAIA,GAAA,CACE,UAAA,EAAY,OAAA,EACZ,QAAA,EAAU,MAAA;EAoCZ,KAAA,CAAA;EAIA,GAAA,CAAI,EAAA,6BAA+B,YAAA;mEAxCjB,yBAAA,CAAA,cAAA;;;;;cAgDP,aAAA,GAAa,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAGzB,yBAAA,CAHyB,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,OAAA;;;;;cASb,cAAA,GAAc,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAA6C,yBAAA,CAA7C,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAAA,yBAAA,CAAA,cAAA,GAAA,yBAAA,CAAA,WAAA;;;;;AAzE3B;;iBAiFgB,sBAAA,CACd,QAAA,GAAW,OAAA,GAAU,WAAA,OACnB,WAAA,GAAc,gBAAA;;;;;;iBAgBF,6BAAA,mBACI,MAAA,kBAAA,CAElB,OAAA,EAAS,WAAA,CAAY,SAAA,KACrB,WAAA,YACC,WAAA,CAAY,SAAA"}
+{"version":3,"file":"messages.d.cts","names":[],"sources":["../../src/ui/messages.ts"],"mappings":";;;;;;;;;AAwBA;;KAAY,qBAAA,MAA2B,CAAA;EAAY,QAAA;AAAA,IAC/C,IAAA,CAAK,CAAA;EAAmB,QAAA,EAAU,WAAA;AAAA,IAClC,CAAA;;;;;KAMQ,uBAAA,MAA6B,CAAA,SAAU,WAAA,cAC/C,WAAA,CAAY,qBAAA,CAAsB,CAAA,OAClC,CAAA;AAAA,cA4IS,mBAAA;EACX,MAAA,EAAQ,MAAA;IAGJ,KAAA,GAAQ,gBAAA,GAAmB,WAAA;IAC3B,QAAA,GAAW,MAAA;IACX,KAAA;EAAA;EAIJ,WAAA,CAAA;EAIA,GAAA,CACE,UAAA,EAAY,OAAA,EACZ,QAAA,EAAU,MAAA;EAqCZ,KAAA,CAAA;EAIA,GAAA,CAAI,EAAA,6BAA+B,YAAA;mEAzCjB,yBAAA,CAAA,cAAA;;;;;cAiDP,aAAA,GAAa,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAGzB,yBAAA,CAHyB,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,OAAA;;;;;cASb,cAAA,GAAc,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAA6C,yBAAA,CAA7C,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAAA,yBAAA,CAAA,cAAA,GAAA,yBAAA,CAAA,WAAA;;;;;AA1E3B;;iBAkFgB,sBAAA,CACd,QAAA,GAAW,OAAA,GAAU,WAAA,OACnB,WAAA,GAAc,gBAAA;;;;;;iBAgBF,6BAAA,mBACI,MAAA,kBAAA,CAElB,OAAA,EAAS,WAAA,CAAY,SAAA,KACrB,WAAA,YACC,WAAA,CAAY,SAAA"}

package/dist/ui/messages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"messages.d.ts","names":[],"sources":["../../src/ui/messages.ts"],"mappings":";;;;;;;;;AAqBA;;KAAY,qBAAA,MAA2B,CAAA;EAAY,QAAA;AAAA,IAC/C,IAAA,CAAK,CAAA;EAAmB,QAAA,EAAU,WAAA;AAAA,IAClC,CAAA;;;;;KAMQ,uBAAA,MAA6B,CAAA,SAAU,WAAA,cAC/C,WAAA,CAAY,qBAAA,CAAsB,CAAA,OAClC,CAAA;AAAA,cA0CS,mBAAA;EACX,MAAA,EAAQ,MAAA;IAGJ,KAAA,GAAQ,gBAAA,GAAmB,WAAA;IAC3B,QAAA,GAAW,MAAA;IACX,KAAA;EAAA;EAIJ,WAAA,CAAA;EAIA,GAAA,CACE,UAAA,EAAY,OAAA,EACZ,QAAA,EAAU,MAAA;EAoCZ,KAAA,CAAA;EAIA,GAAA,CAAI,EAAA,6BAA+B,YAAA;mEAxCjB,yBAAA,CAAA,cAAA;;;;;cAgDP,aAAA,GAAa,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAGzB,yBAAA,CAHyB,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,OAAA;;;;;cASb,cAAA,GAAc,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAA6C,yBAAA,CAA7C,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAAA,yBAAA,CAAA,cAAA,GAAA,yBAAA,CAAA,WAAA;;;;;AAzE3B;;iBAiFgB,sBAAA,CACd,QAAA,GAAW,OAAA,GAAU,WAAA,OACnB,WAAA,GAAc,gBAAA;;;;;;iBAgBF,6BAAA,mBACI,MAAA,kBAAA,CAElB,OAAA,EAAS,WAAA,CAAY,SAAA,KACrB,WAAA,YACC,WAAA,CAAY,SAAA"}
+{"version":3,"file":"messages.d.ts","names":[],"sources":["../../src/ui/messages.ts"],"mappings":";;;;;;;;;AAwBA;;KAAY,qBAAA,MAA2B,CAAA;EAAY,QAAA;AAAA,IAC/C,IAAA,CAAK,CAAA;EAAmB,QAAA,EAAU,WAAA;AAAA,IAClC,CAAA;;;;;KAMQ,uBAAA,MAA6B,CAAA,SAAU,WAAA,cAC/C,WAAA,CAAY,qBAAA,CAAsB,CAAA,OAClC,CAAA;AAAA,cA4IS,mBAAA;EACX,MAAA,EAAQ,MAAA;IAGJ,KAAA,GAAQ,gBAAA,GAAmB,WAAA;IAC3B,QAAA,GAAW,MAAA;IACX,KAAA;EAAA;EAIJ,WAAA,CAAA;EAIA,GAAA,CACE,UAAA,EAAY,OAAA,EACZ,QAAA,EAAU,MAAA;EAqCZ,KAAA,CAAA;EAIA,GAAA,CAAI,EAAA,6BAA+B,YAAA;mEAzCjB,yBAAA,CAAA,cAAA;;;;;cAiDP,aAAA,GAAa,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAGzB,yBAAA,CAHyB,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,OAAA;;;;;cASb,cAAA,GAAc,KAAA,EAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAA6C,yBAAA,CAA7C,cAAA,GAAA,yBAAA,CAAA,WAAA,MAAA,WAAA,CAAA,yBAAA,CAAA,gBAAA,CAAA,yBAAA,CAAA,cAAA,GAAA,yBAAA,CAAA,WAAA;;;;;AA1E3B;;iBAkFgB,sBAAA,CACd,QAAA,GAAW,OAAA,GAAU,WAAA,OACnB,WAAA,GAAc,gBAAA;;;;;;iBAgBF,6BAAA,mBACI,MAAA,kBAAA,CAElB,OAAA,EAAS,WAAA,CAAY,SAAA,KACrB,WAAA,YACC,WAAA,CAAY,SAAA"}

package/dist/ui/messages.js

@@ -1,18 +1,18 @@
-import { AIMessageChunk, HumanMessageChunk, RemoveMessage, SystemMessageChunk, ToolMessageChunk, coerceMessageLikeToMessage, convertToChunk, isBaseMessageChunk } from "@langchain/core/messages";
+import { AIMessage, AIMessageChunk, BaseMessageChunk, HumanMessage, HumanMessageChunk, RemoveMessage, SystemMessage, SystemMessageChunk, ToolMessage, ToolMessageChunk, coerceMessageLikeToMessage, convertToChunk } from "@langchain/core/messages";
 //#region src/ui/messages.ts
 function tryConvertToChunk(message) {
 	try {
-		if (isBaseMessageChunk(message)) return message;
+		if (BaseMessageChunk.isInstance(message)) return message;
 		return convertToChunk(message);
 	} catch {
 		return null;
 	}
 }
 function tryCoerceMessageLikeToMessage(message) {
-	if (message.type === "human" || message.type === "user") return new HumanMessageChunk(message);
-	if (message.type === "ai" || message.type === "assistant") return new AIMessageChunk(message);
-	if (message.type === "system") return new SystemMessageChunk(message);
-	if (message.type === "tool" && "tool_call_id" in message) return new ToolMessageChunk({
+	if (message.type === "human" || message.type === "user") return new HumanMessage(message);
+	if (message.type === "ai" || message.type === "assistant") return new AIMessage(normalizeAIMessageToolCalls(message));
+	if (message.type === "system") return new SystemMessage(message);
+	if (message.type === "tool" && "tool_call_id" in message) return new ToolMessage({
 		...message,
 		tool_call_id: message.tool_call_id
 	});
@@ -22,6 +22,48 @@ function tryCoerceMessageLikeToMessage(message) {
 	});
 	return coerceMessageLikeToMessage(message);
 }
+function tryCoerceMessageLikeToChunk(message) {
+	if (message.type === "human" || message.type === "user") return new HumanMessageChunk(message);
+	if (message.type === "ai" || message.type === "assistant") return new AIMessageChunk(normalizeAIMessageToolCalls(message));
+	if (message.type === "system") return new SystemMessageChunk(message);
+	if (message.type === "tool" && "tool_call_id" in message) return new ToolMessageChunk({
+		...message,
+		tool_call_id: message.tool_call_id
+	});
+	return tryCoerceMessageLikeToMessage(message);
+}
+function normalizeAIMessageToolCalls(message) {
+	const record = message;
+	if (Array.isArray(record.tool_calls) && record.tool_calls.length > 0) return message;
+	const toolCalls = extractToolCallsFromContent(record.content);
+	if (toolCalls.length === 0) return message;
+	return {
+		...message,
+		tool_calls: toolCalls
+	};
+}
+function extractToolCallsFromContent(content) {
+	if (!Array.isArray(content)) return [];
+	return content.flatMap((block) => {
+		if (block == null || typeof block !== "object") return [];
+		const record = block;
+		if (record.type !== "tool_call" && record.type !== "tool_use") return [];
+		return [{
+			id: record.id ?? "",
+			name: record.name ?? "",
+			args: normalizeToolCallArgs(record.args ?? record.input),
+			type: "tool_call"
+		}];
+	});
+}
+function normalizeToolCallArgs(value) {
+	if (value != null && typeof value === "object" && !Array.isArray(value)) return value;
+	if (typeof value === "string" && value.length > 0) try {
+		const parsed = JSON.parse(value);
+		if (parsed != null && typeof parsed === "object" && !Array.isArray(parsed)) return parsed;
+	} catch {}
+	return {};
+}
 var MessageTupleManager = class {
 	chunks = {};
 	constructor() {
@@ -29,7 +71,7 @@ var MessageTupleManager = class {
 	}
 	add(serialized, metadata) {
 		if (serialized.type.endsWith("MessageChunk")) serialized.type = serialized.type.slice(0, -12).toLowerCase();
-		const message = tryCoerceMessageLikeToMessage(serialized);
+		const message = tryCoerceMessageLikeToChunk(serialized);
 		const chunk = tryConvertToChunk(message);
 		const { id } = chunk ?? message;
 		if (!id) {
@@ -40,7 +82,7 @@ var MessageTupleManager = class {
 		this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;
 		if (chunk) {
 			const prev = this.chunks[id].chunk;
-			this.chunks[id].chunk = (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;
+			this.chunks[id].chunk = (BaseMessageChunk.isInstance(prev) ? prev : null)?.concat(chunk) ?? chunk;
 		} else this.chunks[id].chunk = message;
 		return id;
 	}
@@ -98,6 +140,6 @@ function ensureHistoryMessageInstances(history, messagesKey = "messages") {
 	});
 }
 //#endregion
-export { MessageTupleManager, ensureHistoryMessageInstances, ensureMessageInstances, toMessageClass, toMessageDict };
+export { MessageTupleManager, ensureHistoryMessageInstances, ensureMessageInstances, toMessageClass, toMessageDict, tryCoerceMessageLikeToMessage };
 
 //# sourceMappingURL=messages.js.map

package/dist/ui/messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"messages.js","names":[],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  isBaseMessageChunk,\n  HumanMessageChunk,\n  SystemMessageChunk,\n  AIMessageChunk,\n  ToolMessageChunk,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends { messages: unknown }\n  ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (isBaseMessageChunk(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(message);\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToMessage(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>,\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    if (state.values == null) return state;\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;AAiCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,MAAI,mBAAmB,QAAQ,CAAE,QAAO;AACxC,SAAO,eAAe,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAI,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAI,eAAe,QAAQ;AAGpC,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAI,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAI,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAI,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,QAAO,2BAA2B,QAAQ;;AAG5C,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,8BAA8B,WAAW;EACzD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,SACb,mBAAmB,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAAI;QAE7D,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;AAC5B,MAAI,MAAM,UAAU,KAAM,QAAO;EACjC,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}
+{"version":3,"file":"messages.js","names":[],"sources":["../../src/ui/messages.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  BaseMessageChunk,\n  RemoveMessage,\n  convertToChunk,\n  coerceMessageLikeToMessage,\n  HumanMessageChunk,\n  HumanMessage,\n  SystemMessageChunk,\n  SystemMessage,\n  AIMessageChunk,\n  AIMessage,\n  ToolMessageChunk,\n  ToolMessage,\n} from \"@langchain/core/messages\";\n\nimport type { Message } from \"../types.messages.js\";\nimport type { ThreadState } from \"../schema.js\";\n\n/**\n * Replaces the `messages` property in a state type with `BaseMessage[]`.\n * Used by framework SDKs to reflect that `ensureHistoryMessageInstances`\n * converts plain message objects to class instances at runtime.\n */\nexport type StateWithBaseMessages<S> = S extends { messages: unknown }\n  ? Omit<S, \"messages\"> & { messages: BaseMessage[] }\n  : S;\n\n/**\n * Maps a `ThreadState<StateType>[]` so that the `messages` field inside\n * `values` is typed as `BaseMessage[]` instead of `Message[]`.\n */\nexport type HistoryWithBaseMessages<T> = T extends ThreadState<infer S>[]\n  ? ThreadState<StateWithBaseMessages<S>>[]\n  : T;\n\nexport function tryConvertToChunk(\n  message: BaseMessage | BaseMessageChunk\n): BaseMessageChunk | null {\n  try {\n    if (BaseMessageChunk.isInstance(message)) return message;\n    return convertToChunk(message);\n  } catch {\n    return null;\n  }\n}\n\nexport function tryCoerceMessageLikeToMessage(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessage(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessage(normalizeAIMessageToolCalls(message));\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessage(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessage({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  if (message.type === \"remove\" && message.id != null) {\n    return new RemoveMessage({ ...message, id: message.id });\n  }\n\n  return coerceMessageLikeToMessage(message);\n}\n\nfunction tryCoerceMessageLikeToChunk(\n  message: Omit<Message, \"type\"> & { type: string }\n): BaseMessage | BaseMessageChunk {\n  if (message.type === \"human\" || message.type === \"user\") {\n    return new HumanMessageChunk(message);\n  }\n\n  if (message.type === \"ai\" || message.type === \"assistant\") {\n    return new AIMessageChunk(normalizeAIMessageToolCalls(message));\n  }\n\n  if (message.type === \"system\") {\n    return new SystemMessageChunk(message);\n  }\n\n  if (message.type === \"tool\" && \"tool_call_id\" in message) {\n    return new ToolMessageChunk({\n      ...message,\n      tool_call_id: message.tool_call_id as string,\n    });\n  }\n\n  return tryCoerceMessageLikeToMessage(message);\n}\n\ntype ToolCallLike = {\n  id?: string;\n  name?: string;\n  args?: unknown;\n  input?: unknown;\n};\n\nfunction normalizeAIMessageToolCalls<\n  T extends Omit<Message, \"type\"> & { type: string },\n>(message: T): T {\n  const record = message as T & {\n    content?: unknown;\n    tool_calls?: unknown;\n  };\n  if (Array.isArray(record.tool_calls) && record.tool_calls.length > 0) {\n    return message;\n  }\n\n  const toolCalls = extractToolCallsFromContent(record.content);\n  if (toolCalls.length === 0) return message;\n  return {\n    ...message,\n    tool_calls: toolCalls,\n  };\n}\n\nfunction extractToolCallsFromContent(content: unknown) {\n  if (!Array.isArray(content)) return [];\n  return content.flatMap(\n    (\n      block\n    ): Array<{\n      id: string;\n      name: string;\n      args: Record<string, unknown>;\n      type: \"tool_call\";\n    }> => {\n      if (block == null || typeof block !== \"object\") return [];\n      const record = block as ToolCallLike & { type?: unknown };\n      if (record.type !== \"tool_call\" && record.type !== \"tool_use\") return [];\n      return [\n        {\n          id: record.id ?? \"\",\n          name: record.name ?? \"\",\n          args: normalizeToolCallArgs(record.args ?? record.input),\n          type: \"tool_call\",\n        },\n      ];\n    }\n  );\n}\n\nfunction normalizeToolCallArgs(value: unknown): Record<string, unknown> {\n  if (value != null && typeof value === \"object\" && !Array.isArray(value)) {\n    return value as Record<string, unknown>;\n  }\n  if (typeof value === \"string\" && value.length > 0) {\n    try {\n      const parsed = JSON.parse(value);\n      if (\n        parsed != null &&\n        typeof parsed === \"object\" &&\n        !Array.isArray(parsed)\n      ) {\n        return parsed as Record<string, unknown>;\n      }\n    } catch {\n      // Streaming input fragments are expected to be invalid until finalized.\n    }\n  }\n  return {};\n}\n\nexport class MessageTupleManager {\n  chunks: Record<\n    string,\n    {\n      chunk?: BaseMessageChunk | BaseMessage;\n      metadata?: Record<string, unknown>;\n      index?: number;\n    }\n  > = {};\n\n  constructor() {\n    this.chunks = {};\n  }\n\n  add(\n    serialized: Message,\n    metadata: Record<string, unknown> | undefined\n  ): string | null {\n    // TODO: this is sometimes sent from the API\n    // figure out how to prevent this or move this to LC.js\n    if (serialized.type.endsWith(\"MessageChunk\")) {\n      // eslint-disable-next-line no-param-reassign\n      serialized.type = serialized.type\n        .slice(0, -\"MessageChunk\".length)\n        .toLowerCase() as Message[\"type\"];\n    }\n\n    const message = tryCoerceMessageLikeToChunk(serialized);\n    const chunk = tryConvertToChunk(message);\n\n    const { id } = chunk ?? message;\n    if (!id) {\n      console.warn(\n        \"No message ID found for chunk, ignoring in state\",\n        serialized\n      );\n      return null;\n    }\n\n    this.chunks[id] ??= {};\n    this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;\n    if (chunk) {\n      const prev = this.chunks[id].chunk;\n      this.chunks[id].chunk =\n        (BaseMessageChunk.isInstance(prev) ? prev : null)?.concat(chunk) ??\n        chunk;\n    } else {\n      this.chunks[id].chunk = message;\n    }\n\n    return id;\n  }\n\n  clear() {\n    this.chunks = {};\n  }\n\n  get(id: string | null | undefined, defaultIndex?: number) {\n    if (id == null) return null;\n    if (this.chunks[id] == null) return null;\n    if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;\n    return this.chunks[id];\n  }\n}\n\nexport const toMessageDict = (chunk: BaseMessage): Message => {\n  const { type, data } = chunk.toDict();\n  return { ...data, type } as Message;\n};\n\n/**\n * Identity converter that keeps @langchain/core class instances.\n * Used by framework SDKs to expose BaseMessage instances instead of plain dicts.\n */\nexport const toMessageClass = (chunk: BaseMessage): BaseMessage => chunk;\n\n/**\n * Ensures all messages in an array are BaseMessage class instances.\n * Messages that are already class instances pass through unchanged.\n * Plain message objects (e.g. from API values/history) are converted\n * via {@link tryCoerceMessageLikeToMessage}.\n */\nexport function ensureMessageInstances(\n  messages: (Message | BaseMessage)[]\n): (BaseMessage | BaseMessageChunk)[] {\n  return messages.map((msg) => {\n    if (typeof (msg as BaseMessage).getType === \"function\") {\n      return msg as BaseMessage;\n    }\n    return tryCoerceMessageLikeToMessage(\n      msg as Omit<Message, \"type\"> & { type: string }\n    );\n  });\n}\n\n/**\n * Converts plain message objects within each history state's values\n * to proper BaseMessage class instances. Returns a new array with\n * shallow-copied states whose messages have been coerced.\n */\nexport function ensureHistoryMessageInstances<\n  StateType extends Record<string, unknown>,\n>(\n  history: ThreadState<StateType>[],\n  messagesKey: string = \"messages\"\n): ThreadState<StateType>[] {\n  return history.map((state) => {\n    if (state.values == null) return state;\n    const messages = state.values[messagesKey];\n    if (!Array.isArray(messages)) return state;\n    return {\n      ...state,\n      values: {\n        ...state.values,\n        [messagesKey]: ensureMessageInstances(messages),\n      },\n    };\n  });\n}\n"],"mappings":";;AAoCA,SAAgB,kBACd,SACyB;AACzB,KAAI;AACF,MAAI,iBAAiB,WAAW,QAAQ,CAAE,QAAO;AACjD,SAAO,eAAe,QAAQ;SACxB;AACN,SAAO;;;AAIX,SAAgB,8BACd,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAI,aAAa,QAAQ;AAGlC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAI,UAAU,4BAA4B,QAAQ,CAAC;AAG5D,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAI,cAAc,QAAQ;AAGnC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAI,YAAY;EACrB,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,KAAI,QAAQ,SAAS,YAAY,QAAQ,MAAM,KAC7C,QAAO,IAAI,cAAc;EAAE,GAAG;EAAS,IAAI,QAAQ;EAAI,CAAC;AAG1D,QAAO,2BAA2B,QAAQ;;AAG5C,SAAS,4BACP,SACgC;AAChC,KAAI,QAAQ,SAAS,WAAW,QAAQ,SAAS,OAC/C,QAAO,IAAI,kBAAkB,QAAQ;AAGvC,KAAI,QAAQ,SAAS,QAAQ,QAAQ,SAAS,YAC5C,QAAO,IAAI,eAAe,4BAA4B,QAAQ,CAAC;AAGjE,KAAI,QAAQ,SAAS,SACnB,QAAO,IAAI,mBAAmB,QAAQ;AAGxC,KAAI,QAAQ,SAAS,UAAU,kBAAkB,QAC/C,QAAO,IAAI,iBAAiB;EAC1B,GAAG;EACH,cAAc,QAAQ;EACvB,CAAC;AAGJ,QAAO,8BAA8B,QAAQ;;AAU/C,SAAS,4BAEP,SAAe;CACf,MAAM,SAAS;AAIf,KAAI,MAAM,QAAQ,OAAO,WAAW,IAAI,OAAO,WAAW,SAAS,EACjE,QAAO;CAGT,MAAM,YAAY,4BAA4B,OAAO,QAAQ;AAC7D,KAAI,UAAU,WAAW,EAAG,QAAO;AACnC,QAAO;EACL,GAAG;EACH,YAAY;EACb;;AAGH,SAAS,4BAA4B,SAAkB;AACrD,KAAI,CAAC,MAAM,QAAQ,QAAQ,CAAE,QAAO,EAAE;AACtC,QAAO,QAAQ,SAEX,UAMI;AACJ,MAAI,SAAS,QAAQ,OAAO,UAAU,SAAU,QAAO,EAAE;EACzD,MAAM,SAAS;AACf,MAAI,OAAO,SAAS,eAAe,OAAO,SAAS,WAAY,QAAO,EAAE;AACxE,SAAO,CACL;GACE,IAAI,OAAO,MAAM;GACjB,MAAM,OAAO,QAAQ;GACrB,MAAM,sBAAsB,OAAO,QAAQ,OAAO,MAAM;GACxD,MAAM;GACP,CACF;GAEJ;;AAGH,SAAS,sBAAsB,OAAyC;AACtE,KAAI,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACrE,QAAO;AAET,KAAI,OAAO,UAAU,YAAY,MAAM,SAAS,EAC9C,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,MAAM;AAChC,MACE,UAAU,QACV,OAAO,WAAW,YAClB,CAAC,MAAM,QAAQ,OAAO,CAEtB,QAAO;SAEH;AAIV,QAAO,EAAE;;AAGX,IAAa,sBAAb,MAAiC;CAC/B,SAOI,EAAE;CAEN,cAAc;AACZ,OAAK,SAAS,EAAE;;CAGlB,IACE,YACA,UACe;AAGf,MAAI,WAAW,KAAK,SAAS,eAAe,CAE1C,YAAW,OAAO,WAAW,KAC1B,MAAM,GAAG,IAAuB,CAChC,aAAa;EAGlB,MAAM,UAAU,4BAA4B,WAAW;EACvD,MAAM,QAAQ,kBAAkB,QAAQ;EAExC,MAAM,EAAE,OAAO,SAAS;AACxB,MAAI,CAAC,IAAI;AACP,WAAQ,KACN,oDACA,WACD;AACD,UAAO;;AAGT,OAAK,OAAO,QAAQ,EAAE;AACtB,OAAK,OAAO,IAAI,WAAW,YAAY,KAAK,OAAO,IAAI;AACvD,MAAI,OAAO;GACT,MAAM,OAAO,KAAK,OAAO,IAAI;AAC7B,QAAK,OAAO,IAAI,SACb,iBAAiB,WAAW,KAAK,GAAG,OAAO,OAAO,OAAO,MAAM,IAChE;QAEF,MAAK,OAAO,IAAI,QAAQ;AAG1B,SAAO;;CAGT,QAAQ;AACN,OAAK,SAAS,EAAE;;CAGlB,IAAI,IAA+B,cAAuB;AACxD,MAAI,MAAM,KAAM,QAAO;AACvB,MAAI,KAAK,OAAO,OAAO,KAAM,QAAO;AACpC,MAAI,gBAAgB,KAAM,MAAK,OAAO,IAAI,UAAU;AACpD,SAAO,KAAK,OAAO;;;AAIvB,MAAa,iBAAiB,UAAgC;CAC5D,MAAM,EAAE,MAAM,SAAS,MAAM,QAAQ;AACrC,QAAO;EAAE,GAAG;EAAM;EAAM;;;;;;AAO1B,MAAa,kBAAkB,UAAoC;;;;;;;AAQnE,SAAgB,uBACd,UACoC;AACpC,QAAO,SAAS,KAAK,QAAQ;AAC3B,MAAI,OAAQ,IAAoB,YAAY,WAC1C,QAAO;AAET,SAAO,8BACL,IACD;GACD;;;;;;;AAQJ,SAAgB,8BAGd,SACA,cAAsB,YACI;AAC1B,QAAO,QAAQ,KAAK,UAAU;AAC5B,MAAI,MAAM,UAAU,KAAM,QAAO;EACjC,MAAM,WAAW,MAAM,OAAO;AAC9B,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;AACrC,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG,MAAM;KACR,cAAc,uBAAuB,SAAS;IAChD;GACF;GACD"}

package/dist/ui/orchestrator-custom.cjs

@@ -1,5 +1,5 @@
-const require_headless_tools = require("../headless-tools.cjs");
 const require_messages = require("./messages.cjs");
+const require_headless_tools = require("../headless-tools.cjs");
 const require_tools = require("../utils/tools.cjs");
 const require_manager = require("./manager.cjs");
 const require_interrupts = require("./interrupts.cjs");

package/dist/ui/orchestrator-custom.js

@@ -1,5 +1,5 @@
-import { flushPendingHeadlessToolInterrupts } from "../headless-tools.js";
 import { MessageTupleManager, ensureMessageInstances, toMessageClass } from "./messages.js";
+import { flushPendingHeadlessToolInterrupts } from "../headless-tools.js";
 import { getToolCallsWithResults } from "../utils/tools.js";
 import { StreamManager } from "./manager.js";
 import { extractInterrupts, userFacingInterruptsFromValuesArray } from "./interrupts.js";

package/dist/ui/orchestrator.cjs

@@ -1,6 +1,6 @@
-const require_errors = require("./errors.cjs");
-const require_headless_tools = require("../headless-tools.cjs");
 const require_messages = require("./messages.cjs");
+const require_headless_tools = require("../headless-tools.cjs");
+const require_errors = require("./errors.cjs");
 const require_tools = require("../utils/tools.cjs");
 const require_manager = require("./manager.cjs");
 const require_utils = require("./utils.cjs");

package/dist/ui/orchestrator.d.cts

@@ -2,7 +2,7 @@ import { Interrupt, ThreadState } from "../schema.cjs";
 import { DefaultToolCall, Message, ToolCallWithResult } from "../types.messages.cjs";
 import { StreamMode } from "../types.stream.cjs";
 import { StreamEvent } from "../types.cjs";
-import { Client } from "../client.cjs";
+import { Client } from "../client/index.cjs";
 import { BagTemplate } from "../types.template.cjs";
 import { AnyStreamOptions, GetConfigurableType, GetInterruptType, MessageMetadata, SubagentStreamInterface, SubmitOptions, UseStreamThread } from "./types.cjs";
 import { Sequence } from "./branching.cjs";

package/dist/ui/orchestrator.d.ts

@@ -2,7 +2,7 @@ import { Interrupt, ThreadState } from "../schema.js";
 import { DefaultToolCall, Message, ToolCallWithResult } from "../types.messages.js";
 import { StreamMode } from "../types.stream.js";
 import { StreamEvent } from "../types.js";
-import { Client } from "../client.js";
+import { Client } from "../client/index.js";
 import { BagTemplate } from "../types.template.js";
 import { AnyStreamOptions, GetConfigurableType, GetInterruptType, MessageMetadata, SubagentStreamInterface, SubmitOptions, UseStreamThread } from "./types.js";
 import { Sequence } from "./branching.js";

package/dist/ui/orchestrator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"orchestrator.d.ts","names":[],"sources":["../../src/ui/orchestrator.ts"],"mappings":";;;;;;;;;;;;;;;;;AA0GA;UAAiB,qBAAA;EACf,SAAA,IAAa,MAAA;EACb,cAAA;EACA,cAAA;AAAA;;;;;AAaF;;;;;;cAAa,kBAAA,mBACO,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA;EAAA;WAEjB,MAAA,EAAQ,aAAA,CAAc,SAAA,EAAW,GAAA;EAAA,SAEjC,cAAA,EAAgB,mBAAA;EAAA,SAEhB,WAAA,EAAa,kBAAA,CACpB,SAAA,EACA,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA;EAAA,SAOtC,YAAA;EAPO;;;;;;;;EA+ChB,WAAA,CACE,OAAA,EAAS,gBAAA,CAAiB,SAAA,EAAW,GAAA,GACrC,SAAA,EAAW,qBAAA;EA0JsB;;;;;;;EAlGnC,SAAA,CAAU,QAAA;EAgQM;;;;;;EAnPhB,WAAA,CAAA;EA+R6C;;;EAAA,IA5QzC,QAAA,CAAA;EAwSuB;;;;;;EA9R3B,WAAA,CAAY,KAAA;EAuWO;;;;;EAAA,IA/Sf,WAAA,CAAA,GAAe,eAAA,CAAgB,SAAA;EAkUnB;;;;EAxRhB,YAAA,CAAa,QAAA;EAoUI;;;;EAAA,IA1Tb,MAAA,CAAA;EAqV2B;;;;;EA5U/B,SAAA,CAAU,KAAA;EAiWsB;;;;;EAAA,IAtV5B,aAAA,CAAA;gBAxE8B,QAAA;;;;;;;;;;EA+wBhB;;;;;EAAA,IAprBd,aAAA,CAAA,GAAiB,SAAA;EA/SH;;;;EAAA,IA2Td,YAAA,CAAA;;;;;MAiBA,YAAA,CAAA,GAAgB,SAAA;EAvUX;;;EAAA,IA8UL,WAAA,CAAA;EA3UF;;;;EAAA,IAmVE,MAAA,CAAA,GAAU,SAAA;EA3UL;;;;EAAA,IAmVL,KAAA,CAAA;EA1SF;;;EAAA,IAiTE,SAAA,CAAA;EAxPM;;;;EAAA,IAgQN,QAAA,CAAA,GAAY,OAAA;EA9JZ;;;;EAAA,IAsKA,gBAAA,CAAA,GAAoB,WAAA;EAlHpB;;;;;EAAA,IA4HA,SAAA,CAAA,GAAS,kBAAA,CAVsB,eAAA;;;;;;;;EAsBnC,YAAA,CAAa,OAAA,EAAS,OAAA,GAAO,kBAAA,CAAA,eAAA;EAjGzB;;;;;;EAAA,IA+GA,UAAA,CAAA,GAAc,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAnE/B;;;;;EAAA,IA+FV,SAAA,CAAA,GAAa,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAhEpB;;;;;;EAAA,IA8EpB,WAAA,CAAA,GAAW,WAAA;EAxDc;;;;EAAA,IAwEzB,eAAA,CAAA;EA1DyC;;;;;;EAAA,IAoEzC,uBAAA,CAAA,GA1BW,QAAA;EAgBX;;;;EAAA,IAuBA,eAAA,CAAA;;;;;;EAiBJ;;;;;;;;EAAA,mBAAA,CACE,OAAA,EAAS,OAAA,EACT,KAAA,YACC,eAAA,CAAgB,SAAA;EAmBH;;;EAAA,IAAZ,YAAA,CAAA,YAAY,UAAA,CAAA,SAAA,EAAA,aAAA,CAAA,SAAA,EAAA,mBAAA,CAAA,GAAA;EAOZ;;;EAAA,IAAA,SAAA,CAAA;EAwBE;;;;;;;;EAZA,eAAA,CAAgB,EAAA,WAAa,OAAA;EA0CL;;;EA9BxB,UAAA,CAAA,GAAc,OAAA;EAwCD;;;EAAA,IA3Bf,SAAA,CAAA,GAAa,GAAA,SAAY,uBAAA;EAsC7B;;;EAAA,IA/BI,eAAA,CAAA,GAAmB,uBAAA;EA+BgB;;;;;;EArBvC,WAAA,CAAY,UAAA,WAAkB,uBAAA,CAAA,MAAA,mBAAA,eAAA;EA6GxB;;;;;;EAnGN,kBAAA,CAAmB,IAAA,WAAY,uBAAA,CAAA,MAAA,mBAAA,eAAA;EAyGzB;;;;;;;EA9FN,qBAAA,CAAsB,SAAA,WAAiB,uBAAA,CAAA,MAAA,mBAAA,eAAA;EAqL7B;;;;;;EA3KV,4BAAA,CAAA,GAAgC,eAAA;EA4KoC;;;;;;EArIpE,eAAA,CAAA,GAAmB,KAAA,EAAO,UAAA;EA0UiB;;;;;EA7T3C,IAAA,CAAA;EAmYA;;;;;;;;;EAzWM,UAAA,CACJ,KAAA,UACA,WAAA,WACA,WAAA;IACE,UAAA,GAAa,UAAA,GAAa,UAAA;IAC1B,MAAA,IAAU,KAAA;MACR,EAAA;MACA,KAAA,EAAO,WAAA;MACP,IAAA;IAAA;EAAA,IAGH,OAAA;;;;;;;;;;;EAiFH,YAAA,CACE,MAAA,EAAQ,SAAA,EACR,aAAA,GAAgB,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA,KAAK,OAAA;;;;;EAoLpE,UAAA,CAAA;;;;;;;;;;;;EAeM,MAAA,CACJ,MAAA,EAAQ,SAAA,EACR,aAAA,GAAgB,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA,KAC5D,OAAA,CAAQ,UAAA,aAAuB,YAAA;;;;;;;;;EAqElC,YAAA,CAAa,WAAA;;;;;EA8Bb,YAAA,CAAA;;;;;MAiBI,eAAA,CAAA;;;;;;EASJ,OAAA,CAAA;AAAA"}
+{"version":3,"file":"orchestrator.d.ts","names":[],"sources":["../../src/ui/orchestrator.ts"],"mappings":";;;;;;;;;;;;;;;;;;UA0GiB,qBAAA;EACf,SAAA,IAAa,MAAA;EACb,cAAA;EACA,cAAA;AAAA;;;;;;AAaF;;;;;cAAa,kBAAA,mBACO,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA;EAAA;WAEjB,MAAA,EAAQ,aAAA,CAAc,SAAA,EAAW,GAAA;EAAA,SAEjC,cAAA,EAAgB,mBAAA;EAAA,SAEhB,WAAA,EAAa,kBAAA,CACpB,SAAA,EACA,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA;EAAA,SAOtC,YAAA;EARP;;;;;;;;EAgDF,WAAA,CACE,OAAA,EAAS,gBAAA,CAAiB,SAAA,EAAW,GAAA,GACrC,SAAA,EAAW,qBAAA;EAAA;;;;;;;EAwDb,SAAA,CAAU,QAAA;EAyOI;;;;;;EA5Nd,WAAA,CAAA;EAiR6B;;;EAAA,IA9PzB,QAAA,CAAA;EAwSwC;;;;;;EA9R5C,WAAA,CAAY,KAAA;EAqWD;;;;;EAAA,IA7SP,WAAA,CAAA,GAAe,eAAA,CAAgB,SAAA;EAkUnB;;;;EAxRhB,YAAA,CAAa,QAAA;EAoUgB;;;;EAAA,IA1TzB,MAAA,CAAA;EA2U0B;;;;;EAlU9B,SAAA,CAAU,KAAA;EAuV6B;;;;;EAAA,IA5UnC,aAAA,CAAA;gBAxE8B,QAAA;;;;;;;;;;EA+wBS;;;;;EAAA,IAprBvC,aAAA,CAAA,GAAiB,SAAA;EA/SrB;;;;EAAA,IA2TI,YAAA,CAAA;EA1TsB;;;;EAAA,IA2UtB,YAAA,CAAA,GAAgB,SAAA;EAzUsB;;;EAAA,IAgVtC,WAAA,CAAA;EA5UkB;;;;EAAA,IAoVlB,MAAA,CAAA,GAAU,SAAA;EAlViC;;;;EAAA,IA0V3C,KAAA,CAAA;EA1SmC;;;EAAA,IAiTnC,SAAA,CAAA;EAxPJ;;;;EAAA,IAgQI,QAAA,CAAA,GAAY,OAAA;EAtNJ;;;;EAAA,IA8NR,gBAAA,CAAA,GAAoB,WAAA;EA5HX;;;;;EAAA,IAsIT,SAAA,CAAA,GAAS,kBAAA,CAVsB,eAAA;;;;;;;;EAsBnC,YAAA,CAAa,OAAA,EAAS,OAAA,GAAO,kBAAA,CAAA,eAAA;;;;;;;MAczB,UAAA,CAAA,GAAc,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAnEzC;;;;;EAAA,IA+FA,SAAA,CAAA,GAAa,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAhExC;;;;;;EAAA,IA8EA,WAAA,CAAA,GAAW,WAAA;EAxDF;;;;EAAA,IAwET,eAAA,CAAA;EA1DwB;;;;;;EAAA,IAoExB,uBAAA,CAAA,GA1BW,QAAA;EAAA;;;;EAAA,IAuCX,eAAA,CAAA;;;;;;;;;;;;;;EAiBJ,mBAAA,CACE,OAAA,EAAS,OAAA,EACT,KAAA,YACC,eAAA,CAAgB,SAAA;EAmBH;;;EAAA,IAAZ,YAAA,CAAA,YAAY,UAAA,CAAA,SAAA,EAAA,aAAA,CAAA,SAAA,EAAA,mBAAA,CAAA,GAAA;EAAA;;;EAAA,IAOZ,SAAA,CAAA;EAY+B;;;;;;;;EAA7B,eAAA,CAAgB,EAAA,WAAa,OAAA;EA0CvB;;;EA9BN,UAAA,CAAA,GAAc,OAAA;EAwCpB;;;EAAA,IA3BI,SAAA,CAAA,GAAa,GAAA,SAAY,uBAAA;EA2BE;;;EAAA,IApB3B,eAAA,CAAA,GAAmB,uBAAA;EA+BgB;;;;;;EArBvC,WAAA,CAAY,UAAA,WAAkB,uBAAA,CAAA,MAAA,mBAAA,eAAA;EAmF9B;;;;;;EAzEA,kBAAA,CAAmB,IAAA,WAAY,uBAAA,CAAA,MAAA,mBAAA,eAAA;EAwG3B;;;;;;;EA7FJ,qBAAA,CAAsB,SAAA,WAAiB,uBAAA,CAAA,MAAA,mBAAA,eAAA;EAoLvC;;;;;;EA1KA,4BAAA,CAAA,GAAgC,eAAA;EA4K9B;;;;;;EArIF,eAAA,CAAA,GAAmB,KAAA,EAAO,UAAA;EA0UM;;;;;EA7ThC,IAAA,CAAA;EA8TkC;;;;;;;;;EApS5B,UAAA,CACJ,KAAA,UACA,WAAA,WACA,WAAA;IACE,UAAA,GAAa,UAAA,GAAa,UAAA;IAC1B,MAAA,IAAU,KAAA;MACR,EAAA;MACA,KAAA,EAAO,WAAA;MACP,IAAA;IAAA;EAAA,IAGH,OAAA;;;;;;;;;;;EAiFH,YAAA,CACE,MAAA,EAAQ,SAAA,EACR,aAAA,GAAgB,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA,KAAK,OAAA;;;;;EAoLpE,UAAA,CAAA;;;;;;;;;;;;EAeM,MAAA,CACJ,MAAA,EAAQ,SAAA,EACR,aAAA,GAAgB,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA,KAC5D,OAAA,CAAQ,UAAA,aAAuB,YAAA;;;;;;;;;EAqElC,YAAA,CAAa,WAAA;;;;;EA8Bb,YAAA,CAAA;;;;;MAiBI,eAAA,CAAA;;;;;;EASJ,OAAA,CAAA;AAAA"}

package/dist/ui/orchestrator.js

@@ -1,6 +1,6 @@
-import { StreamError } from "./errors.js";
-import { flushPendingHeadlessToolInterrupts } from "../headless-tools.js";
 import { MessageTupleManager, ensureHistoryMessageInstances, ensureMessageInstances, toMessageClass } from "./messages.js";
+import { flushPendingHeadlessToolInterrupts } from "../headless-tools.js";
+import { StreamError } from "./errors.js";
 import { getToolCallsWithResults } from "../utils/tools.js";
 import { StreamManager } from "./manager.js";
 import { filterStream, onFinishRequiresThreadState, unique } from "./utils.js";

package/dist/ui/stream/agent.d.cts

@@ -70,7 +70,7 @@ import { BaseStream } from "./base.cjs";
  * It does NOT include subagent streaming features. For those, use
  * {@link UseDeepAgentStream} with `createDeepAgent`.
  */
-interface UseAgentStream<StateType extends Record<string, unknown> = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> extends BaseStream<StateType, ToolCall, Bag> {
+interface UseAgentStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> extends BaseStream<StateType, ToolCall, Bag> {
   /**
    * Tool calls paired with their results.
    *

package/dist/ui/stream/agent.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.cts","names":[],"sources":["../../../src/ui/stream/agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8JA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UA1EiB,cAAA,mBACG,MAAA,oBAA0B,MAAA,8BACjC,eAAA,cACC,WAAA,GAAc,WAAA,UAClB,UAAA,CAAW,SAAA,EAAW,QAAA,EAAU,GAAA;;;;;;;;;;;;;;;;;;EAkBxC,SAAA,EAAW,kBAAA,CAAmB,QAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA2B9B,YAAA,GACE,OAAA,EAAS,SAAA,CAAU,QAAA,MAChB,kBAAA,CAAmB,QAAA;AAAA;;;;;;;;;;;;;;;;;;;;;UAuBT,qBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,gBAAA,CAAiB,SAAA,EAAW,GAAA"}
+{"version":3,"file":"agent.d.cts","names":[],"sources":["../../../src/ui/stream/agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8JA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UA1EiB,cAAA,4BACY,MAAA,8BAChB,eAAA,cACC,WAAA,GAAc,WAAA,UAClB,UAAA,CAAW,SAAA,EAAW,QAAA,EAAU,GAAA;;;;;;;;;;;;;;;;;;EAkBxC,SAAA,EAAW,kBAAA,CAAmB,QAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA2B9B,YAAA,GACE,OAAA,EAAS,SAAA,CAAU,QAAA,MAChB,kBAAA,CAAmB,QAAA;AAAA;;;;;;;;;;;;;;;;;;;;;UAuBT,qBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,gBAAA,CAAiB,SAAA,EAAW,GAAA"}

package/dist/ui/stream/agent.d.ts

@@ -70,7 +70,7 @@ import { BaseStream } from "./base.js";
  * It does NOT include subagent streaming features. For those, use
  * {@link UseDeepAgentStream} with `createDeepAgent`.
  */
-interface UseAgentStream<StateType extends Record<string, unknown> = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> extends BaseStream<StateType, ToolCall, Bag> {
+interface UseAgentStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> extends BaseStream<StateType, ToolCall, Bag> {
   /**
    * Tool calls paired with their results.
    *

package/dist/ui/stream/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","names":[],"sources":["../../../src/ui/stream/agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8JA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UA1EiB,cAAA,mBACG,MAAA,oBAA0B,MAAA,8BACjC,eAAA,cACC,WAAA,GAAc,WAAA,UAClB,UAAA,CAAW,SAAA,EAAW,QAAA,EAAU,GAAA;;;;;;;;;;;;;;;;;;EAkBxC,SAAA,EAAW,kBAAA,CAAmB,QAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA2B9B,YAAA,GACE,OAAA,EAAS,SAAA,CAAU,QAAA,MAChB,kBAAA,CAAmB,QAAA;AAAA;;;;;;;;;;;;;;;;;;;;;UAuBT,qBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,gBAAA,CAAiB,SAAA,EAAW,GAAA"}
+{"version":3,"file":"agent.d.ts","names":[],"sources":["../../../src/ui/stream/agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8JA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UA1EiB,cAAA,4BACY,MAAA,8BAChB,eAAA,cACC,WAAA,GAAc,WAAA,UAClB,UAAA,CAAW,SAAA,EAAW,QAAA,EAAU,GAAA;;;;;;;;;;;;;;;;;;EAkBxC,SAAA,EAAW,kBAAA,CAAmB,QAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA2B9B,YAAA,GACE,OAAA,EAAS,SAAA,CAAU,QAAA,MAChB,kBAAA,CAAmB,QAAA;AAAA;;;;;;;;;;;;;;;;;;;;;UAuBT,qBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,gBAAA,CAAiB,SAAA,EAAW,GAAA"}

package/dist/ui/stream/base.d.cts

@@ -2,13 +2,14 @@ import { Interrupt, ThreadState } from "../../schema.cjs";
 import { DefaultToolCall, Message } from "../../types.messages.cjs";
 import { StreamMode, ToolProgress } from "../../types.stream.cjs";
 import { StreamEvent } from "../../types.cjs";
-import { Client } from "../../client.cjs";
+import { Client } from "../../client/index.cjs";
 import { BagTemplate } from "../../types.template.cjs";
 import { GetConfigurableType, GetInterruptType, GetUpdateType, MessageMetadata, SubmitOptions } from "../types.cjs";
 import { Sequence } from "../branching.cjs";
 import { QueueInterface } from "../queue.cjs";
 
 //#region src/ui/stream/base.d.ts
+type StateRecord<T> = T extends object ? T & Record<string, unknown> : Record<string, unknown>;
 /**
  * Base stream interface shared by all stream types.
  *
@@ -34,7 +35,7 @@ import { QueueInterface } from "../queue.cjs";
  * // - UseDeepAgentStream for DeepAgent (createDeepAgent)
  * ```
  */
-interface BaseStream<StateType extends Record<string, unknown> = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> {
+interface BaseStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> {
   /**
    * The current state values of the stream.
    * Updated as streaming events are received.
@@ -85,7 +86,7 @@ interface BaseStream<StateType extends Record<string, unknown> = Record<string,
    * @param options - Optional configuration for the submission
    * @returns A promise that resolves when the stream completes
    */
-  submit: (values: GetUpdateType<Bag, StateType> | null | undefined, options?: SubmitOptions<StateType, GetConfigurableType<Bag>>) => Promise<void>;
+  submit: (values: GetUpdateType<Bag, StateRecord<StateType>> | null | undefined, options?: SubmitOptions<StateRecord<StateType>, GetConfigurableType<Bag>>) => Promise<void>;
   /**
    * The current branch of the thread.
    * Used for navigating between different conversation branches.
@@ -114,7 +115,7 @@ interface BaseStream<StateType extends Record<string, unknown> = Record<string,
    * @param index - The index of the message in the thread
    * @returns The metadata for the message, or undefined if not found
    */
-  getMessagesMetadata: (message: Message<ToolCall>, index?: number) => MessageMetadata<StateType> | undefined;
+  getMessagesMetadata: (message: Message<ToolCall>, index?: number) => MessageMetadata<StateRecord<StateType>> | undefined;
   /**
    * Progress of tool executions during streaming. Populated when stream mode includes "tools"
    * and tools yield or report progress.
@@ -152,8 +153,8 @@ interface BaseStream<StateType extends Record<string, unknown> = Record<string,
    * Server-side submission queue. Pending runs created via
    * `multitaskStrategy: "enqueue"` when submitting while the agent is busy.
    */
-  queue: QueueInterface<StateType, SubmitOptions<StateType, GetConfigurableType<Bag>>>;
+  queue: QueueInterface<StateRecord<StateType>, SubmitOptions<StateRecord<StateType>, GetConfigurableType<Bag>>>;
 }
 //#endregion
-export { BaseStream };
+export { BaseStream, StateRecord };
 //# sourceMappingURL=base.d.cts.map

package/dist/ui/stream/base.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.cts","names":[],"sources":["../../../src/ui/stream/base.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAkDiB,UAAA,mBACG,MAAA,oBAA0B,MAAA,8BACjC,eAAA,cACC,WAAA,GAAc,WAAA;EAoGL;;;;EA9FrB,MAAA,EAAQ,SAAA;EA2HsB;;;;EArH9B,KAAA;EA0I+C;;;;EApI/C,SAAA;EAkIqB;;;;EA5HrB,eAAA;EAzBW;;;;;EAgCX,QAAA,EAAU,OAAA,CAAQ,QAAA;EAnBlB;;;;;EA0BA,SAAA,EAAW,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAAtC;;;;;EAOA,UAAA,EAAY,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAAjB;;;;EAMtB,IAAA,QAAY,OAAA;EAUF;;;;;;;EADV,MAAA,GACE,MAAA,EAAQ,aAAA,CAAc,GAAA,EAAK,SAAA,sBAC3B,OAAA,GAAU,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA,OACpD,OAAA;EADH;;;;EAOF,MAAA;EAYA;;;;EANA,SAAA,GAAY,MAAA;EAYsB;;;;EANlC,OAAA,EAAS,WAAA,CAAY,SAAA;EAkBnB;;;;EAZF,uBAAA,EAAyB,QAAA,CAAS,SAAA;EAwBlC;;;;;;;;EAdA,mBAAA,GACE,OAAA,EAAS,OAAA,CAAQ,QAAA,GACjB,KAAA,cACG,eAAA,CAAgB,SAAA;EA8BjB;;;;EAxBJ,YAAA,EAAc,YAAA;EAwBA;;;EAnBd,MAAA,EAAQ,MAAA;EA+BO;;;EA1Bf,WAAA;EAkCE;;;;;;;EAzBF,UAAA,GACE,KAAA,UACA,WAAA,WACA,OAAA;IACE,UAAA,GAAa,UAAA,GAAa,UAAA;IAC1B,MAAA,IAAU,KAAA;MACR,EAAA;MACA,KAAA,EAAO,WAAA;MACP,IAAA;IAAA;EAAA,MAGD,OAAA;;;;;EAML,YAAA,GAAe,WAAA;;;;;EAMf,KAAA,EAAO,cAAA,CACL,SAAA,EACA,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA;AAAA"}
+{"version":3,"file":"base.d.cts","names":[],"sources":["../../../src/ui/stream/base.ts"],"mappings":";;;;;;;;;;;KAyBY,WAAA,MAAiB,CAAA,kBACzB,CAAA,GAAI,MAAA,oBACJ,MAAA;;;;;;;;;;AA2BJ;;;;;;;;;;;;;;;;UAAiB,UAAA,4BACY,MAAA,8BAChB,eAAA,cACC,WAAA,GAAc,WAAA;EA6DF;;;;EAvDxB,MAAA,EAAQ,SAAA;EAwDkB;;;;EAlD1B,KAAA;EAqEqB;;;;EA/DrB,SAAA;EAgFW;;;;EA1EX,eAAA;EAuFQ;;;;;EAhFR,QAAA,EAAU,OAAA,CAAQ,QAAA;EAsHhB;;;;;EA/GF,SAAA,EAAW,SAAA,CAAU,gBAAA,CAAiB,GAAA;EA8G/B;;;;;EAvGP,UAAA,EAAY,SAAA,CAAU,gBAAA,CAAiB,GAAA;EA9C5B;;;;EAoDX,IAAA,QAAY,OAAA;EA7CJ;;;;;;;EAsDR,MAAA,GACE,MAAA,EAAQ,aAAA,CAAc,GAAA,EAAK,WAAA,CAAY,SAAA,uBACvC,OAAA,GAAU,aAAA,CAAc,WAAA,CAAY,SAAA,GAAY,mBAAA,CAAoB,GAAA,OACjE,OAAA;EAzBM;;;;EA+BX,MAAA;EAxBsB;;;;EA8BtB,SAAA,GAAY,MAAA;EAdF;;;;EAoBV,OAAA,EAAS,WAAA,CAAY,SAAA;EAnBT;;;;EAyBZ,uBAAA,EAAyB,QAAA,CAAS,SAAA;EAzBhC;;;;;;;;EAmCF,mBAAA,GACE,OAAA,EAAS,OAAA,CAAQ,QAAA,GACjB,KAAA,cACG,eAAA,CAAgB,WAAA,CAAY,SAAA;EAbR;;;;EAmBzB,YAAA,EAAc,YAAA;EARZ;;;EAaF,MAAA,EAAQ,MAAA;EAXyB;;;EAgBjC,WAAA;EALQ;;;;;;;EAcR,UAAA,GACE,KAAA,UACA,WAAA,WACA,OAAA;IACE,UAAA,GAAa,UAAA,GAAa,UAAA;IAC1B,MAAA,IAAU,KAAA;MACR,EAAA;MACA,KAAA,EAAO,WAAA;MACP,IAAA;IAAA;EAAA,MAGD,OAAA;EAAA;;;;EAML,YAAA,GAAe,WAAA;EAOb;;;;EADF,KAAA,EAAO,cAAA,CACL,WAAA,CAAY,SAAA,GACZ,aAAA,CAAc,WAAA,CAAY,SAAA,GAAY,mBAAA,CAAoB,GAAA;AAAA"}

package/dist/ui/stream/base.d.ts

@@ -2,13 +2,14 @@ import { Interrupt, ThreadState } from "../../schema.js";
 import { DefaultToolCall, Message } from "../../types.messages.js";
 import { StreamMode, ToolProgress } from "../../types.stream.js";
 import { StreamEvent } from "../../types.js";
-import { Client } from "../../client.js";
+import { Client } from "../../client/index.js";
 import { BagTemplate } from "../../types.template.js";
 import { GetConfigurableType, GetInterruptType, GetUpdateType, MessageMetadata, SubmitOptions } from "../types.js";
 import { Sequence } from "../branching.js";
 import { QueueInterface } from "../queue.js";
 
 //#region src/ui/stream/base.d.ts
+type StateRecord<T> = T extends object ? T & Record<string, unknown> : Record<string, unknown>;
 /**
  * Base stream interface shared by all stream types.
  *
@@ -34,7 +35,7 @@ import { QueueInterface } from "../queue.js";
  * // - UseDeepAgentStream for DeepAgent (createDeepAgent)
  * ```
  */
-interface BaseStream<StateType extends Record<string, unknown> = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> {
+interface BaseStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> {
   /**
    * The current state values of the stream.
    * Updated as streaming events are received.
@@ -85,7 +86,7 @@ interface BaseStream<StateType extends Record<string, unknown> = Record<string,
    * @param options - Optional configuration for the submission
    * @returns A promise that resolves when the stream completes
    */
-  submit: (values: GetUpdateType<Bag, StateType> | null | undefined, options?: SubmitOptions<StateType, GetConfigurableType<Bag>>) => Promise<void>;
+  submit: (values: GetUpdateType<Bag, StateRecord<StateType>> | null | undefined, options?: SubmitOptions<StateRecord<StateType>, GetConfigurableType<Bag>>) => Promise<void>;
   /**
    * The current branch of the thread.
    * Used for navigating between different conversation branches.
@@ -114,7 +115,7 @@ interface BaseStream<StateType extends Record<string, unknown> = Record<string,
    * @param index - The index of the message in the thread
    * @returns The metadata for the message, or undefined if not found
    */
-  getMessagesMetadata: (message: Message<ToolCall>, index?: number) => MessageMetadata<StateType> | undefined;
+  getMessagesMetadata: (message: Message<ToolCall>, index?: number) => MessageMetadata<StateRecord<StateType>> | undefined;
   /**
    * Progress of tool executions during streaming. Populated when stream mode includes "tools"
    * and tools yield or report progress.
@@ -152,8 +153,8 @@ interface BaseStream<StateType extends Record<string, unknown> = Record<string,
    * Server-side submission queue. Pending runs created via
    * `multitaskStrategy: "enqueue"` when submitting while the agent is busy.
    */
-  queue: QueueInterface<StateType, SubmitOptions<StateType, GetConfigurableType<Bag>>>;
+  queue: QueueInterface<StateRecord<StateType>, SubmitOptions<StateRecord<StateType>, GetConfigurableType<Bag>>>;
 }
 //#endregion
-export { BaseStream };
+export { BaseStream, StateRecord };
 //# sourceMappingURL=base.d.ts.map

package/dist/ui/stream/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","names":[],"sources":["../../../src/ui/stream/base.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAkDiB,UAAA,mBACG,MAAA,oBAA0B,MAAA,8BACjC,eAAA,cACC,WAAA,GAAc,WAAA;EAoGL;;;;EA9FrB,MAAA,EAAQ,SAAA;EA2HsB;;;;EArH9B,KAAA;EA0I+C;;;;EApI/C,SAAA;EAkIqB;;;;EA5HrB,eAAA;EAzBW;;;;;EAgCX,QAAA,EAAU,OAAA,CAAQ,QAAA;EAnBlB;;;;;EA0BA,SAAA,EAAW,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAAtC;;;;;EAOA,UAAA,EAAY,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAAjB;;;;EAMtB,IAAA,QAAY,OAAA;EAUF;;;;;;;EADV,MAAA,GACE,MAAA,EAAQ,aAAA,CAAc,GAAA,EAAK,SAAA,sBAC3B,OAAA,GAAU,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA,OACpD,OAAA;EADH;;;;EAOF,MAAA;EAYA;;;;EANA,SAAA,GAAY,MAAA;EAYsB;;;;EANlC,OAAA,EAAS,WAAA,CAAY,SAAA;EAkBnB;;;;EAZF,uBAAA,EAAyB,QAAA,CAAS,SAAA;EAwBlC;;;;;;;;EAdA,mBAAA,GACE,OAAA,EAAS,OAAA,CAAQ,QAAA,GACjB,KAAA,cACG,eAAA,CAAgB,SAAA;EA8BjB;;;;EAxBJ,YAAA,EAAc,YAAA;EAwBA;;;EAnBd,MAAA,EAAQ,MAAA;EA+BO;;;EA1Bf,WAAA;EAkCE;;;;;;;EAzBF,UAAA,GACE,KAAA,UACA,WAAA,WACA,OAAA;IACE,UAAA,GAAa,UAAA,GAAa,UAAA;IAC1B,MAAA,IAAU,KAAA;MACR,EAAA;MACA,KAAA,EAAO,WAAA;MACP,IAAA;IAAA;EAAA,MAGD,OAAA;;;;;EAML,YAAA,GAAe,WAAA;;;;;EAMf,KAAA,EAAO,cAAA,CACL,SAAA,EACA,aAAA,CAAc,SAAA,EAAW,mBAAA,CAAoB,GAAA;AAAA"}
+{"version":3,"file":"base.d.ts","names":[],"sources":["../../../src/ui/stream/base.ts"],"mappings":";;;;;;;;;;;KAyBY,WAAA,MAAiB,CAAA,kBACzB,CAAA,GAAI,MAAA,oBACJ,MAAA;;;;;;;;;;;AA2BJ;;;;;;;;;;;;;;;UAAiB,UAAA,4BACY,MAAA,8BAChB,eAAA,cACC,WAAA,GAAc,WAAA;EAmDd;;;;EA7CZ,MAAA,EAAQ,SAAA;EAwD8B;;;;EAlDtC,KAAA;EAmDK;;;;EA7CL,SAAA;EAgFmB;;;;EA1EnB,eAAA;EAkFc;;;;;EA3Ed,QAAA,EAAU,OAAA,CAAQ,QAAA;EAsHJ;;;;;EA/Gd,SAAA,EAAW,SAAA,CAAU,gBAAA,CAAiB,GAAA;EAgHpC;;;;;EAzGF,UAAA,EAAY,SAAA,CAAU,gBAAA,CAAiB,GAAA;EA9CvC;;;;EAoDA,IAAA,QAAY,OAAA;EA7CZ;;;;;;;EAsDA,MAAA,GACE,MAAA,EAAQ,aAAA,CAAc,GAAA,EAAK,WAAA,CAAY,SAAA,uBACvC,OAAA,GAAU,aAAA,CAAc,WAAA,CAAY,SAAA,GAAY,mBAAA,CAAoB,GAAA,OACjE,OAAA;EAzBL;;;;EA+BA,MAAA;EAxBY;;;;EA8BZ,SAAA,GAAY,MAAA;EAfZ;;;;EAqBA,OAAA,EAAS,WAAA,CAAY,SAAA;EApBnB;;;;EA0BF,uBAAA,EAAyB,QAAA,CAAS,SAAA;EAzBoC;;;;;;;;EAmCtE,mBAAA,GACE,OAAA,EAAS,OAAA,CAAQ,QAAA,GACjB,KAAA,cACG,eAAA,CAAgB,WAAA,CAAY,SAAA;EAbjC;;;;EAmBA,YAAA,EAAc,YAAA;EARK;;;EAanB,MAAA,EAAQ,MAAA;EAXa;;;EAgBrB,WAAA;EALA;;;;;;;EAcA,UAAA,GACE,KAAA,UACA,WAAA,WACA,OAAA;IACE,UAAA,GAAa,UAAA,GAAa,UAAA;IAC1B,MAAA,IAAU,KAAA;MACR,EAAA;MACA,KAAA,EAAO,WAAA;MACP,IAAA;IAAA;EAAA,MAGD,OAAA;EARH;;;;EAcF,YAAA,GAAe,WAAA;EAMR;;;;EAAP,KAAA,EAAO,cAAA,CACL,WAAA,CAAY,SAAA,GACZ,aAAA,CAAc,WAAA,CAAY,SAAA,GAAY,mBAAA,CAAoB,GAAA;AAAA"}

package/dist/ui/stream/deep-agent.d.cts

@@ -76,7 +76,7 @@ import { UseAgentStream, UseAgentStreamOptions } from "./agent.cjs";
  * It also enables the `filterSubagentMessages` option to exclude subagent
  * messages from the main `messages` array.
  */
-interface UseDeepAgentStream<StateType extends Record<string, unknown> = Record<string, unknown>, ToolCall = DefaultToolCall, SubagentStates extends Record<string, unknown> = DefaultSubagentStates, Bag extends BagTemplate = BagTemplate> extends UseAgentStream<StateType, ToolCall, Bag> {
+interface UseDeepAgentStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, SubagentStates extends Record<string, unknown> = DefaultSubagentStates, Bag extends BagTemplate = BagTemplate> extends UseAgentStream<StateType, ToolCall, Bag> {
   /**
    * All currently active and completed subagent streams.
    *

package/dist/ui/stream/deep-agent.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"deep-agent.d.cts","names":[],"sources":["../../../src/ui/stream/deep-agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAyFiB,kBAAA,mBACG,MAAA,oBAA0B,MAAA,8BACjC,eAAA,yBACY,MAAA,oBAA0B,qBAAA,cACrC,WAAA,GAAc,WAAA,UAClB,cAAA,CAAe,SAAA,EAAW,QAAA,EAAU,GAAA;EAyEjB;;;;;;;;;;;;;;;;;EAvD3B,SAAA,EAAW,GAAA,SAET,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EA8HR;;;;;;;;AAgCJ;;;;;;;;;;EAxIE,eAAA,EAAiB,uBAAA,CACf,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EAwIqB;;;;;;;;;;;;;;;;;;EAnH7B,WAAA,GACE,UAAA,aAEE,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8BZ,kBAAA;;;;;yBAKuB,cAAA,WACnB,IAAA,EAAM,KAAA,GACL,uBAAA,CAAwB,cAAA,CAAe,KAAA,GAAQ,QAAA,EAAU,KAAA;;;;;KAO1D,IAAA,WACC,uBAAA,CAAwB,MAAA,mBAAyB,QAAA;EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BtD,qBAAA,GACE,SAAA,aACG,uBAAA,CACH,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;UA4BO,yBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,qBAAA,CAAsB,SAAA,EAAW,GAAA;;;;;;;;;;;;;;;;;;;;;;EAsBzC,iBAAA;;;;;;;;;;;;;;;;;;;;;;;;;EA0BA,sBAAA;AAAA"}
+{"version":3,"file":"deep-agent.d.cts","names":[],"sources":["../../../src/ui/stream/deep-agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAyFiB,kBAAA,4BACY,MAAA,8BAChB,eAAA,yBACY,MAAA,oBAA0B,qBAAA,cACrC,WAAA,GAAc,WAAA,UAClB,cAAA,CAAe,SAAA,EAAW,QAAA,EAAU,GAAA;EA2EhC;;;;;;;;;;;;;;;;;EAzDZ,SAAA,EAAW,GAAA,SAET,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EAgIR;;;;;;AA8BJ;;;;;;;;;;;;EAxIE,eAAA,EAAiB,uBAAA,CACf,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EAsIU;;;;;;;;;;;;;;;;;;EAjHlB,WAAA,GACE,UAAA,aAEE,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8BZ,kBAAA;;;;;yBAKuB,cAAA,WACnB,IAAA,EAAM,KAAA,GACL,uBAAA,CAAwB,cAAA,CAAe,KAAA,GAAQ,QAAA,EAAU,KAAA;;;;;KAO1D,IAAA,WACC,uBAAA,CAAwB,MAAA,mBAAyB,QAAA;EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BtD,qBAAA,GACE,SAAA,aACG,uBAAA,CACH,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;UA4BO,yBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,qBAAA,CAAsB,SAAA,EAAW,GAAA;;;;;;;;;;;;;;;;;;;;;;EAsBzC,iBAAA;;;;;;;;;;;;;;;;;;;;;;;;;EA0BA,sBAAA;AAAA"}

package/dist/ui/stream/deep-agent.d.ts

@@ -76,7 +76,7 @@ import { UseAgentStream, UseAgentStreamOptions } from "./agent.js";
  * It also enables the `filterSubagentMessages` option to exclude subagent
  * messages from the main `messages` array.
  */
-interface UseDeepAgentStream<StateType extends Record<string, unknown> = Record<string, unknown>, ToolCall = DefaultToolCall, SubagentStates extends Record<string, unknown> = DefaultSubagentStates, Bag extends BagTemplate = BagTemplate> extends UseAgentStream<StateType, ToolCall, Bag> {
+interface UseDeepAgentStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, SubagentStates extends Record<string, unknown> = DefaultSubagentStates, Bag extends BagTemplate = BagTemplate> extends UseAgentStream<StateType, ToolCall, Bag> {
   /**
    * All currently active and completed subagent streams.
    *

package/dist/ui/stream/deep-agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"deep-agent.d.ts","names":[],"sources":["../../../src/ui/stream/deep-agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAyFiB,kBAAA,mBACG,MAAA,oBAA0B,MAAA,8BACjC,eAAA,yBACY,MAAA,oBAA0B,qBAAA,cACrC,WAAA,GAAc,WAAA,UAClB,cAAA,CAAe,SAAA,EAAW,QAAA,EAAU,GAAA;EAyEjB;;;;;;;;;;;;;;;;;EAvD3B,SAAA,EAAW,GAAA,SAET,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EA8HR;;;;;;;;AAgCJ;;;;;;;;;;EAxIE,eAAA,EAAiB,uBAAA,CACf,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EAwIqB;;;;;;;;;;;;;;;;;;EAnH7B,WAAA,GACE,UAAA,aAEE,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8BZ,kBAAA;;;;;yBAKuB,cAAA,WACnB,IAAA,EAAM,KAAA,GACL,uBAAA,CAAwB,cAAA,CAAe,KAAA,GAAQ,QAAA,EAAU,KAAA;;;;;KAO1D,IAAA,WACC,uBAAA,CAAwB,MAAA,mBAAyB,QAAA;EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BtD,qBAAA,GACE,SAAA,aACG,uBAAA,CACH,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;UA4BO,yBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,qBAAA,CAAsB,SAAA,EAAW,GAAA;;;;;;;;;;;;;;;;;;;;;;EAsBzC,iBAAA;;;;;;;;;;;;;;;;;;;;;;;;;EA0BA,sBAAA;AAAA"}
+{"version":3,"file":"deep-agent.d.ts","names":[],"sources":["../../../src/ui/stream/deep-agent.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAyFiB,kBAAA,4BACY,MAAA,8BAChB,eAAA,yBACY,MAAA,oBAA0B,qBAAA,cACrC,WAAA,GAAc,WAAA,UAClB,cAAA,CAAe,SAAA,EAAW,QAAA,EAAU,GAAA;EA2EhC;;;;;;;;;;;;;;;;;EAzDZ,SAAA,EAAW,GAAA,SAET,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EAgIR;;;;;;AA8BJ;;;;;;;;;;;;EAxIE,eAAA,EAAiB,uBAAA,CACf,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;EAsIU;;;;;;;;;;;;;;;;;;EAjHlB,WAAA,GACE,UAAA,aAEE,uBAAA,CACE,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8BZ,kBAAA;;;;;yBAKuB,cAAA,WACnB,IAAA,EAAM,KAAA,GACL,uBAAA,CAAwB,cAAA,CAAe,KAAA,GAAQ,QAAA,EAAU,KAAA;;;;;KAO1D,IAAA,WACC,uBAAA,CAAwB,MAAA,mBAAyB,QAAA;EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BtD,qBAAA,GACE,SAAA,aACG,uBAAA,CACH,cAAA,OAAqB,cAAA,GACrB,QAAA,QACM,cAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;UA4BO,yBAAA,mBACG,MAAA,oBAA0B,MAAA,+BAChC,WAAA,GAAc,WAAA,UAClB,qBAAA,CAAsB,SAAA,EAAW,GAAA;;;;;;;;;;;;;;;;;;;;;;EAsBzC,iBAAA;;;;;;;;;;;;;;;;;;;;;;;;;EA0BA,sBAAA;AAAA"}

package/dist/ui/stream/index.d.cts

@@ -1,7 +1,7 @@
 import { DefaultToolCall } from "../../types.messages.cjs";
 import { BagTemplate } from "../../types.template.cjs";
 import { AgentTypeConfigLike, DeepAgentTypeConfigLike, DefaultSubagentStates, InferAgentState, InferAgentToolCalls, SubagentStateMap, UseStreamOptions } from "../types.cjs";
-import { BaseStream } from "./base.cjs";
+import { BaseStream, StateRecord } from "./base.cjs";
 import { UseAgentStream, UseAgentStreamOptions } from "./agent.cjs";
 import { UseDeepAgentStream, UseDeepAgentStreamOptions } from "./deep-agent.cjs";
 
@@ -33,9 +33,9 @@ type InferStateType<T> = T extends {
   "~agentTypes": unknown;
 } ? InferAgentState<T> : T extends {
   "~RunOutput": infer S;
-} ? S extends Record<string, unknown> ? S : Record<string, unknown> : T extends {
+} ? S extends object ? S : Record<string, unknown> : T extends {
   "~OutputType": infer O;
-} ? O extends Record<string, unknown> ? O : Record<string, unknown> : T extends Record<string, unknown> ? T : Record<string, unknown>;
+} ? O extends object ? O : Record<string, unknown> : [T] extends [object] ? T : Record<string, unknown>;
 /**
  * Infer the node names from a compiled graph.
  *
@@ -152,7 +152,7 @@ type ResolveStreamInterface<T, Bag extends BagTemplate = BagTemplate> = IsDeepAg
  * // AgentOptions.filterSubagentMessages does NOT exist
  * ```
  */
-type ResolveStreamOptions<T, Bag extends BagTemplate = BagTemplate> = IsDeepAgent<T> extends true ? UseDeepAgentStreamOptions<InferStateType<T>, Bag> : IsReactAgent<T> extends true ? UseAgentStreamOptions<InferStateType<T>, Bag> : UseStreamOptions<InferStateType<T>, Bag>;
+type ResolveStreamOptions<T, Bag extends BagTemplate = BagTemplate> = IsDeepAgent<T> extends true ? UseDeepAgentStreamOptions<StateRecord<InferStateType<T>>, Bag> : IsReactAgent<T> extends true ? UseAgentStreamOptions<StateRecord<InferStateType<T>>, Bag> : UseStreamOptions<StateRecord<InferStateType<T>>, Bag>;
 /**
  * Infer the Bag type from an agent, defaulting to the provided Bag.
  *

package/dist/ui/stream/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","names":[],"sources":["../../../src/ui/stream/index.ts"],"mappings":";;;;;;;;;;;KAgDK,WAAA,MAAiB,CAAA;EAAY,iBAAA,EAAmB,uBAAA;AAAA;;;;KAOhD,YAAA,MAAkB,CAAA;EAAY,aAAA,EAAe,mBAAA;AAAA,IAC9C,CAAA;EAAY,iBAAA;AAAA;;;;;;;;;;KAkBJ,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,eAAA,CAAgB,CAAA,IAChB,CAAA;EAAY,YAAA;AAAA,IACV,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,oBACF,CAAA;EAAY,aAAA;AAAA,IACV,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,oBACF,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA;;;;;;;;;;;;;;;;;KAkBE,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACxC,CAAA,kBACE,OAAA,CAAQ,CAAA;;;;;;;;AAFd;;;;;;;;;;KAuBY,oBAAA,MAA0B,CAAA;EAAY,iBAAA;AAAA,IAC9C,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,SAAe,MAAA,qBACjB,MAAA,SAAe,MAAA;;;;;AAJnB;;KAYY,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,mBAAA,CAAoB,CAAA,IACpB,eAAA;;;;;;;KAQQ,mBAAA,MAAyB,CAAA;EAAY,iBAAA;AAAA,IAC7C,gBAAA,CAAiB,CAAA,EAAG,mBAAA,CAAoB,CAAA,KACxC,qBAAA;;;;;;;;;;;AAZJ;;;;;;;;;;;;;;;;;AAUA;;;;;KAwCY,sBAAA,gBAAsC,WAAA,GAAc,WAAA,IAC9D,WAAA,CAAY,CAAA,iBACR,kBAAA,CACE,cAAA,CAAe,CAAA,GACf,cAAA,CAAe,CAAA,GACf,mBAAA,CAAoB,CAAA,GACpB,GAAA,IAEF,YAAA,CAAa,CAAA,iBACX,cAAA,CAAe,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA,IACrD,UAAA,CAAW,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA;;;;;;;;;;;;;;;;AAVzD;;;;;;;;;;;KAsCY,oBAAA,gBAAoC,WAAA,GAAc,WAAA,IAC5D,WAAA,CAAY,CAAA,iBACR,yBAAA,CAA0B,cAAA,CAAe,CAAA,GAAI,GAAA,IAC7C,YAAA,CAAa,CAAA,iBACX,qBAAA,CAAsB,cAAA,CAAe,CAAA,GAAI,GAAA,IACzC,gBAAA,CAAiB,cAAA,CAAe,CAAA,GAAI,GAAA;;;;;;;KAYhC,QAAA,cAAsB,WAAA,GAAc,WAAA,IAAe,CAAA;EAC7D,aAAA;AAAA,IAEE,WAAA,GACA,CAAA"}
+{"version":3,"file":"index.d.cts","names":[],"sources":["../../../src/ui/stream/index.ts"],"mappings":";;;;;;;;;;;KAgDK,WAAA,MAAiB,CAAA;EAAY,iBAAA,EAAmB,uBAAA;AAAA;;;;KAOhD,YAAA,MAAkB,CAAA;EAAY,aAAA,EAAe,mBAAA;AAAA,IAC9C,CAAA;EAAY,iBAAA;AAAA;;;;;;;;;;KAkBJ,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,eAAA,CAAgB,CAAA,IAChB,CAAA;EAAY,YAAA;AAAA,IACV,CAAA,kBACE,CAAA,GACA,MAAA,oBACF,CAAA;EAAY,aAAA;AAAA,IACV,CAAA,kBACE,CAAA,GACA,MAAA,qBACD,CAAA,qBACC,CAAA,GACA,MAAA;;;;;;;;;;;;;;;;;KAkBE,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACxC,CAAA,kBACE,OAAA,CAAQ,CAAA;;AAFd;;;;;;;;;;;;;;;;KAuBY,oBAAA,MAA0B,CAAA;EAAY,iBAAA;AAAA,IAC9C,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,SAAe,MAAA,qBACjB,MAAA,SAAe,MAAA;;;;;;;KAQP,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,mBAAA,CAAoB,CAAA,IACpB,eAAA;;;;;;;KAQQ,mBAAA,MAAyB,CAAA;EAAY,iBAAA;AAAA,IAC7C,gBAAA,CAAiB,CAAA,EAAG,mBAAA,CAAoB,CAAA,KACxC,qBAAA;;;;;AAZJ;;;;;;;;;;;;;;;;;AAUA;;;;;;;;;;;KAwCY,sBAAA,gBAAsC,WAAA,GAAc,WAAA,IAC9D,WAAA,CAAY,CAAA,iBACR,kBAAA,CACE,cAAA,CAAe,CAAA,GACf,cAAA,CAAe,CAAA,GACf,mBAAA,CAAoB,CAAA,GACpB,GAAA,IAEF,YAAA,CAAa,CAAA,iBACX,cAAA,CAAe,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA,IACrD,UAAA,CAAW,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA;;;;;;;;;;AAVzD;;;;;;;;;;;;;;;;;KAsCY,oBAAA,gBAAoC,WAAA,GAAc,WAAA,IAC5D,WAAA,CAAY,CAAA,iBACR,yBAAA,CAA0B,WAAA,CAAY,cAAA,CAAe,CAAA,IAAK,GAAA,IAC1D,YAAA,CAAa,CAAA,iBACX,qBAAA,CAAsB,WAAA,CAAY,cAAA,CAAe,CAAA,IAAK,GAAA,IACtD,gBAAA,CAAiB,WAAA,CAAY,cAAA,CAAe,CAAA,IAAK,GAAA;;;;;;;KAY7C,QAAA,cAAsB,WAAA,GAAc,WAAA,IAAe,CAAA;EAC7D,aAAA;AAAA,IAEE,WAAA,GACA,CAAA"}

package/dist/ui/stream/index.d.ts

@@ -1,7 +1,7 @@
 import { DefaultToolCall } from "../../types.messages.js";
 import { BagTemplate } from "../../types.template.js";
 import { AgentTypeConfigLike, DeepAgentTypeConfigLike, DefaultSubagentStates, InferAgentState, InferAgentToolCalls, SubagentStateMap, UseStreamOptions } from "../types.js";
-import { BaseStream } from "./base.js";
+import { BaseStream, StateRecord } from "./base.js";
 import { UseAgentStream, UseAgentStreamOptions } from "./agent.js";
 import { UseDeepAgentStream, UseDeepAgentStreamOptions } from "./deep-agent.js";
 
@@ -33,9 +33,9 @@ type InferStateType<T> = T extends {
   "~agentTypes": unknown;
 } ? InferAgentState<T> : T extends {
   "~RunOutput": infer S;
-} ? S extends Record<string, unknown> ? S : Record<string, unknown> : T extends {
+} ? S extends object ? S : Record<string, unknown> : T extends {
   "~OutputType": infer O;
-} ? O extends Record<string, unknown> ? O : Record<string, unknown> : T extends Record<string, unknown> ? T : Record<string, unknown>;
+} ? O extends object ? O : Record<string, unknown> : [T] extends [object] ? T : Record<string, unknown>;
 /**
  * Infer the node names from a compiled graph.
  *
@@ -152,7 +152,7 @@ type ResolveStreamInterface<T, Bag extends BagTemplate = BagTemplate> = IsDeepAg
  * // AgentOptions.filterSubagentMessages does NOT exist
  * ```
  */
-type ResolveStreamOptions<T, Bag extends BagTemplate = BagTemplate> = IsDeepAgent<T> extends true ? UseDeepAgentStreamOptions<InferStateType<T>, Bag> : IsReactAgent<T> extends true ? UseAgentStreamOptions<InferStateType<T>, Bag> : UseStreamOptions<InferStateType<T>, Bag>;
+type ResolveStreamOptions<T, Bag extends BagTemplate = BagTemplate> = IsDeepAgent<T> extends true ? UseDeepAgentStreamOptions<StateRecord<InferStateType<T>>, Bag> : IsReactAgent<T> extends true ? UseAgentStreamOptions<StateRecord<InferStateType<T>>, Bag> : UseStreamOptions<StateRecord<InferStateType<T>>, Bag>;
 /**
  * Infer the Bag type from an agent, defaulting to the provided Bag.
  *

package/dist/ui/stream/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../../../src/ui/stream/index.ts"],"mappings":";;;;;;;;;;;KAgDK,WAAA,MAAiB,CAAA;EAAY,iBAAA,EAAmB,uBAAA;AAAA;;;;KAOhD,YAAA,MAAkB,CAAA;EAAY,aAAA,EAAe,mBAAA;AAAA,IAC9C,CAAA;EAAY,iBAAA;AAAA;;;;;;;;;;KAkBJ,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,eAAA,CAAgB,CAAA,IAChB,CAAA;EAAY,YAAA;AAAA,IACV,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,oBACF,CAAA;EAAY,aAAA;AAAA,IACV,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,oBACF,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA;;;;;;;;;;;;;;;;;KAkBE,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACxC,CAAA,kBACE,OAAA,CAAQ,CAAA;;;;;;;;AAFd;;;;;;;;;;KAuBY,oBAAA,MAA0B,CAAA;EAAY,iBAAA;AAAA,IAC9C,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,SAAe,MAAA,qBACjB,MAAA,SAAe,MAAA;;;;;AAJnB;;KAYY,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,mBAAA,CAAoB,CAAA,IACpB,eAAA;;;;;;;KAQQ,mBAAA,MAAyB,CAAA;EAAY,iBAAA;AAAA,IAC7C,gBAAA,CAAiB,CAAA,EAAG,mBAAA,CAAoB,CAAA,KACxC,qBAAA;;;;;;;;;;;AAZJ;;;;;;;;;;;;;;;;;AAUA;;;;;KAwCY,sBAAA,gBAAsC,WAAA,GAAc,WAAA,IAC9D,WAAA,CAAY,CAAA,iBACR,kBAAA,CACE,cAAA,CAAe,CAAA,GACf,cAAA,CAAe,CAAA,GACf,mBAAA,CAAoB,CAAA,GACpB,GAAA,IAEF,YAAA,CAAa,CAAA,iBACX,cAAA,CAAe,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA,IACrD,UAAA,CAAW,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA;;;;;;;;;;;;;;;;AAVzD;;;;;;;;;;;KAsCY,oBAAA,gBAAoC,WAAA,GAAc,WAAA,IAC5D,WAAA,CAAY,CAAA,iBACR,yBAAA,CAA0B,cAAA,CAAe,CAAA,GAAI,GAAA,IAC7C,YAAA,CAAa,CAAA,iBACX,qBAAA,CAAsB,cAAA,CAAe,CAAA,GAAI,GAAA,IACzC,gBAAA,CAAiB,cAAA,CAAe,CAAA,GAAI,GAAA;;;;;;;KAYhC,QAAA,cAAsB,WAAA,GAAc,WAAA,IAAe,CAAA;EAC7D,aAAA;AAAA,IAEE,WAAA,GACA,CAAA"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../../../src/ui/stream/index.ts"],"mappings":";;;;;;;;;;;KAgDK,WAAA,MAAiB,CAAA;EAAY,iBAAA,EAAmB,uBAAA;AAAA;;;;KAOhD,YAAA,MAAkB,CAAA;EAAY,aAAA,EAAe,mBAAA;AAAA,IAC9C,CAAA;EAAY,iBAAA;AAAA;;;;;;;;;;KAkBJ,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,eAAA,CAAgB,CAAA,IAChB,CAAA;EAAY,YAAA;AAAA,IACV,CAAA,kBACE,CAAA,GACA,MAAA,oBACF,CAAA;EAAY,aAAA;AAAA,IACV,CAAA,kBACE,CAAA,GACA,MAAA,qBACD,CAAA,qBACC,CAAA,GACA,MAAA;;;;;;;;;;;;;;;;;KAkBE,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACxC,CAAA,kBACE,OAAA,CAAQ,CAAA;;AAFd;;;;;;;;;;;;;;;;KAuBY,oBAAA,MAA0B,CAAA;EAAY,iBAAA;AAAA,IAC9C,CAAA,SAAU,MAAA,oBACR,CAAA,GACA,MAAA,SAAe,MAAA,qBACjB,MAAA,SAAe,MAAA;;;;;;;KAQP,cAAA,MAAoB,CAAA;EAAY,aAAA;AAAA,IACxC,mBAAA,CAAoB,CAAA,IACpB,eAAA;;;;;;;KAQQ,mBAAA,MAAyB,CAAA;EAAY,iBAAA;AAAA,IAC7C,gBAAA,CAAiB,CAAA,EAAG,mBAAA,CAAoB,CAAA,KACxC,qBAAA;;;;;AAZJ;;;;;;;;;;;;;;;;;AAUA;;;;;;;;;;;KAwCY,sBAAA,gBAAsC,WAAA,GAAc,WAAA,IAC9D,WAAA,CAAY,CAAA,iBACR,kBAAA,CACE,cAAA,CAAe,CAAA,GACf,cAAA,CAAe,CAAA,GACf,mBAAA,CAAoB,CAAA,GACpB,GAAA,IAEF,YAAA,CAAa,CAAA,iBACX,cAAA,CAAe,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA,IACrD,UAAA,CAAW,cAAA,CAAe,CAAA,GAAI,cAAA,CAAe,CAAA,GAAI,GAAA;;;;;;;;;;AAVzD;;;;;;;;;;;;;;;;;KAsCY,oBAAA,gBAAoC,WAAA,GAAc,WAAA,IAC5D,WAAA,CAAY,CAAA,iBACR,yBAAA,CAA0B,WAAA,CAAY,cAAA,CAAe,CAAA,IAAK,GAAA,IAC1D,YAAA,CAAa,CAAA,iBACX,qBAAA,CAAsB,WAAA,CAAY,cAAA,CAAe,CAAA,IAAK,GAAA,IACtD,gBAAA,CAAiB,WAAA,CAAY,cAAA,CAAe,CAAA,IAAK,GAAA;;;;;;;KAY7C,QAAA,cAAsB,WAAA,GAAc,WAAA,IAAe,CAAA;EAC7D,aAAA;AAAA,IAEE,WAAA,GACA,CAAA"}

package/dist/ui/types.d.cts

@@ -2,11 +2,12 @@ import { Checkpoint, Config, Interrupt, Metadata, ThreadState } from "../schema.
 import { AIMessage as AIMessage$1, DefaultToolCall, Message, ToolCallWithResult } from "../types.messages.cjs";
 import { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, EventsStreamEvent, MetadataStreamEvent, StreamMode, TasksStreamEvent, ToolsStreamEvent, UpdatesStreamEvent } from "../types.stream.cjs";
 import { Command, DisconnectMode, Durability, MultitaskStrategy, OnCompletionBehavior } from "../types.cjs";
-import { Client, ClientConfig } from "../client.cjs";
+import { ClientConfig } from "../client/base.cjs";
+import { Client } from "../client/index.cjs";
 import { BagTemplate } from "../types.template.cjs";
 import { AnyHeadlessToolImplementation, OnToolCallback } from "../headless-tools.cjs";
-import { InferInteropZodInput } from "@langchain/core/utils/types";
 import { BaseMessage } from "@langchain/core/messages";
+import { InferInteropZodInput } from "@langchain/core/utils/types";
 
 //#region src/ui/types.d.ts
 /**