@assistant-ui/react 0.0.28 → 0.1.0

package/dist/experimental.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/experimental.ts","../src/runtime/local/useLocalRuntime.tsx","../src/utils/ModelConfigTypes.ts","../src/runtime/utils/idUtils.tsx","../src/runtime/utils/MessageRepository.tsx","../src/runtime/local/LocalRuntime.tsx","../src/context/AssistantContext.ts","../src/context/ThreadContext.ts","../src/context/ComposerContext.ts","../src/context/MessageContext.ts","../src/context/ContentPartContext.ts","../src/model-config/useAssistantInstructions.tsx"],"sourcesContent":["export type {\n  ImageContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n} from \"./utils/AssistantTypes\";\n\nexport type { ModelConfigProvider } from \"./utils/ModelConfigTypes\";\n\nexport { useLocalRuntime } from \"./runtime/local/useLocalRuntime\";\nexport type {\n  ChatModelAdapter,\n  ChatModelRunOptions,\n} from \"./runtime/local/ChatModelAdapter\";\n\nexport * from \"./context\";\nexport { useAssistantInstructions } from \"./model-config/useAssistantInstructions\";\n","\"use client\";\n\nimport { useInsertionEffect, useState } from \"react\";\nimport type { ChatModelAdapter } from \"./ChatModelAdapter\";\nimport { LocalRuntime } from \"./LocalRuntime\";\n\nexport const useLocalRuntime = (adapter: ChatModelAdapter) => {\n  const [runtime] = useState(() => new LocalRuntime(adapter));\n\n  useInsertionEffect(() => {\n    runtime.adapter = adapter;\n  });\n\n  return runtime;\n};\n","\"use client\";\nimport type { z } from \"zod\";\n\nexport type Tool<TArgs> = {\n  description: string;\n  parameters: z.ZodSchema<TArgs>;\n  execute: (args: TArgs) => unknown; // TODO return type\n};\n\nexport type ModelConfig = {\n  system?: string;\n  // biome-ignore lint/suspicious/noExplicitAny: TODO\n  tools?: Record<string, Tool<any>>;\n};\n\nexport type ModelConfigProvider = () => ModelConfig;\n\nexport const mergeModelConfigs = (configs: ModelConfig[]): ModelConfig => {\n  return configs.reduce((acc, config) => {\n    if (config.system) {\n      if (acc.system) {\n        // TODO should the separator be configurable?\n        acc.system += `\\n\\n${config.system}`;\n      } else {\n        acc.system = config.system;\n      }\n    }\n    if (config.tools) {\n      acc.tools = { ...acc.tools, ...config.tools };\n    }\n    return acc;\n  }, {} as ModelConfig);\n};\n","import { customAlphabet } from \"nanoid/non-secure\";\n\nexport const generateId = customAlphabet(\n  \"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz\",\n  7,\n);\n\nconst optimisticPrefix = \"__optimistic__\";\nexport const generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;\nexport const isOptimisticId = (id: string) => id.startsWith(optimisticPrefix);\n","import type { ThreadMessage } from \"../../utils/AssistantTypes\";\nimport { generateOptimisticId } from \"./idUtils\";\n\ntype RepositoryParent = {\n  children: string[];\n};\n\ntype RepositoryMessage = RepositoryParent & {\n  prev: RepositoryMessage | null;\n  current: ThreadMessage;\n  next: RepositoryMessage | null;\n  level: number;\n};\n\nconst findHead = (message: RepositoryMessage): RepositoryMessage => {\n  if (message.next) return findHead(message.next);\n  return message;\n};\n\nexport class MessageRepository {\n  private messages = new Map<string, RepositoryMessage>(); // message_id -> item\n  private head: RepositoryMessage | null = null;\n  private root: RepositoryParent = {\n    children: [],\n  };\n\n  private performOp(\n    newParent: RepositoryMessage | null,\n    child: RepositoryMessage,\n    operation: \"cut\" | \"link\" | \"relink\",\n  ) {\n    const parentOrRoot = child.prev ?? this.root;\n    const newParentOrRoot = newParent ?? this.root;\n\n    if (operation === \"relink\" && parentOrRoot === newParentOrRoot) return;\n\n    // cut\n    if (operation !== \"link\") {\n      parentOrRoot.children = parentOrRoot.children.filter(\n        (m) => m !== child.current.id,\n      );\n\n      if (child.prev?.next === child) {\n        const fallbackId = child.prev.children.at(-1);\n        const fallback = fallbackId ? this.messages.get(fallbackId) : null;\n        if (fallback === undefined) {\n          throw new Error(\n            \"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n        child.prev.next = fallback;\n      }\n    }\n\n    // link\n    if (operation !== \"cut\") {\n      newParentOrRoot.children = [\n        ...newParentOrRoot.children,\n        child.current.id,\n      ];\n\n      if (\n        newParent &&\n        (findHead(child) === this.head || newParent.next === null)\n      ) {\n        newParent.next = child;\n      }\n\n      child.prev = newParent;\n    }\n  }\n  getMessages() {\n    const messages = new Array<ThreadMessage>(this.head?.level ?? 0);\n    for (let current = this.head; current; current = current.prev) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  }\n\n  addOrUpdateMessage(parentId: string | null, message: ThreadMessage) {\n    const existingItem = this.messages.get(message.id);\n    const prev = parentId ? this.messages.get(parentId) : null;\n    if (prev === undefined)\n      throw new Error(\n        \"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    // update existing message\n    if (existingItem) {\n      existingItem.current = message;\n      this.performOp(prev, existingItem, \"relink\");\n      return;\n    }\n\n    // create a new message\n    const newItem: RepositoryMessage = {\n      prev,\n      current: message,\n      next: null,\n      children: [],\n      level: prev ? prev.level + 1 : 0,\n    };\n\n    this.messages.set(message.id, newItem);\n    this.performOp(prev, newItem, \"link\");\n\n    if (this.head === prev) {\n      this.head = newItem;\n    }\n  }\n\n  appendOptimisticMessage(\n    parentId: string | null,\n    message: Omit<ThreadMessage, \"id\" | \"createdAt\">,\n  ) {\n    let optimisticId: string;\n    do {\n      optimisticId = generateOptimisticId();\n    } while (this.messages.has(optimisticId));\n\n    this.addOrUpdateMessage(parentId, {\n      ...message,\n      id: optimisticId,\n      createdAt: new Date(),\n      ...(message.role === \"assistant\" ? { status: \"in_progress\" } : undefined),\n    } as ThreadMessage);\n\n    return optimisticId;\n  }\n\n  deleteMessage(messageId: string, replacementId?: string | null | undefined) {\n    const message = this.messages.get(messageId);\n\n    if (!message)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const replacement =\n      replacementId === undefined\n        ? message.prev // if no replacementId is provided, use the parent\n        : replacementId === null\n          ? null\n          : this.messages.get(replacementId);\n    if (replacement === undefined)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    for (const child of message.children) {\n      const childMessage = this.messages.get(child);\n      if (!childMessage)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.\",\n        );\n      this.performOp(replacement, childMessage, \"relink\");\n    }\n\n    this.performOp(null, message, \"cut\");\n    this.messages.delete(messageId);\n\n    if (this.head === message) {\n      this.head = replacement ? findHead(replacement) : null;\n    }\n  }\n\n  getBranches(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const { children } = message.prev ?? this.root;\n    return children;\n  }\n\n  switchToBranch(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (message.prev) {\n      message.prev.next = message;\n    }\n\n    this.head = findHead(message);\n  }\n\n  resetHead(messageId: string | null) {\n    if (messageId === null) {\n      this.head = null;\n      return;\n    }\n\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    this.head = message;\n    for (\n      let current: RepositoryMessage | null = message;\n      current;\n      current = current.prev\n    ) {\n      if (current.prev) {\n        current.prev.next = current;\n      }\n    }\n  }\n}\n","import type {\n  AppendMessage,\n  AssistantMessage,\n  UserMessage,\n} from \"../../utils/AssistantTypes\";\nimport {\n  type ModelConfigProvider,\n  mergeModelConfigs,\n} from \"../../utils/ModelConfigTypes\";\nimport type { Unsubscribe } from \"../../utils/Unsubscribe\";\nimport type { AssistantRuntime } from \"../core/AssistantRuntime\";\nimport { MessageRepository } from \"../utils/MessageRepository\";\nimport { generateId } from \"../utils/idUtils\";\nimport type { ChatModelAdapter, ChatModelRunResult } from \"./ChatModelAdapter\";\n\nexport class LocalRuntime implements AssistantRuntime {\n  private _subscriptions = new Set<() => void>();\n  private _configProviders = new Set<ModelConfigProvider>();\n\n  private abortController: AbortController | null = null;\n  private repository = new MessageRepository();\n\n  public get messages() {\n    return this.repository.getMessages();\n  }\n  public get isRunning() {\n    return this.abortController != null;\n  }\n\n  constructor(public adapter: ChatModelAdapter) {}\n\n  public getBranches(messageId: string): string[] {\n    return this.repository.getBranches(messageId);\n  }\n\n  public switchToBranch(branchId: string): void {\n    this.repository.switchToBranch(branchId);\n    this.notifySubscribers();\n  }\n\n  public async append(message: AppendMessage): Promise<void> {\n    // add user message\n    const userMessageId = generateId();\n    const userMessage: UserMessage = {\n      id: userMessageId,\n      role: \"user\",\n      content: message.content,\n      createdAt: new Date(),\n    };\n    this.repository.addOrUpdateMessage(message.parentId, userMessage);\n\n    await this.startRun(userMessageId);\n  }\n\n  public async startRun(parentId: string | null): Promise<void> {\n    const id = generateId();\n\n    this.repository.resetHead(parentId);\n    const messages = this.repository.getMessages();\n\n    // add assistant message\n    const message: AssistantMessage = {\n      id,\n      role: \"assistant\",\n      status: \"in_progress\",\n      content: [{ type: \"text\", text: \"\" }],\n      createdAt: new Date(),\n    };\n    this.repository.addOrUpdateMessage(parentId, { ...message });\n\n    // abort existing run\n    this.abortController?.abort();\n    this.abortController = new AbortController();\n\n    this.notifySubscribers();\n\n    try {\n      const updateHandler = ({ content }: ChatModelRunResult) => {\n        message.content = content;\n        this.repository.addOrUpdateMessage(parentId, { ...message });\n        this.notifySubscribers();\n      };\n      const result = await this.adapter.run({\n        messages,\n        abortSignal: this.abortController.signal,\n        config: mergeModelConfigs([...this._configProviders].map((p) => p())),\n        onUpdate: updateHandler,\n      });\n      updateHandler(result);\n\n      message.status = \"done\";\n      this.repository.addOrUpdateMessage(parentId, { ...message });\n    } catch (e) {\n      message.status = \"error\";\n      this.repository.addOrUpdateMessage(parentId, { ...message });\n      console.error(e);\n    } finally {\n      this.abortController = null;\n      this.notifySubscribers();\n    }\n  }\n\n  cancelRun(): void {\n    if (!this.abortController) return;\n\n    this.abortController.abort();\n    this.abortController = null;\n    this.notifySubscribers();\n  }\n\n  private notifySubscribers() {\n    for (const callback of this._subscriptions) callback();\n  }\n\n  public subscribe(callback: () => void): Unsubscribe {\n    this._subscriptions.add(callback);\n    return () => this._subscriptions.delete(callback);\n  }\n\n  registerModelConfigProvider(provider: ModelConfigProvider) {\n    this._configProviders.add(provider);\n    return () => this._configProviders.delete(provider);\n  }\n}\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { AssistantModelConfigState } from \"./stores/AssistantModelConfig\";\n\nexport type AssistantContextValue = {\n  useModelConfig: UseBoundStore<StoreApi<AssistantModelConfigState>>;\n};\n\nexport const AssistantContext = createContext<AssistantContextValue | null>(\n  null,\n);\n\nexport const useAssistantContext = (): AssistantContextValue => {\n  const context = useContext(AssistantContext);\n  if (!context)\n    throw new Error(\n      \"This component must be used within an AssistantRuntimeProvider.\",\n    );\n  return context;\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { ComposerState } from \"./stores/Composer\";\nimport type { ThreadState } from \"./stores/Thread\";\nimport type { ThreadViewportState } from \"./stores/ThreadViewport\";\n\nexport type ThreadContextValue = {\n  useThread: UseBoundStore<StoreApi<ThreadState>>;\n  useComposer: UseBoundStore<StoreApi<ComposerState>>;\n  useViewport: UseBoundStore<StoreApi<ThreadViewportState>>;\n};\n\nexport const ThreadContext = createContext<ThreadContextValue | null>(null);\n\nexport const useThreadContext = (): ThreadContextValue => {\n  const context = useContext(ThreadContext);\n  if (!context)\n    throw new Error(\"This component must be used within an AssistantRuntimeProvider.\");\n  return context;\n};\n","import { useContext, useMemo } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport { MessageContext } from \"./MessageContext\";\nimport { useThreadContext } from \"./ThreadContext\";\nimport type { ComposerState } from \"./stores/Composer\";\nimport type { EditComposerState } from \"./stores/MessageComposer\";\n\nexport type ComposerContextValue = {\n  useComposer: UseBoundStore<StoreApi<EditComposerState | ComposerState>>;\n  type: \"edit\" | \"new\";\n};\n\nexport const useComposerContext = (): ComposerContextValue => {\n  const { useComposer } = useThreadContext();\n  const { useComposer: useEditComposer } = useContext(MessageContext) ?? {};\n  return useMemo(\n    () => ({\n      useComposer: (useEditComposer ?? useComposer) as UseBoundStore<\n        StoreApi<EditComposerState | ComposerState>\n      >,\n      type: useEditComposer ? (\"edit\" as const) : (\"new\" as const),\n    }),\n    [useEditComposer, useComposer],\n  );\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { MessageState } from \"./stores/Message\";\nimport type { EditComposerState } from \"./stores/MessageComposer\";\n\nexport type MessageContextValue = {\n  useMessage: UseBoundStore<StoreApi<MessageState>>;\n  useComposer: UseBoundStore<StoreApi<EditComposerState>>;\n};\n\nexport const MessageContext = createContext<MessageContextValue | null>(null);\n\nexport const useMessageContext = () => {\n  const context = useContext(MessageContext);\n  if (!context)\n    throw new Error(\n      \"This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />.\",\n    );\n  return context;\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { ContentPartState } from \"./stores/ContentPart\";\n\nexport type ContentPartContextValue = {\n  useContentPart: UseBoundStore<StoreApi<ContentPartState>>;\n};\n\nexport const ContentPartContext = createContext<ContentPartContextValue | null>(\n  null,\n);\n\nexport const useContentPartContext = (): ContentPartContextValue => {\n  const context = useContext(ContentPartContext);\n  if (!context)\n    throw new Error(\n      \"This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >.\",\n    );\n  return context;\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\n\nexport const useAssistantInstructions = (instruction: string) => {\n  const { useModelConfig } = useAssistantContext();\n  const addContextProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  useEffect(\n    () =>\n      addContextProvider(() => {\n        return {\n          system: instruction,\n        };\n      }),\n    [addContextProvider, instruction],\n  );\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACEA,mBAA6C;;;ACetC,IAAM,oBAAoB,CAAC,YAAwC;AACxE,SAAO,QAAQ,OAAO,CAAC,KAAK,WAAW;AACrC,QAAI,OAAO,QAAQ;AACjB,UAAI,IAAI,QAAQ;AAEd,YAAI,UAAU;AAAA;AAAA,EAAO,OAAO,MAAM;AAAA,MACpC,OAAO;AACL,YAAI,SAAS,OAAO;AAAA,MACtB;AAAA,IACF;AACA,QAAI,OAAO,OAAO;AAChB,UAAI,QAAQ,EAAE,GAAG,IAAI,OAAO,GAAG,OAAO,MAAM;AAAA,IAC9C;AACA,WAAO;AAAA,EACT,GAAG,CAAC,CAAgB;AACtB;;;AChCA,wBAA+B;AAExB,IAAM,iBAAa;AAAA,EACxB;AAAA,EACA;AACF;AAEA,IAAM,mBAAmB;AAClB,IAAM,uBAAuB,MAAM,GAAG,gBAAgB,GAAG,WAAW,CAAC;;;ACM5E,IAAM,WAAW,CAAC,YAAkD;AAClE,MAAI,QAAQ,KAAM,QAAO,SAAS,QAAQ,IAAI;AAC9C,SAAO;AACT;AAEO,IAAM,oBAAN,MAAwB;AAAA,EACrB,WAAW,oBAAI,IAA+B;AAAA;AAAA,EAC9C,OAAiC;AAAA,EACjC,OAAyB;AAAA,IAC/B,UAAU,CAAC;AAAA,EACb;AAAA,EAEQ,UACN,WACA,OACA,WACA;AACA,UAAM,eAAe,MAAM,QAAQ,KAAK;AACxC,UAAM,kBAAkB,aAAa,KAAK;AAE1C,QAAI,cAAc,YAAY,iBAAiB,gBAAiB;AAGhE,QAAI,cAAc,QAAQ;AACxB,mBAAa,WAAW,aAAa,SAAS;AAAA,QAC5C,CAAC,MAAM,MAAM,MAAM,QAAQ;AAAA,MAC7B;AAEA,UAAI,MAAM,MAAM,SAAS,OAAO;AAC9B,cAAM,aAAa,MAAM,KAAK,SAAS,GAAG,EAAE;AAC5C,cAAM,WAAW,aAAa,KAAK,SAAS,IAAI,UAAU,IAAI;AAC9D,YAAI,aAAa,QAAW;AAC1B,gBAAM,IAAI;AAAA,YACR;AAAA,UACF;AAAA,QACF;AACA,cAAM,KAAK,OAAO;AAAA,MACpB;AAAA,IACF;AAGA,QAAI,cAAc,OAAO;AACvB,sBAAgB,WAAW;AAAA,QACzB,GAAG,gBAAgB;AAAA,QACnB,MAAM,QAAQ;AAAA,MAChB;AAEA,UACE,cACC,SAAS,KAAK,MAAM,KAAK,QAAQ,UAAU,SAAS,OACrD;AACA,kBAAU,OAAO;AAAA,MACnB;AAEA,YAAM,OAAO;AAAA,IACf;AAAA,EACF;AAAA,EACA,cAAc;AACZ,UAAM,WAAW,IAAI,MAAqB,KAAK,MAAM,SAAS,CAAC;AAC/D,aAAS,UAAU,KAAK,MAAM,SAAS,UAAU,QAAQ,MAAM;AAC7D,eAAS,QAAQ,KAAK,IAAI,QAAQ;AAAA,IACpC;AACA,WAAO;AAAA,EACT;AAAA,EAEA,mBAAmB,UAAyB,SAAwB;AAClE,UAAM,eAAe,KAAK,SAAS,IAAI,QAAQ,EAAE;AACjD,UAAM,OAAO,WAAW,KAAK,SAAS,IAAI,QAAQ,IAAI;AACtD,QAAI,SAAS;AACX,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAGF,QAAI,cAAc;AAChB,mBAAa,UAAU;AACvB,WAAK,UAAU,MAAM,cAAc,QAAQ;AAC3C;AAAA,IACF;AAGA,UAAM,UAA6B;AAAA,MACjC;AAAA,MACA,SAAS;AAAA,MACT,MAAM;AAAA,MACN,UAAU,CAAC;AAAA,MACX,OAAO,OAAO,KAAK,QAAQ,IAAI;AAAA,IACjC;AAEA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AACrC,SAAK,UAAU,MAAM,SAAS,MAAM;AAEpC,QAAI,KAAK,SAAS,MAAM;AACtB,WAAK,OAAO;AAAA,IACd;AAAA,EACF;AAAA,EAEA,wBACE,UACA,SACA;AACA,QAAI;AACJ,OAAG;AACD,qBAAe,qBAAqB;AAAA,IACtC,SAAS,KAAK,SAAS,IAAI,YAAY;AAEvC,SAAK,mBAAmB,UAAU;AAAA,MAChC,GAAG;AAAA,MACH,IAAI;AAAA,MACJ,WAAW,oBAAI,KAAK;AAAA,MACpB,GAAI,QAAQ,SAAS,cAAc,EAAE,QAAQ,cAAc,IAAI;AAAA,IACjE,CAAkB;AAElB,WAAO;AAAA,EACT;AAAA,EAEA,cAAc,WAAmB,eAA2C;AAC1E,UAAM,UAAU,KAAK,SAAS,IAAI,SAAS;AAE3C,QAAI,CAAC;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,UAAM,cACJ,kBAAkB,SACd,QAAQ,OACR,kBAAkB,OAChB,OACA,KAAK,SAAS,IAAI,aAAa;AACvC,QAAI,gBAAgB;AAClB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,eAAW,SAAS,QAAQ,UAAU;AACpC,YAAM,eAAe,KAAK,SAAS,IAAI,KAAK;AAC5C,UAAI,CAAC;AACH,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AACF,WAAK,UAAU,aAAa,cAAc,QAAQ;AAAA,IACpD;AAEA,SAAK,UAAU,MAAM,SAAS,KAAK;AACnC,SAAK,SAAS,OAAO,SAAS;AAE9B,QAAI,KAAK,SAAS,SAAS;AACzB,WAAK,OAAO,cAAc,SAAS,WAAW,IAAI;AAAA,IACpD;AAAA,EACF;AAAA,EAEA,YAAY,WAAmB;AAC7B,UAAM,UAAU,KAAK,SAAS,IAAI,SAAS;AAC3C,QAAI,CAAC;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,UAAM,EAAE,SAAS,IAAI,QAAQ,QAAQ,KAAK;AAC1C,WAAO;AAAA,EACT;AAAA,EAEA,eAAe,WAAmB;AAChC,UAAM,UAAU,KAAK,SAAS,IAAI,SAAS;AAC3C,QAAI,CAAC;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,QAAI,QAAQ,MAAM;AAChB,cAAQ,KAAK,OAAO;AAAA,IACtB;AAEA,SAAK,OAAO,SAAS,OAAO;AAAA,EAC9B;AAAA,EAEA,UAAU,WAA0B;AAClC,QAAI,cAAc,MAAM;AACtB,WAAK,OAAO;AACZ;AAAA,IACF;AAEA,UAAM,UAAU,KAAK,SAAS,IAAI,SAAS;AAC3C,QAAI,CAAC;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,SAAK,OAAO;AACZ,aACM,UAAoC,SACxC,SACA,UAAU,QAAQ,MAClB;AACA,UAAI,QAAQ,MAAM;AAChB,gBAAQ,KAAK,OAAO;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AACF;;;ACvMO,IAAM,eAAN,MAA+C;AAAA,EAcpD,YAAmB,SAA2B;AAA3B;AAAA,EAA4B;AAAA,EAbvC,iBAAiB,oBAAI,IAAgB;AAAA,EACrC,mBAAmB,oBAAI,IAAyB;AAAA,EAEhD,kBAA0C;AAAA,EAC1C,aAAa,IAAI,kBAAkB;AAAA,EAE3C,IAAW,WAAW;AACpB,WAAO,KAAK,WAAW,YAAY;AAAA,EACrC;AAAA,EACA,IAAW,YAAY;AACrB,WAAO,KAAK,mBAAmB;AAAA,EACjC;AAAA,EAIO,YAAY,WAA6B;AAC9C,WAAO,KAAK,WAAW,YAAY,SAAS;AAAA,EAC9C;AAAA,EAEO,eAAe,UAAwB;AAC5C,SAAK,WAAW,eAAe,QAAQ;AACvC,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,MAAa,OAAO,SAAuC;AAEzD,UAAM,gBAAgB,WAAW;AACjC,UAAM,cAA2B;AAAA,MAC/B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS,QAAQ;AAAA,MACjB,WAAW,oBAAI,KAAK;AAAA,IACtB;AACA,SAAK,WAAW,mBAAmB,QAAQ,UAAU,WAAW;AAEhE,UAAM,KAAK,SAAS,aAAa;AAAA,EACnC;AAAA,EAEA,MAAa,SAAS,UAAwC;AAC5D,UAAM,KAAK,WAAW;AAEtB,SAAK,WAAW,UAAU,QAAQ;AAClC,UAAM,WAAW,KAAK,WAAW,YAAY;AAG7C,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA,MAAM;AAAA,MACN,QAAQ;AAAA,MACR,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,GAAG,CAAC;AAAA,MACpC,WAAW,oBAAI,KAAK;AAAA,IACtB;AACA,SAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAG3D,SAAK,iBAAiB,MAAM;AAC5B,SAAK,kBAAkB,IAAI,gBAAgB;AAE3C,SAAK,kBAAkB;AAEvB,QAAI;AACF,YAAM,gBAAgB,CAAC,EAAE,QAAQ,MAA0B;AACzD,gBAAQ,UAAU;AAClB,aAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAC3D,aAAK,kBAAkB;AAAA,MACzB;AACA,YAAM,SAAS,MAAM,KAAK,QAAQ,IAAI;AAAA,QACpC;AAAA,QACA,aAAa,KAAK,gBAAgB;AAAA,QAClC,QAAQ,kBAAkB,CAAC,GAAG,KAAK,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAAA,QACpE,UAAU;AAAA,MACZ,CAAC;AACD,oBAAc,MAAM;AAEpB,cAAQ,SAAS;AACjB,WAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAAA,IAC7D,SAAS,GAAG;AACV,cAAQ,SAAS;AACjB,WAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAC3D,cAAQ,MAAM,CAAC;AAAA,IACjB,UAAE;AACA,WAAK,kBAAkB;AACvB,WAAK,kBAAkB;AAAA,IACzB;AAAA,EACF;AAAA,EAEA,YAAkB;AAChB,QAAI,CAAC,KAAK,gBAAiB;AAE3B,SAAK,gBAAgB,MAAM;AAC3B,SAAK,kBAAkB;AACvB,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEQ,oBAAoB;AAC1B,eAAW,YAAY,KAAK,eAAgB,UAAS;AAAA,EACvD;AAAA,EAEO,UAAU,UAAmC;AAClD,SAAK,eAAe,IAAI,QAAQ;AAChC,WAAO,MAAM,KAAK,eAAe,OAAO,QAAQ;AAAA,EAClD;AAAA,EAEA,4BAA4B,UAA+B;AACzD,SAAK,iBAAiB,IAAI,QAAQ;AAClC,WAAO,MAAM,KAAK,iBAAiB,OAAO,QAAQ;AAAA,EACpD;AACF;;;AJrHO,IAAM,kBAAkB,CAAC,YAA8B;AAC5D,QAAM,CAAC,OAAO,QAAI,uBAAS,MAAM,IAAI,aAAa,OAAO,CAAC;AAE1D,uCAAmB,MAAM;AACvB,YAAQ,UAAU;AAAA,EACpB,CAAC;AAED,SAAO;AACT;;;AKdA,IAAAA,gBAA0C;AAQnC,IAAM,uBAAmB;AAAA,EAC9B;AACF;AAEO,IAAM,sBAAsB,MAA6B;AAC9D,QAAM,cAAU,0BAAW,gBAAgB;AAC3C,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ACnBA,IAAAC,gBAA0C;AAYnC,IAAM,oBAAgB,6BAAyC,IAAI;AAEnE,IAAM,mBAAmB,MAA0B;AACxD,QAAM,cAAU,0BAAW,aAAa;AACxC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,iEAAiE;AACnF,SAAO;AACT;;;ACnBA,IAAAC,gBAAoC;;;ACApC,IAAAC,gBAA0C;AAUnC,IAAM,qBAAiB,6BAA0C,IAAI;AAErE,IAAM,oBAAoB,MAAM;AACrC,QAAM,cAAU,0BAAW,cAAc;AACzC,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ADPO,IAAM,qBAAqB,MAA4B;AAC5D,QAAM,EAAE,YAAY,IAAI,iBAAiB;AACzC,QAAM,EAAE,aAAa,gBAAgB,QAAI,0BAAW,cAAc,KAAK,CAAC;AACxE,aAAO;AAAA,IACL,OAAO;AAAA,MACL,aAAc,mBAAmB;AAAA,MAGjC,MAAM,kBAAmB,SAAoB;AAAA,IAC/C;AAAA,IACA,CAAC,iBAAiB,WAAW;AAAA,EAC/B;AACF;;;AExBA,IAAAC,gBAA0C;AAQnC,IAAM,yBAAqB;AAAA,EAChC;AACF;AAEO,IAAM,wBAAwB,MAA+B;AAClE,QAAM,cAAU,0BAAW,kBAAkB;AAC7C,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ACjBA,IAAAC,gBAA0B;AAGnB,IAAM,2BAA2B,CAAC,gBAAwB;AAC/D,QAAM,EAAE,eAAe,IAAI,oBAAoB;AAC/C,QAAM,qBAAqB;AAAA,IACzB,CAAC,MAAM,EAAE;AAAA,EACX;AACA;AAAA,IACE,MACE,mBAAmB,MAAM;AACvB,aAAO;AAAA,QACL,QAAQ;AAAA,MACV;AAAA,IACF,CAAC;AAAA,IACH,CAAC,oBAAoB,WAAW;AAAA,EAClC;AACF;","names":["import_react","import_react","import_react","import_react","import_react","import_react"]}

