@assistant-ui/react 0.0.15 → 0.0.17

package/dist/index.d.mts

@@ -3,6 +3,7 @@ import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { UseChatHelpers, UseAssistantHelpers } from 'ai/react';
 import * as react_jsx_runtime from 'react/jsx-runtime';
+import { Message } from 'ai';
 import { UseBoundStore, StoreApi } from 'zustand';
 
 declare const ThreadRoot: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
@@ -66,7 +67,7 @@ declare const ThreadSuggestion: react.ForwardRefExoticComponent<Pick<Omit<react.
     autoSend?: boolean;
 } & react.RefAttributes<HTMLButtonElement>>;
 
-declare namespace index$4 {
+declare namespace index$5 {
   export { ThreadEmpty as Empty, ThreadIf as If, ThreadMessages as Messages, ThreadRoot as Root, ThreadScrollToBottom as ScrollToBottom, ThreadSuggestion as Suggestion, ThreadViewport as Viewport };
 }
 
@@ -98,7 +99,7 @@ type ComposerIfFilters = {
 type ComposerIfProps = PropsWithChildren<RequireAtLeastOne<ComposerIfFilters>>;
 declare const ComposerIf: FC<ComposerIfProps>;
 
-declare namespace index$3 {
+declare namespace index$4 {
   export { ComposerCancel as Cancel, ComposerIf as If, ComposerInput as Input, ComposerRoot as Root, ComposerSend as Send };
 }
 
@@ -145,6 +146,7 @@ type UserMessage = BaseMessage & {
 type AssistantMessage = BaseMessage & {
     role: "assistant";
     content: AssistantContentPart[];
+    status: "in_progress" | "done" | "error";
 };
 type AppendMessage = {
     parentId: string | null;
@@ -199,8 +201,14 @@ type MessageContentProps = {
 };
 declare const MessageContent: FC<MessageContentProps>;
 
-declare namespace index$2 {
-  export { MessageContent as Content, MessageIf as If, MessageProvider as Provider, MessageRoot as Root };
+declare const MessageInProgress: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+    ref?: ((instance: HTMLDivElement | null) => void) | react.RefObject<HTMLDivElement> | null | undefined;
+} & {
+    asChild?: boolean;
+}, "key" | keyof react.HTMLAttributes<HTMLDivElement> | "asChild"> & react.RefAttributes<HTMLDivElement>>;
+
+declare namespace index$3 {
+  export { MessageContent as Content, MessageIf as If, MessageInProgress as InProgress, MessageProvider as Provider, MessageRoot as Root };
 }
 
 declare const BranchPickerNext: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -227,7 +235,7 @@ declare const BranchPickerRoot: react.ForwardRefExoticComponent<Pick<Omit<react.
     hideWhenSingleBranch?: boolean;
 } & react.RefAttributes<HTMLDivElement>>;
 
-declare namespace index$1 {
+declare namespace index$2 {
   export { BranchPickerCount as Count, BranchPickerNext as Next, BranchPickerNumber as Number, BranchPickerPrevious as Previous, BranchPickerRoot as Root };
 }
 
@@ -262,10 +270,22 @@ declare const ActionBarEdit: react.ForwardRefExoticComponent<Pick<Omit<react.Det
     asChild?: boolean;
 }, "key" | "asChild" | keyof react.ButtonHTMLAttributes<HTMLButtonElement>> & react.RefAttributes<HTMLButtonElement>>;
 
-declare namespace index {
+declare namespace index$1 {
   export { ActionBarCopy as Copy, ActionBarEdit as Edit, ActionBarReload as Reload, ActionBarRoot as Root };
 }
 
+type ContentPartProviderProps = PropsWithChildren<{
+    part: ThreadMessage["content"][number];
+    status: "in_progress" | "done" | "error";
+}>;
+declare const ContentPartProvider: FC<ContentPartProviderProps>;
+
+declare const ContentPartInProgressIndicator: FC;
+
+declare namespace index {
+  export { ContentPartInProgressIndicator as InProgressIndicator, ContentPartProvider as Provider };
+}
+
 type VercelAIAssistantProviderProps = PropsWithChildren<{
     chat: UseChatHelpers;
 } | {
@@ -292,11 +312,16 @@ type RSCMessageConverter<T> = {
 type VercelRSCAssistantProviderProps<T = VercelRSCMessage> = VercelRSCAssistantProviderBaseProps<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, convertMessage, messages: vercelMessages, append: appendCallback, edit, reload, }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
 
+declare const getVercelMessage: (message: ThreadMessage) => Message | undefined;
+declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
+
 type MessageState = {
     message: ThreadMessage;
     parentId: string | null;
     branches: string[];
     isLast: boolean;
+    inProgressIndicator: ReactNode | null;
+    setInProgressIndicator: (value: ReactNode | null) => void;
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
@@ -321,4 +346,4 @@ declare const useGoToNextBranch: () => (() => void) | null;
 
 declare const useGoToPreviousBranch: () => (() => void) | null;
 
-export { index as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$1 as BranchPickerPrimitive, index$3 as ComposerPrimitive, type ImageContentPart, index$2 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$4 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };
+export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, type ImageContentPart, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$5 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, getVercelMessage as unstable_getVercelMessage, getVercelRSCMessage as unstable_getVercelRSCMessage, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };

package/dist/index.d.ts

@@ -3,6 +3,7 @@ import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { UseChatHelpers, UseAssistantHelpers } from 'ai/react';
 import * as react_jsx_runtime from 'react/jsx-runtime';
+import { Message } from 'ai';
 import { UseBoundStore, StoreApi } from 'zustand';
 
 declare const ThreadRoot: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
@@ -66,7 +67,7 @@ declare const ThreadSuggestion: react.ForwardRefExoticComponent<Pick<Omit<react.
     autoSend?: boolean;
 } & react.RefAttributes<HTMLButtonElement>>;
 
-declare namespace index$4 {
+declare namespace index$5 {
   export { ThreadEmpty as Empty, ThreadIf as If, ThreadMessages as Messages, ThreadRoot as Root, ThreadScrollToBottom as ScrollToBottom, ThreadSuggestion as Suggestion, ThreadViewport as Viewport };
 }
 
@@ -98,7 +99,7 @@ type ComposerIfFilters = {
 type ComposerIfProps = PropsWithChildren<RequireAtLeastOne<ComposerIfFilters>>;
 declare const ComposerIf: FC<ComposerIfProps>;
 
-declare namespace index$3 {
+declare namespace index$4 {
   export { ComposerCancel as Cancel, ComposerIf as If, ComposerInput as Input, ComposerRoot as Root, ComposerSend as Send };
 }
 
@@ -145,6 +146,7 @@ type UserMessage = BaseMessage & {
 type AssistantMessage = BaseMessage & {
     role: "assistant";
     content: AssistantContentPart[];
+    status: "in_progress" | "done" | "error";
 };
 type AppendMessage = {
     parentId: string | null;
@@ -199,8 +201,14 @@ type MessageContentProps = {
 };
 declare const MessageContent: FC<MessageContentProps>;
 
-declare namespace index$2 {
-  export { MessageContent as Content, MessageIf as If, MessageProvider as Provider, MessageRoot as Root };
+declare const MessageInProgress: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+    ref?: ((instance: HTMLDivElement | null) => void) | react.RefObject<HTMLDivElement> | null | undefined;
+} & {
+    asChild?: boolean;
+}, "key" | keyof react.HTMLAttributes<HTMLDivElement> | "asChild"> & react.RefAttributes<HTMLDivElement>>;
+
+declare namespace index$3 {
+  export { MessageContent as Content, MessageIf as If, MessageInProgress as InProgress, MessageProvider as Provider, MessageRoot as Root };
 }
 
 declare const BranchPickerNext: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -227,7 +235,7 @@ declare const BranchPickerRoot: react.ForwardRefExoticComponent<Pick<Omit<react.
     hideWhenSingleBranch?: boolean;
 } & react.RefAttributes<HTMLDivElement>>;
 
-declare namespace index$1 {
+declare namespace index$2 {
   export { BranchPickerCount as Count, BranchPickerNext as Next, BranchPickerNumber as Number, BranchPickerPrevious as Previous, BranchPickerRoot as Root };
 }
 
@@ -262,10 +270,22 @@ declare const ActionBarEdit: react.ForwardRefExoticComponent<Pick<Omit<react.Det
     asChild?: boolean;
 }, "key" | "asChild" | keyof react.ButtonHTMLAttributes<HTMLButtonElement>> & react.RefAttributes<HTMLButtonElement>>;
 
-declare namespace index {
+declare namespace index$1 {
   export { ActionBarCopy as Copy, ActionBarEdit as Edit, ActionBarReload as Reload, ActionBarRoot as Root };
 }
 
+type ContentPartProviderProps = PropsWithChildren<{
+    part: ThreadMessage["content"][number];
+    status: "in_progress" | "done" | "error";
+}>;
+declare const ContentPartProvider: FC<ContentPartProviderProps>;
+
+declare const ContentPartInProgressIndicator: FC;
+
+declare namespace index {
+  export { ContentPartInProgressIndicator as InProgressIndicator, ContentPartProvider as Provider };
+}
+
 type VercelAIAssistantProviderProps = PropsWithChildren<{
     chat: UseChatHelpers;
 } | {
@@ -292,11 +312,16 @@ type RSCMessageConverter<T> = {
 type VercelRSCAssistantProviderProps<T = VercelRSCMessage> = VercelRSCAssistantProviderBaseProps<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, convertMessage, messages: vercelMessages, append: appendCallback, edit, reload, }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
 
+declare const getVercelMessage: (message: ThreadMessage) => Message | undefined;
+declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
+
 type MessageState = {
     message: ThreadMessage;
     parentId: string | null;
     branches: string[];
     isLast: boolean;
+    inProgressIndicator: ReactNode | null;
+    setInProgressIndicator: (value: ReactNode | null) => void;
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
@@ -321,4 +346,4 @@ declare const useGoToNextBranch: () => (() => void) | null;
 
 declare const useGoToPreviousBranch: () => (() => void) | null;
 
-export { index as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$1 as BranchPickerPrimitive, index$3 as ComposerPrimitive, type ImageContentPart, index$2 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$4 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };
+export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, type ImageContentPart, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$5 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, getVercelMessage as unstable_getVercelMessage, getVercelRSCMessage as unstable_getVercelRSCMessage, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };