@atbash/sdk 0.3.23 → 0.3.25

package/README.md

@@ -23,10 +23,12 @@ const agent = loadAgent(process.env.ATBASH_AGENT_PRIVKEY!);
 // 2. Submit an action for judgment, before executing it.
 //    The SDK signs the transaction locally and sends it to the judge API.
 //    Private key stays on your machine — never sent over HTTP.
+//    Pass orgName so the SDK auto-resolves the correct chain (public or private).
 const result = await judgeAction(
   "Transfer $50,000 to external wallet 0xabc",
   "Outbound AML check — new recipient, over threshold",
   agent,
+  { orgName: "my_org" },
 );
 
 // 3. Enforce the verdict
@@ -39,19 +41,19 @@ switch (result.verdict) {
     console.log("Held for review:", result.tool_call_id);
     break;
   case "BLOCK":
-    // Refused — agent is jailed in Enforcement tier
+    // Refused — agent is auto-jailed
     throw new Error(`Blocked: ${result.reason}`);
 }
 ```
 
-Before this works, the agent must be onboarded at [atbash.ai](https://atbash.ai/) — assigned to an org, with a policy pack attached, and the org tier set to Audit+ or Enforcement.
+Before this works, the agent must be onboarded at [atbash.ai](https://atbash.ai/) — assigned to an org with an active subscription and a policy pack attached.
 
 ### How it works
 
 `judgeAction()` performs a two-step flow:
 
 1. **Sign locally** — signs the transaction using the agent's private key. The key never leaves your machine.
-2. **Request verdict** — sends the signed transaction, `tool_call_id`, and `agent_pubkey` to the Atbash judge API. The server broadcasts it to the Chromia blockchain and returns a verdict.
+2. **Request verdict** — sends the signed payload to the Atbash judge API, which records it on the Chromia blockchain and returns a verdict.
 
 
 ### Don't have an agent yet?
@@ -86,9 +88,9 @@ Every `judgeAction` call returns one of three verdicts:
 |---------|---------|-------------------------|
 | `ALLOW` | Action is within policy | Proceed with execution |
 | `HOLD` | Requires operator review | Pause — poll `getJudgmentStatus` until resolved |
-| `BLOCK` | Violates a red line | Abort — agent is jailed in Enforcement tier |
+| `BLOCK` | Violates a red line | Abort — agent is auto-jailed |
 
-> **NB:** If your org is on the **Audit** tier, the judge returns `"No verdict"` — actions are logged on-chain for the audit trail but not evaluated by an AI provider. Upgrade to **Audit+** or **Enforcement** at [atbash.ai/risk-engine/settings](https://atbash.ai/risk-engine/settings) for active verdicts.
+> **NB:** If your org has **no active subscription**, the judge returns `"No verdict"` — actions are logged on-chain for the audit trail but not evaluated. Assign a subscription plan at [atbash.ai/risk-engine/settings](https://atbash.ai/risk-engine/settings) for active verdicts. All subscription plans (including Free) get full enforcement.
 
 ## API
 
@@ -118,17 +120,11 @@ interface JudgeOptions {
   model?: string;             // Model override (e.g. "gpt-4o-mini")
   toolName?: string;          // Tool name for audit trail
   toolArgsJson?: string;      // Tool arguments JSON for audit trail
-  chainOpts?: ChainOpts;      // Override Chromia chain connection
-}
-
-interface ChainOpts {
-  nodeUrls?: string[];        // Chromia node URLs (uses the default  nodeurls)
-  blockchainRid?: string;     // Blockchain RID (uses the default chromia rid)
+  orgName?: string;           // Org name — SDK auto-resolves the correct chain
 }
 
 interface JudgeResult {
   verdict: string;       // "ALLOW", "HOLD", or "BLOCK"
-  action_type: string;   // "allow", "hold_for_user_confirm", or "block"
   reason: string;        // Human-readable explanation
   confidence: number;    // 0–1
   provider: string;      // Which provider evaluated the action
@@ -138,20 +134,6 @@ interface JudgeResult {
 }
 ```
 
-### Log tool call (low-level)
-
-```ts
-logToolCall(
-  action: string,
-  context: string,
-  auth: AgentAuth,
-  chainOpts?: ChainOpts,
-  extra?: { toolName?: string; toolArgsJson?: string },
-): Promise<LogToolCallResult>
-```
-
-Sign the transaction locally. Returns `{ success, toolCallId, signedHex?, error? }`. Use this if you need to separate the signing step from the verdict request.
-
 ### Poll judgment status
 
 ```ts
@@ -191,7 +173,6 @@ Functions that sign transactions and write to the Chromia blockchain.
 | Function | Use case |
 |----------|----------|
 | `judgeAction(action, context, auth, opts?)` | Sign locally + request a verdict from the judge API |
-| `logToolCall(action, context, auth, ...)` | Sign the transaction locally without requesting a verdict |
 
 ### Queries
 
@@ -204,7 +185,7 @@ Functions that sign transactions and write to the Chromia blockchain.
 | `getAgentToolCalls(pubkey, maxCount)` | List tool calls for a specific agent |
 | `getToolCallCount()` | Get total number of tool calls on-chain |
 | `getToolCallFull(toolCallId)` | Get full details of a single tool call (verdict, context, timing) |
-| `getOrgTierInfo(orgName)` | Check an org's tier and whether verdicts are enabled |
+| `getOrgSubscription(orgName)` | Check an org's subscription plan, network, and active status |
 | `getAgentDetail(pubkey)` | Get agent metadata (org, status, creation date) |
 | `getAgentPolicy(pubkey)` | Check agent's policy pack and jail status |
 | `getPendingHeldActions(orgName, maxCount)` | List actions waiting for operator approval |
@@ -242,6 +223,7 @@ saveUserConfig({
 // Then use resolve() anywhere
 const agent = loadAgent(resolve("agentKey"));
 const result = await judgeAction("Transfer $500", "finance", agent, {
+  orgName: resolve("orgName"),
   provider: resolve("provider"), // omit to use the on-chain ATBASH judge
 });
 ```
@@ -264,7 +246,7 @@ Config file location: `~/.config/atbash/config.json`
 | `provider` | `ATBASH_PROVIDER` |
 | `providerModel` | `ATBASH_PROVIDER_MODEL` |
 
-> **Advanced:** The SDK connects to the default Atbash Chromia chain. To use a different chain, pass `chainOpts` with custom `nodeUrls` and `blockchainRid` in `JudgeOptions`.
+> **Chain routing:** When you pass `orgName`, the SDK automatically connects to the correct chain for your org's subscription plan. You don't need to configure chain details manually.
 
 ## Secret redaction
 
@@ -285,6 +267,32 @@ Common `kinds`:
 
 Redaction is silent at the consumer level — the SDK's caller still has the original arguments. Only what's sent to the judge (and persisted on chain via the verdict log) is scrubbed.
 