package/dist/experimental.mjs

@@ -0,0 +1,144 @@
+import {
+  useAssistantContext,
+  useComposerContext,
+  useContentPartContext,
+  useMessageContext,
+  useThreadContext
+} from "./chunk-CY4TTHR7.mjs";
+import {
+  MessageRepository,
+  generateId
+} from "./chunk-GQKH2ADD.mjs";
+import {
+  mergeModelConfigs
+} from "./chunk-X4HBDEFP.mjs";
+
+// src/runtime/local/useLocalRuntime.tsx
+import { useInsertionEffect, useState } from "react";
+
+// src/runtime/local/LocalRuntime.tsx
+var LocalRuntime = class {
+  constructor(adapter) {
+    this.adapter = adapter;
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  _configProviders = /* @__PURE__ */ new Set();
+  abortController = null;
+  repository = new MessageRepository();
+  get messages() {
+    return this.repository.getMessages();
+  }
+  get isRunning() {
+    return this.abortController != null;
+  }
+  getBranches(messageId) {
+    return this.repository.getBranches(messageId);
+  }
+  switchToBranch(branchId) {
+    this.repository.switchToBranch(branchId);
+    this.notifySubscribers();
+  }
+  async append(message) {
+    const userMessageId = generateId();
+    const userMessage = {
+      id: userMessageId,
+      role: "user",
+      content: message.content,
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(message.parentId, userMessage);
+    await this.startRun(userMessageId);
+  }
+  async startRun(parentId) {
+    const id = generateId();
+    this.repository.resetHead(parentId);
+    const messages = this.repository.getMessages();
+    const message = {
+      id,
+      role: "assistant",
+      status: "in_progress",
+      content: [{ type: "text", text: "" }],
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(parentId, { ...message });
+    this.abortController?.abort();
+    this.abortController = new AbortController();
+    this.notifySubscribers();
+    try {
+      const updateHandler = ({ content }) => {
+        message.content = content;
+        this.repository.addOrUpdateMessage(parentId, { ...message });
+        this.notifySubscribers();
+      };
+      const result = await this.adapter.run({
+        messages,
+        abortSignal: this.abortController.signal,
+        config: mergeModelConfigs([...this._configProviders].map((p) => p())),
+        onUpdate: updateHandler
+      });
+      updateHandler(result);
+      message.status = "done";
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+    } catch (e) {
+      message.status = "error";
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+      console.error(e);
+    } finally {
+      this.abortController = null;
+      this.notifySubscribers();
+    }
+  }
+  cancelRun() {
+    if (!this.abortController) return;
+    this.abortController.abort();
+    this.abortController = null;
+    this.notifySubscribers();
+  }
+  notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
+  }
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+  registerModelConfigProvider(provider) {
+    this._configProviders.add(provider);
+    return () => this._configProviders.delete(provider);
+  }
+};
+
+// src/runtime/local/useLocalRuntime.tsx
+var useLocalRuntime = (adapter) => {
+  const [runtime] = useState(() => new LocalRuntime(adapter));
+  useInsertionEffect(() => {
+    runtime.adapter = adapter;
+  });
+  return runtime;
+};
+
+// src/model-config/useAssistantInstructions.tsx
+import { useEffect } from "react";
+var useAssistantInstructions = (instruction) => {
+  const { useModelConfig } = useAssistantContext();
+  const addContextProvider = useModelConfig(
+    (s) => s.registerModelConfigProvider
+  );
+  useEffect(
+    () => addContextProvider(() => {
+      return {
+        system: instruction
+      };
+    }),
+    [addContextProvider, instruction]
+  );
+};
+export {
+  useAssistantContext,
+  useAssistantInstructions,
+  useComposerContext,
+  useContentPartContext,
+  useLocalRuntime,
+  useMessageContext,
+  useThreadContext
+};
+//# sourceMappingURL=experimental.mjs.map

package/dist/experimental.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/runtime/local/useLocalRuntime.tsx","../src/runtime/local/LocalRuntime.tsx","../src/model-config/useAssistantInstructions.tsx"],"sourcesContent":["\"use client\";\n\nimport { useInsertionEffect, useState } from \"react\";\nimport type { ChatModelAdapter } from \"./ChatModelAdapter\";\nimport { LocalRuntime } from \"./LocalRuntime\";\n\nexport const useLocalRuntime = (adapter: ChatModelAdapter) => {\n  const [runtime] = useState(() => new LocalRuntime(adapter));\n\n  useInsertionEffect(() => {\n    runtime.adapter = adapter;\n  });\n\n  return runtime;\n};\n","import type {\n  AppendMessage,\n  AssistantMessage,\n  UserMessage,\n} from \"../../utils/AssistantTypes\";\nimport {\n  type ModelConfigProvider,\n  mergeModelConfigs,\n} from \"../../utils/ModelConfigTypes\";\nimport type { Unsubscribe } from \"../../utils/Unsubscribe\";\nimport type { AssistantRuntime } from \"../core/AssistantRuntime\";\nimport { MessageRepository } from \"../utils/MessageRepository\";\nimport { generateId } from \"../utils/idUtils\";\nimport type { ChatModelAdapter, ChatModelRunResult } from \"./ChatModelAdapter\";\n\nexport class LocalRuntime implements AssistantRuntime {\n  private _subscriptions = new Set<() => void>();\n  private _configProviders = new Set<ModelConfigProvider>();\n\n  private abortController: AbortController | null = null;\n  private repository = new MessageRepository();\n\n  public get messages() {\n    return this.repository.getMessages();\n  }\n  public get isRunning() {\n    return this.abortController != null;\n  }\n\n  constructor(public adapter: ChatModelAdapter) {}\n\n  public getBranches(messageId: string): string[] {\n    return this.repository.getBranches(messageId);\n  }\n\n  public switchToBranch(branchId: string): void {\n    this.repository.switchToBranch(branchId);\n    this.notifySubscribers();\n  }\n\n  public async append(message: AppendMessage): Promise<void> {\n    // add user message\n    const userMessageId = generateId();\n    const userMessage: UserMessage = {\n      id: userMessageId,\n      role: \"user\",\n      content: message.content,\n      createdAt: new Date(),\n    };\n    this.repository.addOrUpdateMessage(message.parentId, userMessage);\n\n    await this.startRun(userMessageId);\n  }\n\n  public async startRun(parentId: string | null): Promise<void> {\n    const id = generateId();\n\n    this.repository.resetHead(parentId);\n    const messages = this.repository.getMessages();\n\n    // add assistant message\n    const message: AssistantMessage = {\n      id,\n      role: \"assistant\",\n      status: \"in_progress\",\n      content: [{ type: \"text\", text: \"\" }],\n      createdAt: new Date(),\n    };\n    this.repository.addOrUpdateMessage(parentId, { ...message });\n\n    // abort existing run\n    this.abortController?.abort();\n    this.abortController = new AbortController();\n\n    this.notifySubscribers();\n\n    try {\n      const updateHandler = ({ content }: ChatModelRunResult) => {\n        message.content = content;\n        this.repository.addOrUpdateMessage(parentId, { ...message });\n        this.notifySubscribers();\n      };\n      const result = await this.adapter.run({\n        messages,\n        abortSignal: this.abortController.signal,\n        config: mergeModelConfigs([...this._configProviders].map((p) => p())),\n        onUpdate: updateHandler,\n      });\n      updateHandler(result);\n\n      message.status = \"done\";\n      this.repository.addOrUpdateMessage(parentId, { ...message });\n    } catch (e) {\n      message.status = \"error\";\n      this.repository.addOrUpdateMessage(parentId, { ...message });\n      console.error(e);\n    } finally {\n      this.abortController = null;\n      this.notifySubscribers();\n    }\n  }\n\n  cancelRun(): void {\n    if (!this.abortController) return;\n\n    this.abortController.abort();\n    this.abortController = null;\n    this.notifySubscribers();\n  }\n\n  private notifySubscribers() {\n    for (const callback of this._subscriptions) callback();\n  }\n\n  public subscribe(callback: () => void): Unsubscribe {\n    this._subscriptions.add(callback);\n    return () => this._subscriptions.delete(callback);\n  }\n\n  registerModelConfigProvider(provider: ModelConfigProvider) {\n    this._configProviders.add(provider);\n    return () => this._configProviders.delete(provider);\n  }\n}\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\n\nexport const useAssistantInstructions = (instruction: string) => {\n  const { useModelConfig } = useAssistantContext();\n  const addContextProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  useEffect(\n    () =>\n      addContextProvider(() => {\n        return {\n          system: instruction,\n        };\n      }),\n    [addContextProvider, instruction],\n  );\n};\n"],"mappings":";;;;;;;;;;;;;;;;AAEA,SAAS,oBAAoB,gBAAgB;;;ACatC,IAAM,eAAN,MAA+C;AAAA,EAcpD,YAAmB,SAA2B;AAA3B;AAAA,EAA4B;AAAA,EAbvC,iBAAiB,oBAAI,IAAgB;AAAA,EACrC,mBAAmB,oBAAI,IAAyB;AAAA,EAEhD,kBAA0C;AAAA,EAC1C,aAAa,IAAI,kBAAkB;AAAA,EAE3C,IAAW,WAAW;AACpB,WAAO,KAAK,WAAW,YAAY;AAAA,EACrC;AAAA,EACA,IAAW,YAAY;AACrB,WAAO,KAAK,mBAAmB;AAAA,EACjC;AAAA,EAIO,YAAY,WAA6B;AAC9C,WAAO,KAAK,WAAW,YAAY,SAAS;AAAA,EAC9C;AAAA,EAEO,eAAe,UAAwB;AAC5C,SAAK,WAAW,eAAe,QAAQ;AACvC,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,MAAa,OAAO,SAAuC;AAEzD,UAAM,gBAAgB,WAAW;AACjC,UAAM,cAA2B;AAAA,MAC/B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS,QAAQ;AAAA,MACjB,WAAW,oBAAI,KAAK;AAAA,IACtB;AACA,SAAK,WAAW,mBAAmB,QAAQ,UAAU,WAAW;AAEhE,UAAM,KAAK,SAAS,aAAa;AAAA,EACnC;AAAA,EAEA,MAAa,SAAS,UAAwC;AAC5D,UAAM,KAAK,WAAW;AAEtB,SAAK,WAAW,UAAU,QAAQ;AAClC,UAAM,WAAW,KAAK,WAAW,YAAY;AAG7C,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA,MAAM;AAAA,MACN,QAAQ;AAAA,MACR,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,GAAG,CAAC;AAAA,MACpC,WAAW,oBAAI,KAAK;AAAA,IACtB;AACA,SAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAG3D,SAAK,iBAAiB,MAAM;AAC5B,SAAK,kBAAkB,IAAI,gBAAgB;AAE3C,SAAK,kBAAkB;AAEvB,QAAI;AACF,YAAM,gBAAgB,CAAC,EAAE,QAAQ,MAA0B;AACzD,gBAAQ,UAAU;AAClB,aAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAC3D,aAAK,kBAAkB;AAAA,MACzB;AACA,YAAM,SAAS,MAAM,KAAK,QAAQ,IAAI;AAAA,QACpC;AAAA,QACA,aAAa,KAAK,gBAAgB;AAAA,QAClC,QAAQ,kBAAkB,CAAC,GAAG,KAAK,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAAA,QACpE,UAAU;AAAA,MACZ,CAAC;AACD,oBAAc,MAAM;AAEpB,cAAQ,SAAS;AACjB,WAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAAA,IAC7D,SAAS,GAAG;AACV,cAAQ,SAAS;AACjB,WAAK,WAAW,mBAAmB,UAAU,EAAE,GAAG,QAAQ,CAAC;AAC3D,cAAQ,MAAM,CAAC;AAAA,IACjB,UAAE;AACA,WAAK,kBAAkB;AACvB,WAAK,kBAAkB;AAAA,IACzB;AAAA,EACF;AAAA,EAEA,YAAkB;AAChB,QAAI,CAAC,KAAK,gBAAiB;AAE3B,SAAK,gBAAgB,MAAM;AAC3B,SAAK,kBAAkB;AACvB,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEQ,oBAAoB;AAC1B,eAAW,YAAY,KAAK,eAAgB,UAAS;AAAA,EACvD;AAAA,EAEO,UAAU,UAAmC;AAClD,SAAK,eAAe,IAAI,QAAQ;AAChC,WAAO,MAAM,KAAK,eAAe,OAAO,QAAQ;AAAA,EAClD;AAAA,EAEA,4BAA4B,UAA+B;AACzD,SAAK,iBAAiB,IAAI,QAAQ;AAClC,WAAO,MAAM,KAAK,iBAAiB,OAAO,QAAQ;AAAA,EACpD;AACF;;;ADrHO,IAAM,kBAAkB,CAAC,YAA8B;AAC5D,QAAM,CAAC,OAAO,IAAI,SAAS,MAAM,IAAI,aAAa,OAAO,CAAC;AAE1D,qBAAmB,MAAM;AACvB,YAAQ,UAAU;AAAA,EACpB,CAAC;AAED,SAAO;AACT;;;AEZA,SAAS,iBAAiB;AAGnB,IAAM,2BAA2B,CAAC,gBAAwB;AAC/D,QAAM,EAAE,eAAe,IAAI,oBAAoB;AAC/C,QAAM,qBAAqB;AAAA,IACzB,CAAC,MAAM,EAAE;AAAA,EACX;AACA;AAAA,IACE,MACE,mBAAmB,MAAM;AACvB,aAAO;AAAA,QACL,QAAQ;AAAA,MACV;AAAA,IACF,CAAC;AAAA,IACH,CAAC,oBAAoB,WAAW;AAAA,EAClC;AACF;","names":[]}

