@aomi-labs/react 0.3.5 → 0.3.7

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { WalletTxPayload, WalletEip712Payload, AomiSystemEvent, AomiClient } from '@aomi-labs/client';
-export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, WalletEip712Payload, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
+import { AomiClient, Session, WalletRequest } from '@aomi-labs/client';
+export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, WalletEip712Payload, WalletRequest, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ReactNode, SetStateAction } from 'react';
 import { ThreadMessageLike } from '@assistant-ui/react';
@@ -80,24 +80,37 @@ type InboundEvent = {
     type: string;
     sessionId: string;
     payload?: unknown;
-    status: "pending" | "fetched";
-    timestamp: number;
-};
-type OutboundEvent = {
-    type: string;
-    sessionId: string;
-    payload: unknown;
-    timestamp: number;
 };
 type SSEStatus = "connected" | "connecting" | "disconnected";
 type EventSubscriber = (event: InboundEvent) => void;
-type EventBuffer = {
-    inboundQueue: InboundEvent[];
-    outboundQueue: OutboundEvent[];
+type EventContext = {
+    /** Subscribe to events by type. Returns unsubscribe function. */
+    subscribe: (type: string, callback: EventSubscriber) => () => void;
+    /** Dispatch an event to all matching subscribers (used by orchestrator) */
+    dispatch: (event: InboundEvent) => void;
+    /** Send an outbound system message to backend */
+    sendOutboundSystem: (event: {
+        type: string;
+        sessionId: string;
+        payload: unknown;
+    }) => Promise<void>;
+    /** Current SSE connection status */
     sseStatus: SSEStatus;
-    lastEventId: string | null;
-    subscribers: Map<string, Set<EventSubscriber>>;
 };
+declare function useEventContext(): EventContext;
+type EventContextProviderProps = {
+    children: ReactNode;
+    aomiClient: AomiClient;
+    sessionId: string;
+};
+/**
+ * Simplified EventContext — a pure pub/sub relay.
+ *
+ * SSE subscription and system event unwrapping are now handled by ClientSession
+ * in the orchestrator. This provider just maintains the subscriber registry
+ * and sendOutboundSystem for direct system messages.
+ */
+declare function EventContextProvider({ children, aomiClient, sessionId, }: EventContextProviderProps): react_jsx_runtime.JSX.Element;
 
 type NotificationType = "notice" | "success" | "error" | "wallet";
 type Notification$1 = {
@@ -127,40 +140,26 @@ declare function NotificationContextProvider({ children, }: NotificationContextP
 
 type WalletRequestKind = "transaction" | "eip712_sign";
 type WalletRequestStatus = "pending" | "processing";
-type WalletRequest = {
-    id: string;
-    kind: WalletRequestKind;
-    payload: WalletTxPayload | WalletEip712Payload;
-    status: WalletRequestStatus;
-    timestamp: number;
-};
-type WalletBuffer = {
-    queue: WalletRequest[];
-    nextId: number;
-};
-
 type WalletRequestResult = {
     txHash?: string;
     signature?: string;
     amount?: string;
 };
 type WalletHandlerConfig = {
-    sessionId: string;
-    /** Called after a wallet request is resolved/rejected and the outbound event is sent.
-     *  Used by core.tsx to start polling for the AI's response. */
-    onRequestComplete?: () => void;
+    /** Get the ClientSession for the current thread. */
+    getSession: () => Session | undefined;
 };
 type WalletHandlerApi = {
     /** All queued wallet requests (tx + eip712) */
     pendingRequests: WalletRequest[];
-    /** Mark a request as being processed */
-    startProcessing: (id: string) => void;
-    /** Complete a request successfully — dequeues + sends response to backend */
+    /** Enqueue a wallet request (called by orchestrator on ClientSession events) */
+    enqueueRequest: (request: WalletRequest) => void;
+    /** Complete a request successfully — sends response to backend via ClientSession */
     resolveRequest: (id: string, result: WalletRequestResult) => void;
-    /** Fail a request — dequeues + sends error to backend */
+    /** Fail a request — sends error to backend via ClientSession */
     rejectRequest: (id: string, error?: string) => void;
 };
-declare function useWalletHandler({ sessionId, onRequestComplete, }: WalletHandlerConfig): WalletHandlerApi;
+declare function useWalletHandler({ getSession, }: WalletHandlerConfig): WalletHandlerApi;
 
 type AomiRuntimeApi = {
     /** Current user state (wallet connection, address, chain, etc.) */
@@ -220,7 +219,11 @@ type AomiRuntimeApi = {
     /** Subscribe to inbound events by type. Returns unsubscribe function. */
     subscribe: (type: string, callback: EventSubscriber) => () => void;
     /** Send a system command to the backend */
-    sendSystemCommand: (event: Omit<OutboundEvent, "timestamp">) => Promise<void>;
+    sendSystemCommand: (event: {
+        type: string;
+        sessionId: string;
+        payload: unknown;
+    }) => Promise<void>;
     /** Current SSE connection status */
     sseStatus: SSEStatus;
 };
@@ -254,24 +257,6 @@ type AomiRuntimeApi = {
  */
 declare function useAomiRuntime(): AomiRuntimeApi;
 
-type EventContext = {
-    /** Subscribe to inbound events by type. Returns unsubscribe function. */
-    subscribe: (type: string, callback: EventSubscriber) => () => void;
-    /** Send an outbound event to backend immediately */
-    sendOutboundSystem: (event: Omit<OutboundEvent, "timestamp">) => Promise<void>;
-    /** Dispatch system events from HTTP polling into the event buffer */
-    dispatchInboundSystem: (sessionId: string, events: AomiSystemEvent[]) => void;
-    /** Current SSE connection status */
-    sseStatus: SSEStatus;
-};
-declare function useEventContext(): EventContext;
-type EventContextProviderProps = {
-    children: ReactNode;
-    aomiClient: AomiClient;
-    sessionId: string;
-};
-declare function EventContextProvider({ children, aomiClient, sessionId, }: EventContextProviderProps): react_jsx_runtime.JSX.Element;
-
 type Notification = {
     id: string;
     type: string;
@@ -321,11 +306,17 @@ declare const SUPPORTED_CHAINS: ChainInfo[];
 /** Look up ChainInfo by chain ID. Returns undefined for unknown chains. */
 declare const getChainInfo: (chainId: number | undefined) => ChainInfo | undefined;
 
+/** A stored provider API key (BYOK) */
+type StoredProviderKey = {
+    apiKey: string;
+    keyPrefix: string;
+    label?: string;
+};
 /** Global control state (shared across all threads) */
 type ControlState = {
     /** API key for authenticated requests */
     apiKey: string | null;
-    /** Stable client identifier for this browser tab (associates sessions with secrets) */
+    /** Stable client identifier for this browser profile (associates sessions with secrets) */
     clientId: string | null;
     /** Available models fetched from backend */
     availableModels: string[];
@@ -335,6 +326,8 @@ type ControlState = {
     defaultModel: string | null;
     /** Default app (from authorizedApps) */
     defaultApp: string | null;
+    /** Provider API keys stored locally (BYOK) — keyed by provider name */
+    providerKeys: Record<string, StoredProviderKey>;
 };
 type ControlContextApi = {
     /** Global state (apiKey, clientId, available models/apps) */
@@ -345,6 +338,14 @@ type ControlContextApi = {
     ingestSecrets: (secrets: Record<string, string>) => Promise<Record<string, string>>;
     /** Clear all secrets from the backend vault */
     clearSecrets: () => Promise<void>;
+    /** Store a provider API key (BYOK) in localStorage and ingest into backend vault */
+    setProviderKey: (provider: string, apiKey: string, label?: string) => Promise<void>;
+    /** Remove a provider API key from localStorage and backend vault */
+    removeProviderKey: (provider: string) => Promise<void>;
+    /** Get all stored provider keys (metadata only — keys are in state.providerKeys) */
+    getProviderKeys: () => Record<string, StoredProviderKey>;
+    /** Check if a provider key is stored */
+    hasProviderKey: (provider?: string) => boolean;
     /** Fetch available models from backend */
     getAvailableModels: () => Promise<string[]>;
     /** Fetch authorized apps from backend */
@@ -384,4 +385,4 @@ type ControlContextProviderProps = {
 };
 declare function ControlContextProvider({ children, aomiClient, sessionId, publicKey, getThreadMetadata, updateThreadMetadata, }: ControlContextProviderProps): react_jsx_runtime.JSX.Element;
 
-export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventBuffer, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type OutboundEvent, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type UserState, type WalletBuffer, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequest, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };
+export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type StoredProviderKey, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type UserState, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { WalletTxPayload, WalletEip712Payload, AomiSystemEvent, AomiClient } from '@aomi-labs/client';
-export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, WalletEip712Payload, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
+import { AomiClient, Session, WalletRequest } from '@aomi-labs/client';
+export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, WalletEip712Payload, WalletRequest, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ReactNode, SetStateAction } from 'react';
 import { ThreadMessageLike } from '@assistant-ui/react';
@@ -80,24 +80,37 @@ type InboundEvent = {
     type: string;
     sessionId: string;
     payload?: unknown;
-    status: "pending" | "fetched";
-    timestamp: number;
-};
-type OutboundEvent = {
-    type: string;
-    sessionId: string;
-    payload: unknown;
-    timestamp: number;
 };
 type SSEStatus = "connected" | "connecting" | "disconnected";
 type EventSubscriber = (event: InboundEvent) => void;
-type EventBuffer = {
-    inboundQueue: InboundEvent[];
-    outboundQueue: OutboundEvent[];
+type EventContext = {
+    /** Subscribe to events by type. Returns unsubscribe function. */
+    subscribe: (type: string, callback: EventSubscriber) => () => void;
+    /** Dispatch an event to all matching subscribers (used by orchestrator) */
+    dispatch: (event: InboundEvent) => void;
+    /** Send an outbound system message to backend */
+    sendOutboundSystem: (event: {
+        type: string;
+        sessionId: string;
+        payload: unknown;
+    }) => Promise<void>;
+    /** Current SSE connection status */
     sseStatus: SSEStatus;
-    lastEventId: string | null;
-    subscribers: Map<string, Set<EventSubscriber>>;
 };
+declare function useEventContext(): EventContext;
+type EventContextProviderProps = {
+    children: ReactNode;
+    aomiClient: AomiClient;
+    sessionId: string;
+};
+/**
+ * Simplified EventContext — a pure pub/sub relay.
+ *
+ * SSE subscription and system event unwrapping are now handled by ClientSession
+ * in the orchestrator. This provider just maintains the subscriber registry
+ * and sendOutboundSystem for direct system messages.
+ */
+declare function EventContextProvider({ children, aomiClient, sessionId, }: EventContextProviderProps): react_jsx_runtime.JSX.Element;
 
 type NotificationType = "notice" | "success" | "error" | "wallet";
 type Notification$1 = {
@@ -127,40 +140,26 @@ declare function NotificationContextProvider({ children, }: NotificationContextP
 
 type WalletRequestKind = "transaction" | "eip712_sign";
 type WalletRequestStatus = "pending" | "processing";
-type WalletRequest = {
-    id: string;
-    kind: WalletRequestKind;
-    payload: WalletTxPayload | WalletEip712Payload;
-    status: WalletRequestStatus;
-    timestamp: number;
-};
-type WalletBuffer = {
-    queue: WalletRequest[];
-    nextId: number;
-};
-
 type WalletRequestResult = {
     txHash?: string;
     signature?: string;
     amount?: string;
 };
 type WalletHandlerConfig = {
-    sessionId: string;
-    /** Called after a wallet request is resolved/rejected and the outbound event is sent.
-     *  Used by core.tsx to start polling for the AI's response. */
-    onRequestComplete?: () => void;
+    /** Get the ClientSession for the current thread. */
+    getSession: () => Session | undefined;
 };
 type WalletHandlerApi = {
     /** All queued wallet requests (tx + eip712) */
     pendingRequests: WalletRequest[];
-    /** Mark a request as being processed */
-    startProcessing: (id: string) => void;
-    /** Complete a request successfully — dequeues + sends response to backend */
+    /** Enqueue a wallet request (called by orchestrator on ClientSession events) */
+    enqueueRequest: (request: WalletRequest) => void;
+    /** Complete a request successfully — sends response to backend via ClientSession */
     resolveRequest: (id: string, result: WalletRequestResult) => void;
-    /** Fail a request — dequeues + sends error to backend */
+    /** Fail a request — sends error to backend via ClientSession */
     rejectRequest: (id: string, error?: string) => void;
 };
-declare function useWalletHandler({ sessionId, onRequestComplete, }: WalletHandlerConfig): WalletHandlerApi;
+declare function useWalletHandler({ getSession, }: WalletHandlerConfig): WalletHandlerApi;
 
 type AomiRuntimeApi = {
     /** Current user state (wallet connection, address, chain, etc.) */
@@ -220,7 +219,11 @@ type AomiRuntimeApi = {
     /** Subscribe to inbound events by type. Returns unsubscribe function. */
     subscribe: (type: string, callback: EventSubscriber) => () => void;
     /** Send a system command to the backend */
-    sendSystemCommand: (event: Omit<OutboundEvent, "timestamp">) => Promise<void>;
+    sendSystemCommand: (event: {
+        type: string;
+        sessionId: string;
+        payload: unknown;
+    }) => Promise<void>;
     /** Current SSE connection status */
     sseStatus: SSEStatus;
 };
@@ -254,24 +257,6 @@ type AomiRuntimeApi = {
  */
 declare function useAomiRuntime(): AomiRuntimeApi;
 
-type EventContext = {
-    /** Subscribe to inbound events by type. Returns unsubscribe function. */
-    subscribe: (type: string, callback: EventSubscriber) => () => void;
-    /** Send an outbound event to backend immediately */
-    sendOutboundSystem: (event: Omit<OutboundEvent, "timestamp">) => Promise<void>;
-    /** Dispatch system events from HTTP polling into the event buffer */
-    dispatchInboundSystem: (sessionId: string, events: AomiSystemEvent[]) => void;
-    /** Current SSE connection status */
-    sseStatus: SSEStatus;
-};
-declare function useEventContext(): EventContext;
-type EventContextProviderProps = {
-    children: ReactNode;
-    aomiClient: AomiClient;
-    sessionId: string;
-};
-declare function EventContextProvider({ children, aomiClient, sessionId, }: EventContextProviderProps): react_jsx_runtime.JSX.Element;
-
 type Notification = {
     id: string;
     type: string;
@@ -321,11 +306,17 @@ declare const SUPPORTED_CHAINS: ChainInfo[];
 /** Look up ChainInfo by chain ID. Returns undefined for unknown chains. */
 declare const getChainInfo: (chainId: number | undefined) => ChainInfo | undefined;
 
+/** A stored provider API key (BYOK) */
+type StoredProviderKey = {
+    apiKey: string;
+    keyPrefix: string;
+    label?: string;
+};
 /** Global control state (shared across all threads) */
 type ControlState = {
     /** API key for authenticated requests */
     apiKey: string | null;
-    /** Stable client identifier for this browser tab (associates sessions with secrets) */
+    /** Stable client identifier for this browser profile (associates sessions with secrets) */
     clientId: string | null;
     /** Available models fetched from backend */
     availableModels: string[];
@@ -335,6 +326,8 @@ type ControlState = {
     defaultModel: string | null;
     /** Default app (from authorizedApps) */
     defaultApp: string | null;
+    /** Provider API keys stored locally (BYOK) — keyed by provider name */
+    providerKeys: Record<string, StoredProviderKey>;
 };
 type ControlContextApi = {
     /** Global state (apiKey, clientId, available models/apps) */
@@ -345,6 +338,14 @@ type ControlContextApi = {
     ingestSecrets: (secrets: Record<string, string>) => Promise<Record<string, string>>;
     /** Clear all secrets from the backend vault */
     clearSecrets: () => Promise<void>;
+    /** Store a provider API key (BYOK) in localStorage and ingest into backend vault */
+    setProviderKey: (provider: string, apiKey: string, label?: string) => Promise<void>;
+    /** Remove a provider API key from localStorage and backend vault */
+    removeProviderKey: (provider: string) => Promise<void>;
+    /** Get all stored provider keys (metadata only — keys are in state.providerKeys) */
+    getProviderKeys: () => Record<string, StoredProviderKey>;
+    /** Check if a provider key is stored */
+    hasProviderKey: (provider?: string) => boolean;
     /** Fetch available models from backend */
     getAvailableModels: () => Promise<string[]>;
     /** Fetch authorized apps from backend */
@@ -384,4 +385,4 @@ type ControlContextProviderProps = {
 };
 declare function ControlContextProvider({ children, aomiClient, sessionId, publicKey, getThreadMetadata, updateThreadMetadata, }: ControlContextProviderProps): react_jsx_runtime.JSX.Element;
 
-export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventBuffer, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type OutboundEvent, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type UserState, type WalletBuffer, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequest, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };
+export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type StoredProviderKey, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type UserState, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };