@athenaintel/react 0.10.24 → 0.10.26

package/dist/index.d.ts

@@ -2,6 +2,7 @@ import { AssistantRuntime } from '@assistant-ui/core';
 import { ClassProp } from 'class-variance-authority/types';
 import { ClassValue } from 'clsx';
 import { Collapsible as Collapsible_2 } from 'radix-ui';
+import { ComponentProps } from 'react';
 import { ComponentPropsWithRef } from 'react';
 import { Context } from 'react';
 import { EmptyMessagePartComponent } from '@assistant-ui/react';
@@ -9,6 +10,8 @@ import { FC } from 'react';
 import { ForwardRefExoticComponent } from 'react';
 import { JSX } from 'react/jsx-runtime';
 import { JSXElementConstructor } from 'react';
+import { LangChainMessage } from '@assistant-ui/react-langgraph';
+import { MessagePrimitive } from '@assistant-ui/react';
 import { PersistOptions } from 'zustand/middleware';
 import * as React_2 from 'react';
 import { ReactElement } from 'react';
@@ -19,6 +22,7 @@ import { RefAttributes } from 'react';
 import { Store } from '@tanstack/react-store';
 import { StoreApi } from 'zustand';
 import { TextMessagePartComponent } from '@assistant-ui/react';
+import { ThreadMessage } from '@assistant-ui/react';
 import { ToolCallMessagePartComponent } from '@assistant-ui/react';
 import { ToolCallMessagePartStatus } from '@assistant-ui/react';
 import { Toolkit } from '@assistant-ui/react';
@@ -86,6 +90,13 @@ export declare interface AssetTab {
 
 export declare type AssetType = 'presentation' | 'spreadsheet' | 'document' | 'notebook' | 'unknown';
 
+declare type AthenaAiLangChainMessage = Extract<LangChainMessage, {
+    type: 'ai';
+}> & {
+    usage_metadata?: Record<string, unknown>;
+    response_metadata?: Record<string, unknown>;
+};
+
 export declare const AthenaAssistantActionBar: FC<AthenaAssistantActionBarProps>;
 
 export declare interface AthenaAssistantActionBarProps {
@@ -199,6 +210,10 @@ export declare interface AthenaConfig extends Pick<AthenaProviderConfig, 'apiKey
 
 export declare type AthenaEnvironment = 'production' | 'staging';
 
+declare type AthenaLangChainMessage = Exclude<LangChainMessage, {
+    type: 'ai';
+}> | AthenaAiLangChainMessage;
+
 /**
  * Layout that places the chat on the left and the asset panel on the right.
  * When no assets are open, the chat takes full width. Includes a draggable resize handle.
@@ -572,6 +587,8 @@ declare interface AuthConfig {
     token?: string | null;
 }
 
+export declare const autoCloseInFlightSubgraphMessages: (msgs: readonly unknown[]) => unknown[];
+
 export declare const BrowseToolUI: ToolCallMessagePartComponent;
 
 export declare function Button({ className, variant, size, asChild, ...props }: React_2.ComponentProps<"button"> & VariantProps<typeof buttonVariants> & {
@@ -619,6 +636,8 @@ export declare interface ComposerQuotePreviewProps {
     className?: string;
 }
 
+export declare const convertLangChainToThreadMessages: (messages: AthenaLangChainMessage[], isRunning?: boolean, metadata?: {}) => ThreadMessage[];
+
 export declare function createAssetToolUI(config: {
     icon: React.ElementType;
     assetType: AssetType;
@@ -692,7 +711,12 @@ export declare const GetDatabaseTableSchemaToolUI: ToolCallMessagePartComponent;
 
 export declare function getPostHogInstance<TClient = unknown>(): TClient | null;
 
-/** Fetch thread state (messages) from the LangGraph-compatible endpoint. */
+/**
+ * Fetch persisted thread state from the conversations API.
+ *
+ * This endpoint is the same source used by Athena's conversation sidebar and
+ * supports API-key auth plus BYO SSO session tokens sent as Bearer tokens.
+ */
 export declare function getThreadState(backendUrl: string, auth: AuthConfig, threadId: string): Promise<ThreadStateResponse>;
 
 export declare const isAthenaCitationUrl: ({ href, appUrl, }: {
@@ -748,6 +772,13 @@ declare interface MenuItem {
  */
 declare type MenuScope = string;
 
+export declare const NestedPtcMessages: FC<NestedPtcMessagesProps>;
+
+export declare interface NestedPtcMessagesProps {
+    components: PartsComponents;
+    messageClassName?: string;
+}
+
 export declare function normalizeResult(result: unknown): Record<string, unknown> | null;
 
 export declare const OpenAssetToolUI: ToolCallMessagePartComponent;
@@ -785,6 +816,8 @@ export declare interface ParsedAthenaCitationLink {
     assetType: AssetType | null;
 }
 
+declare type PartsComponents = ComponentProps<typeof MessagePrimitive.Parts>["components"];
+
 export declare interface PostHogConfig {
     /** PostHog API key. When falsy, PostHog is disabled and the provider is a no-op. */
     apiKey?: string;
@@ -794,6 +827,8 @@ export declare interface PostHogConfig {
     debug?: boolean;
 }
 
+export declare const PTC_SUBCALL_DID_NOT_COMPLETE_MESSAGE = "Sub-call did not complete before the parent tool returned.";
+
 declare interface QuoteContextValue {
     quote: QuoteData | null;
     setQuote: (quote: QuoteData | null) => void;
@@ -1016,6 +1051,7 @@ declare interface ThreadStateResponse {
     thread_id: string;
     messages: unknown[];
     status?: string;
+    [key: string]: unknown;
 }
 
 /**