package/dist/index.d.mts

@@ -1,10 +1,11 @@
 import * as react from 'react';
 import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
-import { UseChatHelpers, UseAssistantHelpers } from '@ai-sdk/react';
-import * as react_jsx_runtime from 'react/jsx-runtime';
-import { UseBoundStore, StoreApi } from 'zustand';
-import { Message } from 'ai';
+import { T as TextContentPart, I as ImageContentPart, U as UIContentPart, a as ToolCallContentPart } from './ModelConfigTypes-B9UY4zxv.mjs';
+export { g as AppendContentPart, d as AppendMessage, e as AssistantContentPart, A as AssistantMessage, b as ThreadMessage, f as UserContentPart, c as UserMessage } from './ModelConfigTypes-B9UY4zxv.mjs';
+import { T as ThreadRuntime, A as AssistantRuntime } from './AssistantRuntime-CBMSAJqH.mjs';
+export { U as Unsubscribe } from './AssistantRuntime-CBMSAJqH.mjs';
+import 'zod';
 
 declare const useCopyMessage: ({ copiedDuration }: {
     copiedDuration?: number | undefined;
@@ -133,46 +134,6 @@ type MessageIfProps = RequireAtLeastOne<MessageIfFilters> & {
 };
 declare const MessageIf: FC<MessageIfProps>;
 
-type TextContentPart = {
-    type: "text";
-    text: string;
-};
-type ImageContentPart = {
-    type: "image";
-    image: string;
-};
-type UIContentPart = {
-    type: "ui";
-    display: ReactNode;
-};
-type ToolCallContentPart = {
-    type: "tool-call";
-    name: string;
-    args: object;
-    result?: object;
-};
-type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
-type AssistantContentPart = TextContentPart | UIContentPart | ToolCallContentPart;
-type AppendContentPart = TextContentPart | ImageContentPart;
-type BaseMessage = {
-    id: string;
-    createdAt: Date;
-};
-type UserMessage = BaseMessage & {
-    role: "user";
-    content: UserContentPart[];
-};
-type AssistantMessage = BaseMessage & {
-    role: "assistant";
-    content: AssistantContentPart[];
-    status: "in_progress" | "done" | "error";
-};
-type AppendMessage = {
-    parentId: string | null;
-    content: AppendContentPart[];
-};
-type ThreadMessage = UserMessage | AssistantMessage;
-
 type MessageContentProps = {
     components?: {
         Text?: ComponentType<{
@@ -275,234 +236,13 @@ declare namespace index {
   export { ContentPartInProgressIndicator as InProgressIndicator };
 }
 
-type VercelRSCMessage = {
-    id: string;
-    role: "user" | "assistant";
-    display: ReactNode;
-    createdAt?: Date;
-};
-
-type RSCMessageConverter<T> = {
-    convertMessage: (message: T) => VercelRSCMessage;
-};
-type VercelRSCAdapterBase<T> = {
-    messages: T[];
-    append: (message: AppendMessage) => Promise<void>;
-    edit?: (message: AppendMessage) => Promise<void>;
-    reload?: (parentId: string | null) => Promise<void>;
-    convertMessage?: (message: T) => VercelRSCMessage;
-};
-type VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
-
-declare class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage> implements AssistantRuntime, ReactThreadRuntime {
-    adapter: VercelRSCAdapter<T>;
-    private useAdapter;
-    private _subscriptions;
-    isRunning: boolean;
-    messages: ThreadMessage[];
-    constructor(adapter: VercelRSCAdapter<T>);
-    private withRunning;
-    getBranches(): readonly string[];
-    switchToBranch(): void;
-    append(message: AppendMessage): Promise<void>;
-    startRun(parentId: string | null): Promise<void>;
-    cancelRun(): void;
-    subscribe(callback: () => void): () => void;
-    onAdapterUpdated(): void;
-    private updateData;
-    unstable_synchronizer: () => null;
-}
-
-declare const useVercelRSCRuntime: <T extends WeakKey>(adapter: VercelRSCAdapter<T>) => VercelRSCRuntime<T>;
-
-declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
-
-type VercelHelpers = UseChatHelpers | UseAssistantHelpers;
-
-declare class VercelAIRuntime implements AssistantRuntime, ReactThreadRuntime {
-    vercel: VercelHelpers;
-    private _subscriptions;
-    private repository;
-    private assistantOptimisticId;
-    private useVercel;
-    messages: ThreadMessage[];
-    isRunning: boolean;
-    constructor(vercel: VercelHelpers);
-    getBranches(messageId: string): string[];
-    switchToBranch(branchId: string): void;
-    append(message: AppendMessage): Promise<void>;
-    startRun(parentId: string | null): Promise<void>;
-    cancelRun(): void;
-    subscribe(callback: () => void): () => void;
-    private updateVercelMessages;
-    onVercelUpdated(): void;
-    private updateData;
-    unstable_synchronizer: () => null;
-}
-
-declare const useVercelUseChatRuntime: (chatHelpers: UseChatHelpers) => VercelAIRuntime;
-
-declare const useVercelUseAssistantRuntime: (assistantHelpers: UseAssistantHelpers) => VercelAIRuntime;
-
-declare const getVercelAIMessage: (message: ThreadMessage) => Message[] | undefined;
-
-type ContentPartState = Readonly<{
-    status: "in_progress" | "done" | "error";
-    part: ThreadMessage["content"][number];
-}>;
-
-type MessageState = Readonly<{
-    message: Readonly<ThreadMessage>;
-    parentId: string | null;
-    branches: readonly string[];
-    isLast: boolean;
-    inProgressIndicator: ReactNode | null;
-    setInProgressIndicator: (value: ReactNode | null) => void;
-    isCopied: boolean;
-    setIsCopied: (value: boolean) => void;
-    isHovering: boolean;
-    setIsHovering: (value: boolean) => void;
-}>;
-
-type BaseComposerState = Readonly<{
-    value: string;
-    setValue: (value: string) => void;
-}>;
-
-type EditComposerState = BaseComposerState & Readonly<{
-    isEditing: boolean;
-    edit: () => void;
-    send: () => void;
-    cancel: () => boolean;
-}>;
-
-type ThreadState = {
-    messages: readonly ThreadMessage[];
-    isRunning: boolean;
-    getBranches: (messageId: string) => readonly string[];
-    switchToBranch: (branchId: string) => void;
-    append: (message: AppendMessage) => void;
-    startRun: (parentId: string | null) => void;
-    cancelRun: () => void;
-};
-
-type ComposerState = BaseComposerState & Readonly<{
-    isEditing: true;
-    send: () => void;
-    cancel: () => boolean;
-}>;
-
-type ThreadViewportState = {
-    isAtBottom: boolean;
-    scrollToBottom: () => void;
-    onScrollToBottom: (callback: () => void) => () => void;
-};
-
-type ThreadContextValue = {
-    useThread: UseBoundStore<StoreApi<ThreadState>>;
-    useComposer: UseBoundStore<StoreApi<ComposerState>>;
-    useViewport: UseBoundStore<StoreApi<ThreadViewportState>>;
-};
-declare const useThreadContext: () => ThreadContextValue;
-
-type ComposerContextValue = {
-    useComposer: UseBoundStore<StoreApi<EditComposerState | ComposerState>>;
-    type: "edit" | "new";
-};
-declare const useComposerContext: () => ComposerContextValue;
-
-type MessageContextValue = {
-    useMessage: UseBoundStore<StoreApi<MessageState>>;
-    useComposer: UseBoundStore<StoreApi<EditComposerState>>;
-};
-declare const useMessageContext: () => MessageContextValue;
-
-type ContentPartContextValue = {
-    useContentPart: UseBoundStore<StoreApi<ContentPartState>>;
-};
-declare const useContentPartContext: () => ContentPartContextValue;
-
-type Unsubscribe = () => void;
-type ThreadRuntime = Readonly<ThreadState & {
-    subscribe: (callback: () => void) => Unsubscribe;
-}>;
-
-type AssistantRuntime = ThreadRuntime;
-
 type ReactThreadRuntime = ThreadRuntime & {
     unstable_synchronizer?: ComponentType;
 };
 
-declare class MessageRepository {
-    private messages;
-    private head;
-    private root;
-    private performOp;
-    getMessages(): ThreadMessage[];
-    addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
-    appendOptimisticMessage(parentId: string | null, message: Omit<ThreadMessage, "id" | "createdAt">): string;
-    deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
-    getBranches(messageId: string): string[];
-    switchToBranch(messageId: string): void;
-    resetHead(messageId: string | null): void;
-}
-
-/**
- * @deprecated Will be removed in 0.1.0.
- */
-type VercelAIAssistantProviderProps = PropsWithChildren<{
-    chat: UseChatHelpers;
-} | {
-    assistant: UseAssistantHelpers;
-}>;
-/**
- * @deprecated `const runtime = useVercelUseChatRuntime(chat)` and `<AssistantRuntimeProvider runtime={...} />`. Will be removed in 0.1.0.
- */
-declare const VercelAIAssistantProvider: FC<VercelAIAssistantProviderProps>;
-
-/**
- * @deprecated Will be removed in 0.1.0.
- */
-type VercelRSCAssistantProviderProps<T> = PropsWithChildren<VercelRSCAdapter<T>>;
-/**
- * @deprecated Replaced with `const runtime = useVercelRSCRuntime({ messages, append })` and `<AssistantRuntimeProvider runtime={runtime} />`. Will be removed in 0.1.0.
- */
-declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, ...adapter }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
-
-type ChatModelRunResult = {
-    content: AssistantContentPart[];
-};
-type ChatModelRunOptions = {
-    messages: ThreadMessage[];
-    abortSignal: AbortSignal;
-    onUpdate: (result: ChatModelRunResult) => void;
-};
-type ChatModelAdapter = {
-    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
-};
-
-declare class LocalRuntime implements AssistantRuntime {
-    adapter: ChatModelAdapter;
-    private _subscriptions;
-    private abortController;
-    private repository;
-    get messages(): ThreadMessage[];
-    get isRunning(): boolean;
-    constructor(adapter: ChatModelAdapter);
-    getBranches(messageId: string): string[];
-    switchToBranch(branchId: string): void;
-    append(message: AppendMessage): Promise<void>;
-    startRun(parentId: string | null): Promise<void>;
-    cancelRun(): void;
-    private notifySubscribers;
-    subscribe(callback: () => void): Unsubscribe;
-}
-
-declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
-
 type AssistantRuntimeProviderProps = {
     runtime: AssistantRuntime;
 };
 declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
 
-export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, type AssistantContentPart, type AssistantMessage, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type ReactThreadRuntime, type TextContentPart, type ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, type UserContentPart, type UserMessage, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, type VercelRSCAdapter, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, type VercelRSCMessage, getVercelAIMessage, getVercelRSCMessage, type ChatModelAdapter as unstable_ChatModelAdapter, type ChatModelRunOptions as unstable_ChatModelRunOptions, type ComposerContextValue as unstable_ComposerContextValue, type ComposerState as unstable_ComposerState, type ContentPartContextValue as unstable_ContentPartContextValue, type ContentPartState as unstable_ContentPartState, type EditComposerState as unstable_EditComposerState, type ImageContentPart as unstable_ImageContentPart, type MessageContextValue as unstable_MessageContextValue, MessageRepository as unstable_MessageRepository, type MessageState as unstable_MessageState, type ThreadContextValue as unstable_ThreadContextValue, type ThreadState as unstable_ThreadState, type ThreadViewportState as unstable_ThreadViewportState, type ToolCallContentPart as unstable_ToolCallContentPart, type UIContentPart as unstable_UIContentPart, useComposerContext as unstable_useComposerContext, useContentPartContext as unstable_useContentPartContext, useLocalRuntime as unstable_useLocalRuntime, useMessageContext as unstable_useMessageContext, useThreadContext as unstable_useThreadContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage, useVercelRSCRuntime, useVercelUseAssistantRuntime, useVercelUseChatRuntime };
+export { index$1 as ActionBarPrimitive, AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, index$3 as MessagePrimitive, type ReactThreadRuntime, TextContentPart, index$5 as ThreadPrimitive, ThreadRuntime, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };