@t2000/engine 0.5.8 → 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;

package/dist/index.js

@@ -555,11 +555,11 @@ var balanceCheckTool = buildTool({
           });
         }
       }
-      const sp = context.serverPositions;
       let savings;
       let debt;
       let pendingRewardsUsd;
-      if (sp) {
+      if (context.positionFetcher && context.walletAddress) {
+        const sp = await context.positionFetcher(context.walletAddress);
         savings = sp.savings;
         debt = sp.borrows;
         pendingRewardsUsd = sp.pendingRewards;
@@ -693,7 +693,7 @@ async function fetchProtocolStats(manager, opts) {
 }
 
 // src/tools/savings.ts
-function buildSavingsFromServer(sp) {
+function buildSavingsFromPositions(sp) {
   const positions = [
     ...sp.supplies.map((s) => ({
       protocol: s.protocol,
@@ -741,8 +741,9 @@ var savingsInfoTool = buildTool({
   jsonSchema: { type: "object", properties: {}, required: [] },
   isReadOnly: true,
   async call(_input, context) {
-    if (context.serverPositions) {
-      return { data: buildSavingsFromServer(context.serverPositions) };
+    if (context.positionFetcher && context.walletAddress) {
+      const sp = await context.positionFetcher(context.walletAddress);
+      return { data: buildSavingsFromPositions(sp) };
     }
     if (hasNaviMcp(context)) {
       const savings = await fetchSavings(
@@ -799,9 +800,9 @@ var healthCheckTool = buildTool({
   jsonSchema: { type: "object", properties: {}, required: [] },
   isReadOnly: true,
   async call(_input, context) {
-    if (context.serverPositions) {
-      const sp = context.serverPositions;
-      const hfVal = sp.healthFactor ?? (sp.borrows > 0 ? Infinity : Infinity);
+    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 {
@@ -1936,6 +1937,7 @@ var QueryEngine = class {
   walletAddress;
   suiRpcUrl;
   serverPositions;
+  positionFetcher;
   txMutex = new TxMutex();
   costTracker;
   messages = [];
@@ -1947,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;
@@ -2021,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() {
@@ -2058,6 +2064,7 @@ var QueryEngine = class {
       walletAddress: this.walletAddress,
       suiRpcUrl: this.suiRpcUrl,
       serverPositions: this.serverPositions,
+      positionFetcher: this.positionFetcher,
       signal
     };
     let turns = 0;