@axiom-lattice/react-sdk 1.0.11 → 1.0.13

package/dist/index.d.mts

@@ -1,8 +1,11 @@
 import { Message, MessageChunk } from '@axiom-lattice/protocols';
 export * from '@axiom-lattice/protocols';
-import { ReactNode } from 'react';
+import * as React$1 from 'react';
+import React__default, { ReactNode } from 'react';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { Client } from '@axiom-lattice/client-sdk';
+import { Prompts } from '@ant-design/x';
+import { GetProp } from 'antd';
 
 /**
  * Types for the Axiom Lattice React SDK
@@ -86,6 +89,15 @@ interface ChatState {
      */
     streamingMessage: MessageChunk | null;
 }
+/**
+ * Extended chat state with agent state
+ */
+interface ChatStateWithAgent extends ChatState {
+    /**
+     * Current agent state (when enableReturnStateWhenStreamCompleted is true)
+     */
+    agentState: AgentState | null;
+}
 /**
  * Chat hook options
  */
@@ -98,6 +110,19 @@ interface UseChatOptions {
      * Initial messages to populate the chat with
      */
     initialMessages?: Message[];
+    /**
+     * Whether to return and update agent state when streaming is completed
+     */
+    enableReturnStateWhenStreamCompleted?: boolean;
+    /**
+     * Whether to automatically resume streaming after loading messages
+     * When enabled, the hook will continuously receive new messages as they arrive
+     */
+    enableResumeStream?: boolean;
+    /**
+     * Polling interval for resume stream in milliseconds (default: 100ms)
+     */
+    resumeStreamPollInterval?: number;
 }
 /**
  * Thread hook state
@@ -196,12 +221,13 @@ interface StreamEventHandlerOptions {
  * - Receiving streaming and non-streaming responses
  * - Loading message history
  * - Managing the chat state
+ * - Optionally returning agent state when streaming completes
  *
  * @param threadId - Thread ID to use for chat
  * @param options - Chat options
  * @returns Chat state and operations
  */
