@aomi-labs/client 0.1.15 → 0.1.17

package/README.md

@@ -126,6 +126,7 @@ npx @aomi-labs/client chat "swap 1 ETH" --verbose        # stream tool calls + r
 npx @aomi-labs/client app list                           # list available apps
 npx @aomi-labs/client model list                         # list available models
 npx @aomi-labs/client model set claude-sonnet-4          # switch the current session model
+npx @aomi-labs/client session new                        # create a fresh active session
 npx @aomi-labs/client secret list                        # list configured secret handles
 npx @aomi-labs/client --secret ALCHEMY_API_KEY=...       # ingest a secret for the active session
 npx @aomi-labs/client log                                # show full conversation history
@@ -149,6 +150,31 @@ npx @aomi-labs/client chat "send 0 ETH to myself" \
 The address is persisted in the state file, so subsequent commands in the same
 session don't need it again.
 
+### Chain selection
+
+Use `--chain <id>` for the current command when the task is chain-specific:
+
+```bash
+$ npx @aomi-labs/client chat "swap 1 POL for USDC on Polygon" --chain 137
+```
+
+Use `AOMI_CHAIN_ID` when several consecutive commands should share the same
+chain context.
+
+### Fresh sessions
+
+Use `--new-session` when you want a command to start a fresh backend/local
+session instead of reusing the currently active one:
+
+```bash
+$ npx @aomi-labs/client chat "show my balances" --new-session
+$ npx @aomi-labs/client --secret ALCHEMY_API_KEY=... --new-session
+$ npx @aomi-labs/client session new
+```
+
+This is useful when starting a new operator flow or a new external chat thread
+and you do not want stale session state to bleed into the next run.
+
 ### Model selection
 
 The CLI can discover and switch backend models for the active session:

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.15",
+  version: "0.1.17",
   description: "Platform-agnostic TypeScript client for the Aomi backend API",
   type: "module",
   main: "./dist/index.cjs",
@@ -139,6 +139,12 @@ function parseSecret(value, secrets) {
     secrets[value.slice(0, eqIdx)] = value.slice(eqIdx + 1);
   }
 }
