@aws-amplify/ui-react-ai 0.4.0 → 1.1.0

package/dist/types/components/AIConversation/AIConversation.d.ts

@@ -1,8 +1,5 @@
 import { AIConversation as AIConversationType, AIConversationInput, AIConversationProps } from './types';
 interface AIConversationBaseProps extends AIConversationProps, AIConversationInput {
 }
-/**
- * @experimental
- */
 export declare const AIConversation: AIConversationType<AIConversationBaseProps>;
 export {};

package/dist/types/components/AIConversation/AIConversationProvider.d.ts

@@ -3,4 +3,4 @@ import { AIConversationInput, AIConversationProps } from './types';
 export interface AIConversationProviderProps extends AIConversationInput, AIConversationProps {
     children?: React.ReactNode;
 }
-export declare const AIConversationProvider: ({ actions, allowAttachments, avatars, children, controls, displayText, elements, handleSendMessage, isLoading, messages, responseComponents, suggestedPrompts, variant, welcomeMessage, }: AIConversationProviderProps) => React.JSX.Element;
+export declare const AIConversationProvider: ({ aiContext, actions, allowAttachments, avatars, children, controls, displayText, handleSendMessage, isLoading, maxAttachmentSize, maxAttachments, messages, messageRenderer, responseComponents, suggestedPrompts, variant, welcomeMessage, FallbackResponseComponent, }: AIConversationProviderProps) => React.JSX.Element;

package/dist/types/components/AIConversation/context/AIContextContext.d.ts

@@ -0,0 +1,6 @@
+import React from 'react';
+export declare const AIContextContext: React.Context<(() => object) | undefined>;
+export declare const AIContextProvider: ({ children, aiContext, }: {
+    children?: React.ReactNode;
+    aiContext?: (() => object) | undefined;
+}) => JSX.Element;

package/dist/types/components/AIConversation/context/AttachmentContext.d.ts

@@ -1,6 +1,6 @@
 import * as React from 'react';
-export declare const AttachmentContext: React.Context<boolean>;
-export declare const AttachmentProvider: ({ children, allowAttachments, }: {
-    children?: React.ReactNode;
-    allowAttachments?: boolean | undefined;
-}) => JSX.Element;
+import { AIConversationInput } from '../types';
+export interface AttachmentContextProps extends Pick<AIConversationInput, 'allowAttachments' | 'maxAttachments' | 'maxAttachmentSize'> {
+}
+export declare const AttachmentContext: React.Context<Required<AttachmentContextProps>>;
+export declare const AttachmentProvider: ({ children, allowAttachments, maxAttachmentSize, maxAttachments, }: React.PropsWithChildren<AttachmentContextProps>) => JSX.Element;

package/dist/types/components/AIConversation/context/ControlsContext.d.ts

@@ -1,18 +1,20 @@
 import React from 'react';
-import { ConversationInputContext } from './ConversationInputContext';
+import { ConversationInputContextProps } from './ConversationInputContext';
 import { SuggestedPrompt } from '../types';
 import { ConversationMessage } from '../../../types';
 export interface ControlsContextProps {
     Form?: React.ComponentType<{
         handleSubmit: (e: React.FormEvent<HTMLFormElement>) => void;
         allowAttachments?: boolean;
-    } & Required<ConversationInputContext>>;
+        isLoading?: boolean;
+        onValidate: (files: File[]) => Promise<void>;
+    } & ConversationInputContextProps>;
     MessageList?: React.ComponentType<{
         messages: ConversationMessage[];
     }>;
     PromptList?: React.ComponentType<{
         suggestedPrompts?: SuggestedPrompt[];
-        setInput: ConversationInputContext['setInput'];
+        setInput: ConversationInputContextProps['setInput'];
     }>;
 }
 export declare const ControlsContext: React.Context<ControlsContextProps | undefined>;

package/dist/types/components/AIConversation/context/ConversationInputContext.d.ts

@@ -3,11 +3,13 @@ export interface ConversationInput {
     text?: string;
     files?: File[];
 }
