@t2000/sdk 0.15.2 → 0.15.3

package/dist/index.js

@@ -3054,9 +3054,12 @@ To access invested funds: t2000 invest sell ${params.amount} ${asset}`,
   }
   async balance() {
     const bal = await queryBalance(this.client, this._address);
+    const earningAssets = new Set(
+      this.portfolio.getPositions().filter((p) => p.earning).map((p) => p.asset)
+    );
     try {
       const positions = await this.positions();
-      const savings = positions.positions.filter((p) => p.type === "save").reduce((sum, p) => sum + p.amount, 0);
+      const savings = positions.positions.filter((p) => p.type === "save").filter((p) => !earningAssets.has(p.asset)).reduce((sum, p) => sum + p.amount, 0);
       const debt = positions.positions.filter((p) => p.type === "borrow").reduce((sum, p) => sum + p.amount, 0);
       bal.savings = savings;
       bal.debt = debt;
@@ -3084,7 +3087,14 @@ To access invested funds: t2000 invest sell ${params.amount} ${asset}`,
       for (const pos of portfolioPositions) {
         if (!(pos.asset in INVESTMENT_ASSETS)) continue;
         const price = assetPrices[pos.asset] ?? 0;
-        if (pos.asset === "SUI") {
+        if (pos.earning) {
+          investmentValue += pos.totalAmount * price;
+          investmentCostBasis += pos.costBasis;
+          if (pos.asset === "SUI") {
+            const gasSui = Math.max(0, bal.gasReserve.sui);
+            bal.gasReserve = { sui: gasSui, usdEquiv: gasSui * price };
+          }
+        } else if (pos.asset === "SUI") {
           const actualHeld = Math.min(pos.totalAmount, bal.gasReserve.sui);
           investmentValue += actualHeld * price;
           if (actualHeld < pos.totalAmount && pos.totalAmount > 0) {
@@ -4017,7 +4027,7 @@ To sell investment: t2000 invest sell ${params.amount} ${fromAsset}`,
       const currentPrice = prices[pos.asset] ?? 0;
       let totalAmount = pos.totalAmount;
       let costBasis = pos.costBasis;
-      if (pos.asset in INVESTMENT_ASSETS) {
+      if (pos.asset in INVESTMENT_ASSETS && !pos.earning) {
         try {
           const assetInfo = SUPPORTED_ASSETS[pos.asset];
           const bal = await this.client.getBalance({ owner: this._address, coinType: assetInfo.type });