@aws-amplify/ui-react-ai 0.3.2 → 1.0.0

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

@@ -1,22 +1,5 @@
-import * as React from 'react';
-import { AIConversationInput, AIConversationProps } from './types';
+import { AIConversation as AIConversationType, AIConversationInput, AIConversationProps } from './types';
 interface AIConversationBaseProps extends AIConversationProps, AIConversationInput {
 }
-declare function AIConversationBase({ actions, avatars, controls, handleSendMessage, messages, responseComponents, suggestedPrompts, variant, isLoading, displayText, allowAttachments, }: AIConversationBaseProps): JSX.Element;
-/**
- * @experimental
- */
-export declare const AIConversation: typeof AIConversationBase & {
-    MessageList: React.ComponentType<{
-        messages: import("@aws-amplify/data-schema/dist/esm/ai/ConversationType").ConversationMessage[];
-    }> | undefined;
-    PromptList: React.ComponentType<{
-        suggestedPrompts?: import("./types").SuggestedPrompt[] | undefined;
-        setInput: React.Dispatch<React.SetStateAction<import("./context").ConversationInput | undefined>> | undefined;
-    }> | undefined;
-    Form: NonNullable<React.ComponentType<{
-        handleSubmit: (e: React.FormEvent<HTMLFormElement>) => void;
-        allowAttachments?: boolean | undefined;
-    } & Required<import("./context").ConversationInputContext>> | undefined>;
-};
+export declare const AIConversation: AIConversationType<AIConversationBaseProps>;
 export {};

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

@@ -1,7 +1,6 @@
 import React from 'react';
 import { AIConversationInput, AIConversationProps } from './types';