-export interface ConversationInputContext {
+export interface ConversationInputContextProps {
     input?: ConversationInput;
     setInput?: React.Dispatch<React.SetStateAction<ConversationInput | undefined>>;
+    error?: string;
+    setError?: React.Dispatch<React.SetStateAction<string | undefined>>;
 }
-export declare const ConversationInputContext: React.Context<ConversationInputContext>;
+export declare const ConversationInputContext: React.Context<ConversationInputContextProps>;
 export declare const ConversationInputContextProvider: ({ children, }: {
     children?: React.ReactNode;
 }) => JSX.Element;

package/dist/types/components/AIConversation/context/DisplayTextContext.d.ts

@@ -1,5 +1,5 @@
 /// <reference types="react" />
 import { ConversationDisplayText } from '../displayText';
-export declare const ConversationDisplayTextContext: import("react").Context<Required<ConversationDisplayText> | undefined>, ConversationDisplayTextProvider: import("react").ComponentType<import("react").PropsWithChildren<Required<ConversationDisplayText>>>, useConversationDisplayText: (params?: {
+export declare const ConversationDisplayTextContext: import("react").Context<Required<ConversationDisplayText>>, ConversationDisplayTextProvider: import("react").ComponentType<import("react").PropsWithChildren<Required<ConversationDisplayText>>>, useConversationDisplayText: (params?: {
     errorMessage?: string | undefined;
 } | undefined) => Required<ConversationDisplayText>;

package/dist/types/components/AIConversation/context/FallbackComponentContext.d.ts

@@ -0,0 +1,7 @@
+import React from 'react';
+import { AIConversationInput } from '../types';
+export declare const FallbackComponentContext: React.Context<React.ComponentType<any> | undefined>;
+export declare const FallbackComponentProvider: ({ children, FallbackComponent, }: {
+    children?: React.ReactNode;
+    FallbackComponent?: AIConversationInput['FallbackResponseComponent'];
+}) => JSX.Element;

package/dist/types/components/AIConversation/context/MessageRenderContext.d.ts

@@ -1,5 +1,5 @@
 /// <reference types="react" />
 import { MessageRenderer } from '../types';
-export declare const MessageRendererContext: import("react").Context<MessageRenderer | undefined>, MessageRendererProvider: import("react").ComponentType<import("react").PropsWithChildren<MessageRenderer>>, useMessageRenderer: (params?: {
+export declare const MessageRendererContext: import("react").Context<MessageRenderer>, MessageRendererProvider: import("react").ComponentType<import("react").PropsWithChildren<MessageRenderer>>, useMessageRenderer: (params?: {
     errorMessage?: string | undefined;
 } | undefined) => MessageRenderer;

package/dist/types/components/AIConversation/context/ResponseComponentsContext.d.ts

@@ -1,9 +1,9 @@
 import React from 'react';
-import { ResponseComponents } from '../types';
-import { ToolConfiguration } from '../../../types';
+import { ToolConfiguration, ResponseComponents } from '../../../types';
 type ResponseComponentsContextProps = ResponseComponents | undefined;
 export declare const RESPONSE_COMPONENT_PREFIX = "AMPLIFY_UI_";
 export declare const ResponseComponentsContext: React.Context<ResponseComponentsContextProps>;
+export declare const prependResponseComponents: (responseComponents?: ResponseComponents) => ResponseComponents | undefined;
 export declare const ResponseComponentsProvider: ({ children, responseComponents, }: {
     children?: React.ReactNode;
     responseComponents?: ResponseComponents | undefined;

package/dist/types/components/AIConversation/context/elements/IconElement.d.ts

@@ -2,5 +2,5 @@ import React from 'react';
 export type IconElementProps = React.ComponentProps<typeof BaseIconElement>;
 export type IconVariant = 'attach' | 'close' | 'image' | 'send-message' | 'user-avatar';
 export declare const DEFAULT_ICON_PATHS: Record<IconVariant, string>;
-export declare const BaseIconElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, IconVariant, React.SVGProps<SVGSVGElement>>, SVGSVGElement>;
-export declare const IconElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<Omit<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, IconVariant, React.SVGProps<SVGSVGElement>>, "ref"> & React.RefAttributes<SVGSVGElement>, SVGSVGElement>;
+export declare const BaseIconElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, IconVariant, React.SVGProps<SVGSVGElement>>, SVGSVGElement>;
+export declare const IconElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<Omit<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, IconVariant, React.SVGProps<SVGSVGElement>>, "ref"> & React.RefAttributes<SVGSVGElement>, SVGSVGElement>;

package/dist/types/components/AIConversation/context/elements/definitions.d.ts

@@ -14,22 +14,22 @@ export interface AIConversationElements {
     UnorderedList: typeof UnorderedListElement;
     View: typeof ViewElement;
 }
-export declare const LabelElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<"htmlFor", string, import("react").DetailedHTMLProps<import("react").LabelHTMLAttributes<HTMLLabelElement>, HTMLLabelElement>>, HTMLLabelElement>;
-export declare const TextElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLParagraphElement>, HTMLParagraphElement>>, HTMLParagraphElement>;
-export declare const UnorderedListElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLUListElement>, HTMLUListElement>>, HTMLUListElement>;
-export declare const ListItemElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, string, import("react").DetailedHTMLProps<import("react").LiHTMLAttributes<HTMLLIElement>, HTMLLIElement>>, HTMLLIElement>;
-export declare const HeadingElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLHeadingElement>, HTMLHeadingElement>>, HTMLHeadingElement>;
+export declare const LabelElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<"htmlFor", string, import("react").DetailedHTMLProps<import("react").LabelHTMLAttributes<HTMLLabelElement>, HTMLLabelElement>>, HTMLLabelElement>;
+export declare const TextElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLParagraphElement>, HTMLParagraphElement>>, HTMLParagraphElement>;
+export declare const UnorderedListElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLUListElement>, HTMLUListElement>>, HTMLUListElement>;
+export declare const ListItemElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, string, import("react").DetailedHTMLProps<import("react").LiHTMLAttributes<HTMLLIElement>, HTMLLIElement>>, HTMLLIElement>;
+export declare const HeadingElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLHeadingElement>, HTMLHeadingElement>>, HTMLHeadingElement>;
 export type IconElementProps = React.ComponentProps<typeof IconElement>;
 type ImageElementProps = 'src' | 'alt';
-export declare const ImageElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<ImageElementProps, string, import("react").DetailedHTMLProps<import("react").ImgHTMLAttributes<HTMLImageElement>, HTMLImageElement>>, HTMLImageElement>;
-export declare const InputElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<"type", string, import("react").DetailedHTMLProps<import("react").InputHTMLAttributes<HTMLInputElement>, HTMLInputElement>>, HTMLInputElement>;
+export declare const ImageElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<ImageElementProps, string, import("react").DetailedHTMLProps<import("react").ImgHTMLAttributes<HTMLImageElement>, HTMLImageElement>>, HTMLImageElement>;
+export declare const InputElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<"type", string, import("react").DetailedHTMLProps<import("react").InputHTMLAttributes<HTMLInputElement>, HTMLInputElement>>, HTMLInputElement>;
 type ButtonElementProps = 'disabled' | 'onClick' | 'type' | 'tabIndex';
 type ButtonElementVariant = 'attach' | 'remove' | 'send-message';
-export declare const ButtonElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<ButtonElementProps, ButtonElementVariant, import("react").DetailedHTMLProps<import("react").ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>>, HTMLButtonElement>;
+export declare const ButtonElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<ButtonElementProps, ButtonElementVariant, import("react").DetailedHTMLProps<import("react").ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>>, HTMLButtonElement>;
 type ViewElementProps = 'onFocus' | 'tabIndex' | 'onKeyDown';
-export declare const ViewElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<ViewElementProps, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLDivElement>, HTMLDivElement>>, HTMLDivElement>;
-export declare const SpanElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLSpanElement>, HTMLSpanElement>>, HTMLSpanElement>;
-type TextAreaElementProps = 'id' | 'name' | 'onChange' | 'placeholder' | 'autoFocus' | 'onKeyDown';
-export declare const TextAreaElement: import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElement<import("@aws-amplify/ui-react-core/dist/types/elements/types").BaseElementProps<TextAreaElementProps, string, import("react").DetailedHTMLProps<import("react").TextareaHTMLAttributes<HTMLTextAreaElement>, HTMLTextAreaElement>>, HTMLTextAreaElement>;
+export declare const ViewElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<ViewElementProps, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLDivElement>, HTMLDivElement>>, HTMLDivElement>;
+export declare const SpanElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<never, string, import("react").DetailedHTMLProps<import("react").HTMLAttributes<HTMLSpanElement>, HTMLSpanElement>>, HTMLSpanElement>;
+type TextAreaElementProps = 'id' | 'name' | 'onChange' | 'placeholder' | 'autoFocus' | 'onCompositionStart' | 'onCompositionEnd' | 'onKeyDown';
+export declare const TextAreaElement: import("@aws-amplify/ui-react-core/elements").BaseElementWithRef<import("@aws-amplify/ui-react-core/elements").BaseElementWithRefProps<TextAreaElementProps, string, import("react").DetailedHTMLProps<import("react").TextareaHTMLAttributes<HTMLTextAreaElement>, HTMLTextAreaElement>>, HTMLTextAreaElement>;
 export declare const AIConversationElements: AIConversationElements;
 export {};

package/dist/types/components/AIConversation/context/index.d.ts

@@ -1,6 +1,7 @@
+export { AIContextContext, AIContextProvider } from './AIContextContext';
 export { ActionsContext, ActionsProvider } from './ActionsContext';
 export { AvatarsContext, AvatarsProvider } from './AvatarsContext';
-export { ConversationInputContext, ConversationInput, ConversationInputContextProvider, } from './ConversationInputContext';
+export { ConversationInputContextProps, ConversationInputContext, ConversationInput, ConversationInputContextProvider, } from './ConversationInputContext';
 export { MessagesContext, RoleContext, MessagesProvider, } from './MessagesContext';
 export { SuggestedPromptsContext, SuggestedPromptProvider, } from './SuggestedPromptsContext';
 export { MessageVariantContext, MessageVariantProvider, } from './MessageVariantContext';
@@ -10,6 +11,7 @@ export { LoadingContextProvider } from './LoadingContext';
 export { ResponseComponentsProvider, RESPONSE_COMPONENT_PREFIX, } from './ResponseComponentsContext';
 export { SendMessageContextProvider } from './SendMessageContext';
 export { MessageRendererProvider, MessageRendererContext, useMessageRenderer, } from './MessageRenderContext';
-export { AttachmentProvider, AttachmentContext } from './AttachmentContext';
+export { AttachmentProvider, AttachmentContext, AttachmentContextProps, } from './AttachmentContext';
 export { WelcomeMessageContext, WelcomeMessageProvider, } from './WelcomeMessageContext';
+export { FallbackComponentContext, FallbackComponentProvider, } from './FallbackComponentContext';
 export * from './elements';

package/dist/types/components/AIConversation/createAIConversation.d.ts

@@ -1,7 +1,4 @@
 import { AIConversationInput, AIConversation } from './types';
-/**
- * @experimental
- */
 export declare function createAIConversation(input?: AIConversationInput): {
     AIConversation: AIConversation;
 };

package/dist/types/components/AIConversation/displayText.d.ts

@@ -1,6 +1,8 @@
 import { DisplayTextTemplate } from '@aws-amplify/ui';
 export type ConversationDisplayText = {
     getMessageTimestampText?: (date: Date) => string;
+    getMaxAttachmentErrorText?: (count: number) => string;
+    getAttachmentSizeErrorText?: (sizeText: string) => string;
 };
 export declare const defaultAIConversationDisplayTextEn: Required<AIConversationDisplayText>;
 export type AIConversationDisplayText = DisplayTextTemplate<ConversationDisplayText>;

package/dist/types/components/AIConversation/index.d.ts

@@ -1,4 +1,5 @@
 import { createAIConversation } from './createAIConversation';
 import { AIConversation } from './AIConversation';
-import { Avatars, CustomAction, ResponseComponent, SuggestedPrompt } from './types';
+import { Avatars, CustomAction, SuggestedPrompt } from './types';
+import { ResponseComponent } from '../../types';
 export { createAIConversation, AIConversation, Avatars, CustomAction, ResponseComponent, SuggestedPrompt, };

package/dist/types/components/AIConversation/types.d.ts

@@ -1,9 +1,8 @@
 import React from 'react';
-import { AIConversationElements } from './context/elements';
 import { ActionsBarControl, AvatarControl, FormControl, MessagesControl, PromptControl } from './views';
 import { DisplayTextTemplate } from '@aws-amplify/ui';
 import { AIConversationDisplayText } from './displayText';
-import { ConversationMessage, ImageContentBlock, SendMessage, TextContentBlock } from '../../types';
+import { ConversationMessage, SendMessage, ResponseComponents, TextContentBlock, ImageContentBlock } from '../../types';
 import { ControlsContextProps } from './context/ControlsContext';
 import { AIConversationProviderProps } from './AIConversationProvider';
 export interface Controls {
@@ -14,15 +13,17 @@ export interface Controls {
     SuggestedPrompts: PromptControl;
 }
 export interface AIConversationInput {
-    elements?: Partial<AIConversationElements>;
     displayText?: DisplayTextTemplate<AIConversationDisplayText>;
     welcomeMessage?: React.ReactNode;
     suggestedPrompts?: SuggestedPrompt[];
     actions?: CustomAction[];
     responseComponents?: ResponseComponents;
+    FallbackResponseComponent?: React.ComponentType<any>;
     variant?: MessageVariant;
     controls?: ControlsContextProps;
     allowAttachments?: boolean;
+    maxAttachments?: number;
+    maxAttachmentSize?: number;
     messageRenderer?: MessageRenderer;
 }
 export interface AIConversationProps {
@@ -30,6 +31,7 @@ export interface AIConversationProps {
     handleSendMessage: SendMessage;
     avatars?: Avatars;
     isLoading?: boolean;
+    aiContext?: () => object;
 }
 export interface AIConversation<PropsType extends AIConversationProps = AIConversationProps> {
     (props: PropsType): JSX.Element;
@@ -56,30 +58,10 @@ export interface Avatars {
     ai?: Avatar;
 }
 export interface CustomAction {
-    displayName: string;
     handler: (message: ConversationMessage) => void;
-    icon: React.ReactNode;
+    component?: React.ReactNode;
 }
 export interface SuggestedPrompt {
     component?: React.ReactNode;
     inputText: string;
 }
-type JSONType = 'string' | 'number' | 'integer' | 'boolean' | 'object' | 'array' | 'null' | 'any';
-interface ResponseComponentProp {
-    type: JSONType;
-    enum?: string[];
-    description?: string;
-    required?: boolean;
-}
-interface ResponseComponentPropMap {
-    [key: string]: ResponseComponentProp;
-}
-export interface ResponseComponent {
-    component: React.ComponentType<any>;
-    description?: string;
-    props: ResponseComponentPropMap;
-}
-export interface ResponseComponents {
-    [key: string]: ResponseComponent;
-}
-export {};

package/dist/types/components/AIConversation/utils.d.ts

@@ -2,3 +2,13 @@ import { ImageContentBlock } from '../../types';
 export declare function formatDate(date: Date): string;
 export declare function convertBufferToBase64(buffer: ArrayBuffer, format: ImageContentBlock['format']): string;
 export declare function getImageTypeFromMimeType(mimeType: string): 'png' | 'jpeg' | 'gif' | 'webp';
+export declare function attachmentsValidator({ files, maxAttachments, maxAttachmentSize, }: {
+    files: File[];
+    maxAttachments: number;
+    maxAttachmentSize: number;
+}): Promise<{
+    acceptedFiles: File[];
+    rejectedFiles: File[];
+    hasMaxAttachmentSizeError: boolean;
+    hasMaxAttachmentsError: boolean;
+}>;

package/dist/types/components/AIConversation/views/Controls/MessagesControl.d.ts

@@ -4,13 +4,10 @@ import { ActionsBarControl } from './ActionsBarControl';
 import { AvatarControl } from './AvatarControl';
 import { ConversationMessage } from '../../../../types';
 export declare const MessageControl: MessageControl;
-interface MessageControl<T extends Partial<AIConversationElements> = AIConversationElements> {
+interface MessageControl {
     (props: {
         message: ConversationMessage;
     }): JSX.Element;
-    Container: T['View'];
-    MediaContent: T['Image'];
-    TextContent: T['Text'];
 }
 export declare const MessagesControl: MessagesControl;
 export interface MessagesControl {
@@ -21,6 +18,5 @@ export interface MessagesControl {
     HeaderContainer: AIConversationElements['View'];
     Layout: AIConversationElements['View'];
     Message: MessageControl;
-    Separator: AIConversationElements['Span'];
 }
 export {};

package/dist/types/components/AIConversation/views/default/Attachments.d.ts

@@ -1,6 +1,6 @@
 /// <reference types="react" />
-import { ConversationInputContext } from '../../context';
+import { ConversationInputContextProps } from '../../context';
 export declare const Attachments: ({ files, setInput, }: {
     files?: File[] | undefined;
-    setInput: ConversationInputContext['setInput'];
+    setInput: ConversationInputContextProps['setInput'];
 }) => JSX.Element | null;

package/dist/types/components/AIConversation/views/default/Form.d.ts

@@ -1,2 +1,2 @@
 import { ControlsContextProps } from '../../context/ControlsContext';
-export declare const Form: NonNullable<ControlsContextProps['Form']>;
+export declare const Form: Required<ControlsContextProps>['Form'];

package/dist/types/components/AIConversation/views/default/MessageList.d.ts

@@ -1,2 +1,2 @@
 import { ControlsContextProps } from '../../context/ControlsContext';
-export declare const MessageList: ControlsContextProps['MessageList'];
+export declare const MessageList: Required<ControlsContextProps>['MessageList'];

package/dist/types/components/AIConversation/views/default/PromptList.d.ts

@@ -1,2 +1,2 @@
 import { ControlsContextProps } from '../../context/ControlsContext';
-export declare const PromptList: ControlsContextProps['PromptList'];
+export declare const PromptList: Required<ControlsContextProps>['PromptList'];

package/dist/types/hooks/contentFromEvents.d.ts

@@ -0,0 +1,2 @@
+import { ConversationMessage, ConversationStreamEvent } from '../types';
+export declare const contentFromEvents: (contentBlocks?: ConversationStreamEvent[][]) => ConversationMessage['content'];

package/dist/types/hooks/createAIHooks.d.ts

@@ -4,8 +4,5 @@ import { getSchema } from '../types';
 type UseAIHooks<Client extends Record<'generations' | 'conversations', Record<string, any>>, Schema extends Record<any, any>> = {
     useAIConversation: UseAIConversationHook<Extract<keyof Client['conversations'], string>>;
 } & UseAIGenerationHookWrapper<keyof Client['generations'], Schema>;
-/**
- * @experimental
- */
 export declare function createAIHooks<Client extends Record<'generations' | 'conversations', Record<string, any>>, Schema extends getSchema<Client>>(_client: Client): UseAIHooks<Client, Schema>;
 export {};

package/dist/types/hooks/exhaustivelyListMessages.d.ts

@@ -0,0 +1,8 @@
+import { Conversation, ConversationMessage } from '../types';
+interface ExhaustivelyListMessagesParams {
+    conversation: Conversation;
+    messages?: ConversationMessage[];
+    nextToken?: string;
+}
+export declare function exhaustivelyListMessages({ conversation, messages, nextToken, }: ExhaustivelyListMessagesParams): ReturnType<Conversation['listMessages']>;
+export {};

package/dist/types/hooks/index.d.ts

@@ -1,3 +1,2 @@
-import { AIContextProvider, AIContext } from './AIContextProvider';
 import { createAIHooks } from './createAIHooks';
-export { AIContextProvider, AIContext, createAIHooks };
+export { createAIHooks };

package/dist/types/hooks/shared.d.ts

@@ -0,0 +1,23 @@
+import { DataState } from '@aws-amplify/ui-react-core';
+import { GraphQLFormattedError } from '../types';
+export type DataClientState<T> = Omit<DataState<T>, 'message'> & {
+    messages?: GraphQLFormattedError[];
+};
+export type DataClientResponse<T> = {
+    data: T | null;
+    errors?: GraphQLFormattedError[];
+};
+export declare const INITIAL_STATE: {
+    hasError: boolean;
+    isLoading: boolean;
+    messages: undefined;
+};
+export declare const LOADING_STATE: {
+    hasError: boolean;
+    isLoading: boolean;
+    messages: undefined;
+};
+export declare const ERROR_STATE: {
+    hasError: boolean;
+    isLoading: boolean;
+};

package/dist/types/hooks/useAIConversation.d.ts

@@ -1,12 +1,14 @@
-import { DataState } from '@aws-amplify/ui-react-core';
-import { ConversationMessage, ConversationRoute, SendMessage } from '../types';
+import { Conversation, ConversationMessage, ConversationRoute, SendMessage } from '../types';
+import { DataClientState } from './shared';
 interface UseAIConversationInput {
     id?: string;
-    onResponse?: (message: ConversationMessage) => void;
+    onMessage?: (message: ConversationMessage) => void;
+    onInitialize?: (conversation: Conversation) => void;
 }
 interface AIConversationState {
     messages: ConversationMessage[];
+    conversation?: Conversation;
 }
-export type UseAIConversationHook<T extends string> = (routeName: T, input?: UseAIConversationInput) => [DataState<AIConversationState>, SendMessage];
+export type UseAIConversationHook<T extends string> = (routeName: T, input?: UseAIConversationInput) => [DataClientState<AIConversationState>, SendMessage];
 export declare function createUseAIConversation<T extends Record<'conversations', Record<string, ConversationRoute>>>(client: T): UseAIConversationHook<Extract<keyof T['conversations'], string>>;
 export {};

package/dist/types/hooks/useAIGeneration.d.ts

@@ -1,26 +1,16 @@
-import { DataState } from '@aws-amplify/ui-react-core';
 import { V6Client } from '@aws-amplify/api-graphql';
 import { getSchema } from '../types';
+import { DataClientState } from './shared';
 export interface UseAIGenerationHookWrapper<Key extends keyof AIGenerationClient<Schema>['generations'], Schema extends Record<any, any>> {
     useAIGeneration: <U extends Key>(routeName: U) => [
-        Awaited<GenerationState<Schema[U]['returnType']>>,
+        Awaited<DataClientState<Schema[U]['returnType']>>,
         (input: Schema[U]['args']) => void
     ];
 }
 export type UseAIGenerationHook<Key extends keyof AIGenerationClient<Schema>['generations'], Schema extends Record<any, any>> = (routeName: Key) => [
-    Awaited<GenerationState<Schema[Key]['returnType']>>,
+    Awaited<DataClientState<Schema[Key]['returnType']>>,
     (input: Schema[Key]['args']) => void
 ];
 type AIGenerationClient<T extends Record<any, any>> = Pick<V6Client<T>, 'generations'>;
-interface GraphQLFormattedError {
-    readonly message: string;
-    readonly errorType: string;
-    readonly errorInfo: null | {
-        [key: string]: unknown;
-    };
-}
-type GenerationState<T> = Omit<DataState<T>, 'message'> & {
-    messages?: GraphQLFormattedError[];
-};
 export declare function createUseAIGeneration<Client extends Record<'generations' | 'conversations', Record<string, any>>, Schema extends getSchema<Client>>(client: Client): UseAIGenerationHook<keyof Client['generations'], Client>;
 export {};

package/dist/types/index.d.ts

@@ -1,3 +1,3 @@
 export { createAIConversation, AIConversation, Avatars, CustomAction, ResponseComponent, SuggestedPrompt, } from './components';
-export { createAIHooks, AIContextProvider } from './hooks';
+export { createAIHooks } from './hooks';
 export { ConversationMessage, ConversationMessageContent, SendMessage, SendMesageParameters, } from './types';

package/dist/types/types.d.ts

@@ -1,10 +1,16 @@
+/// <reference types="react" />
 import { V6Client } from '@aws-amplify/api-graphql';
 export type ConversationRoute = V6Client<any>['conversations'][string];
 export type Conversation = NonNullable<Awaited<ReturnType<ConversationRoute['create']>>['data']>;
-export type ConversationMessage = NonNullable<Awaited<ReturnType<Conversation['sendMessage']>>['data']>;
+export type ConversationStreamEvent = Parameters<Parameters<Conversation['onStreamEvent']>[0]['next']>[0];
+export type ConversationMessage = NonNullable<Awaited<ReturnType<Conversation['sendMessage']>>['data']> & {
+    isLoading?: boolean;
+};
 export type ConversationMessageContent = ConversationMessage['content'][number];
 export type TextContentBlock = NonNullable<ConversationMessageContent['text']>;
 export type ImageContentBlock = NonNullable<ConversationMessageContent['image']>;
+export type ToolUseContent = NonNullable<ConversationMessageContent['toolUse']>;
+export type ToolResultContent = NonNullable<ConversationMessageContent['toolResult']>;
 export type InputContent = Exclude<Parameters<Conversation['sendMessage']>[0], string>['content'][number];
 export type SendMessageContent = Exclude<Parameters<Conversation['sendMessage']>[0], string>['content'];
 export type SendMessageContext = Exclude<Parameters<Conversation['sendMessage']>[0], string>['aiContext'];
@@ -17,4 +23,29 @@ export interface SendMesageParameters {
 export type SendMessage = (input: SendMesageParameters) => void;
 type AIClient<T extends Record<any, any>> = Pick<V6Client<T>, 'generations' | 'conversations'>;
 export type getSchema<T> = T extends AIClient<infer Schema> ? Schema : never;
+export interface GraphQLFormattedError {
+    readonly message: string;
+    readonly errorType: string;
+    readonly errorInfo?: null | {
+        [key: string]: unknown;
+    };
+}
+type JSONType = 'string' | 'number' | 'integer' | 'boolean' | 'object' | 'array' | 'null' | 'any';
+interface ResponseComponentProp {
+    type: JSONType;
+    enum?: string[];
+    description?: string;
+    required?: boolean;
+}
+interface ResponseComponentPropMap {
+    [key: string]: ResponseComponentProp;
+}
+export interface ResponseComponent {
+    component: React.ComponentType<any>;
+    description?: string;
+    props: ResponseComponentPropMap;
+}
+export interface ResponseComponents {
+    [key: string]: ResponseComponent;
+}
 export {};

package/dist/types/version.d.ts

@@ -1 +1 @@
-export declare const VERSION = "0.4.0";
+export declare const VERSION = "1.1.0";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-amplify/ui-react-ai",
-  "version": "0.4.0",
+  "version": "1.1.0",
   "main": "dist/index.js",
   "module": "dist/esm/index.mjs",
   "exports": {
@@ -42,15 +42,15 @@
     "typecheck": "tsc --noEmit"
   },
   "peerDependencies": {
-    "@aws-amplify/api-graphql": "^4.3.0",
-    "aws-amplify": "^6.6.5",
+    "@aws-amplify/api-graphql": "unstable",
+    "aws-amplify": "^6.9.0",
     "react": "^16.14.0 || ^17.0 || ^18.0",
     "react-dom": "^16.14.0 || ^17.0 || ^18.0"
   },
   "dependencies": {
-    "@aws-amplify/ui": "^6.6.5",
-    "@aws-amplify/ui-react": "^6.5.5",
-    "@aws-amplify/ui-react-core": "^3.0.29"
+    "@aws-amplify/ui": "^6.7.1",
+    "@aws-amplify/ui-react": "^6.7.1",
+    "@aws-amplify/ui-react-core": "^3.1.1"
   },
   "devDependencies": {
     "@types/jest-when": "^3.5.0",