+function normalizePrivateKey(value) {
+  if (value === void 0) return void 0;
+  const trimmed = value.trim();
+  if (!trimmed) return void 0;
+  return trimmed.startsWith("0x") ? trimmed : `0x${trimmed}`;
+}
 function parseArgs(argv) {
   const raw = argv.slice(2);
   const command = raw[0] && !raw[0].startsWith("-") ? raw[0] : void 0;
@@ -200,8 +206,11 @@ function getConfig(parsed) {
     apiKey: (_e = parsed.flags["api-key"]) != null ? _e : process.env.AOMI_API_KEY,
     app: (_g = (_f = parsed.flags["app"]) != null ? _f : process.env.AOMI_APP) != null ? _g : "default",
     model: (_h = parsed.flags["model"]) != null ? _h : process.env.AOMI_MODEL,
+    freshSession: parsed.flags["new-session"] === "true",
     publicKey: (_i = parsed.flags["public-key"]) != null ? _i : process.env.AOMI_PUBLIC_KEY,
-    privateKey: (_j = parsed.flags["private-key"]) != null ? _j : process.env.PRIVATE_KEY,
+    privateKey: normalizePrivateKey(
+      (_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,
@@ -278,6 +287,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,
@@ -286,6 +296,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()
@@ -1148,6 +1159,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() {
@@ -1220,20 +1255,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;
@@ -1391,7 +1412,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;
@@ -1408,8 +1429,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,
@@ -1434,7 +1456,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);
@@ -1458,7 +1481,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);
@@ -1580,6 +1604,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 : {};
@@ -1609,7 +1637,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;
@@ -1639,7 +1667,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);
@@ -1760,9 +1789,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;
@@ -1771,23 +1797,33 @@ 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
-function getOrCreateSession(runtime) {
+function buildSessionState(runtime) {
   const { config } = runtime;
-  let state = readState();
+  return {
+    sessionId: crypto.randomUUID(),
+    baseUrl: config.baseUrl,
+    app: config.app,
+    model: config.model,
+    apiKey: config.apiKey,
+    publicKey: config.publicKey,
+    chainId: config.chain,
+    clientId: crypto.randomUUID()
+  };
+}
+function createFreshSessionState(runtime) {
+  const state = buildSessionState(runtime);
+  writeState(state);
+  return state;
+}
+function getOrCreateSession(runtime, options = {}) {
+  const { config } = runtime;
+  let state = options.fresh || config.freshSession ? createFreshSessionState(runtime) : readState();
   if (!state) {
-    state = {
-      sessionId: crypto.randomUUID(),
-      baseUrl: config.baseUrl,
-      app: config.app,
-      apiKey: config.apiKey,
-      publicKey: config.publicKey,
-      chainId: config.chain
-    };
-    writeState(state);
+    state = createFreshSessionState(runtime);
   } else {
     let changed = false;
     if (config.baseUrl !== state.baseUrl) {
@@ -1810,21 +1846,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) {
@@ -1967,13 +2005,13 @@ async function chatCommand(runtime) {
     fatal("Usage: aomi chat <message>");
   }
   const verbose = runtime.parsed.flags["verbose"] === "true" || runtime.parsed.flags["v"] === "true";
-  const { session, state } = getOrCreateSession(runtime);
+  const { session, state } = getOrCreateSession(runtime, {
+    fresh: runtime.config.freshSession
+  });
   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();
@@ -2360,7 +2398,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(
         {
@@ -2640,7 +2678,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 : [];
@@ -2722,7 +2760,9 @@ async function ingestSecretsCommand(runtime) {
   if (secretEntries.length === 0) {
     fatal("Usage: aomi --secret NAME=value [NAME=value ...]");
   }
-  const { session, state } = getOrCreateSession(runtime);
+  const { session, state } = getOrCreateSession(runtime, {
+    fresh: runtime.config.freshSession
+  });
   try {
     const handles = await ingestSecretsIfPresent(
       runtime,
@@ -2767,6 +2807,11 @@ async function secretCommand(runtime) {
   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);
@@ -2788,7 +2833,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",
@@ -2850,6 +2895,12 @@ async function sessionsCommand(_runtime) {
 async function sessionCommand(runtime) {
   const subcommand = runtime.parsed.positional[0];
   const selector = runtime.parsed.positional[1];
+  if (subcommand === "new") {
+    const state = createFreshSessionState(runtime);
+    console.log(`Active session set to ${state.sessionId} (new).`);
+    printDataFileLocation();
+    return;
+  }
   if (subcommand === "resume") {
     if (!selector) {
       fatal("Usage: aomi session resume <session-id|session-N|N>");
@@ -2885,7 +2936,7 @@ async function sessionCommand(runtime) {
     return;
   }
   fatal(
-    "Usage: aomi session list\n       aomi session resume <session-id|session-N|N>\n       aomi session delete <session-id|session-N|N>"
+    "Usage: aomi session list\n       aomi session new\n       aomi session resume <session-id|session-N|N>\n       aomi session delete <session-id|session-N|N>"
   );
 }
 
@@ -3490,10 +3541,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) {
@@ -3820,6 +3868,7 @@ Usage:
   aomi model set <rig>  Set the active model for the current session
   aomi chain list       List supported chains
   aomi session list     List local sessions with metadata
+  aomi session new      Start a fresh local/backend session and make it active
   aomi session resume <id>
                         Resume a local session (session-id or session-N)
   aomi session delete <id>
@@ -3840,6 +3889,8 @@ Options:
   --api-key <key>       API key for non-default apps
   --app <name>          App (default: "default")
   --model <rig>         Set the active model for this session
+  --new-session         Create a fresh active session for this command
+  --chain <id>          Active chain for chat/session context
   --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
@@ -3867,6 +3918,7 @@ Environment (overridden by flags):
   AOMI_API_KEY          API key
   AOMI_APP              App
   AOMI_MODEL            Model rig
+  AOMI_CHAIN_ID         Active chain for chat/session context
   AOMI_PUBLIC_KEY       Wallet address
   AOMI_AA_PROVIDER      AA provider: alchemy | pimlico
   AOMI_AA_MODE          AA mode: 4337 | 7702

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,