+## High-level client
+
+For framework integrations, `createAtbashClient` wraps key loading, secret redaction, and verdict handling into a single `auditToolCall` method:
+
+```ts
+import { createAtbashClient } from "@atbash/sdk";
+
+const atbash = createAtbashClient({
+  orgName: "my_org",
+  keyPair: { privKey: process.env.ATBASH_AGENT_KEY!, pubKey: "" },
+  failClosed: true,  // block on errors (default: true)
+});
+
+const decision = await atbash.auditToolCall({
+  toolName: "send_email",
+  args: { to: "user@example.com", subject: "Reset" },
+  context: "Password reset flow",
+});
+
+if (!decision.allow) {
+  console.log(`${decision.verdict}: ${decision.reason}`);
+}
+```
+
+The client auto-resolves the correct chain from the org's subscription on the first call and caches the result. Secret redaction runs automatically before signing.
+
 ## Integration patterns
 
 ### Pre-execution gate
@@ -334,8 +342,8 @@ API error 404: {"error":"Agent not registered..."}
 |---|---|---|
 | `API error 404: Agent not registered` | Agent not onboarded | [atbash.ai/risk-engine/agents](https://atbash.ai/risk-engine/agents) |
 | `API error 400: Agent has no policy` | No policy attached to agent | [atbash.ai/risk-engine/agents](https://atbash.ai/risk-engine/agents) |
-| `Agent is jailed` | BLOCK verdict triggered auto-jail (Enforcement tier) | [atbash.ai/risk-engine/agents](https://atbash.ai/risk-engine/agents) |
-| `Org tier does not support verdicts` | Org is on Audit tier | [atbash.ai/risk-engine/settings](https://atbash.ai/risk-engine/settings) |
+| `Agent is jailed` | BLOCK verdict triggered auto-jail | [atbash.ai/risk-engine/agents](https://atbash.ai/risk-engine/agents) |
+| `Verdicts are disabled` | Org has no active subscription | [atbash.ai/risk-engine/settings](https://atbash.ai/risk-engine/settings) |
 | `API error 400: action is required` | Empty action string | Fix caller |
 | `API error 502: Incorrect API key` | Invalid provider API key | Check saved key at [atbash.ai/risk-engine/settings](https://atbash.ai/risk-engine/settings) |
 

package/dist/index.cjs

@@ -47,7 +47,7 @@ __export(index_exports, {
   getConfigPath: () => getConfigPath,
   getHeldActionReviews: () => getHeldActionReviews,
   getJudgmentStatus: () => getJudgmentStatus,
-  getOrgTierInfo: () => getOrgTierInfo,
+  getOrgSubscription: () => getOrgSubscription,
   getOrgToolCalls: () => getOrgToolCalls,
   getPendingHeldActions: () => getPendingHeldActions,
   getSafetyStats: () => getSafetyStats,
@@ -59,7 +59,6 @@ __export(index_exports, {
   loadAgent: () => loadAgent,
   loadAgentFromFile: () => loadAgentFromFile,
   loadUserConfig: () => loadUserConfig,
-  logToolCall: () => logToolCall,
   normalizeForMatching: () => normalizeForMatching,
   resolve: () => resolve,
   resolveKeyPath: () => resolveKeyPath,
@@ -197,7 +196,37 @@ var DEFAULT_CHROMIA_NODE_URLS = [
   "https://node1.testnet.chromia.com:7740",
   "https://node3.testnet.chromia.com:7740"
 ];
-var DEFAULT_BLOCKCHAIN_RID = "619E62A5FD01BDDBFF0EF7A3821B5C88145B0E64A870400EAB4065C6A2779981";
+var DEFAULT_BLOCKCHAIN_RID = "B91106947F1EAED7B5D789C7D35755330A8A7DD7CB990D59366114EFFB79ED10";
+var DEFAULT_PRIVATE_NODE_URLS = [
+  "https://node0-pvn-testnet.dynamic.chromia.dev"
+];
+var DEFAULT_PRIVATE_BLOCKCHAIN_RID = "431AE6A5695D157D74194A61AB4D0B6A98C99AFEEF186FC885CDA4A3BAAB800E";
+var PUBLIC_CHAIN = {
+  network: "public",
+  blockchainRid: DEFAULT_BLOCKCHAIN_RID,
+  nodeUrls: DEFAULT_CHROMIA_NODE_URLS
+};
+var PRIVATE_CHAIN = {
+  network: "private",
+  blockchainRid: DEFAULT_PRIVATE_BLOCKCHAIN_RID,
+  nodeUrls: DEFAULT_PRIVATE_NODE_URLS
+};
+function chainForNetwork(network) {
+  return network === "private" ? PRIVATE_CHAIN : PUBLIC_CHAIN;
+}
+function resolveChainOpts(chainOpts) {
+  if (chainOpts?.network) {
+    const chain = chainForNetwork(chainOpts.network);
+    return {
+      nodeUrls: chainOpts.nodeUrls ?? chain.nodeUrls,
+      blockchainRid: chainOpts.blockchainRid ?? chain.blockchainRid
+    };
+  }
+  return {
+    nodeUrls: chainOpts?.nodeUrls ?? DEFAULT_CHROMIA_NODE_URLS,
+    blockchainRid: chainOpts?.blockchainRid ?? DEFAULT_BLOCKCHAIN_RID
+  };
+}
 function isValidPrivateKey(hex) {
   return /^[0-9a-fA-F]{64}$/.test(hex);
 }
@@ -236,14 +265,30 @@ function toPubkeyHex(val) {
 function baseUrl(opts) {
   return opts?.endpoint || DEFAULT_ENDPOINT;
 }
+var AUTH_BEARER_REFRESH_MS = 4 * 60 * 1e3;
+var bearerCache = /* @__PURE__ */ new Map();
+async function getOrCreateAuthBearer(auth) {
+  const now = Date.now();
+  const cached = bearerCache.get(auth.pubkey);
+  if (cached && now - cached.issuedAt < AUTH_BEARER_REFRESH_MS) {
+    return cached.hex;
+  }
+  const nonce = `auth-${now.toString(36)}-${(0, import_crypto.randomBytes)(4).toString("hex")}`;
+  const hex = await buildSignedTx(
+    "log_tool_call",
+    [nonce, `auth:${now}`, "", "auth-bearer", ""],
+    auth
+  );
+  bearerCache.set(auth.pubkey, { hex, issuedAt: now });
+  return hex;
+}
 function generateToolCallId() {
   const ts = Date.now();
   const rand = (0, import_crypto.randomBytes)(4).toString("hex");
   return `tc-${ts}-${rand}`;
 }
 async function buildSignedTx(opName, args, auth, chainOpts) {
-  const nodeUrls = chainOpts?.nodeUrls ?? DEFAULT_CHROMIA_NODE_URLS;
-  const blockchainRid = chainOpts?.blockchainRid ?? DEFAULT_BLOCKCHAIN_RID;
+  const { nodeUrls, blockchainRid } = resolveChainOpts(chainOpts);
   const client = await createClient({ nodeUrlPool: nodeUrls, blockchainRid });
   const privKeyBuf = Buffer.from(auth.privkey, "hex");
   const keyPair = encryption2.makeKeyPair(privKeyBuf);
@@ -260,11 +305,13 @@ async function buildSignedTx(opName, args, auth, chainOpts) {
   );
   return Buffer.from(signed).toString("hex");
 }
-async function checkAgentExists(pubkey, opts) {
+async function _checkAgentExists(pubkey, opts, chainOpts) {
   const start = performance.now();
   recordCall("checkAgentExists", void 0, pubkey);
   try {
-    const url = `${baseUrl(opts)}/api/ai/exists?pubkey=${encodeURIComponent(pubkey)}`;
+    const network = chainOpts?.network;
+    let url = `${baseUrl(opts)}/api/ai/exists?pubkey=${encodeURIComponent(pubkey)}`;
+    if (network) url += `&network=${encodeURIComponent(network)}`;
     const data = await getJson(url, opts);
     recordDuration("checkAgentExists", performance.now() - start, "success");
     return Boolean(data.registered);
@@ -273,10 +320,13 @@ async function checkAgentExists(pubkey, opts) {
     throw err;
   }
 }
+async function checkAgentExists(pubkey, opts) {
+  return _checkAgentExists(pubkey, opts);
+}
 async function logToolCall(action, context, auth, chainOpts, extra, clientOpts) {
   const start = performance.now();
   recordCall("logToolCall", void 0, auth.pubkey);
-  const exists = await checkAgentExists(auth.pubkey, clientOpts);
+  const exists = await _checkAgentExists(auth.pubkey, clientOpts, chainOpts);
   if (!exists) {
     recordDuration("logToolCall", performance.now() - start, "error");
     return {
@@ -345,9 +395,13 @@ function enrichError(status, body, statusText, opts) {
   return new Error(message);
 }
 async function postJson(url, body, opts) {
+  const headers = { "Content-Type": "application/json" };
+  if (opts?.auth) {
+    headers["Authorization"] = `Bearer ${await getOrCreateAuthBearer(opts.auth)}`;
+  }
   const resp = await fetch(url, {
     method: "POST",
-    headers: { "Content-Type": "application/json" },
+    headers,
     body: JSON.stringify(body),
     signal: opts?.timeout ? AbortSignal.timeout(opts.timeout) : void 0
   });
@@ -359,9 +413,13 @@ async function postJson(url, body, opts) {
   return ct.includes("application/json") ? resp.json() : {};
 }
 async function getJson(url, opts) {
+  const headers = { Accept: "application/json" };
+  if (opts?.auth) {
+    headers["Authorization"] = `Bearer ${await getOrCreateAuthBearer(opts.auth)}`;
+  }
   const resp = await fetch(url, {
     method: "GET",
-    headers: { Accept: "application/json" },
+    headers,
     signal: opts?.timeout ? AbortSignal.timeout(opts.timeout) : void 0
   });
   if (!resp.ok) {
@@ -402,11 +460,16 @@ async function judgeAction(action, context = "", auth, opts) {
     throw new Error("action is required and cannot be empty.");
   }
   try {
+    let chainOpts = opts?.chainOpts;
+    if (opts?.orgName && !chainOpts?.blockchainRid) {
+      const resolved = await resolveChainForOrg(opts.orgName, opts);
+      chainOpts = { ...chainOpts, network: resolved.network };
+    }
     const logResult = await logToolCall(
       action,
       context,
       auth,
-      opts?.chainOpts,
+      chainOpts,
       { toolName: opts?.toolName, toolArgsJson: opts?.toolArgsJson },
       opts
     );
@@ -420,7 +483,7 @@ async function judgeAction(action, context = "", auth, opts) {
         "judge_action",
         [judgmentId, action, context || "", ""],
         auth,
-        opts?.chainOpts
+        chainOpts
       );
     }
     const url = `${baseUrl(opts)}/api/v1/judge`;
@@ -565,21 +628,52 @@ async function getToolCallFull(toolCallId, opts) {
     throw err;
   }
 }
-async function getOrgTierInfo(orgName, opts) {
+function coerceOrgSubscription(row, orgName) {
+  if (!row || typeof row !== "object") return null;
+  const r = row;
+  return {
+    org_name: String(r.org_name ?? orgName),
+    subscription_name: String(r.subscription_name ?? ""),
+    agent_number: Number(r.agent_number ?? 0),
+    is_private_blockchain: Boolean(r.is_private_blockchain),
+    monthly_price: Number(r.monthly_price ?? 0),
+    yearly_price: Number(r.yearly_price ?? 0),
+    duration_months: Number(r.duration_months ?? 0),
+    assigned_at: Number(r.assigned_at ?? 0),
+    expires_at: Number(r.expires_at ?? 0),
+    is_active: Boolean(r.is_active)
+  };
+}
+async function getOrgSubscription(orgName, opts) {
   const start = performance.now();
-  recordCall("getOrgTierInfo");
+  recordCall("getOrgSubscription");
   try {
     const result = await getJson(
-      riskEngineUrl("org-tier-info", { org: orgName }, opts),
+      riskEngineUrl("org-subscription", { org: orgName }, opts),
       opts
     );
-    recordDuration("getOrgTierInfo", performance.now() - start, "success");
-    return result;
+    recordDuration("getOrgSubscription", performance.now() - start, "success");
+    return coerceOrgSubscription(result, orgName);
   } catch (err) {
-    recordDuration("getOrgTierInfo", performance.now() - start, "error");
+    recordDuration("getOrgSubscription", performance.now() - start, "error");
     throw err;
   }
 }
+var _chainCache = /* @__PURE__ */ new Map();
+async function resolveChainForOrg(orgName, opts) {
+  const cached = _chainCache.get(orgName);
+  if (cached) return cached;
+  try {
+    const sub = await getOrgSubscription(orgName, opts);
+    if (sub?.is_private_blockchain) {
+      _chainCache.set(orgName, PRIVATE_CHAIN);
+      return PRIVATE_CHAIN;
+    }
+  } catch {
+  }
+  _chainCache.set(orgName, PUBLIC_CHAIN);
+  return PUBLIC_CHAIN;
+}
 async function getPendingHeldActions(orgName, maxCount, opts) {
   const start = performance.now();
   recordCall("getPendingHeldActions");
@@ -835,6 +929,8 @@ function createAtbashClient(config = {}) {
   const validated = validateJudgeEndpoint(config.judge);
   const failClosed = config.failClosed !== false;
   const logger = config.logger ?? {};
+  const orgName = config.orgName;
+  let resolvedChain = null;
   const inlineKeyPair = config.keyPair;
   const keyPath = inlineKeyPair ? null : config.keyPath;
   if (validated.url !== DEFAULT_ENDPOINT) {
@@ -884,12 +980,23 @@ function createAtbashClient(config = {}) {
         });
       }
       try {
+        if (!resolvedChain && orgName) {
+          resolvedChain = await resolveChainForOrg(orgName, { endpoint: validated.url });
+          config.nodeUrls = resolvedChain.nodeUrls;
+          config.blockchainRid = resolvedChain.blockchainRid;
+          logger.info?.("[atbash] resolved network from subscription", {
+            org: orgName,
+            network: resolvedChain.network,
+            brid: resolvedChain.blockchainRid
+          });
+        }
         logger.info?.("[atbash] judge API called", { tool: toolName });
         const result = await judgeAction(actionText, contextText, agent, {
           endpoint: validated.url,
           verifyPubKey: validated.verifyPubKey ?? void 0,
           toolName,
           toolArgsJson: argsJson,
+          orgName,
           chainOpts: {
             nodeUrls: config.nodeUrls,
             blockchainRid: config.blockchainRid
@@ -977,6 +1084,7 @@ var ENV_MAP = {
   orgName: "ATBASH_ORG_NAME",
   judgeEndpoint: "ATBASH_ENDPOINT",
   blockchainRid: "ATBASH_BLOCKCHAIN_RID",
+  network: "ATBASH_NETWORK",
   provider: "ATBASH_PROVIDER",
   providerModel: "ATBASH_PROVIDER_MODEL"
 };
@@ -1482,7 +1590,7 @@ function deduplicateAnomalies(anomalies) {
   getConfigPath,
   getHeldActionReviews,
   getJudgmentStatus,
-  getOrgTierInfo,
+  getOrgSubscription,
   getOrgToolCalls,
   getPendingHeldActions,
   getSafetyStats,
@@ -1494,7 +1602,6 @@ function deduplicateAnomalies(anomalies) {
   loadAgent,
   loadAgentFromFile,
   loadUserConfig,
-  logToolCall,
   normalizeForMatching,
   resolve,
   resolveKeyPath,

package/dist/index.d.cts

@@ -1,11 +1,24 @@
 type Verdict = "ALLOW" | "HOLD" | "BLOCK" | "No verdict";
 type Provider = "openai" | "google" | "microsoft" | "custom" | (string & {});
-type Tier = "audit" | "audit_plus" | "enforcement" | (string & {});
 type ActionType = "allow" | "hold_for_user_confirm" | "block" | (string & {});
 type PubkeyValue = string | Buffer | {
     data: number[];
 };
 type JudgmentStatusState = "pending" | "answered" | "error";
+interface Subscription {
+    subscription_name: string;
+    agent_number: number;
+    is_private_blockchain: boolean;
+    monthly_price: number;
+    yearly_price: number;
+}
+interface OrgSubscription extends Subscription {
+    org_name: string;
+    duration_months: number;
+    assigned_at: number;
+    expires_at: number;
+    is_active: boolean;
+}
 interface AgentAuth {
     pubkey: string;
     privkey: string;
@@ -13,17 +26,12 @@ interface AgentAuth {
 interface ClientOpts {
     endpoint?: string;
     timeout?: number;
+    auth?: AgentAuth;
 }
 interface ChainOpts {
     nodeUrls?: string[];
     blockchainRid?: string;
 }
-interface LogToolCallResult {
-    success: boolean;
-    toolCallId: string | null;
-    signedHex?: string;
-    error?: string;
-}
 interface JudgeResult {
     verdict: Verdict;
     action_type: ActionType;
@@ -39,6 +47,7 @@ interface JudgeOptions extends ClientOpts {
     model?: string;
     toolName?: string;
     toolArgsJson?: string;
+    orgName?: string;
     chainOpts?: ChainOpts;
     verifyPubKey?: string;
 }
@@ -51,12 +60,6 @@ interface JudgmentStatus {
     cached?: boolean;
     responseTimeMs?: number;
 }
-interface TierInfo {
-    org_name: string;
-    tier: Tier;
-    verdict_enabled: boolean;
-    enforcement_enabled: boolean;
-}
 interface ToolCallRecord {
     tool_call_id: string;
     agent_pubkey: PubkeyValue;
@@ -179,6 +182,7 @@ interface AtbashClientConfig {
     judge?: JudgeEndpointConfig;
     nodeUrls?: string[];
     blockchainRid?: string;
+    orgName?: string;
     keyPath?: string;
     keyPair?: {
         privKey: string;
@@ -193,7 +197,7 @@ interface AtbashClientConfig {
 
 declare const DEFAULT_ENDPOINT = "https://atbash.ai";
 declare const DEFAULT_CHROMIA_NODE_URLS: string[];
-declare const DEFAULT_BLOCKCHAIN_RID = "619E62A5FD01BDDBFF0EF7A3821B5C88145B0E64A870400EAB4065C6A2779981";
+declare const DEFAULT_BLOCKCHAIN_RID = "B91106947F1EAED7B5D789C7D35755330A8A7DD7CB990D59366114EFFB79ED10";
 declare function isValidPrivateKey(hex: string): boolean;
 declare function derivePublicKey(privKeyHex: string): string;
 declare function generateKeyPair(): {
@@ -202,22 +206,7 @@ declare function generateKeyPair(): {
 };
 declare function loadAgent(privkey: string): AgentAuth;
 declare function toPubkeyHex(val: unknown): string;
-/**
- * Check if an agent is onboarded before signing anything.
- * Calls GET /api/ai/exists?pubkey=<66-hex>
- */
 declare function checkAgentExists(pubkey: string, opts?: ClientOpts): Promise<boolean>;
-/**
- * Sign `log_tool_call` locally and return the signed transaction hex.
- *
- * Checks that the agent is onboarded before signing. The private key
- * is used locally — never sent over the network. The server will
- * broadcast the signed transaction to the chain.
- */
-declare function logToolCall(action: string, context: string, auth: AgentAuth, chainOpts?: ChainOpts, extra?: {
-    toolName?: string;
-    toolArgsJson?: string;
-}, clientOpts?: ClientOpts): Promise<LogToolCallResult>;
 declare function judgeAction(action: string, context: string | undefined, auth: AgentAuth, opts?: JudgeOptions): Promise<JudgeResult>;
 declare function getJudgmentStatus(judgmentId: string, agentPubkey: string, opts?: ClientOpts): Promise<JudgmentStatus>;
 declare function getToolCalls(maxCount: number, opts?: ClientOpts): Promise<ToolCallRecord[]>;
@@ -225,7 +214,7 @@ declare function getOrgToolCalls(orgName: string, maxCount: number, opts?: Clien
 declare function getAgentToolCalls(agentPubkey: string, maxCount: number, opts?: ClientOpts): Promise<ToolCallRecord[]>;
 declare function getToolCallCount(opts?: ClientOpts): Promise<number>;
 declare function getToolCallFull(toolCallId: string, opts?: ClientOpts): Promise<ToolCallFull | null>;
-declare function getOrgTierInfo(orgName: string, opts?: ClientOpts): Promise<TierInfo | null>;
+declare function getOrgSubscription(orgName: string, opts?: ClientOpts): Promise<OrgSubscription | null>;
 declare function getPendingHeldActions(orgName: string, maxCount: number, opts?: ClientOpts): Promise<HeldAction[]>;
 declare function getHeldActionReviews(orgName: string, maxCount: number, opts?: ClientOpts): Promise<HeldActionReview[]>;
 declare function getAgentDetail(agentPubkey: string, opts?: ClientOpts): Promise<Record<string, unknown>>;
@@ -277,6 +266,7 @@ interface AtbashUserConfig {
     orgName?: string;
     judgeEndpoint?: string;
     blockchainRid?: string;
+    network?: string;
     provider?: string;
     providerModel?: string;
 }
@@ -346,4 +336,4 @@ declare function normalizeForMatching(input: string): string;
  */
 declare function containsEvasionCharacters(input: string): boolean;
 
-export { type ActionType, type AgentAuth, type AgentPolicy, type AnomalySeverity, type AnomalyType, type AtbashClient, type AtbashClientConfig, type AtbashUserConfig, type ChainOpts, type ClientOpts, type ClientSource, DEFAULT_BLOCKCHAIN_RID, DEFAULT_CHROMIA_NODE_URLS, DEFAULT_ENDPOINT, type Decision, type DecisionVerdict, type HeldAction, type HeldActionReview, type JudgeEndpointConfig, type JudgeOptions, type JudgeResult, type JudgmentStatus, type JudgmentStatusState, type LogToolCallResult, type MemoryAnomaly, type MemoryDiffResult, type MemoryEntry, type MemoryScanOptions, type MemoryScanResult, type MemoryScanVerdict, type MemorySnapshot, type Provider, type PubkeyValue, type TelemetryConfig, type Tier, type TierInfo, type ToolCallFull, type ToolCallInput, type ToolCallRecord, type ValidatedEndpoint, type Verdict, checkAgentExists, containsEvasionCharacters, createAtbashClient, createMemorySnapshot, derivePublicKey, diffMemorySnapshots, generateKeyPair, getAgentDetail, getAgentPolicy, getAgentToolCalls, getConfigDir, getConfigPath, getHeldActionReviews, getJudgmentStatus, getOrgTierInfo, getOrgToolCalls, getPendingHeldActions, getSafetyStats, getToolCallCount, getToolCallFull, getToolCalls, isValidPrivateKey, judgeAction, loadAgent, loadAgentFromFile, loadUserConfig, logToolCall, normalizeForMatching, resolve, resolveKeyPath, saveUserConfig, scanMemory, scanMemoryBatch, setupTelemetry, shutdownTelemetry, toPubkeyHex, validateJudgeEndpoint, verifyJudgeResponseSignature };
+export { type ActionType, type AgentAuth, type AgentPolicy, type AnomalySeverity, type AnomalyType, type AtbashClient, type AtbashClientConfig, type AtbashUserConfig, type ClientOpts, type ClientSource, DEFAULT_BLOCKCHAIN_RID, DEFAULT_CHROMIA_NODE_URLS, DEFAULT_ENDPOINT, type Decision, type DecisionVerdict, type HeldAction, type HeldActionReview, type JudgeEndpointConfig, type JudgeOptions, type JudgeResult, type JudgmentStatus, type JudgmentStatusState, type MemoryAnomaly, type MemoryDiffResult, type MemoryEntry, type MemoryScanOptions, type MemoryScanResult, type MemoryScanVerdict, type MemorySnapshot, type OrgSubscription, type Provider, type PubkeyValue, type TelemetryConfig, type ToolCallFull, type ToolCallInput, type ToolCallRecord, type ValidatedEndpoint, type Verdict, checkAgentExists, containsEvasionCharacters, createAtbashClient, createMemorySnapshot, derivePublicKey, diffMemorySnapshots, generateKeyPair, getAgentDetail, getAgentPolicy, getAgentToolCalls, getConfigDir, getConfigPath, getHeldActionReviews, getJudgmentStatus, getOrgSubscription, getOrgToolCalls, getPendingHeldActions, getSafetyStats, getToolCallCount, getToolCallFull, getToolCalls, isValidPrivateKey, judgeAction, loadAgent, loadAgentFromFile, loadUserConfig, normalizeForMatching, resolve, resolveKeyPath, saveUserConfig, scanMemory, scanMemoryBatch, setupTelemetry, shutdownTelemetry, toPubkeyHex, validateJudgeEndpoint, verifyJudgeResponseSignature };

package/dist/index.d.ts

@@ -1,11 +1,24 @@
 type Verdict = "ALLOW" | "HOLD" | "BLOCK" | "No verdict";
 type Provider = "openai" | "google" | "microsoft" | "custom" | (string & {});
-type Tier = "audit" | "audit_plus" | "enforcement" | (string & {});
 type ActionType = "allow" | "hold_for_user_confirm" | "block" | (string & {});
 type PubkeyValue = string | Buffer | {
     data: number[];
 };
 type JudgmentStatusState = "pending" | "answered" | "error";
+interface Subscription {
+    subscription_name: string;
+    agent_number: number;
+    is_private_blockchain: boolean;
+    monthly_price: number;
+    yearly_price: number;
+}
+interface OrgSubscription extends Subscription {
+    org_name: string;
+    duration_months: number;
+    assigned_at: number;
+    expires_at: number;
+    is_active: boolean;
+}
 interface AgentAuth {
     pubkey: string;
     privkey: string;
@@ -13,17 +26,12 @@ interface AgentAuth {
 interface ClientOpts {
     endpoint?: string;
     timeout?: number;
+    auth?: AgentAuth;
 }
 interface ChainOpts {
     nodeUrls?: string[];
     blockchainRid?: string;
 }
-interface LogToolCallResult {
-    success: boolean;
-    toolCallId: string | null;
-    signedHex?: string;
-    error?: string;
-}
 interface JudgeResult {
     verdict: Verdict;
     action_type: ActionType;
@@ -39,6 +47,7 @@ interface JudgeOptions extends ClientOpts {
     model?: string;
     toolName?: string;
     toolArgsJson?: string;
+    orgName?: string;
     chainOpts?: ChainOpts;
     verifyPubKey?: string;
 }
@@ -51,12 +60,6 @@ interface JudgmentStatus {
     cached?: boolean;
     responseTimeMs?: number;
 }
-interface TierInfo {
-    org_name: string;
-    tier: Tier;
-    verdict_enabled: boolean;
-    enforcement_enabled: boolean;
-}
 interface ToolCallRecord {
     tool_call_id: string;
     agent_pubkey: PubkeyValue;
@@ -179,6 +182,7 @@ interface AtbashClientConfig {
     judge?: JudgeEndpointConfig;
     nodeUrls?: string[];
     blockchainRid?: string;
+    orgName?: string;
     keyPath?: string;
     keyPair?: {
         privKey: string;
@@ -193,7 +197,7 @@ interface AtbashClientConfig {
 
 declare const DEFAULT_ENDPOINT = "https://atbash.ai";
 declare const DEFAULT_CHROMIA_NODE_URLS: string[];
-declare const DEFAULT_BLOCKCHAIN_RID = "619E62A5FD01BDDBFF0EF7A3821B5C88145B0E64A870400EAB4065C6A2779981";
+declare const DEFAULT_BLOCKCHAIN_RID = "B91106947F1EAED7B5D789C7D35755330A8A7DD7CB990D59366114EFFB79ED10";
 declare function isValidPrivateKey(hex: string): boolean;
 declare function derivePublicKey(privKeyHex: string): string;
 declare function generateKeyPair(): {
@@ -202,22 +206,7 @@ declare function generateKeyPair(): {
 };
 declare function loadAgent(privkey: string): AgentAuth;
 declare function toPubkeyHex(val: unknown): string;
-/**
- * Check if an agent is onboarded before signing anything.
- * Calls GET /api/ai/exists?pubkey=<66-hex>
- */
 declare function checkAgentExists(pubkey: string, opts?: ClientOpts): Promise<boolean>;
-/**
- * Sign `log_tool_call` locally and return the signed transaction hex.
- *
- * Checks that the agent is onboarded before signing. The private key
- * is used locally — never sent over the network. The server will
- * broadcast the signed transaction to the chain.
- */
-declare function logToolCall(action: string, context: string, auth: AgentAuth, chainOpts?: ChainOpts, extra?: {
-    toolName?: string;
-    toolArgsJson?: string;
-}, clientOpts?: ClientOpts): Promise<LogToolCallResult>;
 declare function judgeAction(action: string, context: string | undefined, auth: AgentAuth, opts?: JudgeOptions): Promise<JudgeResult>;
 declare function getJudgmentStatus(judgmentId: string, agentPubkey: string, opts?: ClientOpts): Promise<JudgmentStatus>;
 declare function getToolCalls(maxCount: number, opts?: ClientOpts): Promise<ToolCallRecord[]>;
@@ -225,7 +214,7 @@ declare function getOrgToolCalls(orgName: string, maxCount: number, opts?: Clien
 declare function getAgentToolCalls(agentPubkey: string, maxCount: number, opts?: ClientOpts): Promise<ToolCallRecord[]>;
 declare function getToolCallCount(opts?: ClientOpts): Promise<number>;
 declare function getToolCallFull(toolCallId: string, opts?: ClientOpts): Promise<ToolCallFull | null>;
-declare function getOrgTierInfo(orgName: string, opts?: ClientOpts): Promise<TierInfo | null>;
+declare function getOrgSubscription(orgName: string, opts?: ClientOpts): Promise<OrgSubscription | null>;
 declare function getPendingHeldActions(orgName: string, maxCount: number, opts?: ClientOpts): Promise<HeldAction[]>;
 declare function getHeldActionReviews(orgName: string, maxCount: number, opts?: ClientOpts): Promise<HeldActionReview[]>;
 declare function getAgentDetail(agentPubkey: string, opts?: ClientOpts): Promise<Record<string, unknown>>;
@@ -277,6 +266,7 @@ interface AtbashUserConfig {
     orgName?: string;
     judgeEndpoint?: string;
     blockchainRid?: string;
+    network?: string;
     provider?: string;
     providerModel?: string;
 }
@@ -346,4 +336,4 @@ declare function normalizeForMatching(input: string): string;
  */
 declare function containsEvasionCharacters(input: string): boolean;
 
-export { type ActionType, type AgentAuth, type AgentPolicy, type AnomalySeverity, type AnomalyType, type AtbashClient, type AtbashClientConfig, type AtbashUserConfig, type ChainOpts, type ClientOpts, type ClientSource, DEFAULT_BLOCKCHAIN_RID, DEFAULT_CHROMIA_NODE_URLS, DEFAULT_ENDPOINT, type Decision, type DecisionVerdict, type HeldAction, type HeldActionReview, type JudgeEndpointConfig, type JudgeOptions, type JudgeResult, type JudgmentStatus, type JudgmentStatusState, type LogToolCallResult, type MemoryAnomaly, type MemoryDiffResult, type MemoryEntry, type MemoryScanOptions, type MemoryScanResult, type MemoryScanVerdict, type MemorySnapshot, type Provider, type PubkeyValue, type TelemetryConfig, type Tier, type TierInfo, type ToolCallFull, type ToolCallInput, type ToolCallRecord, type ValidatedEndpoint, type Verdict, checkAgentExists, containsEvasionCharacters, createAtbashClient, createMemorySnapshot, derivePublicKey, diffMemorySnapshots, generateKeyPair, getAgentDetail, getAgentPolicy, getAgentToolCalls, getConfigDir, getConfigPath, getHeldActionReviews, getJudgmentStatus, getOrgTierInfo, getOrgToolCalls, getPendingHeldActions, getSafetyStats, getToolCallCount, getToolCallFull, getToolCalls, isValidPrivateKey, judgeAction, loadAgent, loadAgentFromFile, loadUserConfig, logToolCall, normalizeForMatching, resolve, resolveKeyPath, saveUserConfig, scanMemory, scanMemoryBatch, setupTelemetry, shutdownTelemetry, toPubkeyHex, validateJudgeEndpoint, verifyJudgeResponseSignature };
+export { type ActionType, type AgentAuth, type AgentPolicy, type AnomalySeverity, type AnomalyType, type AtbashClient, type AtbashClientConfig, type AtbashUserConfig, type ClientOpts, type ClientSource, DEFAULT_BLOCKCHAIN_RID, DEFAULT_CHROMIA_NODE_URLS, DEFAULT_ENDPOINT, type Decision, type DecisionVerdict, type HeldAction, type HeldActionReview, type JudgeEndpointConfig, type JudgeOptions, type JudgeResult, type JudgmentStatus, type JudgmentStatusState, type MemoryAnomaly, type MemoryDiffResult, type MemoryEntry, type MemoryScanOptions, type MemoryScanResult, type MemoryScanVerdict, type MemorySnapshot, type OrgSubscription, type Provider, type PubkeyValue, type TelemetryConfig, type ToolCallFull, type ToolCallInput, type ToolCallRecord, type ValidatedEndpoint, type Verdict, checkAgentExists, containsEvasionCharacters, createAtbashClient, createMemorySnapshot, derivePublicKey, diffMemorySnapshots, generateKeyPair, getAgentDetail, getAgentPolicy, getAgentToolCalls, getConfigDir, getConfigPath, getHeldActionReviews, getJudgmentStatus, getOrgSubscription, getOrgToolCalls, getPendingHeldActions, getSafetyStats, getToolCallCount, getToolCallFull, getToolCalls, isValidPrivateKey, judgeAction, loadAgent, loadAgentFromFile, loadUserConfig, normalizeForMatching, resolve, resolveKeyPath, saveUserConfig, scanMemory, scanMemoryBatch, setupTelemetry, shutdownTelemetry, toPubkeyHex, validateJudgeEndpoint, verifyJudgeResponseSignature };

package/dist/index.js

@@ -121,7 +121,37 @@ var DEFAULT_CHROMIA_NODE_URLS = [
   "https://node1.testnet.chromia.com:7740",
   "https://node3.testnet.chromia.com:7740"
 ];
-var DEFAULT_BLOCKCHAIN_RID = "619E62A5FD01BDDBFF0EF7A3821B5C88145B0E64A870400EAB4065C6A2779981";
+var DEFAULT_BLOCKCHAIN_RID = "B91106947F1EAED7B5D789C7D35755330A8A7DD7CB990D59366114EFFB79ED10";
+var DEFAULT_PRIVATE_NODE_URLS = [
+  "https://node0-pvn-testnet.dynamic.chromia.dev"
+];
+var DEFAULT_PRIVATE_BLOCKCHAIN_RID = "431AE6A5695D157D74194A61AB4D0B6A98C99AFEEF186FC885CDA4A3BAAB800E";
+var PUBLIC_CHAIN = {
+  network: "public",
+  blockchainRid: DEFAULT_BLOCKCHAIN_RID,
+  nodeUrls: DEFAULT_CHROMIA_NODE_URLS
+};
+var PRIVATE_CHAIN = {
+  network: "private",
+  blockchainRid: DEFAULT_PRIVATE_BLOCKCHAIN_RID,
+  nodeUrls: DEFAULT_PRIVATE_NODE_URLS
+};
+function chainForNetwork(network) {
+  return network === "private" ? PRIVATE_CHAIN : PUBLIC_CHAIN;
+}
+function resolveChainOpts(chainOpts) {
+  if (chainOpts?.network) {
+    const chain = chainForNetwork(chainOpts.network);
+    return {
+      nodeUrls: chainOpts.nodeUrls ?? chain.nodeUrls,
+      blockchainRid: chainOpts.blockchainRid ?? chain.blockchainRid
+    };
+  }
+  return {
+    nodeUrls: chainOpts?.nodeUrls ?? DEFAULT_CHROMIA_NODE_URLS,
+    blockchainRid: chainOpts?.blockchainRid ?? DEFAULT_BLOCKCHAIN_RID
+  };
+}
 function isValidPrivateKey(hex) {
   return /^[0-9a-fA-F]{64}$/.test(hex);
 }
@@ -160,14 +190,30 @@ function toPubkeyHex(val) {
 function baseUrl(opts) {
   return opts?.endpoint || DEFAULT_ENDPOINT;
 }
+var AUTH_BEARER_REFRESH_MS = 4 * 60 * 1e3;
+var bearerCache = /* @__PURE__ */ new Map();
+async function getOrCreateAuthBearer(auth) {
+  const now = Date.now();
+  const cached = bearerCache.get(auth.pubkey);
+  if (cached && now - cached.issuedAt < AUTH_BEARER_REFRESH_MS) {
+    return cached.hex;
+  }
+  const nonce = `auth-${now.toString(36)}-${randomBytes(4).toString("hex")}`;
+  const hex = await buildSignedTx(
+    "log_tool_call",
+    [nonce, `auth:${now}`, "", "auth-bearer", ""],
+    auth
+  );
+  bearerCache.set(auth.pubkey, { hex, issuedAt: now });
+  return hex;
+}
 function generateToolCallId() {
   const ts = Date.now();
   const rand = randomBytes(4).toString("hex");
   return `tc-${ts}-${rand}`;
 }
 async function buildSignedTx(opName, args, auth, chainOpts) {
-  const nodeUrls = chainOpts?.nodeUrls ?? DEFAULT_CHROMIA_NODE_URLS;
-  const blockchainRid = chainOpts?.blockchainRid ?? DEFAULT_BLOCKCHAIN_RID;
+  const { nodeUrls, blockchainRid } = resolveChainOpts(chainOpts);
   const client = await createClient({ nodeUrlPool: nodeUrls, blockchainRid });
   const privKeyBuf = Buffer.from(auth.privkey, "hex");
   const keyPair = encryption2.makeKeyPair(privKeyBuf);
@@ -184,11 +230,13 @@ async function buildSignedTx(opName, args, auth, chainOpts) {
   );
   return Buffer.from(signed).toString("hex");
 }
-async function checkAgentExists(pubkey, opts) {
+async function _checkAgentExists(pubkey, opts, chainOpts) {
   const start = performance.now();
   recordCall("checkAgentExists", void 0, pubkey);
   try {
-    const url = `${baseUrl(opts)}/api/ai/exists?pubkey=${encodeURIComponent(pubkey)}`;
+    const network = chainOpts?.network;
+    let url = `${baseUrl(opts)}/api/ai/exists?pubkey=${encodeURIComponent(pubkey)}`;
+    if (network) url += `&network=${encodeURIComponent(network)}`;
     const data = await getJson(url, opts);
     recordDuration("checkAgentExists", performance.now() - start, "success");
     return Boolean(data.registered);
@@ -197,10 +245,13 @@ async function checkAgentExists(pubkey, opts) {
     throw err;
   }
 }
+async function checkAgentExists(pubkey, opts) {
+  return _checkAgentExists(pubkey, opts);
+}
 async function logToolCall(action, context, auth, chainOpts, extra, clientOpts) {
   const start = performance.now();
   recordCall("logToolCall", void 0, auth.pubkey);
-  const exists = await checkAgentExists(auth.pubkey, clientOpts);
+  const exists = await _checkAgentExists(auth.pubkey, clientOpts, chainOpts);
   if (!exists) {
     recordDuration("logToolCall", performance.now() - start, "error");
     return {
@@ -269,9 +320,13 @@ function enrichError(status, body, statusText, opts) {
   return new Error(message);
 }
 async function postJson(url, body, opts) {
+  const headers = { "Content-Type": "application/json" };
+  if (opts?.auth) {
+    headers["Authorization"] = `Bearer ${await getOrCreateAuthBearer(opts.auth)}`;
+  }
   const resp = await fetch(url, {
     method: "POST",
-    headers: { "Content-Type": "application/json" },
+    headers,
     body: JSON.stringify(body),
     signal: opts?.timeout ? AbortSignal.timeout(opts.timeout) : void 0
   });
@@ -283,9 +338,13 @@ async function postJson(url, body, opts) {
   return ct.includes("application/json") ? resp.json() : {};
 }
 async function getJson(url, opts) {
+  const headers = { Accept: "application/json" };
+  if (opts?.auth) {
+    headers["Authorization"] = `Bearer ${await getOrCreateAuthBearer(opts.auth)}`;
+  }
   const resp = await fetch(url, {
     method: "GET",
-    headers: { Accept: "application/json" },
+    headers,
     signal: opts?.timeout ? AbortSignal.timeout(opts.timeout) : void 0
   });
   if (!resp.ok) {
@@ -326,11 +385,16 @@ async function judgeAction(action, context = "", auth, opts) {
     throw new Error("action is required and cannot be empty.");
   }
   try {
+    let chainOpts = opts?.chainOpts;
+    if (opts?.orgName && !chainOpts?.blockchainRid) {
+      const resolved = await resolveChainForOrg(opts.orgName, opts);
+      chainOpts = { ...chainOpts, network: resolved.network };
+    }
     const logResult = await logToolCall(
       action,
       context,
       auth,
-      opts?.chainOpts,
+      chainOpts,
       { toolName: opts?.toolName, toolArgsJson: opts?.toolArgsJson },
       opts
     );
@@ -344,7 +408,7 @@ async function judgeAction(action, context = "", auth, opts) {
         "judge_action",
         [judgmentId, action, context || "", ""],
         auth,
-        opts?.chainOpts
+        chainOpts
       );
     }
     const url = `${baseUrl(opts)}/api/v1/judge`;
@@ -489,21 +553,52 @@ async function getToolCallFull(toolCallId, opts) {
     throw err;
   }
 }
-async function getOrgTierInfo(orgName, opts) {
+function coerceOrgSubscription(row, orgName) {
+  if (!row || typeof row !== "object") return null;
+  const r = row;
+  return {
+    org_name: String(r.org_name ?? orgName),
+    subscription_name: String(r.subscription_name ?? ""),
+    agent_number: Number(r.agent_number ?? 0),
+    is_private_blockchain: Boolean(r.is_private_blockchain),
+    monthly_price: Number(r.monthly_price ?? 0),
+    yearly_price: Number(r.yearly_price ?? 0),
+    duration_months: Number(r.duration_months ?? 0),
+    assigned_at: Number(r.assigned_at ?? 0),
+    expires_at: Number(r.expires_at ?? 0),
+    is_active: Boolean(r.is_active)
+  };
+}
+async function getOrgSubscription(orgName, opts) {
   const start = performance.now();
-  recordCall("getOrgTierInfo");
+  recordCall("getOrgSubscription");
   try {
     const result = await getJson(
-      riskEngineUrl("org-tier-info", { org: orgName }, opts),
+      riskEngineUrl("org-subscription", { org: orgName }, opts),
       opts
     );
-    recordDuration("getOrgTierInfo", performance.now() - start, "success");
-    return result;
+    recordDuration("getOrgSubscription", performance.now() - start, "success");
+    return coerceOrgSubscription(result, orgName);
   } catch (err) {
-    recordDuration("getOrgTierInfo", performance.now() - start, "error");
+    recordDuration("getOrgSubscription", performance.now() - start, "error");
     throw err;
   }
 }
+var _chainCache = /* @__PURE__ */ new Map();
+async function resolveChainForOrg(orgName, opts) {
+  const cached = _chainCache.get(orgName);
+  if (cached) return cached;
+  try {
+    const sub = await getOrgSubscription(orgName, opts);
+    if (sub?.is_private_blockchain) {
+      _chainCache.set(orgName, PRIVATE_CHAIN);
+      return PRIVATE_CHAIN;
+    }
+  } catch {
+  }
+  _chainCache.set(orgName, PUBLIC_CHAIN);
+  return PUBLIC_CHAIN;
+}
 async function getPendingHeldActions(orgName, maxCount, opts) {
   const start = performance.now();
   recordCall("getPendingHeldActions");
@@ -759,6 +854,8 @@ function createAtbashClient(config = {}) {
   const validated = validateJudgeEndpoint(config.judge);
   const failClosed = config.failClosed !== false;
   const logger = config.logger ?? {};
+  const orgName = config.orgName;
+  let resolvedChain = null;
   const inlineKeyPair = config.keyPair;
   const keyPath = inlineKeyPair ? null : config.keyPath;
   if (validated.url !== DEFAULT_ENDPOINT) {
@@ -808,12 +905,23 @@ function createAtbashClient(config = {}) {
         });
       }
       try {
+        if (!resolvedChain && orgName) {
+          resolvedChain = await resolveChainForOrg(orgName, { endpoint: validated.url });
+          config.nodeUrls = resolvedChain.nodeUrls;
+          config.blockchainRid = resolvedChain.blockchainRid;
+          logger.info?.("[atbash] resolved network from subscription", {
+            org: orgName,
+            network: resolvedChain.network,
+            brid: resolvedChain.blockchainRid
+          });
+        }
         logger.info?.("[atbash] judge API called", { tool: toolName });
         const result = await judgeAction(actionText, contextText, agent, {
           endpoint: validated.url,
           verifyPubKey: validated.verifyPubKey ?? void 0,
           toolName,
           toolArgsJson: argsJson,
+          orgName,
           chainOpts: {
             nodeUrls: config.nodeUrls,
             blockchainRid: config.blockchainRid
@@ -901,6 +1009,7 @@ var ENV_MAP = {
   orgName: "ATBASH_ORG_NAME",
   judgeEndpoint: "ATBASH_ENDPOINT",
   blockchainRid: "ATBASH_BLOCKCHAIN_RID",
+  network: "ATBASH_NETWORK",
   provider: "ATBASH_PROVIDER",
   providerModel: "ATBASH_PROVIDER_MODEL"
 };
@@ -1405,7 +1514,7 @@ export {
   getConfigPath,
   getHeldActionReviews,
   getJudgmentStatus,
-  getOrgTierInfo,
+  getOrgSubscription,
   getOrgToolCalls,
   getPendingHeldActions,
   getSafetyStats,
@@ -1417,7 +1526,6 @@ export {
   loadAgent,
   loadAgentFromFile,
   loadUserConfig,
-  logToolCall,
   normalizeForMatching,
   resolve,
   resolveKeyPath,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atbash/sdk",
-  "version": "0.3.23",
+  "version": "0.3.25",
   "description": "Atbash SDK — control boundary before the last irreversible step in an agent workflow",
   "homepage": "https://atbash.ai",
   "author": "Atbash",