@t2000/engine 0.5.9 → 0.5.10

package/dist/index.d.ts

@@ -92,6 +92,8 @@ interface ToolContext {
     walletAddress?: string;
     suiRpcUrl?: string;
     serverPositions?: ServerPositionData;
+    /** Fresh on-chain position reader — bypasses MCP caching. If provided, read tools prefer this. */
+    positionFetcher?: (address: string) => Promise<ServerPositionData>;
     signal?: AbortSignal;
 }
 interface ServerPositionData {
@@ -138,6 +140,8 @@ interface EngineConfig {
     walletAddress?: string;
     suiRpcUrl?: string;
     serverPositions?: ServerPositionData;
+    /** Fresh on-chain position reader — called per tool invocation, bypasses MCP caching. */
+    positionFetcher?: (address: string) => Promise<ServerPositionData>;
     tools?: Tool[];
     systemPrompt?: string;
     model?: string;
@@ -248,6 +252,7 @@ declare class QueryEngine {
     private readonly walletAddress;
     private readonly suiRpcUrl;
     private serverPositions;
+    private readonly positionFetcher;
     private readonly txMutex;
     private readonly costTracker;
     private messages;
@@ -788,12 +793,12 @@ declare const balanceCheckTool: Tool<{}, {
 declare const savingsInfoTool: Tool<{}, SavingsResult>;
 
 declare const healthCheckTool: Tool<{}, {
-    status: string;
     healthFactor: number;
     supplied: number;
     borrowed: number;
     maxBorrow: number;
     liquidationThreshold: number;
+    status: string;
 }>;
 
 declare const ratesInfoTool: Tool<{}, _t2000_sdk.RatesResult>;

package/dist/index.js

@@ -555,11 +555,21 @@ var balanceCheckTool = buildTool({
           });
         }
       }
-      const posEntries = transformPositions(positions);
-      const rewardEntries = transformRewards(rewards);
-      const savings = posEntries.filter((p) => p.type === "supply").reduce((sum, p) => sum + p.valueUsd, 0);
-      const debt = posEntries.filter((p) => p.type === "borrow").reduce((sum, p) => sum + p.valueUsd, 0);
-      const pendingRewardsUsd = rewardEntries.reduce((sum, r) => sum + r.valueUsd, 0);
+      let savings;
+      let debt;
+      let pendingRewardsUsd;
+      if (context.positionFetcher && context.walletAddress) {
+        const sp = await context.positionFetcher(context.walletAddress);
+        savings = sp.savings;
+        debt = sp.borrows;
+        pendingRewardsUsd = sp.pendingRewards;
+      } else {
+        const posEntries = transformPositions(positions);
+        const rewardEntries = transformRewards(rewards);
+        savings = posEntries.filter((p) => p.type === "supply").reduce((sum, p) => sum + p.valueUsd, 0);
+        debt = posEntries.filter((p) => p.type === "borrow").reduce((sum, p) => sum + p.valueUsd, 0);
+        pendingRewardsUsd = rewardEntries.reduce((sum, r) => sum + r.valueUsd, 0);
+      }
       const bal = {
         available: availableUsd,
         savings,
@@ -683,6 +693,47 @@ async function fetchProtocolStats(manager, opts) {
 }
 
 // src/tools/savings.ts
+function buildSavingsFromPositions(sp) {
+  const positions = [
+    ...sp.supplies.map((s) => ({
+      protocol: s.protocol,
+      type: "supply",
+      symbol: s.asset,
+      amount: s.amount,
+      valueUsd: s.amountUsd,
+      apy: s.apy,
+      liquidationThreshold: 0
+    })),
+    ...sp.borrows_detail.map((b) => ({
+      protocol: b.protocol,
+      type: "borrow",
+      symbol: b.asset,
+      amount: b.amount,
+      valueUsd: b.amountUsd,
+      apy: b.apy,
+      liquidationThreshold: 0
+    }))
+  ];
+  const supplied = sp.savings;
+  const weightedApy = supplied > 0 ? sp.savingsRate : 0;
+  const dailyEarning = supplied * weightedApy / 365;
+  return {
+    positions,
+    earnings: {
+      totalYieldEarned: 0,
+      currentApy: weightedApy,
+      dailyEarning,
+      supplied
+    },
+    fundStatus: {
+      supplied,
+      apy: weightedApy,
+      earnedToday: dailyEarning,
+      earnedAllTime: 0,
+      projectedMonthly: dailyEarning * 30
+    }
+  };
+}
 var savingsInfoTool = buildTool({
   name: "savings_info",
   description: "Get detailed savings positions and earnings: current deposits by protocol, APY, total yield earned, daily earning rate, and projected monthly returns.",
@@ -690,6 +741,10 @@ var savingsInfoTool = buildTool({
   jsonSchema: { type: "object", properties: {}, required: [] },
   isReadOnly: true,
   async call(_input, context) {
+    if (context.positionFetcher && context.walletAddress) {
+      const sp = await context.positionFetcher(context.walletAddress);
+      return { data: buildSavingsFromPositions(sp) };
+    }
     if (hasNaviMcp(context)) {
       const savings = await fetchSavings(
         getMcpManager(context),
@@ -745,6 +800,23 @@ var healthCheckTool = buildTool({
   jsonSchema: { type: "object", properties: {}, required: [] },
   isReadOnly: true,
   async call(_input, context) {
+    if (context.positionFetcher && context.walletAddress) {
+      const sp = await context.positionFetcher(context.walletAddress);
+      const hfVal = sp.healthFactor ?? (sp.borrows > 0 ? 0 : Infinity);
+      const status2 = hfStatus(hfVal);
+      const displayHf = Number.isFinite(hfVal) ? hfVal.toFixed(2) : "\u221E";
+      return {
+        data: {
+          healthFactor: hfVal,
+          supplied: sp.savings,
+          borrowed: sp.borrows,
+          maxBorrow: sp.maxBorrow,
+          liquidationThreshold: 0,
+          status: status2
+        },
+        displayText: `Health Factor: ${displayHf} (${status2})`
+      };
+    }
     if (hasNaviMcp(context)) {
       const hf2 = await fetchHealthFactor(
         getMcpManager(context),
@@ -1865,6 +1937,7 @@ var QueryEngine = class {
   walletAddress;
   suiRpcUrl;
   serverPositions;
+  positionFetcher;
   txMutex = new TxMutex();
   costTracker;
   messages = [];
@@ -1876,6 +1949,7 @@ var QueryEngine = class {
     this.walletAddress = config.walletAddress;
     this.suiRpcUrl = config.suiRpcUrl;
     this.serverPositions = config.serverPositions;
+    this.positionFetcher = config.positionFetcher;
     this.model = config.model;
     this.maxTurns = config.maxTurns ?? DEFAULT_MAX_TURNS;
     this.maxTokens = config.maxTokens ?? DEFAULT_MAX_TOKENS;
@@ -1950,6 +2024,9 @@ var QueryEngine = class {
       yield { type: "turn_complete", stopReason: "end_turn" };
       return;
     }
+    if (this.mcpManager && typeof this.mcpManager.cache?.invalidate === "function") {
+      this.mcpManager.cache.invalidate();
+    }
     yield* this.agentLoop(null, signal, false);
   }
   interrupt() {
@@ -1987,6 +2064,7 @@ var QueryEngine = class {
       walletAddress: this.walletAddress,
       suiRpcUrl: this.suiRpcUrl,
       serverPositions: this.serverPositions,
+      positionFetcher: this.positionFetcher,
       signal
     };
     let turns = 0;