@owney/sdk 0.2.18 → 0.3.0-beta.0

package/README.md

@@ -280,18 +280,12 @@ console.log(apy.walletAddress); // "0x..."
 console.log(apy.weightedApyAfterFee); // 5.2
 // Per chain + token breakdown, when the agent reports it
 console.log(apy.weightedApyAfterFeeDetails?.apyPerAsset[8453]?.USDC); // 5.4
-// Same per-(chain, asset) breakdown surfaced under a unified key
-console.log(apy.apyByChainAndAsset?.[8453]?.USDC); // 5.4
 
 // All agents (balance-weighted)
 const allApy = await sdk.getAccountApy({ days: "30D" });
 console.log(allApy.totalApy); // "5.6"
 console.log(allApy.agentApy.zyfai.weightedApyAfterFee); // 6.0
 console.log(allApy.agentApy.sail.weightedApyAfterFee); // 4.0
-// Per (chain, asset), balance-weighted across all agents.
-// Cells with 0/missing APY for a given agent are excluded from that cell.
-console.log(allApy.apyByChainAndAsset[8453]?.USDC); // 5.3
-console.log(allApy.apyByChainAndAsset[42161]?.USDT); // 4.7
 ```
 
 ### `getHistory({ agentId?, filters? })`
@@ -566,21 +560,15 @@ interface OwneyEarnings {
   agentEarnings: Record<AgentId, AgentEarnings>;
 }
 
-type ApyByChainAndAsset = Partial<
-  Record<OwneySupportedChainId, Partial<Record<OwneySupportedTokens, number>>>
->;
-
 interface AccountAgentApy {
   walletAddress: string;
   weightedApyAfterFee?: number;
   weightedApyAfterFeeDetails?: AgentApyDetails;
-  apyByChainAndAsset?: ApyByChainAndAsset;
 }
 
 interface OwneyAccountApy {
   totalApy: string;
   agentApy: Record<AgentId, AccountAgentApy>;
-  apyByChainAndAsset: ApyByChainAndAsset;
 }
 
 type HistoryAction = "Rebalance" | "Deposit" | "Top up" | "Withdraw" | "Earned";

package/dist/index.cjs

@@ -36,7 +36,6 @@ var P_SUPPORTED_CHAINS = {
   chainIds: [8453, 42161],
   chains: ["BASE", "ARBITRUM"]
 };
-var SUPPORTED_TOKENS = ["USDC", "USDT"];
 var SUPPORTED_CHAIN_IDS = P_SUPPORTED_CHAINS.chainIds;
 var SUPPORTED_CHAINS = P_SUPPORTED_CHAINS.chains;
 var SupportedAssets = [
@@ -210,31 +209,10 @@ function mapEarnings(raw, smartWallet) {
     tokens
   };
 }
-function mapWeightedApyByChain(raw) {
-  if (!raw)
-    return void 0;
-  const out = {};
-  for (const [chainKey, tokenApy] of Object.entries(raw)) {
-    const chainId = Number(chainKey);
-    if (!SUPPORTED_CHAIN_IDS.includes(chainId))
-      continue;
-    const perAsset = {};
-    for (const [symbol, apy] of Object.entries(tokenApy)) {
-      const token = symbol;
-      if (!SUPPORTED_TOKENS.includes(token))
-        continue;
-      perAsset[token] = apy;
-    }
-    if (Object.keys(perAsset).length > 0)
-      out[chainId] = perAsset;
-  }
-  return Object.keys(out).length > 0 ? out : void 0;
-}
 function mapApyHistory(raw) {
   return {
     walletAddress: raw.walletAddress,
-    weightedApyAfterFee: raw.weightedApyAfterFee ? sumTokenValues(raw.weightedApyAfterFee) : void 0,
-    apyByChainAndAsset: mapWeightedApyByChain(raw.weightedApyAfterFeeByChain)
+    weightedApyAfterFee: raw.weightedApyAfterFee ? sumTokenValues(raw.weightedApyAfterFee) : void 0
   };
 }
 function mapHistory(raw, chainId) {
@@ -578,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,
@@ -602,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
@@ -634,51 +641,16 @@ function sumWithdrawnAmount(results) {
   return Object.values(results).reduce((sum, r) => sum + BigInt(r.amount), 0n);
 }
 
-// src/lib/helpers/account-apy-helper.ts
-function aggregateApyByChainAndAsset(agentApys, agentBalances) {
-  const sums = {};
-  const weights = {};
-  for (const id of Object.keys(agentApys)) {
-    const cells = agentApys[id].apyByChainAndAsset;
-    const balance = agentBalances[id] ?? 0;
-    if (!cells || balance <= 0)
-      continue;
-    for (const [chainKey, perAsset] of Object.entries(cells)) {
-      if (!perAsset)
-        continue;
-      const chainId = Number(chainKey);
-      for (const [asset, apyValue] of Object.entries(perAsset)) {
-        const apy = Number(apyValue ?? 0);
-        if (apy === 0)
-          continue;
-        sums[chainId] ??= {};
-        weights[chainId] ??= {};
-        sums[chainId][asset] = (sums[chainId][asset] ?? 0) + apy * balance;
-        weights[chainId][asset] = (weights[chainId][asset] ?? 0) + balance;
-      }
-    }
-  }
-  const out = {};
-  for (const chainKey of Object.keys(sums)) {
-    const chainId = Number(chainKey);
-    const perAssetOut = {};
-    for (const asset of Object.keys(sums[chainId])) {
-      const w = weights[chainId][asset];
-      if (w > 0)
-        perAssetOut[asset] = sums[chainId][asset] / w;
-    }
-    if (Object.keys(perAssetOut).length > 0) {
-      out[chainId] = perAssetOut;
-    }
-  }
-  return out;
-}
-
 // 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;
@@ -715,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
@@ -768,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(
@@ -817,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) => ({
@@ -855,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) => {
@@ -883,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,
@@ -973,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
       );
@@ -1041,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(
@@ -1063,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++) {
@@ -1080,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) {
@@ -1246,11 +1269,9 @@ var OwneySDK = class {
         totalApy += agentApy * weight;
       }
     }
-    const apyByChainAndAsset = aggregateApyByChainAndAsset(results, balances);
     return {
       totalApy: String(totalApy),
-      agentApy: results,
-      apyByChainAndAsset
+      agentApy: results
     };
   }
   /**

package/dist/index.d.cts

@@ -139,17 +139,14 @@ interface OwneyEarnings {
     totalEarnings: string;
     agentEarnings: Record<AgentId, AgentEarnings>;
 }
-type ApyByChainAndAsset = Partial<Record<OwneySupportedChainId, Partial<Record<OwneySupportedTokens, number>>>>;
 interface AccountAgentApy {
     walletAddress: string;
     weightedApyAfterFee?: number;
     weightedApyAfterFeeDetails?: AgentApyDetails;
-    apyByChainAndAsset?: ApyByChainAndAsset;
 }
 interface OwneyAccountApy {
     totalApy: string;
     agentApy: Record<AgentId, AccountAgentApy>;
-    apyByChainAndAsset: ApyByChainAndAsset;
 }
 type HistoryAction = "Rebalance" | "Deposit" | "Top up" | "Withdraw" | "Earned";
 interface HistoryTransaction {
@@ -211,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?;
@@ -227,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
@@ -349,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>;
@@ -374,4 +378,4 @@ declare class AgentChainIncompatibleError extends OwneyError {
     }>, connectedChainId: number);
 }
 
-export { AccountAgentApy, AccountApyOptions, AgentApy, AgentApyDetails, AgentBalance, AgentChainIncompatibleError, AgentEarnings, AgentHistoryEntry, AgentId, AgentNotFoundError, AgentSupportedAsset, AgentSupportedAssets, AgentUserProfile, AgentWithdrawResult, AgentsApyOptions, ApyByChainAndAsset, Asset, ConnectionState, DailyApyDays, DepositCallback, DepositOptions, HistoryAction, HistoryFilters, HistoryOptions, HistoryTransaction, IAgent, NotConnectedError, OwneyAccountApy, OwneyAgentAllocation, OwneyAgentApy, OwneyAgentHistory, OwneyAgentStatus, OwneyBalances, OwneyDepositResult, OwneyEarnings, OwneyError, OwneyErrorCode, OwneyMultiDepositResult, OwneySDK, OwneySDKConfig, OwneySupportedChainId, OwneySupportedChains, OwneySupportedTokens, OwneyToken, OwneyUserProfile, OwneyWithdrawResult, RebalanceLog, WithdrawOptions };
+export { AccountAgentApy, AccountApyOptions, AgentApy, AgentApyDetails, AgentBalance, AgentChainIncompatibleError, AgentEarnings, AgentHistoryEntry, AgentId, AgentNotFoundError, AgentSupportedAsset, AgentSupportedAssets, AgentUserProfile, AgentWithdrawResult, AgentsApyOptions, Asset, ConnectionState, DailyApyDays, DepositCallback, DepositOptions, HistoryAction, HistoryFilters, HistoryOptions, HistoryTransaction, IAgent, NotConnectedError, OwneyAccountApy, OwneyAgentAllocation, OwneyAgentApy, OwneyAgentHistory, OwneyAgentStatus, OwneyBalances, OwneyDepositResult, OwneyEarnings, OwneyError, OwneyErrorCode, OwneyMultiDepositResult, OwneySDK, OwneySDKConfig, OwneySupportedChainId, OwneySupportedChains, OwneySupportedTokens, OwneyToken, OwneyUserProfile, OwneyWithdrawResult, RebalanceLog, WithdrawOptions };

package/dist/index.d.ts

@@ -139,17 +139,14 @@ interface OwneyEarnings {
     totalEarnings: string;
     agentEarnings: Record<AgentId, AgentEarnings>;
 }
-type ApyByChainAndAsset = Partial<Record<OwneySupportedChainId, Partial<Record<OwneySupportedTokens, number>>>>;
 interface AccountAgentApy {
     walletAddress: string;
     weightedApyAfterFee?: number;
     weightedApyAfterFeeDetails?: AgentApyDetails;
-    apyByChainAndAsset?: ApyByChainAndAsset;
 }
 interface OwneyAccountApy {
     totalApy: string;
     agentApy: Record<AgentId, AccountAgentApy>;
-    apyByChainAndAsset: ApyByChainAndAsset;
 }
 type HistoryAction = "Rebalance" | "Deposit" | "Top up" | "Withdraw" | "Earned";
 interface HistoryTransaction {
@@ -211,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?;
@@ -227,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
@@ -349,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>;
@@ -374,4 +378,4 @@ declare class AgentChainIncompatibleError extends OwneyError {
     }>, connectedChainId: number);
 }
 
-export { AccountAgentApy, AccountApyOptions, AgentApy, AgentApyDetails, AgentBalance, AgentChainIncompatibleError, AgentEarnings, AgentHistoryEntry, AgentId, AgentNotFoundError, AgentSupportedAsset, AgentSupportedAssets, AgentUserProfile, AgentWithdrawResult, AgentsApyOptions, ApyByChainAndAsset, Asset, ConnectionState, DailyApyDays, DepositCallback, DepositOptions, HistoryAction, HistoryFilters, HistoryOptions, HistoryTransaction, IAgent, NotConnectedError, OwneyAccountApy, OwneyAgentAllocation, OwneyAgentApy, OwneyAgentHistory, OwneyAgentStatus, OwneyBalances, OwneyDepositResult, OwneyEarnings, OwneyError, OwneyErrorCode, OwneyMultiDepositResult, OwneySDK, OwneySDKConfig, OwneySupportedChainId, OwneySupportedChains, OwneySupportedTokens, OwneyToken, OwneyUserProfile, OwneyWithdrawResult, RebalanceLog, WithdrawOptions };
+export { AccountAgentApy, AccountApyOptions, AgentApy, AgentApyDetails, AgentBalance, AgentChainIncompatibleError, AgentEarnings, AgentHistoryEntry, AgentId, AgentNotFoundError, AgentSupportedAsset, AgentSupportedAssets, AgentUserProfile, AgentWithdrawResult, AgentsApyOptions, Asset, ConnectionState, DailyApyDays, DepositCallback, DepositOptions, HistoryAction, HistoryFilters, HistoryOptions, HistoryTransaction, IAgent, NotConnectedError, OwneyAccountApy, OwneyAgentAllocation, OwneyAgentApy, OwneyAgentHistory, OwneyAgentStatus, OwneyBalances, OwneyDepositResult, OwneyEarnings, OwneyError, OwneyErrorCode, OwneyMultiDepositResult, OwneySDK, OwneySDKConfig, OwneySupportedChainId, OwneySupportedChains, OwneySupportedTokens, OwneyToken, OwneyUserProfile, OwneyWithdrawResult, RebalanceLog, WithdrawOptions };

package/dist/index.js

@@ -6,7 +6,6 @@ var P_SUPPORTED_CHAINS = {
   chainIds: [8453, 42161],
   chains: ["BASE", "ARBITRUM"]
 };
-var SUPPORTED_TOKENS = ["USDC", "USDT"];
 var SUPPORTED_CHAIN_IDS = P_SUPPORTED_CHAINS.chainIds;
 var SUPPORTED_CHAINS = P_SUPPORTED_CHAINS.chains;
 var SupportedAssets = [
@@ -180,31 +179,10 @@ function mapEarnings(raw, smartWallet) {
     tokens
   };
 }
-function mapWeightedApyByChain(raw) {
-  if (!raw)
-    return void 0;
-  const out = {};
-  for (const [chainKey, tokenApy] of Object.entries(raw)) {
-    const chainId = Number(chainKey);
-    if (!SUPPORTED_CHAIN_IDS.includes(chainId))
-      continue;
-    const perAsset = {};
-    for (const [symbol, apy] of Object.entries(tokenApy)) {
-      const token = symbol;
-      if (!SUPPORTED_TOKENS.includes(token))
-        continue;
-      perAsset[token] = apy;
-    }
-    if (Object.keys(perAsset).length > 0)
-      out[chainId] = perAsset;
-  }
-  return Object.keys(out).length > 0 ? out : void 0;
-}
 function mapApyHistory(raw) {
   return {
     walletAddress: raw.walletAddress,
-    weightedApyAfterFee: raw.weightedApyAfterFee ? sumTokenValues(raw.weightedApyAfterFee) : void 0,
-    apyByChainAndAsset: mapWeightedApyByChain(raw.weightedApyAfterFeeByChain)
+    weightedApyAfterFee: raw.weightedApyAfterFee ? sumTokenValues(raw.weightedApyAfterFee) : void 0
   };
 }
 function mapHistory(raw, chainId) {
@@ -548,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,
@@ -572,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
@@ -604,51 +611,16 @@ function sumWithdrawnAmount(results) {
   return Object.values(results).reduce((sum, r) => sum + BigInt(r.amount), 0n);
 }
 
-// src/lib/helpers/account-apy-helper.ts
-function aggregateApyByChainAndAsset(agentApys, agentBalances) {
-  const sums = {};
-  const weights = {};
-  for (const id of Object.keys(agentApys)) {
-    const cells = agentApys[id].apyByChainAndAsset;
-    const balance = agentBalances[id] ?? 0;
-    if (!cells || balance <= 0)
-      continue;
-    for (const [chainKey, perAsset] of Object.entries(cells)) {
-      if (!perAsset)
-        continue;
-      const chainId = Number(chainKey);
-      for (const [asset, apyValue] of Object.entries(perAsset)) {
-        const apy = Number(apyValue ?? 0);
-        if (apy === 0)
-          continue;
-        sums[chainId] ??= {};
-        weights[chainId] ??= {};
-        sums[chainId][asset] = (sums[chainId][asset] ?? 0) + apy * balance;
-        weights[chainId][asset] = (weights[chainId][asset] ?? 0) + balance;
-      }
-    }
-  }
-  const out = {};
-  for (const chainKey of Object.keys(sums)) {
-    const chainId = Number(chainKey);
-    const perAssetOut = {};
-    for (const asset of Object.keys(sums[chainId])) {
-      const w = weights[chainId][asset];
-      if (w > 0)
-        perAssetOut[asset] = sums[chainId][asset] / w;
-    }
-    if (Object.keys(perAssetOut).length > 0) {
-      out[chainId] = perAssetOut;
-    }
-  }
-  return out;
-}
-
 // 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;
@@ -685,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
@@ -738,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(
@@ -787,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) => ({
@@ -825,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) => {
@@ -853,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,
@@ -943,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
       );
@@ -1011,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(
@@ -1033,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++) {
@@ -1050,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) {
@@ -1216,11 +1239,9 @@ var OwneySDK = class {
         totalApy += agentApy * weight;
       }
     }
-    const apyByChainAndAsset = aggregateApyByChainAndAsset(results, balances);
     return {
       totalApy: String(totalApy),
-      agentApy: results,
-      apyByChainAndAsset
+      agentApy: results
     };
   }
   /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@owney/sdk",
-  "version": "0.2.18",
+  "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"