@aomi-labs/client 0.1.11 → 0.1.13

package/dist/cli.js

@@ -729,6 +729,22 @@ function createSseSubscriber({
 // src/client.ts
 var SESSION_ID_HEADER = "X-Session-Id";
 var API_KEY_HEADER = "X-API-Key";
+function joinApiPath(baseUrl, path) {
+  const normalizedBase = baseUrl === "/" ? "" : baseUrl.replace(/\/+$/, "");
+  const normalizedPath = path.startsWith("/") ? path : `/${path}`;
+  return `${normalizedBase}${normalizedPath}` || normalizedPath;
+}
+function buildApiUrl(baseUrl, path, query) {
+  const url = joinApiPath(baseUrl, path);
+  if (!query) return url;
+  const params = new URLSearchParams();
+  for (const [key, value] of Object.entries(query)) {
+    if (value === void 0) continue;
+    params.set(key, value);
+  }
+  const queryString = params.toString();
+  return queryString ? `${url}?${queryString}` : url;
+}
 function toQueryString(payload) {
   const params = new URLSearchParams();
   for (const [key, value] of Object.entries(payload)) {
@@ -776,12 +792,12 @@ var AomiClient = class {
   /**
    * Fetch current session state (messages, processing status, title).
    */
-  async fetchState(sessionId, userState) {
-    const url = new URL("/api/state", this.baseUrl);
-    if (userState) {
-      url.searchParams.set("user_state", JSON.stringify(userState));
-    }
-    const response = await fetch(url.toString(), {
+  async fetchState(sessionId, userState, clientId) {
+    const url = buildApiUrl(this.baseUrl, "/api/state", {
+      user_state: userState ? JSON.stringify(userState) : void 0,
+      client_id: clientId
+    });
+    const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
     if (!response.ok) {
@@ -803,6 +819,9 @@ var AomiClient = class {
     if (options == null ? void 0 : options.userState) {
       payload.user_state = JSON.stringify(options.userState);
     }
+    if (options == null ? void 0 : options.clientId) {
+      payload.client_id = options.clientId;
+    }
     return postState(
       this.baseUrl,
       "/api/chat",
@@ -834,6 +853,40 @@ var AomiClient = class {
     );
   }
   // ===========================================================================
+  // Secrets
+  // ===========================================================================
+  /**
+   * Ingest secrets for a client. Returns opaque `$SECRET:<name>` handles.
+   * Call this once at page load (or when secrets change) with a stable
+   * client_id for the browser tab. The same client_id should be passed
+   * to `sendMessage` / `fetchState` so sessions get associated.
+   */
+  async ingestSecrets(clientId, secrets) {
+    const url = joinApiPath(this.baseUrl, "/api/secrets");
+    const response = await fetch(url, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ client_id: clientId, secrets })
+    });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
+  /**
+   * Clear all secrets for a client (e.g. on page unload or logout).
+   */
+  async clearSecrets(clientId) {
+    const url = buildApiUrl(this.baseUrl, "/api/secrets", {
+      client_id: clientId
+    });
+    const response = await fetch(url, { method: "DELETE" });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
+  // ===========================================================================
   // SSE (Real-time Updates)
   // ===========================================================================
   /**
@@ -851,7 +904,9 @@ var AomiClient = class {
    * List all threads for a wallet address.
    */
   async listThreads(publicKey) {
-    const url = `${this.baseUrl}/api/sessions?public_key=${encodeURIComponent(publicKey)}`;
+    const url = buildApiUrl(this.baseUrl, "/api/sessions", {
+      public_key: publicKey
+    });
     const response = await fetch(url);
     if (!response.ok) {
       throw new Error(`Failed to fetch threads: HTTP ${response.status}`);
@@ -862,7 +917,10 @@ var AomiClient = class {
    * Get a single thread by ID.
    */
   async getThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}`
+    );
     const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
@@ -877,7 +935,7 @@ var AomiClient = class {
   async createThread(threadId, publicKey) {
     const body = {};
     if (publicKey) body.public_key = publicKey;
-    const url = `${this.baseUrl}/api/sessions`;
+    const url = buildApiUrl(this.baseUrl, "/api/sessions");
     const response = await fetch(url, {
       method: "POST",
       headers: withSessionHeader(threadId, {
@@ -894,7 +952,10 @@ var AomiClient = class {
    * Delete a thread by ID.
    */
   async deleteThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}`
+    );
     const response = await fetch(url, {
       method: "DELETE",
       headers: withSessionHeader(sessionId)
@@ -907,7 +968,10 @@ var AomiClient = class {
    * Rename a thread.
    */
   async renameThread(sessionId, newTitle) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}`
+    );
     const response = await fetch(url, {
       method: "PATCH",
       headers: withSessionHeader(sessionId, {
@@ -923,7 +987,10 @@ var AomiClient = class {
    * Archive a thread.
    */
   async archiveThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}/archive`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}/archive`
+    );
     const response = await fetch(url, {
       method: "POST",
       headers: withSessionHeader(sessionId)
@@ -936,7 +1003,10 @@ var AomiClient = class {
    * Unarchive a thread.
    */
   async unarchiveThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}/unarchive`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}/unarchive`
+    );
     const response = await fetch(url, {
       method: "POST",
       headers: withSessionHeader(sessionId)
@@ -952,11 +1022,10 @@ var AomiClient = class {
    * Get system events for a session.
    */
   async getSystemEvents(sessionId, count) {
-    const url = new URL("/api/events", this.baseUrl);
-    if (count !== void 0) {
-      url.searchParams.set("count", String(count));
-    }
-    const response = await fetch(url.toString(), {
+    const url = buildApiUrl(this.baseUrl, "/api/events", {
+      count: count !== void 0 ? String(count) : void 0
+    });
+    const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
     if (!response.ok) {
@@ -973,16 +1042,15 @@ var AomiClient = class {
    */
   async getApps(sessionId, options) {
     var _a3;
-    const url = new URL("/api/control/apps", this.baseUrl);
-    if (options == null ? void 0 : options.publicKey) {
-      url.searchParams.set("public_key", options.publicKey);
-    }
+    const url = buildApiUrl(this.baseUrl, "/api/control/apps", {
+      public_key: options == null ? void 0 : options.publicKey
+    });
     const apiKey = (_a3 = options == null ? void 0 : options.apiKey) != null ? _a3 : this.apiKey;
     const headers = new Headers(withSessionHeader(sessionId));
     if (apiKey) {
       headers.set(API_KEY_HEADER, apiKey);
     }
-    const response = await fetch(url.toString(), { headers });
+    const response = await fetch(url, { headers });
     if (!response.ok) {
       throw new Error(`Failed to get apps: HTTP ${response.status}`);
     }
@@ -993,13 +1061,13 @@ var AomiClient = class {
    */
   async getModels(sessionId, options) {
     var _a3;
-    const url = new URL("/api/control/models", this.baseUrl);
+    const url = buildApiUrl(this.baseUrl, "/api/control/models");
     const apiKey = (_a3 = options == null ? void 0 : options.apiKey) != null ? _a3 : this.apiKey;
     const headers = new Headers(withSessionHeader(sessionId));
     if (apiKey) {
       headers.set(API_KEY_HEADER, apiKey);
     }
-    const response = await fetch(url.toString(), {
+    const response = await fetch(url, {
       headers
     });
     if (!response.ok) {
@@ -1228,6 +1296,9 @@ function toViemSignTypedDataArgs(payload) {
 }
 
 // src/session.ts
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
 function sortJson(value) {
   if (Array.isArray(value)) {
     return value.map((entry) => sortJson(entry));
@@ -1450,6 +1521,30 @@ var ClientSession = class extends TypedEventEmitter {
       this.publicKey = address;
     }
   }
+  addExtValue(key, value) {
+    var _a3;
+    const current = (_a3 = this.userState) != null ? _a3 : {};
+    const currentExt = isRecord(current["ext"]) ? current["ext"] : {};
+    this.resolveUserState(__spreadProps(__spreadValues({}, current), {
+      ext: __spreadProps(__spreadValues({}, currentExt), {
+        [key]: value
+      })
+    }));
+  }
+  removeExtValue(key) {
+    if (!this.userState) return;
+    const currentExt = this.userState["ext"];
+    if (!isRecord(currentExt)) return;
+    const nextExt = __spreadValues({}, currentExt);
+    delete nextExt[key];
+    const nextState = __spreadValues({}, this.userState);
+    if (Object.keys(nextExt).length === 0) {
+      delete nextState["ext"];
+    } else {
+      nextState["ext"] = nextExt;
+    }
+    this.resolveUserState(nextState);
+  }
   resolveWallet(address, chainId) {
     this.resolveUserState({ address, chainId: chainId != null ? chainId : 1, isConnected: true });
   }

package/dist/index.cjs

@@ -254,6 +254,22 @@ function createSseSubscriber({
 // src/client.ts
 var SESSION_ID_HEADER = "X-Session-Id";
 var API_KEY_HEADER = "X-API-Key";
+function joinApiPath(baseUrl, path) {
+  const normalizedBase = baseUrl === "/" ? "" : baseUrl.replace(/\/+$/, "");
+  const normalizedPath = path.startsWith("/") ? path : `/${path}`;
+  return `${normalizedBase}${normalizedPath}` || normalizedPath;
+}
+function buildApiUrl(baseUrl, path, query) {
+  const url = joinApiPath(baseUrl, path);
+  if (!query) return url;
+  const params = new URLSearchParams();
+  for (const [key, value] of Object.entries(query)) {
+    if (value === void 0) continue;
+    params.set(key, value);
+  }
+  const queryString = params.toString();
+  return queryString ? `${url}?${queryString}` : url;
+}
 function toQueryString(payload) {
   const params = new URLSearchParams();
   for (const [key, value] of Object.entries(payload)) {
@@ -301,12 +317,12 @@ var AomiClient = class {
   /**
    * Fetch current session state (messages, processing status, title).
    */
-  async fetchState(sessionId, userState) {
-    const url = new URL("/api/state", this.baseUrl);
-    if (userState) {
-      url.searchParams.set("user_state", JSON.stringify(userState));
-    }
-    const response = await fetch(url.toString(), {
+  async fetchState(sessionId, userState, clientId) {
+    const url = buildApiUrl(this.baseUrl, "/api/state", {
+      user_state: userState ? JSON.stringify(userState) : void 0,
+      client_id: clientId
+    });
+    const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
     if (!response.ok) {
@@ -328,6 +344,9 @@ var AomiClient = class {
     if (options == null ? void 0 : options.userState) {
       payload.user_state = JSON.stringify(options.userState);
     }
+    if (options == null ? void 0 : options.clientId) {
+      payload.client_id = options.clientId;
+    }
     return postState(
       this.baseUrl,
       "/api/chat",
@@ -359,6 +378,40 @@ var AomiClient = class {
     );
   }
   // ===========================================================================
+  // Secrets
+  // ===========================================================================
+  /**
+   * Ingest secrets for a client. Returns opaque `$SECRET:<name>` handles.
+   * Call this once at page load (or when secrets change) with a stable
+   * client_id for the browser tab. The same client_id should be passed
+   * to `sendMessage` / `fetchState` so sessions get associated.
+   */
+  async ingestSecrets(clientId, secrets) {
+    const url = joinApiPath(this.baseUrl, "/api/secrets");
+    const response = await fetch(url, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ client_id: clientId, secrets })
+    });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
+  /**
+   * Clear all secrets for a client (e.g. on page unload or logout).
+   */
+  async clearSecrets(clientId) {
+    const url = buildApiUrl(this.baseUrl, "/api/secrets", {
+      client_id: clientId
+    });
+    const response = await fetch(url, { method: "DELETE" });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
+  // ===========================================================================
   // SSE (Real-time Updates)
   // ===========================================================================
   /**
@@ -376,7 +429,9 @@ var AomiClient = class {
    * List all threads for a wallet address.
    */
   async listThreads(publicKey) {
-    const url = `${this.baseUrl}/api/sessions?public_key=${encodeURIComponent(publicKey)}`;
+    const url = buildApiUrl(this.baseUrl, "/api/sessions", {
+      public_key: publicKey
+    });
     const response = await fetch(url);
     if (!response.ok) {
       throw new Error(`Failed to fetch threads: HTTP ${response.status}`);
@@ -387,7 +442,10 @@ var AomiClient = class {
    * Get a single thread by ID.
    */
   async getThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}`
+    );
     const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
@@ -402,7 +460,7 @@ var AomiClient = class {
   async createThread(threadId, publicKey) {
     const body = {};
     if (publicKey) body.public_key = publicKey;
-    const url = `${this.baseUrl}/api/sessions`;
+    const url = buildApiUrl(this.baseUrl, "/api/sessions");
     const response = await fetch(url, {
       method: "POST",
       headers: withSessionHeader(threadId, {
@@ -419,7 +477,10 @@ var AomiClient = class {
    * Delete a thread by ID.
    */
   async deleteThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}`
+    );
     const response = await fetch(url, {
       method: "DELETE",
       headers: withSessionHeader(sessionId)
@@ -432,7 +493,10 @@ var AomiClient = class {
    * Rename a thread.
    */
   async renameThread(sessionId, newTitle) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}`
+    );
     const response = await fetch(url, {
       method: "PATCH",
       headers: withSessionHeader(sessionId, {
@@ -448,7 +512,10 @@ var AomiClient = class {
    * Archive a thread.
    */
   async archiveThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}/archive`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}/archive`
+    );
     const response = await fetch(url, {
       method: "POST",
       headers: withSessionHeader(sessionId)
@@ -461,7 +528,10 @@ var AomiClient = class {
    * Unarchive a thread.
    */
   async unarchiveThread(sessionId) {
-    const url = `${this.baseUrl}/api/sessions/${encodeURIComponent(sessionId)}/unarchive`;
+    const url = buildApiUrl(
+      this.baseUrl,
+      `/api/sessions/${encodeURIComponent(sessionId)}/unarchive`
+    );
     const response = await fetch(url, {
       method: "POST",
       headers: withSessionHeader(sessionId)
@@ -477,11 +547,10 @@ var AomiClient = class {
    * Get system events for a session.
    */
   async getSystemEvents(sessionId, count) {
-    const url = new URL("/api/events", this.baseUrl);
-    if (count !== void 0) {
-      url.searchParams.set("count", String(count));
-    }
-    const response = await fetch(url.toString(), {
+    const url = buildApiUrl(this.baseUrl, "/api/events", {
+      count: count !== void 0 ? String(count) : void 0
+    });
+    const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
     if (!response.ok) {
@@ -498,16 +567,15 @@ var AomiClient = class {
    */
   async getApps(sessionId, options) {
     var _a;
-    const url = new URL("/api/control/apps", this.baseUrl);
-    if (options == null ? void 0 : options.publicKey) {
-      url.searchParams.set("public_key", options.publicKey);
-    }
+    const url = buildApiUrl(this.baseUrl, "/api/control/apps", {
+      public_key: options == null ? void 0 : options.publicKey
+    });
     const apiKey = (_a = options == null ? void 0 : options.apiKey) != null ? _a : this.apiKey;
     const headers = new Headers(withSessionHeader(sessionId));
     if (apiKey) {
       headers.set(API_KEY_HEADER, apiKey);
     }
-    const response = await fetch(url.toString(), { headers });
+    const response = await fetch(url, { headers });
     if (!response.ok) {
       throw new Error(`Failed to get apps: HTTP ${response.status}`);
     }
@@ -518,13 +586,13 @@ var AomiClient = class {
    */
   async getModels(sessionId, options) {
     var _a;
-    const url = new URL("/api/control/models", this.baseUrl);
+    const url = buildApiUrl(this.baseUrl, "/api/control/models");
     const apiKey = (_a = options == null ? void 0 : options.apiKey) != null ? _a : this.apiKey;
     const headers = new Headers(withSessionHeader(sessionId));
     if (apiKey) {
       headers.set(API_KEY_HEADER, apiKey);
     }
-    const response = await fetch(url.toString(), {
+    const response = await fetch(url, {
       headers
     });
     if (!response.ok) {
@@ -753,6 +821,9 @@ function toViemSignTypedDataArgs(payload) {
 }
 
 // src/session.ts
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
 function sortJson(value) {
   if (Array.isArray(value)) {
     return value.map((entry) => sortJson(entry));
@@ -975,6 +1046,30 @@ var ClientSession = class extends TypedEventEmitter {
       this.publicKey = address;
     }
   }
+  addExtValue(key, value) {
+    var _a;
+    const current = (_a = this.userState) != null ? _a : {};
+    const currentExt = isRecord(current["ext"]) ? current["ext"] : {};
+    this.resolveUserState(__spreadProps(__spreadValues({}, current), {
+      ext: __spreadProps(__spreadValues({}, currentExt), {
+        [key]: value
+      })
+    }));
+  }
+  removeExtValue(key) {
+    if (!this.userState) return;
+    const currentExt = this.userState["ext"];
+    if (!isRecord(currentExt)) return;
+    const nextExt = __spreadValues({}, currentExt);
+    delete nextExt[key];
+    const nextState = __spreadValues({}, this.userState);
+    if (Object.keys(nextExt).length === 0) {
+      delete nextState["ext"];
+    } else {
+      nextState["ext"] = nextExt;
+    }
+    this.resolveUserState(nextState);
+  }
   resolveWallet(address, chainId) {
     this.resolveUserState({ address, chainId: chainId != null ? chainId : 1, isConnected: true });
   }