@aomi-labs/client 0.1.2 → 0.1.6

package/dist/index.d.cts

@@ -12,7 +12,7 @@ type Logger = {
 type AomiClientOptions = {
     /** Base URL of the Aomi backend (e.g. "https://aomi.dev") */
     baseUrl: string;
-    /** Default API key for non-default namespaces */
+    /** Default API key for non-default apps */
     apiKey?: string;
     /** Optional logger for debug output (default: silent) */
     logger?: Logger;
@@ -33,6 +33,7 @@ interface AomiStateResponse {
     system_events?: AomiSystemEvent[] | null;
     title?: string | null;
     is_processing?: boolean;
+    user_state?: UserState | null;
 }
 /**
  * POST /api/chat
@@ -43,6 +44,7 @@ interface AomiChatResponse {
     system_events?: AomiSystemEvent[] | null;
     title?: string | null;
     is_processing?: boolean;
+    user_state?: UserState | null;
 }
 /**
  * POST /api/system
@@ -133,7 +135,7 @@ declare class AomiClient {
      * Send a chat message and return updated session state.
      */
     sendMessage(sessionId: string, message: string, options?: {
-        namespace?: string;
+        app?: string;
         publicKey?: string;
         apiKey?: string;
         userState?: UserState;
@@ -185,9 +187,9 @@ declare class AomiClient {
      */
     getSystemEvents(sessionId: string, count?: number): Promise<AomiSystemEvent[]>;
     /**
-     * Get available namespaces.
+     * Get available apps.
      */
-    getNamespaces(sessionId: string, options?: {
+    getApps(sessionId: string, options?: {
         publicKey?: string;
         apiKey?: string;
     }): Promise<string[]>;
@@ -201,7 +203,7 @@ declare class AomiClient {
      * Set the model for a session.
      */
     setModel(sessionId: string, rig: string, options?: {
-        namespace?: string;
+        app?: string;
         apiKey?: string;
     }): Promise<{
         success: boolean;
@@ -295,8 +297,8 @@ type SendResult = {
 type SessionOptions = {
     /** Session ID. Auto-generated (crypto.randomUUID) if omitted. */
     sessionId?: string;
-    /** Namespace for chat messages. Default: "default" */
-    namespace?: string;
+    /** App for chat messages. Default: "default" */
+    app?: string;
     /** User public key (wallet address). */
     publicKey?: string;
     /** API key override. */
@@ -350,12 +352,12 @@ type SessionEventMap = {
         payload: unknown;
     };
 };
-declare class Session extends TypedEventEmitter<SessionEventMap> {
+declare class ClientSession extends TypedEventEmitter<SessionEventMap> {
     /** The underlying low-level client. */
     readonly client: AomiClient;
     /** The session (thread) ID. */
     readonly sessionId: string;
-    private namespace;
+    private app;
     private publicKey?;
     private apiKey?;
     private userState?;
@@ -412,6 +414,9 @@ declare class Session extends TypedEventEmitter<SessionEventMap> {
     getPendingRequests(): WalletRequest[];
     /** Whether the AI is currently processing. */
     getIsProcessing(): boolean;
+    resolveUserState(userState: UserState): void;
+    resolveWallet(address: string, chainId?: number): void;
+    syncUserState(): Promise<AomiStateResponse>;
     private startPolling;
     private stopPolling;
     private pollTick;
@@ -423,6 +428,7 @@ declare class Session extends TypedEventEmitter<SessionEventMap> {
     private sendSystemEvent;
     private resolvePending;
     private assertOpen;
+    private assertUserStateAligned;
 }
 
 type UnwrappedEvent = {
@@ -440,4 +446,4 @@ type UnwrappedEvent = {
  */
 declare function unwrapSystemEvent(event: AomiSystemEvent): UnwrappedEvent | null;
 
-export { type AomiChatResponse, AomiClient, type AomiClientOptions, type AomiCreateThreadResponse, type AomiInterruptResponse, type AomiMessage, type AomiSSEEvent, type AomiSSEEventType, type AomiStateResponse, type AomiSystemEvent, type AomiSystemResponse, type AomiThread, type Logger, type SendResult, Session, type SessionEventMap, type SessionOptions, TypedEventEmitter, type UnwrappedEvent, type UserState, type WalletEip712Payload, type WalletRequest, type WalletRequestKind, type WalletRequestResult, type WalletTxPayload, isAsyncCallback, isInlineCall, isSystemError, isSystemNotice, normalizeEip712Payload, normalizeTxPayload, unwrapSystemEvent };
+export { type AomiChatResponse, AomiClient, type AomiClientOptions, type AomiCreateThreadResponse, type AomiInterruptResponse, type AomiMessage, type AomiSSEEvent, type AomiSSEEventType, type AomiStateResponse, type AomiSystemEvent, type AomiSystemResponse, type AomiThread, type Logger, type SendResult, ClientSession as Session, type SessionEventMap, type SessionOptions, TypedEventEmitter, type UnwrappedEvent, type UserState, type WalletEip712Payload, type WalletRequest, type WalletRequestKind, type WalletRequestResult, type WalletTxPayload, isAsyncCallback, isInlineCall, isSystemError, isSystemNotice, normalizeEip712Payload, normalizeTxPayload, unwrapSystemEvent };

package/dist/index.d.ts

@@ -12,7 +12,7 @@ type Logger = {
 type AomiClientOptions = {
     /** Base URL of the Aomi backend (e.g. "https://aomi.dev") */
     baseUrl: string;
-    /** Default API key for non-default namespaces */
+    /** Default API key for non-default apps */
     apiKey?: string;
     /** Optional logger for debug output (default: silent) */
     logger?: Logger;
@@ -33,6 +33,7 @@ interface AomiStateResponse {
     system_events?: AomiSystemEvent[] | null;
     title?: string | null;
     is_processing?: boolean;
+    user_state?: UserState | null;
 }
 /**
  * POST /api/chat
@@ -43,6 +44,7 @@ interface AomiChatResponse {
     system_events?: AomiSystemEvent[] | null;
     title?: string | null;
     is_processing?: boolean;
+    user_state?: UserState | null;
 }
 /**
  * POST /api/system
@@ -133,7 +135,7 @@ declare class AomiClient {
      * Send a chat message and return updated session state.
      */
     sendMessage(sessionId: string, message: string, options?: {
-        namespace?: string;
+        app?: string;
         publicKey?: string;
         apiKey?: string;
         userState?: UserState;
@@ -185,9 +187,9 @@ declare class AomiClient {
      */
     getSystemEvents(sessionId: string, count?: number): Promise<AomiSystemEvent[]>;
     /**
-     * Get available namespaces.
+     * Get available apps.
      */
-    getNamespaces(sessionId: string, options?: {
+    getApps(sessionId: string, options?: {
         publicKey?: string;
         apiKey?: string;
     }): Promise<string[]>;
@@ -201,7 +203,7 @@ declare class AomiClient {
      * Set the model for a session.
      */
     setModel(sessionId: string, rig: string, options?: {
-        namespace?: string;
+        app?: string;
         apiKey?: string;
     }): Promise<{
         success: boolean;
@@ -295,8 +297,8 @@ type SendResult = {
 type SessionOptions = {
     /** Session ID. Auto-generated (crypto.randomUUID) if omitted. */
     sessionId?: string;
-    /** Namespace for chat messages. Default: "default" */
-    namespace?: string;
+    /** App for chat messages. Default: "default" */
+    app?: string;
     /** User public key (wallet address). */
     publicKey?: string;
     /** API key override. */
@@ -350,12 +352,12 @@ type SessionEventMap = {
         payload: unknown;
     };
 };
-declare class Session extends TypedEventEmitter<SessionEventMap> {
+declare class ClientSession extends TypedEventEmitter<SessionEventMap> {
     /** The underlying low-level client. */
     readonly client: AomiClient;
     /** The session (thread) ID. */
     readonly sessionId: string;
-    private namespace;
+    private app;
     private publicKey?;
     private apiKey?;
     private userState?;
@@ -412,6 +414,9 @@ declare class Session extends TypedEventEmitter<SessionEventMap> {
     getPendingRequests(): WalletRequest[];
     /** Whether the AI is currently processing. */
     getIsProcessing(): boolean;
+    resolveUserState(userState: UserState): void;
+    resolveWallet(address: string, chainId?: number): void;
+    syncUserState(): Promise<AomiStateResponse>;
     private startPolling;
     private stopPolling;
     private pollTick;
@@ -423,6 +428,7 @@ declare class Session extends TypedEventEmitter<SessionEventMap> {
     private sendSystemEvent;
     private resolvePending;
     private assertOpen;
+    private assertUserStateAligned;
 }
 
 type UnwrappedEvent = {
@@ -440,4 +446,4 @@ type UnwrappedEvent = {
  */
 declare function unwrapSystemEvent(event: AomiSystemEvent): UnwrappedEvent | null;
 
-export { type AomiChatResponse, AomiClient, type AomiClientOptions, type AomiCreateThreadResponse, type AomiInterruptResponse, type AomiMessage, type AomiSSEEvent, type AomiSSEEventType, type AomiStateResponse, type AomiSystemEvent, type AomiSystemResponse, type AomiThread, type Logger, type SendResult, Session, type SessionEventMap, type SessionOptions, TypedEventEmitter, type UnwrappedEvent, type UserState, type WalletEip712Payload, type WalletRequest, type WalletRequestKind, type WalletRequestResult, type WalletTxPayload, isAsyncCallback, isInlineCall, isSystemError, isSystemNotice, normalizeEip712Payload, normalizeTxPayload, unwrapSystemEvent };
+export { type AomiChatResponse, AomiClient, type AomiClientOptions, type AomiCreateThreadResponse, type AomiInterruptResponse, type AomiMessage, type AomiSSEEvent, type AomiSSEEventType, type AomiStateResponse, type AomiSystemEvent, type AomiSystemResponse, type AomiThread, type Logger, type SendResult, ClientSession as Session, type SessionEventMap, type SessionOptions, TypedEventEmitter, type UnwrappedEvent, type UserState, type WalletEip712Payload, type WalletRequest, type WalletRequestKind, type WalletRequestResult, type WalletTxPayload, isAsyncCallback, isInlineCall, isSystemError, isSystemNotice, normalizeEip712Payload, normalizeTxPayload, unwrapSystemEvent };

package/dist/index.js

@@ -240,9 +240,9 @@ var AomiClient = class {
    */
   async sendMessage(sessionId, message, options) {
     var _a, _b;
-    const namespace = (_a = options == null ? void 0 : options.namespace) != null ? _a : "default";
+    const app = (_a = options == null ? void 0 : options.app) != null ? _a : "default";
     const apiKey = (_b = options == null ? void 0 : options.apiKey) != null ? _b : this.apiKey;
-    const payload = { message, namespace };
+    const payload = { message, app };
     if (options == null ? void 0 : options.publicKey) {
       payload.public_key = options.publicKey;
     }
@@ -415,11 +415,11 @@ var AomiClient = class {
   // Control API
   // ===========================================================================
   /**
-   * Get available namespaces.
+   * Get available apps.
    */
-  async getNamespaces(sessionId, options) {
+  async getApps(sessionId, options) {
     var _a;
-    const url = new URL("/api/control/namespaces", this.baseUrl);
+    const url = new URL("/api/control/apps", this.baseUrl);
     if (options == null ? void 0 : options.publicKey) {
       url.searchParams.set("public_key", options.publicKey);
     }
@@ -430,7 +430,7 @@ var AomiClient = class {
     }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
-      throw new Error(`Failed to get namespaces: HTTP ${response.status}`);
+      throw new Error(`Failed to get apps: HTTP ${response.status}`);
     }
     return await response.json();
   }
@@ -460,8 +460,8 @@ var AomiClient = class {
     var _a;
     const apiKey = (_a = options == null ? void 0 : options.apiKey) != null ? _a : this.apiKey;
     const payload = { rig };
-    if (options == null ? void 0 : options.namespace) {
-      payload.namespace = options.namespace;
+    if (options == null ? void 0 : options.app) {
+      payload.app = options.app;
     }
     return postState(this.baseUrl, "/api/control/model", payload, sessionId, apiKey);
   }
@@ -642,7 +642,38 @@ function normalizeEip712Payload(payload) {
 }
 
 // src/session.ts
-var Session = class extends TypedEventEmitter {
+function sortJson(value) {
+  if (Array.isArray(value)) {
+    return value.map((entry) => sortJson(entry));
+  }
+  if (value && typeof value === "object") {
+    return Object.keys(value).sort().reduce((acc, key) => {
+      acc[key] = sortJson(value[key]);
+      return acc;
+    }, {});
+  }
+  return value;
+}
+function isSubsetMatch(expected, actual) {
+  if (Array.isArray(expected)) {
+    if (!Array.isArray(actual) || expected.length !== actual.length) {
+      return false;
+    }
+    return expected.every(
+      (entry, index) => isSubsetMatch(entry, actual[index])
+    );
+  }
+  if (expected && typeof expected === "object") {
+    if (!actual || typeof actual !== "object" || Array.isArray(actual)) {
+      return false;
+    }
+    return Object.entries(expected).every(
+      ([key, value]) => isSubsetMatch(value, actual[key])
+    );
+  }
+  return expected === actual;
+}
+var ClientSession = class extends TypedEventEmitter {
   constructor(clientOrOptions, sessionOptions) {
     var _a, _b, _c;
     super();
@@ -658,7 +689,7 @@ var Session = class extends TypedEventEmitter {
     this.pendingResolve = null;
     this.client = clientOrOptions instanceof AomiClient ? clientOrOptions : new AomiClient(clientOrOptions);
     this.sessionId = (_a = sessionOptions == null ? void 0 : sessionOptions.sessionId) != null ? _a : crypto.randomUUID();
-    this.namespace = (_b = sessionOptions == null ? void 0 : sessionOptions.namespace) != null ? _b : "default";
+    this.app = (_b = sessionOptions == null ? void 0 : sessionOptions.app) != null ? _b : "default";
     this.publicKey = sessionOptions == null ? void 0 : sessionOptions.publicKey;
     this.apiKey = sessionOptions == null ? void 0 : sessionOptions.apiKey;
     this.userState = sessionOptions == null ? void 0 : sessionOptions.userState;
@@ -684,11 +715,12 @@ var Session = class extends TypedEventEmitter {
   async send(message) {
     this.assertOpen();
     const response = await this.client.sendMessage(this.sessionId, message, {
-      namespace: this.namespace,
+      app: this.app,
       publicKey: this.publicKey,
       apiKey: this.apiKey,
       userState: this.userState
     });
+    this.assertUserStateAligned(response.user_state);
     this.applyState(response);
     if (!response.is_processing && this.walletRequests.length === 0) {
       return { messages: this._messages, title: this._title };
@@ -707,11 +739,12 @@ var Session = class extends TypedEventEmitter {
   async sendAsync(message) {
     this.assertOpen();
     const response = await this.client.sendMessage(this.sessionId, message, {
-      namespace: this.namespace,
+      app: this.app,
       publicKey: this.publicKey,
       apiKey: this.apiKey,
       userState: this.userState
     });
+    this.assertUserStateAligned(response.user_state);
     this.applyState(response);
     if (response.is_processing) {
       this._isProcessing = true;
@@ -824,6 +857,23 @@ var Session = class extends TypedEventEmitter {
   getIsProcessing() {
     return this._isProcessing;
   }
+  resolveUserState(userState) {
+    this.userState = userState;
+    const address = userState["address"];
+    if (typeof address === "string" && address.length > 0) {
+      this.publicKey = address;
+    }
+  }
+  resolveWallet(address, chainId) {
+    this.resolveUserState({ address, chainId: chainId != null ? chainId : 1, isConnected: true });
+  }
+  async syncUserState() {
+    this.assertOpen();
+    const state = await this.client.fetchState(this.sessionId, this.userState);
+    this.assertUserStateAligned(state.user_state);
+    this.applyState(state);
+    return state;
+  }
   // ===========================================================================
   // Internal — Polling (ported from PollingController)
   // ===========================================================================
@@ -852,6 +902,7 @@ var Session = class extends TypedEventEmitter {
         this.userState
       );
       if (!this.pollTimer) return;
+      this.assertUserStateAligned(state.user_state);
       this.applyState(state);
       if (!state.is_processing && this.walletRequests.length === 0) {
         this.stopPolling();
@@ -953,10 +1004,22 @@ var Session = class extends TypedEventEmitter {
       throw new Error("Session is closed");
     }
   }
+  assertUserStateAligned(actualUserState) {
+    if (!this.userState || !actualUserState) {
+      return;
+    }
+    if (!isSubsetMatch(this.userState, actualUserState)) {
+      const expected = JSON.stringify(sortJson(this.userState));
+      const actual = JSON.stringify(sortJson(actualUserState));
+      throw new Error(
+        `Backend user_state mismatch. expected subset=${expected} actual=${actual}`
+      );
+    }
+  }
 };
 export {
   AomiClient,
-  Session,
+  ClientSession as Session,
   TypedEventEmitter,
   isAsyncCallback,
   isInlineCall,