@tanstack/ai-client 0.28.0 → 0.29.0

package/src/message-date-normalizer.ts

@@ -0,0 +1,29 @@
+import type { UIMessage } from './types'
+
+function validDate(value: unknown): Date | undefined {
+  if (value instanceof Date)
+    return Number.isNaN(value.getTime()) ? undefined : value
+  if (typeof value !== 'string') return undefined
+  const date = new Date(value)
+  return Number.isNaN(date.getTime()) ? undefined : date
+}
+
+export function normalizeMessageDates(message: UIMessage): UIMessage {
+  const messageDate = validDate(message.createdAt)
+  const parts = message.parts.map((part) => {
+    if (part.type !== 'tool-result') return part
+    const date = validDate(part.createdAt)
+    const { createdAt: _ignored, ...rest } = part
+    return date ? { ...rest, createdAt: date } : rest
+  })
+  const { createdAt: _ignored, ...rest } = message
+  return messageDate
+    ? { ...rest, parts, createdAt: messageDate }
+    : { ...rest, parts }
+}
+
+export function normalizeMessagesDates(
+  messages: Array<UIMessage>,
+): Array<UIMessage> {
+  return messages.map(normalizeMessageDates)
+}

package/src/storage-adapters.ts

@@ -1,4 +1,5 @@
-import type { ChatPersistedState, ChatStorageAdapter } from './types'
+import type { ChatPersistedState, ChatStorageAdapter, UIMessage } from './types'
+import { normalizeMessagesDates } from './message-date-normalizer'
 
 export interface WebStoragePersistenceOptions {
   keyPrefix?: string
@@ -45,6 +46,16 @@ function stringifyJson(value: ChatPersistedState): string {
   return serialized
 }
 
+function reviveMessageCreatedAt(message: UIMessage): UIMessage {
+  return normalizeMessagesDates([message]).at(0) ?? message
+}
+
+function revivePersistedState(state: ChatPersistedState): ChatPersistedState {
+  if (state == null || typeof state !== 'object') return state
+  if (!Array.isArray(state.messages)) return state
+  return { ...state, messages: state.messages.map(reviveMessageCreatedAt) }
+}
+
 function createWebStoragePersistence(
   storageName: 'localStorage' | 'sessionStorage',
   options: WebStoragePersistenceOptions,
@@ -69,7 +80,7 @@ function createWebStoragePersistence(
   return {
     getItem(id) {
       const item = getStorage().getItem(key(id))
-      return item === null ? null : deserialize(item)
+      return item === null ? null : revivePersistedState(deserialize(item))
     },
     setItem(id, value) {
       getStorage().setItem(key(id), serialize(value))

package/src/types.ts

@@ -595,10 +595,14 @@ export type ToolCallPart<TTools extends ReadonlyArray<AnyClientTool> = any> =
 
 export interface ToolResultPart {
   type: 'tool-result'
+  id?: string
+  name?: string
   toolCallId: string
   content: string | Array<ContentPart>
   state: ToolResultState
   error?: string // Error message if state is "error"
+  metadata?: Record<string, unknown>
+  createdAt?: Date
 }
 
 export interface ThinkingPart {
@@ -639,6 +643,7 @@ export interface UIMessage<
 > {
   id: string
   role: 'system' | 'user' | 'assistant'
+  name?: string
   parts: Array<MessagePart<TTools, TData>>
   createdAt?: Date
   /**