-declare function useChat(threadId: string | null, options?: UseChatOptions): ChatState & {
+declare function useChat<T extends UseChatOptions>(threadId: string | null, options?: T): (T["enableReturnStateWhenStreamCompleted"] extends true ? ChatStateWithAgent : ChatState) & {
     sendMessage: (data: {
         input?: {
             message: string;
@@ -284,4 +310,114 @@ declare function AxiomLatticeProvider({ config, children, }: AxiomLatticeProvide
  */
 declare function useAxiomLattice(): Client;
 
-export { type AgentState, AxiomLatticeProvider, type AxiomLatticeProviderProps, type ChatResponse, type ChatState, type ClientConfig, type StreamEventHandlerOptions, type Thread, type ThreadState, type UseAgentStateOptions, type UseAgentStateReturn, type UseChatOptions, useAgentGraph, useAgentState, useAxiomLattice, useChat, useThread };
+declare const MDResponse: ({ content, context, embeddedLink, interactive, userData, noGenUI, eventHandler, }: {
+    context?: any;
+    content: string;
+    embeddedLink?: boolean;
+    interactive?: boolean;
+    userData?: any;
+    noGenUI?: boolean;
+    eventHandler?: (action: string, data: any, message: string, agent?: string) => void;
+}) => react_jsx_runtime.JSX.Element;
+declare const MDViewFormItem: ({ value }: {
+    value?: string;
+}) => react_jsx_runtime.JSX.Element;
+
+declare const MDMermaid: ({ children }: any) => react_jsx_runtime.JSX.Element;
+
+type ElementEventHandler = (action: string | "__open_side_app", data: {
+    component_key?: string;
+    data?: any;
+    message?: string;
+    action?: string;
+    size?: "small" | "middle" | "large";
+    config?: {
+        thread_id?: string;
+        work_log_id?: string;
+    };
+}, message: string, agent?: string) => void;
+type ElementProps<T = any> = {
+    component_key: string;
+    data: T;
+    eventHandler: ElementEventHandler;
+    interactive?: boolean;
+    default_open_in_side_app?: boolean;
+};
+
+interface ElementMeta {
+    card_view: React.FC<ElementProps>;
+    side_app_view?: React.FC<ElementProps>;
+    action?: (data: any) => void;
+}
+declare const elements: Record<string, ElementMeta>;
+declare const getElement: (language: string | undefined) => ElementMeta | null;
+declare const regsiterElement: (language: string, ElementMeta: ElementMeta) => ElementMeta;
+
+interface ChatingProps {
+    name?: string;
+    description?: string;
+    tenant_id: string;
+    messages: any[];
+    isLoading: boolean;
+    sendMessage: (params: any) => void;
+    stopStreaming: () => void;
+    default_submit_message?: string;
+    avatar?: string;
+    attachment_placeholder?: React__default.ReactNode;
+    onOpenSidePanel: (data: any) => void;
+    onReminderClick: () => void;
+    styles: any;
+    reminderCount: number;
+    handleMDResponseEvent: (action: string, data?: any, message?: string, agent?: string) => void;
+    senderPromptsItems: GetProp<typeof Prompts, "items">;
+    extra?: React__default.ReactNode;
+    extraMeta?: Array<{
+        id: string;
+    }>;
+}
+declare const Chating: React__default.FC<ChatingProps>;
+
+interface UIMessage {
+    id: string;
+    run_id?: string;
+    content: string;
+    name?: string;
+    files?: AttachFile[];
+    role: "human" | "ai" | "tool";
+    items?: UIMessage[];
+    type: "message" | "action" | "message_chunk" | "start_thinking" | "thinking_chunk" | "end_thinking";
+    parent_id?: number | string;
+    status?: string;
+    think_id?: string;
+    think_type?: string;
+}
+interface AttachFile {
+    name: string;
+    id: string;
+}
+
+declare const ThinkingChain: ({ message }: {
+    message: UIMessage;
+}) => react_jsx_runtime.JSX.Element;
+declare const ThinkingChainGroup: ({ message }: {
+    message: UIMessage;
+}) => react_jsx_runtime.JSX.Element;
+
+interface SideAppViewBrowserProps {
+    open_uri: {
+        component_key: string;
+        data: any;
+        message?: string;
+        size?: "small" | "middle" | "large" | "full";
+    };
+    eventHandler: (action: string, data: any, message: string, agent?: string) => void;
+    onClose: () => void;
+    onChangeSize: (size: "small" | "middle" | "large" | "full") => void;
+}
+declare const SideAppViewBrowser: React__default.FC<SideAppViewBrowserProps>;
+
+declare const chatContext: React$1.Context<{
+    eventHandler: (component_key: string, data: any, message: string) => void;
+}>;
+
+export { type AgentState, type AttachFile, AxiomLatticeProvider, type AxiomLatticeProviderProps, type ChatResponse, type ChatState, type ChatStateWithAgent, Chating, type ClientConfig, type ElementEventHandler, type ElementProps, MDMermaid, MDResponse, MDViewFormItem, SideAppViewBrowser, type StreamEventHandlerOptions, ThinkingChain, ThinkingChainGroup, type Thread, type ThreadState, type UIMessage, type UseAgentStateOptions, type UseAgentStateReturn, type UseChatOptions, chatContext, elements, getElement, regsiterElement, useAgentGraph, useAgentState, useAxiomLattice, useChat, useThread };

package/dist/index.d.ts

@@ -1,8 +1,11 @@
 import { Message, MessageChunk } from '@axiom-lattice/protocols';
 export * from '@axiom-lattice/protocols';
-import { ReactNode } from 'react';
+import * as React$1 from 'react';
+import React__default, { ReactNode } from 'react';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { Client } from '@axiom-lattice/client-sdk';
+import { Prompts } from '@ant-design/x';
+import { GetProp } from 'antd';
 
 /**
  * Types for the Axiom Lattice React SDK
@@ -86,6 +89,15 @@ interface ChatState {
      */
     streamingMessage: MessageChunk | null;
 }
+/**
+ * Extended chat state with agent state
+ */
+interface ChatStateWithAgent extends ChatState {
+    /**
+     * Current agent state (when enableReturnStateWhenStreamCompleted is true)
+     */
+    agentState: AgentState | null;
+}
 /**
  * Chat hook options
  */
@@ -98,6 +110,19 @@ interface UseChatOptions {
      * Initial messages to populate the chat with
      */
     initialMessages?: Message[];
+    /**
+     * Whether to return and update agent state when streaming is completed
+     */
+    enableReturnStateWhenStreamCompleted?: boolean;
+    /**
+     * Whether to automatically resume streaming after loading messages
+     * When enabled, the hook will continuously receive new messages as they arrive
+     */
+    enableResumeStream?: boolean;
+    /**
+     * Polling interval for resume stream in milliseconds (default: 100ms)
+     */
+    resumeStreamPollInterval?: number;
 }
 /**
  * Thread hook state
@@ -196,12 +221,13 @@ interface StreamEventHandlerOptions {
  * - Receiving streaming and non-streaming responses
  * - Loading message history
  * - Managing the chat state
+ * - Optionally returning agent state when streaming completes
  *
  * @param threadId - Thread ID to use for chat
  * @param options - Chat options
  * @returns Chat state and operations
  */
-declare function useChat(threadId: string | null, options?: UseChatOptions): ChatState & {
+declare function useChat<T extends UseChatOptions>(threadId: string | null, options?: T): (T["enableReturnStateWhenStreamCompleted"] extends true ? ChatStateWithAgent : ChatState) & {
     sendMessage: (data: {
         input?: {
             message: string;
@@ -284,4 +310,114 @@ declare function AxiomLatticeProvider({ config, children, }: AxiomLatticeProvide
  */
 declare function useAxiomLattice(): Client;
 
-export { type AgentState, AxiomLatticeProvider, type AxiomLatticeProviderProps, type ChatResponse, type ChatState, type ClientConfig, type StreamEventHandlerOptions, type Thread, type ThreadState, type UseAgentStateOptions, type UseAgentStateReturn, type UseChatOptions, useAgentGraph, useAgentState, useAxiomLattice, useChat, useThread };
+declare const MDResponse: ({ content, context, embeddedLink, interactive, userData, noGenUI, eventHandler, }: {
+    context?: any;
+    content: string;
+    embeddedLink?: boolean;
+    interactive?: boolean;
+    userData?: any;
+    noGenUI?: boolean;
+    eventHandler?: (action: string, data: any, message: string, agent?: string) => void;
+}) => react_jsx_runtime.JSX.Element;
+declare const MDViewFormItem: ({ value }: {
+    value?: string;
+}) => react_jsx_runtime.JSX.Element;
+
+declare const MDMermaid: ({ children }: any) => react_jsx_runtime.JSX.Element;
+
+type ElementEventHandler = (action: string | "__open_side_app", data: {
+    component_key?: string;
+    data?: any;
+    message?: string;
+    action?: string;
+    size?: "small" | "middle" | "large";
+    config?: {
+        thread_id?: string;
+        work_log_id?: string;
+    };
+}, message: string, agent?: string) => void;
+type ElementProps<T = any> = {
+    component_key: string;
+    data: T;
+    eventHandler: ElementEventHandler;
+    interactive?: boolean;
+    default_open_in_side_app?: boolean;
+};
+
+interface ElementMeta {
+    card_view: React.FC<ElementProps>;
+    side_app_view?: React.FC<ElementProps>;
+    action?: (data: any) => void;
+}
+declare const elements: Record<string, ElementMeta>;
+declare const getElement: (language: string | undefined) => ElementMeta | null;
+declare const regsiterElement: (language: string, ElementMeta: ElementMeta) => ElementMeta;
+
+interface ChatingProps {
+    name?: string;
+    description?: string;
+    tenant_id: string;
+    messages: any[];
+    isLoading: boolean;
+    sendMessage: (params: any) => void;
+    stopStreaming: () => void;
+    default_submit_message?: string;
+    avatar?: string;
+    attachment_placeholder?: React__default.ReactNode;
+    onOpenSidePanel: (data: any) => void;
+    onReminderClick: () => void;
+    styles: any;
+    reminderCount: number;
+    handleMDResponseEvent: (action: string, data?: any, message?: string, agent?: string) => void;
+    senderPromptsItems: GetProp<typeof Prompts, "items">;
+    extra?: React__default.ReactNode;
+    extraMeta?: Array<{
+        id: string;
+    }>;
+}
+declare const Chating: React__default.FC<ChatingProps>;
+
+interface UIMessage {
+    id: string;
+    run_id?: string;
+    content: string;
+    name?: string;
+    files?: AttachFile[];
+    role: "human" | "ai" | "tool";
+    items?: UIMessage[];
+    type: "message" | "action" | "message_chunk" | "start_thinking" | "thinking_chunk" | "end_thinking";
+    parent_id?: number | string;
+    status?: string;
+    think_id?: string;
+    think_type?: string;
+}
+interface AttachFile {
+    name: string;
+    id: string;
+}
+
+declare const ThinkingChain: ({ message }: {
+    message: UIMessage;
+}) => react_jsx_runtime.JSX.Element;
+declare const ThinkingChainGroup: ({ message }: {
+    message: UIMessage;
+}) => react_jsx_runtime.JSX.Element;
+
+interface SideAppViewBrowserProps {
+    open_uri: {
+        component_key: string;
+        data: any;
+        message?: string;
+        size?: "small" | "middle" | "large" | "full";
+    };
+    eventHandler: (action: string, data: any, message: string, agent?: string) => void;
+    onClose: () => void;
+    onChangeSize: (size: "small" | "middle" | "large" | "full") => void;
+}
+declare const SideAppViewBrowser: React__default.FC<SideAppViewBrowserProps>;
+
+declare const chatContext: React$1.Context<{
+    eventHandler: (component_key: string, data: any, message: string) => void;
+}>;
+
+export { type AgentState, type AttachFile, AxiomLatticeProvider, type AxiomLatticeProviderProps, type ChatResponse, type ChatState, type ChatStateWithAgent, Chating, type ClientConfig, type ElementEventHandler, type ElementProps, MDMermaid, MDResponse, MDViewFormItem, SideAppViewBrowser, type StreamEventHandlerOptions, ThinkingChain, ThinkingChainGroup, type Thread, type ThreadState, type UIMessage, type UseAgentStateOptions, type UseAgentStateReturn, type UseChatOptions, chatContext, elements, getElement, regsiterElement, useAgentGraph, useAgentState, useAxiomLattice, useChat, useThread };