@owney/sdk 0.2.17 → 0.3.0-beta.0

package/dist/index.cjs

@@ -556,6 +556,19 @@ async function fetchAgentKeys(apiKey) {
 }
 
 // src/lib/helpers/withdraw-helper.ts
+var import_viem = require("viem");
+function projectAgentBalancesForAsset(agents, aggregated, chainId, asset, decimals) {
+  const target = asset.toUpperCase();
+  return agents.map((agent) => {
+    const agentBalance = aggregated[agent.id];
+    const tokenBalance = agentBalance?.tokens.find(
+      (t) => t.chainId === chainId && t.asset.toUpperCase() === target
+    );
+    if (!tokenBalance)
+      return { agent, balance: 0n };
+    return { agent, balance: (0, import_viem.parseUnits)(tokenBalance.amount, decimals) };
+  });
+}
 function planProportionalShares(balances, requested, totalAvailable) {
   const plans = balances.map(({ agent, balance }) => ({
     agent,
@@ -580,8 +593,24 @@ function planProportionalShares(balances, requested, totalAvailable) {
   }
   return plans;
 }
-function redistributeShare(plans, fromIndex, amount) {
-  const candidates = plans.slice(fromIndex + 1).filter((c) => c.balance - c.planned > 0n);
+function planDisabledDrain(disabled, requested) {
+  const sorted = [...disabled].filter((d) => d.balance > 0n).sort((a, b) => b.balance > a.balance ? 1 : b.balance < a.balance ? -1 : 0);
+  const plans = [];
+  let remaining = requested;
+  for (const { agent, balance } of sorted) {
+    if (remaining === 0n) {
+      plans.push({ agent, balance, planned: 0n });
+      continue;
+    }
+    const take = balance < remaining ? balance : remaining;
+    plans.push({ agent, balance, planned: take });
+    remaining -= take;
+  }
+  return { plans, remaining };
+}
+function redistributeShare(plans, fromIndex, amount, candidatePool) {
+  const pool = candidatePool ?? plans.slice(fromIndex + 1);
+  const candidates = pool.filter((c) => c.balance - c.planned > 0n);
   const totalHeadroom = candidates.reduce(
     (s, c) => s + (c.balance - c.planned),
     0n
@@ -613,10 +642,15 @@ function sumWithdrawnAmount(results) {
 }
 
 // src/client.ts
-var import_viem = require("viem");
 var OwneySDK = class {
   agents = /* @__PURE__ */ new Map();
   activeAgents = /* @__PURE__ */ new Set();
+  /**
+   * Agents whose routing flag is_enabled === false. They remain in
+   * `this.agents` and may still be added to `activeAgents` for read +
+   * withdraw, but they are excluded from new deposits / activations.
+   */
+  disabledAgents = /* @__PURE__ */ new Set();
   state = null;
   apiKey;
   zyfaiRpcUrls;
@@ -653,8 +687,12 @@ var OwneySDK = class {
       await agent.disconnect();
     }
     this.activeAgents.clear();
+    this.disabledAgents.clear();
     this.state = null;
   }
+  isAgentDisabled(agentId) {
+    return this.disabledAgents.has(agentId);
+  }
   /**
    * Check if the SDK has an active connection state.
    * @returns `true` if `connect()` has been called and `disconnect()` has not
@@ -706,11 +744,15 @@ var OwneySDK = class {
     if (!this.initializingAgentsPromise) {
       this.initializingAgentsPromise = (async () => {
         const agentKeys = await fetchAgentKeys(this.apiKey);
-        for (const { key, agent_type } of agentKeys) {
+        this.disabledAgents.clear();
+        for (const { key, agent_type, is_enabled } of agentKeys) {
           const agent = this.createAgent(agent_type, key);
           if (!agent)
             continue;
           this.agents.set(agent_type, agent);
+          if (is_enabled === false) {
+            this.disabledAgents.add(agent_type);
+          }
         }
         if (this.agents.size === 0) {
           throw new OwneyError(
@@ -755,6 +797,16 @@ var OwneySDK = class {
       }
       const uniqueIds = [...new Set(agentId)];
       const agents = uniqueIds.map((id) => this.getAgent(id));
+      const disabledRequested = uniqueIds.filter(
+        (id) => this.isAgentDisabled(id)
+      );
+      if (disabledRequested.length > 0) {
+        throw new OwneyError(
+          "AGENT_DISABLED",
+          `Cannot activate disabled agent(s): ${disabledRequested.join(", ")}.`,
+          { agentIds: disabledRequested }
+        );
+      }
       const incompatibleAgents = agents.filter(
         (agent) => !agent.supportedChainIds.includes(chainId)
       ).map((agent) => ({
@@ -793,8 +845,11 @@ var OwneySDK = class {
       this.activeAgents.add(agent.id);
     }
     state.chainId = chainId;
+    const enabledCompatible = compatible.filter(
+      (agent) => !this.isAgentDisabled(agent.id)
+    );
     Promise.all(
-      compatible.map(async (agent) => {
+      enabledCompatible.map(async (agent) => {
         await agent.activateAgent(state, chainId);
       })
     ).catch((error) => {
@@ -821,13 +876,22 @@ var OwneySDK = class {
     const state = this.requireState();
     const chainId = this.requireChainId();
     if (agentId) {
+      if (this.isAgentDisabled(agentId)) {
+        throw new OwneyError(
+          "AGENT_DISABLED",
+          `Cannot deposit into disabled agent "${agentId}". Existing balances remain visible and withdrawable.`,
+          { agentId }
+        );
+      }
       const agent = this.getAgent(agentId);
       this.validateAssetSupport(agent, chainId, asset);
       this.validateMinDepositAmount(agent, chainId, asset, amount);
       return agent.deposit(state, chainId, amount, depositCallback);
     }
     const totalAmount = BigInt(amount);
-    const eligibleAgents = this.getEligibleAgents(chainId, asset);
+    const eligibleAgents = this.getEligibleAgents(chainId, asset, {
+      excludeDisabled: true
+    });
     const agentAmounts = this.splitDepositAmount(
       totalAmount,
       eligibleAgents,
@@ -911,8 +975,11 @@ var OwneySDK = class {
       );
     }
   }
-  getEligibleAgents(chainId, asset) {
+  getEligibleAgents(chainId, asset, opts) {
     const eligible = [...this.agents.values()].filter((agent) => {
+      if (opts?.excludeDisabled && this.isAgentDisabled(agent.id)) {
+        return false;
+      }
       const chainAssets = agent.supportedAssets.find(
         (sa) => sa.chainId === chainId
       );
@@ -979,16 +1046,13 @@ var OwneySDK = class {
     }
     const requested = BigInt(amount);
     const aggregated = await this.getBalances();
-    const balances = eligibleAgents.map((agent) => {
-      const agentBalance = aggregated.agentBalances[agent.id];
-      const tokenBalance = agentBalance?.tokens.find(
-        (t) => t.chainId === chainId && t.asset.toUpperCase() === asset.toUpperCase()
-      );
-      if (!tokenBalance)
-        return { agent, balance: 0n };
-      const balance = (0, import_viem.parseUnits)(tokenBalance.amount, assetInfo.decimals);
-      return { agent, balance };
-    });
+    const balances = projectAgentBalancesForAsset(
+      eligibleAgents,
+      aggregated.agentBalances,
+      chainId,
+      asset,
+      assetInfo.decimals
+    );
     const totalAvailable = balances.reduce((sum, x) => sum + x.balance, 0n);
     if (totalAvailable < requested) {
       throw new OwneyError(
@@ -1001,7 +1065,26 @@ var OwneySDK = class {
         }
       );
     }
-    const plans = planProportionalShares(balances, requested, totalAvailable);
+    const disabledBalances = balances.filter(
+      (b) => this.isAgentDisabled(b.agent.id)
+    );
+    const enabledBalances = balances.filter(
+      (b) => !this.isAgentDisabled(b.agent.id)
+    );
+    const { plans: disabledPlans, remaining: afterDrain } = planDisabledDrain(
+      disabledBalances,
+      requested
+    );
+    const enabledTotal = enabledBalances.reduce(
+      (s, b) => s + b.balance,
+      0n
+    );
+    const enabledPlans = afterDrain > 0n ? planProportionalShares(enabledBalances, afterDrain, enabledTotal) : enabledBalances.map(({ agent, balance }) => ({
+      agent,
+      balance,
+      planned: 0n
+    }));
+    const plans = [...disabledPlans, ...enabledPlans];
     const results = {};
     const agentErrors = {};
     for (let i = 0; i < plans.length; i++) {
@@ -1018,7 +1101,9 @@ var OwneySDK = class {
       } catch (err) {
         console.error(`withdraw failed for agent "${p.agent.id}":`, err);
         agentErrors[p.agent.id] = err instanceof Error ? err.message : String(err);
-        redistributeShare(plans, i, p.planned);
+        const failedAmount = p.planned;
+        p.planned = 0n;
+        redistributeShare(plans, i, failedAmount);
       }
     }
     if (Object.keys(results).length === 0) {

package/dist/index.d.cts

@@ -208,6 +208,12 @@ interface OwneyAgentAllocation {
 declare class OwneySDK {
     private agents;
     private activeAgents;
+    /**
+     * Agents whose routing flag is_enabled === false. They remain in
+     * `this.agents` and may still be added to `activeAgents` for read +
+     * withdraw, but they are excluded from new deposits / activations.
+     */
+    private disabledAgents;
     private state;
     private apiKey;
     private zyfaiRpcUrls?;
@@ -224,6 +230,7 @@ declare class OwneySDK {
      * After calling this, `connect()` must be called again before using agent methods.
      */
     disconnect(): Promise<void>;
+    private isAgentDisabled;
     /**
      * Check if the SDK has an active connection state.
      * @returns `true` if `connect()` has been called and `disconnect()` has not
@@ -346,7 +353,7 @@ declare class OwneySDK {
     getAgentAllocation(): Promise<OwneyAgentAllocation>;
 }
 
-type OwneyErrorCode = "NOT_CONNECTED" | "NO_ACTIVE_CHAIN" | "WALLET_NO_ACCOUNTS" | "WALLET_ADDRESS_REQUIRED" | "AGENT_NOT_FOUND" | "AGENT_CHAIN_INCOMPATIBLE" | "AGENT_EMPTY_LIST" | "CHAIN_UNSUPPORTED" | "CHAIN_NO_COMPATIBLE_AGENTS" | "ASSET_UNSUPPORTED" | "ASSET_NO_COMPATIBLE_AGENTS" | "DEPOSIT_AMOUNT_BELOW_MINIMUM" | "DEPOSIT_CALLBACK_REQUIRED" | "DEPOSIT_CALLBACK_INVALID" | "DEPOSIT_NO_PERMITTED_TOKENS" | "WITHDRAW_NO_PERMITTED_TOKENS" | "WITHDRAW_INSUFFICIENT_BALANCE" | "WITHDRAW_ALL_FAILED" | "WITHDRAW_PARTIAL_FAILURE" | "WITHDRAW_FAILED" | "API_ROUTING_ERROR" | "API_ROUTING_FAILED" | "API_SAIL_ERROR" | "API_SAIL_TIMEOUT" | "API_NO_AGENTS" | "BALANCE_ALL_FAILED" | "ALLOCATION_ALL_FAILED" | "VALIDATION_INVALID_DAYS";
+type OwneyErrorCode = "NOT_CONNECTED" | "NO_ACTIVE_CHAIN" | "WALLET_NO_ACCOUNTS" | "WALLET_ADDRESS_REQUIRED" | "AGENT_NOT_FOUND" | "AGENT_CHAIN_INCOMPATIBLE" | "AGENT_EMPTY_LIST" | "AGENT_DISABLED" | "CHAIN_UNSUPPORTED" | "CHAIN_NO_COMPATIBLE_AGENTS" | "ASSET_UNSUPPORTED" | "ASSET_NO_COMPATIBLE_AGENTS" | "DEPOSIT_AMOUNT_BELOW_MINIMUM" | "DEPOSIT_CALLBACK_REQUIRED" | "DEPOSIT_CALLBACK_INVALID" | "DEPOSIT_NO_PERMITTED_TOKENS" | "WITHDRAW_NO_PERMITTED_TOKENS" | "WITHDRAW_INSUFFICIENT_BALANCE" | "WITHDRAW_ALL_FAILED" | "WITHDRAW_PARTIAL_FAILURE" | "WITHDRAW_FAILED" | "API_ROUTING_ERROR" | "API_ROUTING_FAILED" | "API_SAIL_ERROR" | "API_SAIL_TIMEOUT" | "API_NO_AGENTS" | "BALANCE_ALL_FAILED" | "ALLOCATION_ALL_FAILED" | "VALIDATION_INVALID_DAYS";
 declare class OwneyError extends Error {
     readonly code: OwneyErrorCode;
     readonly details?: Record<string, unknown>;

package/dist/index.d.ts

@@ -208,6 +208,12 @@ interface OwneyAgentAllocation {
 declare class OwneySDK {
     private agents;
     private activeAgents;
+    /**
+     * Agents whose routing flag is_enabled === false. They remain in
+     * `this.agents` and may still be added to `activeAgents` for read +
+     * withdraw, but they are excluded from new deposits / activations.
+     */
+    private disabledAgents;
     private state;
     private apiKey;
     private zyfaiRpcUrls?;
@@ -224,6 +230,7 @@ declare class OwneySDK {
      * After calling this, `connect()` must be called again before using agent methods.
      */
     disconnect(): Promise<void>;
+    private isAgentDisabled;
     /**
      * Check if the SDK has an active connection state.
      * @returns `true` if `connect()` has been called and `disconnect()` has not
@@ -346,7 +353,7 @@ declare class OwneySDK {
     getAgentAllocation(): Promise<OwneyAgentAllocation>;
 }
 
-type OwneyErrorCode = "NOT_CONNECTED" | "NO_ACTIVE_CHAIN" | "WALLET_NO_ACCOUNTS" | "WALLET_ADDRESS_REQUIRED" | "AGENT_NOT_FOUND" | "AGENT_CHAIN_INCOMPATIBLE" | "AGENT_EMPTY_LIST" | "CHAIN_UNSUPPORTED" | "CHAIN_NO_COMPATIBLE_AGENTS" | "ASSET_UNSUPPORTED" | "ASSET_NO_COMPATIBLE_AGENTS" | "DEPOSIT_AMOUNT_BELOW_MINIMUM" | "DEPOSIT_CALLBACK_REQUIRED" | "DEPOSIT_CALLBACK_INVALID" | "DEPOSIT_NO_PERMITTED_TOKENS" | "WITHDRAW_NO_PERMITTED_TOKENS" | "WITHDRAW_INSUFFICIENT_BALANCE" | "WITHDRAW_ALL_FAILED" | "WITHDRAW_PARTIAL_FAILURE" | "WITHDRAW_FAILED" | "API_ROUTING_ERROR" | "API_ROUTING_FAILED" | "API_SAIL_ERROR" | "API_SAIL_TIMEOUT" | "API_NO_AGENTS" | "BALANCE_ALL_FAILED" | "ALLOCATION_ALL_FAILED" | "VALIDATION_INVALID_DAYS";
+type OwneyErrorCode = "NOT_CONNECTED" | "NO_ACTIVE_CHAIN" | "WALLET_NO_ACCOUNTS" | "WALLET_ADDRESS_REQUIRED" | "AGENT_NOT_FOUND" | "AGENT_CHAIN_INCOMPATIBLE" | "AGENT_EMPTY_LIST" | "AGENT_DISABLED" | "CHAIN_UNSUPPORTED" | "CHAIN_NO_COMPATIBLE_AGENTS" | "ASSET_UNSUPPORTED" | "ASSET_NO_COMPATIBLE_AGENTS" | "DEPOSIT_AMOUNT_BELOW_MINIMUM" | "DEPOSIT_CALLBACK_REQUIRED" | "DEPOSIT_CALLBACK_INVALID" | "DEPOSIT_NO_PERMITTED_TOKENS" | "WITHDRAW_NO_PERMITTED_TOKENS" | "WITHDRAW_INSUFFICIENT_BALANCE" | "WITHDRAW_ALL_FAILED" | "WITHDRAW_PARTIAL_FAILURE" | "WITHDRAW_FAILED" | "API_ROUTING_ERROR" | "API_ROUTING_FAILED" | "API_SAIL_ERROR" | "API_SAIL_TIMEOUT" | "API_NO_AGENTS" | "BALANCE_ALL_FAILED" | "ALLOCATION_ALL_FAILED" | "VALIDATION_INVALID_DAYS";
 declare class OwneyError extends Error {
     readonly code: OwneyErrorCode;
     readonly details?: Record<string, unknown>;

package/dist/index.js

@@ -526,6 +526,19 @@ async function fetchAgentKeys(apiKey) {
 }
 
 // src/lib/helpers/withdraw-helper.ts
+import { parseUnits } from "viem";
+function projectAgentBalancesForAsset(agents, aggregated, chainId, asset, decimals) {
+  const target = asset.toUpperCase();
+  return agents.map((agent) => {
+    const agentBalance = aggregated[agent.id];
+    const tokenBalance = agentBalance?.tokens.find(
+      (t) => t.chainId === chainId && t.asset.toUpperCase() === target
+    );
+    if (!tokenBalance)
+      return { agent, balance: 0n };
+    return { agent, balance: parseUnits(tokenBalance.amount, decimals) };
+  });
+}
 function planProportionalShares(balances, requested, totalAvailable) {
   const plans = balances.map(({ agent, balance }) => ({
     agent,
@@ -550,8 +563,24 @@ function planProportionalShares(balances, requested, totalAvailable) {
   }
   return plans;
 }
-function redistributeShare(plans, fromIndex, amount) {
-  const candidates = plans.slice(fromIndex + 1).filter((c) => c.balance - c.planned > 0n);
+function planDisabledDrain(disabled, requested) {
+  const sorted = [...disabled].filter((d) => d.balance > 0n).sort((a, b) => b.balance > a.balance ? 1 : b.balance < a.balance ? -1 : 0);
+  const plans = [];
+  let remaining = requested;
+  for (const { agent, balance } of sorted) {
+    if (remaining === 0n) {
+      plans.push({ agent, balance, planned: 0n });
+      continue;
+    }
+    const take = balance < remaining ? balance : remaining;
+    plans.push({ agent, balance, planned: take });
+    remaining -= take;
+  }
+  return { plans, remaining };
+}
+function redistributeShare(plans, fromIndex, amount, candidatePool) {
+  const pool = candidatePool ?? plans.slice(fromIndex + 1);
+  const candidates = pool.filter((c) => c.balance - c.planned > 0n);
   const totalHeadroom = candidates.reduce(
     (s, c) => s + (c.balance - c.planned),
     0n
@@ -583,10 +612,15 @@ function sumWithdrawnAmount(results) {
 }
 
 // src/client.ts
-import { parseUnits } from "viem";
 var OwneySDK = class {
   agents = /* @__PURE__ */ new Map();
   activeAgents = /* @__PURE__ */ new Set();
+  /**
+   * Agents whose routing flag is_enabled === false. They remain in
+   * `this.agents` and may still be added to `activeAgents` for read +
+   * withdraw, but they are excluded from new deposits / activations.
+   */
+  disabledAgents = /* @__PURE__ */ new Set();
   state = null;
   apiKey;
   zyfaiRpcUrls;
@@ -623,8 +657,12 @@ var OwneySDK = class {
       await agent.disconnect();
     }
     this.activeAgents.clear();
+    this.disabledAgents.clear();
     this.state = null;
   }
+  isAgentDisabled(agentId) {
+    return this.disabledAgents.has(agentId);
+  }
   /**
    * Check if the SDK has an active connection state.
    * @returns `true` if `connect()` has been called and `disconnect()` has not
@@ -676,11 +714,15 @@ var OwneySDK = class {
     if (!this.initializingAgentsPromise) {
       this.initializingAgentsPromise = (async () => {
         const agentKeys = await fetchAgentKeys(this.apiKey);
-        for (const { key, agent_type } of agentKeys) {
+        this.disabledAgents.clear();
+        for (const { key, agent_type, is_enabled } of agentKeys) {
           const agent = this.createAgent(agent_type, key);
           if (!agent)
             continue;
           this.agents.set(agent_type, agent);
+          if (is_enabled === false) {
+            this.disabledAgents.add(agent_type);
+          }
         }
         if (this.agents.size === 0) {
           throw new OwneyError(
@@ -725,6 +767,16 @@ var OwneySDK = class {
       }
       const uniqueIds = [...new Set(agentId)];
       const agents = uniqueIds.map((id) => this.getAgent(id));
+      const disabledRequested = uniqueIds.filter(
+        (id) => this.isAgentDisabled(id)
+      );
+      if (disabledRequested.length > 0) {
+        throw new OwneyError(
+          "AGENT_DISABLED",
+          `Cannot activate disabled agent(s): ${disabledRequested.join(", ")}.`,
+          { agentIds: disabledRequested }
+        );
+      }
       const incompatibleAgents = agents.filter(
         (agent) => !agent.supportedChainIds.includes(chainId)
       ).map((agent) => ({
@@ -763,8 +815,11 @@ var OwneySDK = class {
       this.activeAgents.add(agent.id);
     }
     state.chainId = chainId;
+    const enabledCompatible = compatible.filter(
+      (agent) => !this.isAgentDisabled(agent.id)
+    );
     Promise.all(
-      compatible.map(async (agent) => {
+      enabledCompatible.map(async (agent) => {
         await agent.activateAgent(state, chainId);
       })
     ).catch((error) => {
@@ -791,13 +846,22 @@ var OwneySDK = class {
     const state = this.requireState();
     const chainId = this.requireChainId();
     if (agentId) {
+      if (this.isAgentDisabled(agentId)) {
+        throw new OwneyError(
+          "AGENT_DISABLED",
+          `Cannot deposit into disabled agent "${agentId}". Existing balances remain visible and withdrawable.`,
+          { agentId }
+        );
+      }
       const agent = this.getAgent(agentId);
       this.validateAssetSupport(agent, chainId, asset);
       this.validateMinDepositAmount(agent, chainId, asset, amount);
       return agent.deposit(state, chainId, amount, depositCallback);
     }
     const totalAmount = BigInt(amount);
-    const eligibleAgents = this.getEligibleAgents(chainId, asset);
+    const eligibleAgents = this.getEligibleAgents(chainId, asset, {
+      excludeDisabled: true
+    });
     const agentAmounts = this.splitDepositAmount(
       totalAmount,
       eligibleAgents,
@@ -881,8 +945,11 @@ var OwneySDK = class {
       );
     }
   }
-  getEligibleAgents(chainId, asset) {
+  getEligibleAgents(chainId, asset, opts) {
     const eligible = [...this.agents.values()].filter((agent) => {
+      if (opts?.excludeDisabled && this.isAgentDisabled(agent.id)) {
+        return false;
+      }
       const chainAssets = agent.supportedAssets.find(
         (sa) => sa.chainId === chainId
       );
@@ -949,16 +1016,13 @@ var OwneySDK = class {
     }
     const requested = BigInt(amount);
     const aggregated = await this.getBalances();
-    const balances = eligibleAgents.map((agent) => {
-      const agentBalance = aggregated.agentBalances[agent.id];
-      const tokenBalance = agentBalance?.tokens.find(
-        (t) => t.chainId === chainId && t.asset.toUpperCase() === asset.toUpperCase()
-      );
-      if (!tokenBalance)
-        return { agent, balance: 0n };
-      const balance = parseUnits(tokenBalance.amount, assetInfo.decimals);
-      return { agent, balance };
-    });
+    const balances = projectAgentBalancesForAsset(
+      eligibleAgents,
+      aggregated.agentBalances,
+      chainId,
+      asset,
+      assetInfo.decimals
+    );
     const totalAvailable = balances.reduce((sum, x) => sum + x.balance, 0n);
     if (totalAvailable < requested) {
       throw new OwneyError(
@@ -971,7 +1035,26 @@ var OwneySDK = class {
         }
       );
     }
-    const plans = planProportionalShares(balances, requested, totalAvailable);
+    const disabledBalances = balances.filter(
+      (b) => this.isAgentDisabled(b.agent.id)
+    );
+    const enabledBalances = balances.filter(
+      (b) => !this.isAgentDisabled(b.agent.id)
+    );
+    const { plans: disabledPlans, remaining: afterDrain } = planDisabledDrain(
+      disabledBalances,
+      requested
+    );
+    const enabledTotal = enabledBalances.reduce(
+      (s, b) => s + b.balance,
+      0n
+    );
+    const enabledPlans = afterDrain > 0n ? planProportionalShares(enabledBalances, afterDrain, enabledTotal) : enabledBalances.map(({ agent, balance }) => ({
+      agent,
+      balance,
+      planned: 0n
+    }));
+    const plans = [...disabledPlans, ...enabledPlans];
     const results = {};
     const agentErrors = {};
     for (let i = 0; i < plans.length; i++) {
@@ -988,7 +1071,9 @@ var OwneySDK = class {
       } catch (err) {
         console.error(`withdraw failed for agent "${p.agent.id}":`, err);
         agentErrors[p.agent.id] = err instanceof Error ? err.message : String(err);
-        redistributeShare(plans, i, p.planned);
+        const failedAmount = p.planned;
+        p.planned = 0n;
+        redistributeShare(plans, i, failedAmount);
       }
     }
     if (Object.keys(results).length === 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@owney/sdk",
-  "version": "0.2.17",
+  "version": "0.3.0-beta.0",
   "type": "module",
   "main": "dist/index.cjs",
   "module": "dist/index.js",
@@ -17,16 +17,19 @@
   ],
   "scripts": {
     "build": "tsup",
-    "dev": "tsup --watch"
+    "dev": "tsup --watch",
+    "test": "vitest run",
+    "test:watch": "vitest"
   },
   "dependencies": {
-    "@zyfai/sdk": "^0.2.34",
+    "@zyfai/sdk": "^0.2.33",
     "viem": "^2.48.1"
   },
   "devDependencies": {
     "dotenv": "^17.3.1",
     "tsup": "^7.0.0",
-    "typescript": "^5.0.0"
+    "typescript": "^5.0.0",
+    "vitest": "^4.1.5"
   },
   "publishConfig": {
     "access": "public"