-interface AIConversationProviderProps extends AIConversationInput, AIConversationProps {
+export interface AIConversationProviderProps extends AIConversationInput, AIConversationProps {
     children?: React.ReactNode;
 }
-export declare const AIConversationProvider: ({ elements, actions, suggestedPrompts, responseComponents, variant, controls, displayText, allowAttachments, messages, handleSendMessage, avatars, isLoading, children, }: AIConversationProviderProps) => React.JSX.Element;
-export {};
+export declare const AIConversationProvider: ({ aiContext, actions, allowAttachments, avatars, children, controls, displayText, handleSendMessage, isLoading, 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/ControlsContext.d.ts

@@ -6,6 +6,7 @@ export interface ControlsContextProps {
     Form?: React.ComponentType<{
         handleSubmit: (e: React.FormEvent<HTMLFormElement>) => void;
         allowAttachments?: boolean;
+        isLoading?: boolean;
     } & Required<ConversationInputContext>>;
     MessageList?: React.ComponentType<{
         messages: ConversationMessage[];

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

@@ -0,0 +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?: {
+    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/WelcomeMessageContext.d.ts

@@ -0,0 +1,8 @@
+import * as React from 'react';
+type WelcomeMessageContextProps = React.ReactNode | undefined;
+export declare const WelcomeMessageContext: React.Context<WelcomeMessageContextProps>;
+export declare const WelcomeMessageProvider: ({ children, welcomeMessage, }: {
+    children?: React.ReactNode;
+    welcomeMessage?: React.ReactNode;
+}) => JSX.Element;
+export {};

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

@@ -29,7 +29,7 @@ export declare const ButtonElement: import("@aws-amplify/ui-react-core/dist/type
 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';
+type TextAreaElementProps = 'id' | 'name' | 'onChange' | 'placeholder' | 'autoFocus' | 'onCompositionStart' | 'onCompositionEnd' | '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 AIConversationElements: AIConversationElements;
 export {};

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

@@ -1,3 +1,4 @@
+export { AIContextContext, AIContextProvider } from './AIContextContext';
 export { ActionsContext, ActionsProvider } from './ActionsContext';
 export { AvatarsContext, AvatarsProvider } from './AvatarsContext';
 export { ConversationInputContext, ConversationInput, ConversationInputContextProvider, } from './ConversationInputContext';
@@ -9,4 +10,8 @@ export { ControlsContext, ControlsContextProps, ControlsProvider, } from './Cont
 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 { 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/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,43 +1,52 @@
 import React from 'react';
-import { AIConversationElements } from './context/elements';
-import { ActionsBarControl, AvatarControl, FieldControl, HeaderControl, MessagesControl, PromptControl } from './views';
+import { ActionsBarControl, AvatarControl, FormControl, MessagesControl, PromptControl } from './views';
 import { DisplayTextTemplate } from '@aws-amplify/ui';
 import { AIConversationDisplayText } from './displayText';
-import { ConversationMessage, SendMessage } from '../../types';
+import { ConversationMessage, SendMessage, ResponseComponents, TextContentBlock, ImageContentBlock } from '../../types';
 import { ControlsContextProps } from './context/ControlsContext';
+import { AIConversationProviderProps } from './AIConversationProvider';
 export interface Controls {
     Avatars: AvatarControl;
     ActionsBar: ActionsBarControl;
-    Field: FieldControl;
-    Header: HeaderControl;
+    Form: FormControl;
     Messages: MessagesControl;
     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;
+    messageRenderer?: MessageRenderer;
 }
 export interface AIConversationProps {
     messages: ConversationMessage[];
     handleSendMessage: SendMessage;
     avatars?: Avatars;
     isLoading?: boolean;
+    aiContext?: () => object;
 }
-export interface AIConversation {
-    (props: AIConversationProps): JSX.Element;
-    Conversation: () => React.JSX.Element;
-    Controls: Controls;
-    Provider: (props: {
-        children?: React.ReactNode;
-    } & Pick<AIConversationProps, 'messages' | 'avatars' | 'handleSendMessage'>) => React.JSX.Element;
+export interface AIConversation<PropsType extends AIConversationProps = AIConversationProps> {
+    (props: PropsType): JSX.Element;
+    DefaultMessage: () => JSX.Element | undefined;
+    Messages: () => JSX.Element;
+    Form: () => JSX.Element;
+    Provider: (props: AIConversationProviderProps) => React.JSX.Element;
 }
 export type MessageVariant = 'bubble' | 'default';
+export interface MessageRenderer {
+    text?: (input: {
+        text: TextContentBlock;
+    }) => React.JSX.Element;
+    image?: (input: {
+        image: ImageContentBlock;
+    }) => React.JSX.Element;
+}
 export interface Avatar {
     username?: string;
     avatar?: React.ReactNode;
@@ -47,31 +56,10 @@ export interface Avatars {
     ai?: Avatar;
 }
 export interface CustomAction {
-    displayName: string;
     handler: (message: ConversationMessage) => void;
-    icon: React.ReactNode;
+    component?: React.ReactNode;
 }
 export interface SuggestedPrompt {
-    icon?: React.ReactNode;
-    header: string;
+    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

@@ -1,4 +1,4 @@
-import { ImageContent } from '../../types';
+import { ImageContentBlock } from '../../types';
 export declare function formatDate(date: Date): string;
-export declare function convertBufferToBase64(buffer: ArrayBuffer, format: ImageContent['format']): string;
+export declare function convertBufferToBase64(buffer: ArrayBuffer, format: ImageContentBlock['format']): string;
 export declare function getImageTypeFromMimeType(mimeType: string): 'png' | 'jpeg' | 'gif' | 'webp';

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

@@ -0,0 +1,2 @@
+/// <reference types="react" />
+export declare const DefaultMessageControl: () => JSX.Element | undefined;

package/dist/types/components/AIConversation/views/Controls/{FieldControl.d.ts → FormControl.d.ts}

@@ -1,8 +1,8 @@
 import React from 'react';
 import { AIConversationElements } from '../../context/elements';
 import { AttachFileControl } from './AttachFileControl';
-export declare const FieldControl: FieldControl;
-export interface FieldControl {
+export declare const FormControl: FormControl;
+export interface FormControl {
     (): React.JSX.Element;
     AttachFile: AttachFileControl;
     InputContainer: AIConversationElements['View'];

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

@@ -1,28 +1,22 @@
-import React from 'react';
+/// <reference types="react" />
 import { AIConversationElements } from '../../context/elements';
 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 {
-    (props: {
-        renderMessage?: (message: ConversationMessage) => React.ReactNode;
-    }): JSX.Element;
+    (): JSX.Element;
     ActionsBar: ActionsBarControl;
     Avatar: AvatarControl;
     Container: AIConversationElements['View'];
     HeaderContainer: AIConversationElements['View'];
     Layout: AIConversationElements['View'];
     Message: MessageControl;
-    Separator: AIConversationElements['Span'];
 }
 export {};

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

@@ -1,12 +1,9 @@
 import React from 'react';
 import { AIConversationElements } from '../../context';
 export declare const PromptControl: PromptControl;
-export declare const AutoHidablePromptControl: () => JSX.Element | undefined;
 export interface PromptControl {
     (): React.JSX.Element;
     Container: AIConversationElements['View'];
-    Header: AIConversationElements['Heading'];
-    Icon: AIConversationElements['Icon'];
     PromptGroup: AIConversationElements['View'];
     PromptCard: AIConversationElements['Button'];
 }

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

@@ -1,7 +1,6 @@
 import { ActionsBarControl } from './ActionsBarControl';
 import { AvatarControl } from './AvatarControl';
-import { HeaderControl } from './HeaderControl';
-import { FieldControl } from './FieldControl';
+import { FormControl } from './FormControl';
 import { MessagesControl } from './MessagesControl';
-import { AutoHidablePromptControl, PromptControl } from './PromptControl';
-export { ActionsBarControl, AvatarControl, HeaderControl, FieldControl, MessagesControl, PromptControl, AutoHidablePromptControl, };
+import { PromptControl } from './PromptControl';
+export { ActionsBarControl, AvatarControl, FormControl, MessagesControl, PromptControl, };

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/components/AIConversation/views/index.d.ts

@@ -1,3 +1,2 @@
-import Conversation from './ConversationView';
-import { ActionsBarControl, AvatarControl, FieldControl, HeaderControl, MessagesControl, PromptControl } from './Controls';
-export { ActionsBarControl, AvatarControl, Conversation, FieldControl, HeaderControl, MessagesControl, PromptControl, };
+import { ActionsBarControl, AvatarControl, FormControl, MessagesControl, PromptControl } from './Controls';
+export { ActionsBarControl, AvatarControl, FormControl, MessagesControl, PromptControl, };

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,14 +1,20 @@
+/// <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 TextContent = NonNullable<ConversationMessageContent['text']>;
-export type ImageContent = NonNullable<ConversationMessageContent['image']>;
-export type InputContent = Parameters<Conversation['sendMessage']>[0]['content'][number];
-export type SendMessageContent = Parameters<Conversation['sendMessage']>[0]['content'];
-export type SendMessageContext = Parameters<Conversation['sendMessage']>[0]['aiContext'];
-export type ToolConfiguration = NonNullable<Parameters<Conversation['sendMessage']>[0]['toolConfiguration']>;
+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'];
+export type ToolConfiguration = NonNullable<Exclude<Parameters<Conversation['sendMessage']>[0], string>['toolConfiguration']>;
 export interface SendMesageParameters {
     content: SendMessageContent;
     aiContext?: SendMessageContext;
@@ -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

@@ -0,0 +1 @@
+export declare const VERSION = "1.0.0";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-amplify/ui-react-ai",
-  "version": "0.3.2",
+  "version": "1.0.0",
   "main": "dist/index.js",
   "module": "dist/esm/index.mjs",
   "exports": {
@@ -43,17 +43,31 @@
   },
   "peerDependencies": {
     "@aws-amplify/api-graphql": "^4.3.0",
-    "aws-amplify": "^6.6.0",
+    "aws-amplify": "^6.6.5",
     "react": "^16.14.0 || ^17.0 || ^18.0",
     "react-dom": "^16.14.0 || ^17.0 || ^18.0"
   },
   "dependencies": {
-    "@aws-amplify/ui": "^6.6.4",
-    "@aws-amplify/ui-react": "^6.5.4",
-    "@aws-amplify/ui-react-core": "^3.0.28"
+    "@aws-amplify/ui": "^6.6.6",
+    "@aws-amplify/ui-react": "^6.6.0",
+    "@aws-amplify/ui-react-core": "^3.0.30"
   },
   "devDependencies": {
     "@types/jest-when": "^3.5.0",
     "jest-when": "^3.5.1"
-  }
+  },
+  "size-limit": [
+    {
+      "name": "AIConversation",
+      "path": "dist/esm/index.mjs",
+      "import": "{ AIConversation }",
+      "limit": "25 kB"
+    },
+    {
+      "name": "createAIConversation",
+      "path": "dist/esm/index.mjs",
+      "import": "{ createAIConversation }",
+      "limit": "7 kB"
+    }
+  ]
 }