@aomi-labs/client 0.1.14 → 0.1.16

package/dist/cli.js

@@ -22,7 +22,7 @@ var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 // package.json
 var package_default = {
   name: "@aomi-labs/client",
-  version: "0.1.14",
+  version: "0.1.16",
   description: "Platform-agnostic TypeScript client for the Aomi backend API",
   type: "module",
   main: "./dist/index.cjs",
@@ -133,22 +133,38 @@ function parseAAMode(value) {
   }
   fatal("Unsupported AA mode. Use `4337` or `7702`.");
 }
+function parseSecret(value, secrets) {
+  const eqIdx = value.indexOf("=");
+  if (eqIdx > 0) {
+    secrets[value.slice(0, eqIdx)] = value.slice(eqIdx + 1);
+  }
+}
 function parseArgs(argv) {
   const raw = argv.slice(2);
   const command = raw[0] && !raw[0].startsWith("-") ? raw[0] : void 0;
   const rest = command ? raw.slice(1) : raw;
   const positional = [];
   const flags = {};
+  const secrets = {};
   for (let i = 0; i < rest.length; i++) {
     const arg = rest[i];
     if (arg.startsWith("--") && arg.includes("=")) {
       const [key, ...val] = arg.slice(2).split("=");
-      flags[key] = val.join("=");
+      const value = val.join("=");
+      if (key === "secret") {
+        parseSecret(value, secrets);
+      } else {
+        flags[key] = value;
+      }
     } else if (arg.startsWith("--")) {
       const key = arg.slice(2);
       const next = rest[i + 1];
       if (next && !next.startsWith("-")) {
-        flags[key] = next;
+        if (key === "secret") {
+          parseSecret(next, secrets);
+        } else {
+          flags[key] = next;
+        }
         i++;
       } else {
         flags[key] = "true";
@@ -159,7 +175,7 @@ function parseArgs(argv) {
       positional.push(arg);
     }
   }
-  return { command, positional, flags };
+  return { command, positional, flags, secrets };
 }
 function getConfig(parsed) {
   var _a3, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
@@ -188,6 +204,7 @@ function getConfig(parsed) {
     privateKey: (_j = parsed.flags["private-key"]) != null ? _j : process.env.PRIVATE_KEY,
     chainRpcUrl: (_k = parsed.flags["rpc-url"]) != null ? _k : process.env.CHAIN_RPC_URL,
     chain: parseChainId((_l = parsed.flags["chain"]) != null ? _l : process.env.AOMI_CHAIN_ID),
+    secrets: parsed.secrets,
     execution,
     aaProvider,
     aaMode
@@ -238,6 +255,7 @@ function toSessionFilePath(localId) {
 function toCliSessionState(stored) {
   return {
     sessionId: stored.sessionId,
+    clientId: stored.clientId,
     baseUrl: stored.baseUrl,
     app: stored.app,
     model: stored.model,
@@ -245,7 +263,8 @@ function toCliSessionState(stored) {
     publicKey: stored.publicKey,
     chainId: stored.chainId,
     pendingTxs: stored.pendingTxs,
-    signedTxs: stored.signedTxs
+    signedTxs: stored.signedTxs,
+    secretHandles: stored.secretHandles
   };
 }
 function readStoredSession(path) {
@@ -259,6 +278,7 @@ function readStoredSession(path) {
     const fallbackLocalId = (_a3 = parseSessionFileLocalId(basename(path))) != null ? _a3 : 0;
     return {
       sessionId: parsed.sessionId,
+      clientId: parsed.clientId,
       baseUrl: parsed.baseUrl,
       app: parsed.app,
       model: parsed.model,
@@ -267,6 +287,7 @@ function readStoredSession(path) {
       chainId: parsed.chainId,
       pendingTxs: parsed.pendingTxs,
       signedTxs: parsed.signedTxs,
+      secretHandles: parsed.secretHandles,
       localId: typeof parsed.localId === "number" && parsed.localId > 0 ? parsed.localId : fallbackLocalId,
       createdAt: typeof parsed.createdAt === "number" && parsed.createdAt > 0 ? parsed.createdAt : Date.now(),
       updatedAt: typeof parsed.updatedAt === "number" && parsed.updatedAt > 0 ? parsed.updatedAt : Date.now()
@@ -1129,6 +1150,30 @@ var AomiClient = class {
   }
 };
 
+// src/types.ts
+var CLIENT_TYPE_TS_CLI = "ts_cli";
+function addUserStateExt(userState, key, value) {
+  const currentExt = userState["ext"];
+  const extRecord = typeof currentExt === "object" && currentExt !== null && !Array.isArray(currentExt) ? currentExt : {};
+  return __spreadProps(__spreadValues({}, userState), {
+    ext: __spreadProps(__spreadValues({}, extRecord), {
+      [key]: value
+    })
+  });
+}
+function isInlineCall(event) {
+  return "InlineCall" in event;
+}
+function isSystemNotice(event) {
+  return "SystemNotice" in event;
+}
+function isSystemError(event) {
+  return "SystemError" in event;
+}
+function isAsyncCallback(event) {
+  return "AsyncCallback" in event;
+}
+
 // src/event-emitter.ts
 var TypedEventEmitter = class {
   constructor() {
@@ -1201,20 +1246,6 @@ var TypedEventEmitter = class {
   }
 };
 
-// src/types.ts
-function isInlineCall(event) {
-  return "InlineCall" in event;
-}
-function isSystemNotice(event) {
-  return "SystemNotice" in event;
-}
-function isSystemError(event) {
-  return "SystemError" in event;
-}
-function isAsyncCallback(event) {
-  return "AsyncCallback" in event;
-}
-
 // src/event-unwrap.ts
 function unwrapSystemEvent(event) {
   var _a3;
@@ -1372,7 +1403,7 @@ function isSubsetMatch(expected, actual) {
 }
 var ClientSession = class extends TypedEventEmitter {
   constructor(clientOrOptions, sessionOptions) {
-    var _a3, _b, _c;
+    var _a3, _b, _c, _d, _e;
     super();
     // Internal state
     this.pollTimer = null;
@@ -1389,8 +1420,9 @@ var ClientSession = class extends TypedEventEmitter {
     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;
-    this.pollIntervalMs = (_c = sessionOptions == null ? void 0 : sessionOptions.pollIntervalMs) != null ? _c : 500;
+    this.userState = (sessionOptions == null ? void 0 : sessionOptions.clientType) ? addUserStateExt((_c = sessionOptions == null ? void 0 : sessionOptions.userState) != null ? _c : {}, "client_type", sessionOptions.clientType) : sessionOptions == null ? void 0 : sessionOptions.userState;
+    this.clientId = (_d = sessionOptions == null ? void 0 : sessionOptions.clientId) != null ? _d : crypto.randomUUID();
+    this.pollIntervalMs = (_e = sessionOptions == null ? void 0 : sessionOptions.pollIntervalMs) != null ? _e : 500;
     this.logger = sessionOptions == null ? void 0 : sessionOptions.logger;
     this.unsubscribeSSE = this.client.subscribeSSE(
       this.sessionId,
@@ -1415,7 +1447,8 @@ var ClientSession = class extends TypedEventEmitter {
       app: this.app,
       publicKey: this.publicKey,
       apiKey: this.apiKey,
-      userState: this.userState
+      userState: this.userState,
+      clientId: this.clientId
     });
     this.assertUserStateAligned(response.user_state);
     this.applyState(response);
@@ -1439,7 +1472,8 @@ var ClientSession = class extends TypedEventEmitter {
       app: this.app,
       publicKey: this.publicKey,
       apiKey: this.apiKey,
-      userState: this.userState
+      userState: this.userState,
+      clientId: this.clientId
     });
     this.assertUserStateAligned(response.user_state);
     this.applyState(response);
@@ -1561,6 +1595,10 @@ var ClientSession = class extends TypedEventEmitter {
       this.publicKey = address;
     }
   }
+  setClientType(clientType) {
+    var _a3;
+    this.resolveUserState(addUserStateExt((_a3 = this.userState) != null ? _a3 : {}, "client_type", clientType));
+  }
   addExtValue(key, value) {
     var _a3;
     const current = (_a3 = this.userState) != null ? _a3 : {};
@@ -1590,7 +1628,7 @@ var ClientSession = class extends TypedEventEmitter {
   }
   async syncUserState() {
     this.assertOpen();
-    const state = await this.client.fetchState(this.sessionId, this.userState);
+    const state = await this.client.fetchState(this.sessionId, this.userState, this.clientId);
     this.assertUserStateAligned(state.user_state);
     this.applyState(state);
     return state;
@@ -1620,7 +1658,8 @@ var ClientSession = class extends TypedEventEmitter {
     try {
       const state = await this.client.fetchState(
         this.sessionId,
-        this.userState
+        this.userState,
+        this.clientId
       );
       if (!this.pollTimer) return;
       this.assertUserStateAligned(state.user_state);
@@ -1741,9 +1780,6 @@ var ClientSession = class extends TypedEventEmitter {
 
 // src/cli/user-state.ts
 function buildCliUserState(publicKey, chainId) {
-  if (publicKey === void 0 && chainId === void 0) {
-    return void 0;
-  }
   const userState = {};
   if (publicKey !== void 0) {
     userState.address = publicKey;
@@ -1752,7 +1788,7 @@ function buildCliUserState(publicKey, chainId) {
   if (chainId !== void 0) {
     userState.chainId = chainId;
   }
-  return userState;
+  return addUserStateExt(userState, "client_type", CLIENT_TYPE_TS_CLI);
 }
 
 // src/cli/context.ts
@@ -1766,7 +1802,8 @@ function getOrCreateSession(runtime) {
       app: config.app,
       apiKey: config.apiKey,
       publicKey: config.publicKey,
-      chainId: config.chain
+      chainId: config.chain,
+      clientId: crypto.randomUUID()
     };
     writeState(state);
   } else {
@@ -1791,21 +1828,23 @@ function getOrCreateSession(runtime) {
       state.chainId = config.chain;
       changed = true;
     }
+    if (!state.clientId) {
+      state.clientId = crypto.randomUUID();
+      changed = true;
+    }
     if (changed) writeState(state);
   }
   const session = new ClientSession(
     { baseUrl: state.baseUrl, apiKey: state.apiKey },
     {
       sessionId: state.sessionId,
+      clientId: state.clientId,
       app: state.app,
       apiKey: state.apiKey,
       publicKey: state.publicKey
     }
   );
-  const userState = buildCliUserState(state.publicKey, state.chainId);
-  if (userState) {
-    session.resolveUserState(userState);
-  }
+  session.resolveUserState(buildCliUserState(state.publicKey, state.chainId));
   return { session, state };
 }
 function createControlClient(runtime) {
@@ -1822,6 +1861,22 @@ async function applyModelSelection(session, state, model) {
   state.model = model;
   writeState(state);
 }
+async function ingestSecretsIfPresent(runtime, state, client) {
+  var _a3;
+  const secrets = runtime.config.secrets;
+  if (Object.keys(secrets).length === 0) return {};
+  if (!state.clientId) {
+    state.clientId = crypto.randomUUID();
+    writeState(state);
+  }
+  const response = await client.ingestSecrets(
+    state.clientId,
+    secrets
+  );
+  state.secretHandles = __spreadValues(__spreadValues({}, (_a3 = state.secretHandles) != null ? _a3 : {}), response.handles);
+  writeState(state);
+  return response.handles;
+}
 async function applyRequestedModelIfPresent(runtime, session, state) {
   const requestedModel = runtime.config.model;
   if (!requestedModel || requestedModel === state.model) {
@@ -1934,11 +1989,9 @@ async function chatCommand(runtime) {
   const verbose = runtime.parsed.flags["verbose"] === "true" || runtime.parsed.flags["v"] === "true";
   const { session, state } = getOrCreateSession(runtime);
   try {
+    await ingestSecretsIfPresent(runtime, state, session.client);
     await applyRequestedModelIfPresent(runtime, session, state);
-    const userState = buildCliUserState(state.publicKey, state.chainId);
-    if (userState) {
-      session.resolveUserState(userState);
-    }
+    session.resolveUserState(buildCliUserState(state.publicKey, state.chainId));
     const capturedRequests = [];
     let printedAgentCount = 0;
     const seenToolResults = /* @__PURE__ */ new Set();
@@ -2325,7 +2378,7 @@ async function statusCommand(runtime) {
   }
   const { session, state } = getOrCreateSession(runtime);
   try {
-    const apiState = await session.client.fetchState(state.sessionId);
+    const apiState = await session.client.fetchState(state.sessionId, void 0, state.clientId);
     console.log(
       JSON.stringify(
         {
@@ -2605,7 +2658,7 @@ async function logCommand(runtime) {
   }
   const { session, state } = getOrCreateSession(runtime);
   try {
-    const apiState = await session.client.fetchState(state.sessionId);
+    const apiState = await session.client.fetchState(state.sessionId, void 0, state.clientId);
     const messages = (_a3 = apiState.messages) != null ? _a3 : [];
     const pendingTxs = (_b = state.pendingTxs) != null ? _b : [];
     const signedTxs = (_c = state.signedTxs) != null ? _c : [];
@@ -2681,6 +2734,75 @@ function closeCommand(runtime) {
   console.log("Session closed");
 }
 
+// src/cli/commands/secrets.ts
+async function ingestSecretsCommand(runtime) {
+  const secretEntries = Object.entries(runtime.config.secrets);
+  if (secretEntries.length === 0) {
+    fatal("Usage: aomi --secret NAME=value [NAME=value ...]");
+  }
+  const { session, state } = getOrCreateSession(runtime);
+  try {
+    const handles = await ingestSecretsIfPresent(
+      runtime,
+      state,
+      session.client
+    );
+    const names = Object.keys(handles).sort();
+    console.log(
+      `Configured ${names.length} secret${names.length === 1 ? "" : "s"} for session ${state.sessionId}.`
+    );
+    for (const name of names) {
+      console.log(`${name}  ${handles[name]}`);
+    }
+    printDataFileLocation();
+  } finally {
+    session.close();
+  }
+}
+async function secretCommand(runtime) {
+  var _a3;
+  const subcommand = runtime.parsed.positional[0];
+  if (!subcommand || subcommand === "list") {
+    const state = readState();
+    if (!state) {
+      console.log("No active session");
+      printDataFileLocation();
+      return;
+    }
+    const secretHandles = (_a3 = state.secretHandles) != null ? _a3 : {};
+    const names = Object.keys(secretHandles).sort();
+    if (names.length === 0) {
+      console.log("No secrets configured.");
+      printDataFileLocation();
+      return;
+    }
+    for (const name of names) {
+      console.log(`${name}  ${secretHandles[name]}`);
+    }
+    printDataFileLocation();
+    return;
+  }
+  if (subcommand === "clear") {
+    const { session, state } = getOrCreateSession(runtime);
+    try {
+      if (!state.clientId) {
+        console.log("No secrets configured.");
+        printDataFileLocation();
+        return;
+      }
+      await session.client.clearSecrets(state.clientId);
+      state.secretHandles = {};
+      writeState(state);
+      console.log("Cleared all secrets for the active session.");
+      printDataFileLocation();
+    } finally {
+      session.close();
+    }
+    return;
+  }
+  fatal("Usage: aomi secret list\n       aomi secret clear");
+}
+
 // src/cli/commands/sessions.ts
 async function fetchRemoteSessionStats(record) {
   var _a3, _b;
@@ -2689,7 +2811,7 @@ async function fetchRemoteSessionStats(record) {
     apiKey: record.state.apiKey
   });
   try {
-    const apiState = await client.fetchState(record.sessionId);
+    const apiState = await client.fetchState(record.sessionId, void 0, record.state.clientId);
     const messages = (_a3 = apiState.messages) != null ? _a3 : [];
     return {
       topic: (_b = apiState.title) != null ? _b : "Untitled Session",
@@ -3391,10 +3513,7 @@ function createSessionFromState(state) {
       publicKey: state.publicKey
     }
   );
-  const userState = buildCliUserState(state.publicKey, state.chainId);
-  if (userState) {
-    session.resolveUserState(userState);
-  }
+  session.resolveUserState(buildCliUserState(state.publicKey, state.chainId));
   return session;
 }
 async function persistResolvedSignerState(session, state, address, chainId) {
@@ -3729,6 +3848,8 @@ Usage:
   aomi tx               List pending and signed transactions
   aomi sign <tx-id> [<tx-id> ...] [--eoa | --aa] [--aa-provider <name>] [--aa-mode <mode>]
                         Sign and submit a pending transaction
+  aomi secret list      List configured secrets for the active session
+  aomi secret clear     Clear all secrets for the active session
   aomi status           Show current session state
   aomi events           List system events
   aomi close            Close the current session
@@ -3742,6 +3863,7 @@ Options:
   --public-key <addr>   Wallet address (so the agent knows your wallet)
   --private-key <key>   Hex private key for signing
   --rpc-url <url>       RPC URL for transaction submission
+  --secret NAME=value   Ingest a secret (repeatable, e.g. --secret X_API_KEY=abc)
   --verbose, -v         Show tool calls and streaming output (for chat)
   --version, -V         Print the installed CLI version
 
@@ -3777,8 +3899,9 @@ Environment (overridden by flags):
 `.trim());
 }
 async function main(runtime) {
-  var _a3, _b;
-  const command = (_b = (_a3 = runtime.parsed.command) != null ? _a3 : runtime.parsed.flags["version"] || runtime.parsed.flags["V"] ? "version" : void 0) != null ? _b : runtime.parsed.flags["help"] || runtime.parsed.flags["h"] ? "help" : void 0;
+  var _a3, _b, _c;
+  const hasSecrets = Object.keys(runtime.parsed.secrets).length > 0;
+  const command = (_c = (_b = (_a3 = runtime.parsed.command) != null ? _a3 : hasSecrets ? "ingest-secrets" : void 0) != null ? _b : runtime.parsed.flags["version"] || runtime.parsed.flags["V"] ? "version" : void 0) != null ? _c : runtime.parsed.flags["help"] || runtime.parsed.flags["h"] ? "help" : void 0;
   switch (command) {
     case "chat":
       await chatCommand(runtime);
@@ -3810,6 +3933,12 @@ async function main(runtime) {
     case "events":
       await eventsCommand(runtime);
       break;
+    case "secret":
+      await secretCommand(runtime);
+      break;
+    case "ingest-secrets":
+      await ingestSecretsCommand(runtime);
+      break;
     case "close":
       closeCommand(runtime);
       break;

package/dist/index.cjs

@@ -48,10 +48,13 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   AomiClient: () => AomiClient,
+  CLIENT_TYPE_TS_CLI: () => CLIENT_TYPE_TS_CLI,
+  CLIENT_TYPE_WEB_UI: () => CLIENT_TYPE_WEB_UI,
   DEFAULT_AA_CONFIG: () => DEFAULT_AA_CONFIG,
   Session: () => ClientSession,
   TypedEventEmitter: () => TypedEventEmitter,
   adaptSmartAccount: () => adaptSmartAccount,
+  addUserStateExt: () => addUserStateExt,
   buildAAExecutionPlan: () => buildAAExecutionPlan,
   createAAProviderState: () => createAAProviderState,
   createAlchemyAAProvider: () => createAlchemyAAProvider,
@@ -615,6 +618,17 @@ var AomiClient = class {
 };
 
 // src/types.ts
+var CLIENT_TYPE_TS_CLI = "ts_cli";
+var CLIENT_TYPE_WEB_UI = "web_ui";
+function addUserStateExt(userState, key, value) {
+  const currentExt = userState["ext"];
+  const extRecord = typeof currentExt === "object" && currentExt !== null && !Array.isArray(currentExt) ? currentExt : {};
+  return __spreadProps(__spreadValues({}, userState), {
+    ext: __spreadProps(__spreadValues({}, extRecord), {
+      [key]: value
+    })
+  });
+}
 function isInlineCall(event) {
   return "InlineCall" in event;
 }
@@ -857,7 +871,7 @@ function isSubsetMatch(expected, actual) {
 }
 var ClientSession = class extends TypedEventEmitter {
   constructor(clientOrOptions, sessionOptions) {
-    var _a, _b, _c;
+    var _a, _b, _c, _d, _e;
     super();
     // Internal state
     this.pollTimer = null;
@@ -874,8 +888,9 @@ var ClientSession = class extends TypedEventEmitter {
     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;
-    this.pollIntervalMs = (_c = sessionOptions == null ? void 0 : sessionOptions.pollIntervalMs) != null ? _c : 500;
+    this.userState = (sessionOptions == null ? void 0 : sessionOptions.clientType) ? addUserStateExt((_c = sessionOptions == null ? void 0 : sessionOptions.userState) != null ? _c : {}, "client_type", sessionOptions.clientType) : sessionOptions == null ? void 0 : sessionOptions.userState;
+    this.clientId = (_d = sessionOptions == null ? void 0 : sessionOptions.clientId) != null ? _d : crypto.randomUUID();
+    this.pollIntervalMs = (_e = sessionOptions == null ? void 0 : sessionOptions.pollIntervalMs) != null ? _e : 500;
     this.logger = sessionOptions == null ? void 0 : sessionOptions.logger;
     this.unsubscribeSSE = this.client.subscribeSSE(
       this.sessionId,
@@ -900,7 +915,8 @@ var ClientSession = class extends TypedEventEmitter {
       app: this.app,
       publicKey: this.publicKey,
       apiKey: this.apiKey,
-      userState: this.userState
+      userState: this.userState,
+      clientId: this.clientId
     });
     this.assertUserStateAligned(response.user_state);
     this.applyState(response);
@@ -924,7 +940,8 @@ var ClientSession = class extends TypedEventEmitter {
       app: this.app,
       publicKey: this.publicKey,
       apiKey: this.apiKey,
-      userState: this.userState
+      userState: this.userState,
+      clientId: this.clientId
     });
     this.assertUserStateAligned(response.user_state);
     this.applyState(response);
@@ -1046,6 +1063,10 @@ var ClientSession = class extends TypedEventEmitter {
       this.publicKey = address;
     }
   }
+  setClientType(clientType) {
+    var _a;
+    this.resolveUserState(addUserStateExt((_a = this.userState) != null ? _a : {}, "client_type", clientType));
+  }
   addExtValue(key, value) {
     var _a;
     const current = (_a = this.userState) != null ? _a : {};
@@ -1075,7 +1096,7 @@ var ClientSession = class extends TypedEventEmitter {
   }
   async syncUserState() {
     this.assertOpen();
-    const state = await this.client.fetchState(this.sessionId, this.userState);
+    const state = await this.client.fetchState(this.sessionId, this.userState, this.clientId);
     this.assertUserStateAligned(state.user_state);
     this.applyState(state);
     return state;
@@ -1105,7 +1126,8 @@ var ClientSession = class extends TypedEventEmitter {
     try {
       const state = await this.client.fetchState(
         this.sessionId,
-        this.userState
+        this.userState,
+        this.clientId
       );
       if (!this.pollTimer) return;
       this.assertUserStateAligned(state.user_state);
@@ -2029,10 +2051,13 @@ async function createPimlicoAAState(options) {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AomiClient,
+  CLIENT_TYPE_TS_CLI,
+  CLIENT_TYPE_WEB_UI,
   DEFAULT_AA_CONFIG,
   Session,
   TypedEventEmitter,
   adaptSmartAccount,
+  addUserStateExt,
   buildAAExecutionPlan,
   createAAProviderState,
   createAlchemyAAProvider,