@aomi-labs/client 0.1.11 → 0.1.12

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)) {
@@ -777,11 +793,10 @@ 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(), {
+    const url = buildApiUrl(this.baseUrl, "/api/state", {
+      user_state: userState ? JSON.stringify(userState) : void 0
+    });
+    const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
     if (!response.ok) {
@@ -851,7 +866,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 +879,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 +897,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 +914,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 +930,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 +949,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 +965,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 +984,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 +1004,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 +1023,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 +1258,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 +1483,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)) {
@@ -302,11 +318,10 @@ 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(), {
+    const url = buildApiUrl(this.baseUrl, "/api/state", {
+      user_state: userState ? JSON.stringify(userState) : void 0
+    });
+    const response = await fetch(url, {
       headers: withSessionHeader(sessionId)
     });
     if (!response.ok) {
@@ -376,7 +391,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 +404,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 +422,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 +439,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 +455,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 +474,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 +490,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 +509,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 +529,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 +548,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 +783,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 +1008,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 });
   }