@athenaintel/react 0.9.8 → 0.9.10

package/dist/index.d.ts

@@ -3,14 +3,18 @@ import { ClassProp } from 'class-variance-authority/types';
 import { ClassValue } from 'clsx';
 import { Collapsible as Collapsible_2 } from 'radix-ui';
 import { ComponentPropsWithRef } from 'react';
+import { EmptyMessagePartComponent } from '@assistant-ui/react';
 import { FC } from 'react';
 import { ForwardRefExoticComponent } from 'react';
 import { JSX } from 'react/jsx-runtime';
 import * as React_2 from 'react';
 import { ReactNode } from 'react';
+import { ReasoningMessagePartComponent } from '@assistant-ui/react';
+import { ReasoningMessagePartProps } from '@assistant-ui/react';
 import { RefAttributes } from 'react';
 import { Store } from '@tanstack/react-store';
 import { StoreApi } from 'zustand';
+import { TextMessagePartComponent } from '@assistant-ui/react';
 import { ToolCallMessagePartComponent } from '@assistant-ui/react';
 import { ToolCallMessagePartStatus } from '@assistant-ui/react';
 import { Toolkit } from '@assistant-ui/react';
@@ -57,8 +61,37 @@ export declare interface AssetTab {
 
 export declare type AssetType = 'presentation' | 'spreadsheet' | 'document' | 'notebook' | 'unknown';
 
+export declare const AthenaAssistantActionBar: FC<AthenaAssistantActionBarProps>;
+
+export declare interface AthenaAssistantActionBarProps {
+    className?: string;
+}
+
+export declare const AthenaAssistantMessage: FC<AthenaAssistantMessageProps>;
+
+export declare interface AthenaAssistantMessageProps {
+    toolUIs: Record<string, ToolCallMessagePartComponent>;
+    TextComponent?: TextMessagePartComponent;
+    ReasoningComponent?: ReasoningMessagePartComponent;
+    EmptyComponent?: EmptyMessagePartComponent;
+    ActionBarComponent?: FC<AthenaAssistantActionBarProps>;
+}
+
 export declare const AthenaChat: FC<AthenaChatProps>;
 
+export declare interface AthenaChatComponents {
+    UserMessage?: FC;
+    AssistantMessage?: FC;
+    AssistantActionBar?: FC<AthenaAssistantActionBarProps>;
+    messageParts?: AthenaChatMessagePartComponents;
+}
+
+export declare interface AthenaChatMessagePartComponents {
+    Text?: TextMessagePartComponent;
+    Reasoning?: ReasoningMessagePartComponent;
+    Empty?: EmptyMessagePartComponent;
+}
+
 export declare interface AthenaChatProps {
     /** CSS class name for the root element. */
     className?: string;
@@ -76,6 +109,8 @@ export declare interface AthenaChatProps {
      *  For workflow launchers outside AthenaChat, use `useSendMessage()`.
      */
     welcomeSuggestions?: WelcomeSuggestion[];
+    /** Optional component overrides for message containers and key message parts. */
+    components?: AthenaChatComponents;
 }
 
 export declare interface AthenaConfig extends Pick<AthenaProviderConfig, 'apiKey' | 'token'> {
@@ -167,6 +202,12 @@ export declare interface AthenaProviderProps {
     theme?: AthenaTheme;
 }
 
+export declare const AthenaReasoningPart: FC<AthenaReasoningPartProps>;
+
+export declare interface AthenaReasoningPartProps extends ReasoningMessagePartProps {
+    TextComponent?: TextMessagePartComponent;
+}
+
 export declare interface AthenaRuntimeConfig {
     /** URL for the chat streaming endpoint. Defaults to Athena production sync server. */
     apiUrl?: string;
@@ -276,6 +317,12 @@ export declare interface AthenaTheme {
     threadMaxWidth?: string;
 }
 
+export declare const AthenaUserMessage: FC<AthenaUserMessageProps>;
+
+export declare interface AthenaUserMessageProps {
+    TextComponent?: TextMessagePartComponent;
+}
+
 export declare const BrowseToolUI: ToolCallMessagePartComponent;
 
 export declare function Button({ className, variant, size, asChild, ...props }: React_2.ComponentProps<"button"> & VariantProps<typeof buttonVariants> & {