@peridot-agent/agent-kit 0.2.0 → 0.2.1

package/dist/adapters/mcp/server.js

@@ -414,6 +414,121 @@ async function listMarkets(input, config2) {
 
 // src/features/lending/read/get-portfolio.ts
 import { z as z3 } from "zod";
+init_constants();
+
+// src/shared/on-chain-position.ts
+import Decimal from "decimal.js";
+import { createPublicClient, http } from "viem";
+
+// src/shared/abis.ts
+import { parseAbi } from "viem";
+var PTOKEN_ABI = parseAbi([
+  "function mint(uint256 mintAmount) returns (uint256)",
+  "function redeem(uint256 redeemTokens) returns (uint256)",
+  "function redeemUnderlying(uint256 redeemAmount) returns (uint256)",
+  "function borrow(uint256 borrowAmount) returns (uint256)",
+  "function repayBorrow(uint256 repayAmount) returns (uint256)",
+  "function repayBorrowBehalf(address borrower, uint256 repayAmount) returns (uint256)",
+  "function liquidateBorrow(address borrower, uint256 repayAmount, address pTokenCollateral) returns (uint256)",
+  "function balanceOf(address owner) view returns (uint256)",
+  "function borrowBalanceStored(address account) view returns (uint256)",
+  "function supplyRatePerBlock() view returns (uint256)",
+  "function borrowRatePerBlock() view returns (uint256)",
+  "function exchangeRateStored() view returns (uint256)",
+  "function underlying() view returns (address)"
+]);
+var COMPTROLLER_ABI = parseAbi([
+  "function enterMarkets(address[] calldata pTokens) returns (uint256[] memory)",
+  "function exitMarket(address pTokenAddress) returns (uint256)",
+  "function getAccountLiquidity(address account) view returns (uint256 errorCode, uint256 liquidity, uint256 shortfall)",
+  "function getAllMarkets() view returns (address[] memory)",
+  "function markets(address pToken) view returns (bool isListed, uint256 collateralFactorMantissa, bool isComped)",
+  "function checkMembership(address account, address pToken) view returns (bool)"
+]);
+var ERC20_ABI = parseAbi([
+  "function approve(address spender, uint256 amount) returns (bool)",
+  "function allowance(address owner, address spender) view returns (uint256)",
+  "function balanceOf(address owner) view returns (uint256)",
+  "function transfer(address to, uint256 amount) returns (bool)"
+]);
+
+// src/shared/on-chain-position.ts
+init_constants();
+async function readOnChainPosition(address, chainId, config2) {
+  const markets = PERIDOT_MARKETS[chainId];
+  if (!markets) {
+    throw new Error(`No markets configured for chain ${chainId}`);
+  }
+  const marketEntries = Object.entries(markets).filter(
+    (entry) => entry[1] !== void 0
+  );
+  if (marketEntries.length === 0) {
+    return { totalSuppliedUsd: 0, totalBorrowedUsd: 0, assets: [] };
+  }
+  const rpcUrl = config2.rpcUrls?.[chainId] ?? DEFAULT_RPC_URLS[chainId];
+  if (!rpcUrl) {
+    throw new Error(
+      `No RPC URL available for chain ${chainId}. Provide one via config.rpcUrls[${chainId}].`
+    );
+  }
+  const viemClient = createPublicClient({ transport: http(rpcUrl) });
+  const apiClient = new PeridotApiClient(config2);
+  const [metricsData, multicallResults] = await Promise.all([
+    apiClient.getMarketMetrics(),
+    viemClient.multicall({
+      contracts: marketEntries.flatMap(([, pTokenAddress]) => [
+        {
+          address: pTokenAddress,
+          abi: PTOKEN_ABI,
+          functionName: "balanceOf",
+          args: [address]
+        },
+        {
+          address: pTokenAddress,
+          abi: PTOKEN_ABI,
+          functionName: "exchangeRateStored"
+        },
+        {
+          address: pTokenAddress,
+          abi: PTOKEN_ABI,
+          functionName: "borrowBalanceStored",
+          args: [address]
+        }
+      ]),
+      allowFailure: true
+    })
+  ]);
+  const assets = [];
+  let totalSuppliedUsd = 0;
+  let totalBorrowedUsd = 0;
+  for (let i = 0; i < marketEntries.length; i++) {
+    const entry = marketEntries[i];
+    if (!entry) continue;
+    const [symbol] = entry;
+    const base = i * 3;
+    const balanceResult = multicallResults[base];
+    const exchangeRateResult = multicallResults[base + 1];
+    const borrowResult = multicallResults[base + 2];
+    if (!balanceResult || !exchangeRateResult || !borrowResult || balanceResult.status === "failure" || exchangeRateResult.status === "failure" || borrowResult.status === "failure") {
+      continue;
+    }
+    const cTokenBalance = balanceResult.result;
+    const exchangeRate = exchangeRateResult.result;
+    const borrowBalance = borrowResult.result;
+    const underlyingDecimals = ASSET_DECIMALS[symbol] ?? 18;
+    const priceUsd = metricsData[`${symbol}:${chainId}`]?.priceUsd ?? 0;
+    const suppliedTokens = new Decimal(cTokenBalance.toString()).mul(exchangeRate.toString()).div("1e18").div(new Decimal(10).pow(underlyingDecimals)).toNumber();
+    const borrowedTokens = new Decimal(borrowBalance.toString()).div(new Decimal(10).pow(underlyingDecimals)).toNumber();
+    const suppliedUsd = suppliedTokens * priceUsd;
+    const borrowedUsd = borrowedTokens * priceUsd;
+    if (suppliedUsd > 1e-3 || borrowedUsd > 1e-3) {
+      assets.push({ assetId: symbol, suppliedUsd, borrowedUsd, suppliedTokens, borrowedTokens, priceUsd });
+      totalSuppliedUsd += suppliedUsd;
+      totalBorrowedUsd += borrowedUsd;
+    }
+  }
+  return { totalSuppliedUsd, totalBorrowedUsd, assets };
+}
 
 // src/shared/zod-utils.ts
 import { z as z2 } from "zod";
@@ -469,26 +584,47 @@ var Cache = class {
 // src/features/lending/read/get-portfolio.ts
 var portfolioCache = new Cache(3e4);
 var getPortfolioSchema = z3.object({
-  address: evmAddress.describe("The wallet address (0x...) to look up")
+  address: evmAddress.describe("The wallet address (0x...) to look up"),
+  chainId: z3.number().int().default(BSC_MAINNET_CHAIN_ID).refine(isHubChain, { message: "chainId must be a hub chain (56=BSC, 143=Monad, 1868=Somnia)." }).describe("Hub chain ID to query. Defaults to BSC (56).")
 });
 async function getPortfolio(input, config2) {
-  const key = input.address.toLowerCase();
+  const key = `${input.address.toLowerCase()}:${input.chainId}`;
   return portfolioCache.getOrFetch(key, async () => {
-    const client = new PeridotApiClient(config2);
-    const data = await client.getUserPortfolio(input.address);
+    const apiClient = new PeridotApiClient(config2);
+    const [position, apyData] = await Promise.all([
+      readOnChainPosition(input.address, input.chainId, config2),
+      apiClient.getMarketApy(input.chainId)
+    ]);
+    const { totalSuppliedUsd, totalBorrowedUsd, assets } = position;
+    let netApy = 0;
+    if (totalSuppliedUsd > 0) {
+      let weighted = 0;
+      for (const asset of assets) {
+        const apyEntry = apyData[asset.assetId.toLowerCase()]?.[input.chainId];
+        if (apyEntry) {
+          weighted += asset.suppliedUsd * (apyEntry.totalSupplyApy ?? 0);
+          weighted -= asset.borrowedUsd * (apyEntry.netBorrowApy ?? 0);
+        }
+      }
+      netApy = weighted / totalSuppliedUsd;
+    }
     return {
       address: input.address,
       fetchedAt: (/* @__PURE__ */ new Date()).toISOString(),
       portfolio: {
-        currentValue: data.portfolio.currentValue,
-        totalSupplied: data.portfolio.totalSupplied,
-        totalBorrowed: data.portfolio.totalBorrowed,
-        netApy: data.portfolio.netApy,
-        healthFactor: data.portfolio.totalBorrowed > 0 ? data.portfolio.totalSupplied / data.portfolio.totalBorrowed : null
+        currentValue: totalSuppliedUsd - totalBorrowedUsd,
+        totalSupplied: totalSuppliedUsd,
+        totalBorrowed: totalBorrowedUsd,
+        netApy,
+        healthFactor: totalBorrowedUsd > 0 ? totalSuppliedUsd / totalBorrowedUsd : null
       },
-      assets: data.assets,
-      transactions: data.transactions,
-      earnings: data.earnings
+      assets: assets.map((a) => ({
+        assetId: a.assetId,
+        supplied: a.suppliedUsd,
+        borrowed: a.borrowedUsd,
+        net: a.suppliedUsd - a.borrowedUsd,
+        percentage: totalSuppliedUsd > 0 ? (a.suppliedUsd - a.borrowedUsd) / totalSuppliedUsd * 100 : 0
+      }))
     };
   });
 }
@@ -570,39 +706,50 @@ async function getMarketRates(input, config2) {
 
 // src/features/lending/read/get-user-position.ts
 import { z as z6 } from "zod";
+init_constants();
 var getUserPositionSchema = z6.object({
-  address: evmAddress.describe("The wallet address (0x...) to look up")
+  address: evmAddress.describe("The wallet address (0x...) to look up"),
+  chainId: z6.number().int().default(BSC_MAINNET_CHAIN_ID).refine(isHubChain, { message: "chainId must be a hub chain (56=BSC, 143=Monad, 1868=Somnia)." }).describe("Hub chain ID to query. Defaults to BSC (56).")
 });
 async function getUserPosition(input, config2) {
-  const client = new PeridotApiClient(config2);
-  const data = await client.getUserPortfolio(input.address);
-  const { portfolio, assets, transactions } = data;
-  const healthFactor = portfolio.totalBorrowed > 0 ? portfolio.totalSupplied / portfolio.totalBorrowed : null;
+  const apiClient = new PeridotApiClient(config2);
+  const [position, apyData] = await Promise.all([
+    readOnChainPosition(input.address, input.chainId, config2),
+    apiClient.getMarketApy(input.chainId)
+  ]);
+  const { totalSuppliedUsd, totalBorrowedUsd, assets } = position;
+  const healthFactor = totalBorrowedUsd > 0 ? totalSuppliedUsd / totalBorrowedUsd : null;
+  let netApyPct = 0;
+  if (totalSuppliedUsd > 0) {
+    let weightedApy = 0;
+    for (const asset of assets) {
+      const apyEntry = apyData[asset.assetId.toLowerCase()]?.[input.chainId];
+      if (apyEntry) {
+        weightedApy += asset.suppliedUsd * (apyEntry.totalSupplyApy ?? 0);
+        weightedApy -= asset.borrowedUsd * (apyEntry.netBorrowApy ?? 0);
+      }
+    }
+    netApyPct = weightedApy / totalSuppliedUsd;
+  }
   return {
     address: input.address,
-    totalSuppliedUsd: portfolio.totalSupplied,
-    totalBorrowedUsd: portfolio.totalBorrowed,
-    netWorthUsd: portfolio.currentValue,
-    netApyPct: portfolio.netApy,
+    totalSuppliedUsd,
+    totalBorrowedUsd,
+    netWorthUsd: totalSuppliedUsd - totalBorrowedUsd,
+    netApyPct,
     healthFactor,
     assets: assets.map((a) => ({
       assetId: a.assetId,
-      suppliedUsd: a.supplied,
-      borrowedUsd: a.borrowed,
-      netUsd: a.net
+      suppliedUsd: a.suppliedUsd,
+      borrowedUsd: a.borrowedUsd,
+      netUsd: a.suppliedUsd - a.borrowedUsd
     })),
-    transactions: {
-      supplyCount: transactions.supplyCount,
-      borrowCount: transactions.borrowCount,
-      repayCount: transactions.repayCount,
-      redeemCount: transactions.redeemCount
-    },
     fetchedAt: (/* @__PURE__ */ new Date()).toISOString()
   };
 }
 
 // src/features/lending/read/simulate-borrow.ts
-import Decimal from "decimal.js";
+import Decimal2 from "decimal.js";
 import { z as z7 } from "zod";
 init_constants();
 var simulateBorrowSchema = z7.object({
@@ -625,10 +772,10 @@ function classifyRisk(hf) {
   return "liquidatable";
 }
 async function simulateBorrow(input, config2) {
-  const client = new PeridotApiClient(config2);
-  const [portfolioData, metricsData] = await Promise.all([
-    client.getUserPortfolio(input.address),
-    client.getMarketMetrics()
+  const apiClient = new PeridotApiClient(config2);
+  const [position, metricsData] = await Promise.all([
+    readOnChainPosition(input.address, input.chainId, config2),
+    apiClient.getMarketMetrics()
   ]);
   const assetUpper = input.asset.toUpperCase();
   const metricKey = `${assetUpper}:${input.chainId}`;
@@ -640,12 +787,11 @@ async function simulateBorrow(input, config2) {
   if (isNaN(borrowAmountRaw) || borrowAmountRaw <= 0) {
     throw new Error(`Invalid borrow amount: "${input.amount}"`);
   }
-  const borrowAmount = new Decimal(input.amount);
+  const borrowAmount = new Decimal2(input.amount);
   const borrowAmountUsd = borrowAmount.mul(metric.priceUsd).toNumber();
-  const { totalSupplied, totalBorrowed } = portfolioData.portfolio;
-  const currentHF = totalBorrowed > 0 ? new Decimal(totalSupplied).div(totalBorrowed).toNumber() : null;
-  const projectedBorrowedUsd = new Decimal(totalBorrowed).add(borrowAmountUsd);
-  if (totalSupplied === 0) {
+  const { totalSuppliedUsd, totalBorrowedUsd } = position;
+  const currentHF = totalBorrowedUsd > 0 ? new Decimal2(totalSuppliedUsd).div(totalBorrowedUsd).toNumber() : null;
+  if (totalSuppliedUsd === 0) {
     return {
       currentHealthFactor: null,
       projectedHealthFactor: null,
@@ -653,13 +799,14 @@ async function simulateBorrow(input, config2) {
       isSafe: false,
       riskLevel: "liquidatable",
       maxSafeBorrowUsd: 0,
-      warning: "Portfolio shows no supplied collateral (data is sourced from indexed DB snapshots and may not reflect activity from the last few minutes). If you recently supplied assets, wait for the next snapshot update. Otherwise, supply assets and enable them as collateral before borrowing."
+      warning: "No supplied collateral found on-chain for this address. Supply assets and enable them as collateral before borrowing."
     };
   }
-  const projectedHF = new Decimal(totalSupplied).div(projectedBorrowedUsd).toNumber();
-  const maxSafeBorrowUsd = Decimal.max(
+  const projectedBorrowedUsd = new Decimal2(totalBorrowedUsd).add(borrowAmountUsd);
+  const projectedHF = new Decimal2(totalSuppliedUsd).div(projectedBorrowedUsd).toNumber();
+  const maxSafeBorrowUsd = Decimal2.max(
     0,
-    new Decimal(totalSupplied).div(RISK_THRESHOLDS.safe).sub(totalBorrowed)
+    new Decimal2(totalSuppliedUsd).div(RISK_THRESHOLDS.safe).sub(totalBorrowedUsd)
   ).toNumber();
   const riskLevel = classifyRisk(projectedHF);
   const isSafe = projectedHF >= RISK_THRESHOLDS.high;
@@ -683,43 +830,9 @@ async function simulateBorrow(input, config2) {
 }
 
 // src/features/lending/read/get-account-liquidity.ts
-import Decimal2 from "decimal.js";
+import Decimal3 from "decimal.js";
 import { z as z8 } from "zod";
-import { createPublicClient, http } from "viem";
-
-// src/shared/abis.ts
-import { parseAbi } from "viem";
-var PTOKEN_ABI = parseAbi([
-  "function mint(uint256 mintAmount) returns (uint256)",
-  "function redeem(uint256 redeemTokens) returns (uint256)",
-  "function redeemUnderlying(uint256 redeemAmount) returns (uint256)",
-  "function borrow(uint256 borrowAmount) returns (uint256)",
-  "function repayBorrow(uint256 repayAmount) returns (uint256)",
-  "function repayBorrowBehalf(address borrower, uint256 repayAmount) returns (uint256)",
-  "function liquidateBorrow(address borrower, uint256 repayAmount, address pTokenCollateral) returns (uint256)",
-  "function balanceOf(address owner) view returns (uint256)",
-  "function borrowBalanceStored(address account) view returns (uint256)",
-  "function supplyRatePerBlock() view returns (uint256)",
-  "function borrowRatePerBlock() view returns (uint256)",
-  "function exchangeRateStored() view returns (uint256)",
-  "function underlying() view returns (address)"
-]);
-var COMPTROLLER_ABI = parseAbi([
-  "function enterMarkets(address[] calldata pTokens) returns (uint256[] memory)",
-  "function exitMarket(address pTokenAddress) returns (uint256)",
-  "function getAccountLiquidity(address account) view returns (uint256 errorCode, uint256 liquidity, uint256 shortfall)",
-  "function getAllMarkets() view returns (address[] memory)",
-  "function markets(address pToken) view returns (bool isListed, uint256 collateralFactorMantissa, bool isComped)",
-  "function checkMembership(address account, address pToken) view returns (bool)"
-]);
-var ERC20_ABI = parseAbi([
-  "function approve(address spender, uint256 amount) returns (bool)",
-  "function allowance(address owner, address spender) view returns (uint256)",
-  "function balanceOf(address owner) view returns (uint256)",
-  "function transfer(address to, uint256 amount) returns (bool)"
-]);
-
-// src/features/lending/read/get-account-liquidity.ts
+import { createPublicClient as createPublicClient2, http as http2 } from "viem";
 init_constants();
 var getAccountLiquiditySchema = z8.object({
   address: evmAddress.describe("The wallet address to check"),
@@ -733,7 +846,7 @@ async function getAccountLiquidity(input, config2) {
     );
   }
   const controllerAddress = getControllerAddress(input.chainId);
-  const client = createPublicClient({ transport: http(rpcUrl) });
+  const client = createPublicClient2({ transport: http2(rpcUrl) });
   const [error, liquidity, shortfall] = await client.readContract({
     address: controllerAddress,
     abi: COMPTROLLER_ABI,
@@ -743,8 +856,8 @@ async function getAccountLiquidity(input, config2) {
   if (error !== 0n) {
     throw new Error(`Comptroller getAccountLiquidity returned error code ${error.toString()}`);
   }
-  const liquidityUsd = new Decimal2(liquidity.toString()).div("1e18").toNumber();
-  const shortfallUsd = new Decimal2(shortfall.toString()).div("1e18").toNumber();
+  const liquidityUsd = new Decimal3(liquidity.toString()).div("1e18").toNumber();
+  const shortfallUsd = new Decimal3(shortfall.toString()).div("1e18").toNumber();
   return {
     address: input.address,
     chainId: input.chainId,