@exagent/agent 0.1.9 → 0.1.10

package/dist/cli.js

@@ -32,7 +32,7 @@ var path2 = __toESM(require("path"));
 
 // src/runtime.ts
 var import_sdk = require("@exagent/sdk");
-var import_viem2 = require("viem");
+var import_viem3 = require("viem");
 var import_chains2 = require("viem/chains");
 
 // src/llm/openai.ts
@@ -980,10 +980,22 @@ var RiskManager = class {
 };
 
 // src/trading/market.ts
+var import_viem = require("viem");
+var NATIVE_ETH = "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE";
+var USDC_DECIMALS = {
+  "0x833589fcd6edb6e08f4c7c32d4f71b54bda02913": 6,
+  // USDC on Base
+  "0xd9aaec86b65d86f6a7b5b1b0c42ffa531710b6ca": 6
+  // USDbC on Base
+};
 var MarketDataService = class {
   rpcUrl;
+  client;
   constructor(rpcUrl) {
     this.rpcUrl = rpcUrl;
+    this.client = (0, import_viem.createPublicClient)({
+      transport: (0, import_viem.http)(rpcUrl)
+    });
   }
   /**
    * Fetch current market data for the agent
@@ -1005,25 +1017,53 @@ var MarketDataService = class {
   async fetchPrices(tokenAddresses) {
     const prices = {};
     const knownPrices = {
+      // Native ETH (sentinel address)
+      [NATIVE_ETH.toLowerCase()]: 3500,
       // WETH
       "0x4200000000000000000000000000000000000006": 3500,
       // USDC
-      "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913": 1,
+      "0x833589fcd6edb6e08f4c7c32d4f71b54bda02913": 1,
       // USDbC (bridged USDC)
-      "0xd9aAEc86B65D86f6A7B5B1b0c42FFA531710b6CA": 1
+      "0xd9aaec86b65d86f6a7b5b1b0c42ffa531710b6ca": 1
     };
+    prices[NATIVE_ETH.toLowerCase()] = knownPrices[NATIVE_ETH.toLowerCase()];
     for (const address of tokenAddresses) {
       prices[address.toLowerCase()] = knownPrices[address.toLowerCase()] || 0;
     }
     return prices;
   }
   /**
-   * Fetch token balances from chain
+   * Fetch real on-chain balances: native ETH + ERC-20 tokens
    */
   async fetchBalances(walletAddress, tokenAddresses) {
     const balances = {};
-    for (const address of tokenAddresses) {
-      balances[address.toLowerCase()] = 0n;
+    const wallet = walletAddress;
+    try {
+      const nativeBalance = await this.client.getBalance({ address: wallet });
+      balances[NATIVE_ETH.toLowerCase()] = nativeBalance;
+      const erc20Promises = tokenAddresses.map(async (tokenAddress) => {
+        try {
+          const balance = await this.client.readContract({
+            address: tokenAddress,
+            abi: import_viem.erc20Abi,
+            functionName: "balanceOf",
+            args: [wallet]
+          });
+          return { address: tokenAddress.toLowerCase(), balance };
+        } catch (error) {
+          return { address: tokenAddress.toLowerCase(), balance: 0n };
+        }
+      });
+      const results = await Promise.all(erc20Promises);
+      for (const { address, balance } of results) {
+        balances[address] = balance;
+      }
+    } catch (error) {
+      console.error("MarketData: Failed to fetch balances:", error instanceof Error ? error.message : error);
+      balances[NATIVE_ETH.toLowerCase()] = 0n;
+      for (const address of tokenAddresses) {
+        balances[address.toLowerCase()] = 0n;
+      }
     }
     return balances;
   }
@@ -1034,7 +1074,8 @@ var MarketDataService = class {
     let total = 0;
     for (const [address, balance] of Object.entries(balances)) {
       const price = prices[address.toLowerCase()] || 0;
-      const amount = Number(balance) / 1e18;
+      const decimals = USDC_DECIMALS[address.toLowerCase()] || 18;
+      const amount = Number(balance) / Math.pow(10, decimals);
       total += amount * price;
     }
     return total;
@@ -1042,7 +1083,7 @@ var MarketDataService = class {
 };
 
 // src/vault/manager.ts
-var import_viem = require("viem");
+var import_viem2 = require("viem");
 var import_accounts = require("viem/accounts");
 var import_chains = require("viem/chains");
 var ADDRESSES = {
@@ -1145,14 +1186,14 @@ var VaultManager = class {
     this.account = (0, import_accounts.privateKeyToAccount)(config.walletKey);
     this.chain = config.network === "mainnet" ? import_chains.base : import_chains.baseSepolia;
     const rpcUrl = config.network === "mainnet" ? "https://mainnet.base.org" : "https://sepolia.base.org";
-    this.publicClient = (0, import_viem.createPublicClient)({
+    this.publicClient = (0, import_viem2.createPublicClient)({
       chain: this.chain,
-      transport: (0, import_viem.http)(rpcUrl)
+      transport: (0, import_viem2.http)(rpcUrl)
     });
-    this.walletClient = (0, import_viem.createWalletClient)({
+    this.walletClient = (0, import_viem2.createWalletClient)({
       account: this.account,
       chain: this.chain,
-      transport: (0, import_viem.http)(rpcUrl)
+      transport: (0, import_viem2.http)(rpcUrl)
     });
   }
   /**
@@ -1841,9 +1882,9 @@ var AgentRuntime = class {
         if (message.includes("insufficient funds") || message.includes("gas") || message.includes("intrinsic gas too low") || message.includes("exceeds the balance")) {
           const ccUrl = `https://exagent.io/agents/${encodeURIComponent(this.config.name)}/command-center`;
           const chain = this.config.network === "mainnet" ? import_chains2.base : import_chains2.baseSepolia;
-          const publicClientInstance = (0, import_viem2.createPublicClient)({
+          const publicClientInstance = (0, import_viem3.createPublicClient)({
             chain,
-            transport: (0, import_viem2.http)(this.getRpcUrl())
+            transport: (0, import_viem3.http)(this.getRpcUrl())
           });
           console.log("");
           console.log("=== ETH NEEDED FOR GAS ===");
@@ -2108,9 +2149,8 @@ var AgentRuntime = class {
     const marketData = await this.marketData.fetchMarketData(this.client.address, tokens);
     console.log(`Portfolio value: $${marketData.portfolioValue.toFixed(2)}`);
     this.lastPortfolioValue = marketData.portfolioValue;
-    const wethAddr = "0x4200000000000000000000000000000000000006";
-    const ethBal = marketData.balances[wethAddr] || BigInt(0);
-    this.lastEthBalance = (Number(ethBal) / 1e18).toFixed(6);
+    const nativeEthBal = marketData.balances[NATIVE_ETH.toLowerCase()] || BigInt(0);
+    this.lastEthBalance = (Number(nativeEthBal) / 1e18).toFixed(6);
     this.checkFundsLow(marketData);
     let signals;
     try {
@@ -2173,8 +2213,7 @@ var AgentRuntime = class {
    */
   checkFundsLow(marketData) {
     if (!this.relay) return;
-    const wethAddress = "0x4200000000000000000000000000000000000006";
-    const ethBalance = marketData.balances[wethAddress] || BigInt(0);
+    const ethBalance = marketData.balances[NATIVE_ETH.toLowerCase()] || BigInt(0);
     const ethAmount = Number(ethBalance) / 1e18;
     if (ethAmount < FUNDS_LOW_THRESHOLD) {
       this.relay.sendMessage(

package/dist/cli.mjs

@@ -6,7 +6,7 @@ import {
   loadConfig,
   loadSecureEnv,
   validateConfig
-} from "./chunk-CKSCA3O7.mjs";
+} from "./chunk-7GQXIIL2.mjs";
 
 // src/cli.ts
 import { Command } from "commander";

package/dist/index.d.mts

@@ -818,10 +818,15 @@ declare class RiskManager {
 
 /**
  * Market Data Service
- * Fetches prices and portfolio data for strategy analysis
+ * Fetches prices and portfolio data for strategy analysis.
+ *
+ * Queries real on-chain balances for:
+ * - Native ETH (via eth_getBalance)
+ * - ERC-20 tokens (via balanceOf)
  */
 declare class MarketDataService {
     private rpcUrl;
+    private client;
     constructor(rpcUrl: string);
     /**
      * Fetch current market data for the agent
@@ -832,7 +837,7 @@ declare class MarketDataService {
      */
     private fetchPrices;
     /**
-     * Fetch token balances from chain
+     * Fetch real on-chain balances: native ETH + ERC-20 tokens
      */
     private fetchBalances;
     /**

package/dist/index.d.ts

@@ -818,10 +818,15 @@ declare class RiskManager {
 
 /**
  * Market Data Service
- * Fetches prices and portfolio data for strategy analysis
+ * Fetches prices and portfolio data for strategy analysis.
+ *
+ * Queries real on-chain balances for:
+ * - Native ETH (via eth_getBalance)
+ * - ERC-20 tokens (via balanceOf)
  */
 declare class MarketDataService {
     private rpcUrl;
+    private client;
     constructor(rpcUrl: string);
     /**
      * Fetch current market data for the agent
@@ -832,7 +837,7 @@ declare class MarketDataService {
      */
     private fetchPrices;
     /**
-     * Fetch token balances from chain
+     * Fetch real on-chain balances: native ETH + ERC-20 tokens
      */
     private fetchBalances;
     /**

package/dist/index.js

@@ -70,7 +70,7 @@ module.exports = __toCommonJS(index_exports);
 
 // src/runtime.ts
 var import_sdk = require("@exagent/sdk");
-var import_viem2 = require("viem");
+var import_viem3 = require("viem");
 var import_chains2 = require("viem/chains");
 
 // src/llm/openai.ts
@@ -1024,10 +1024,22 @@ var RiskManager = class {
 };
 
 // src/trading/market.ts
+var import_viem = require("viem");
+var NATIVE_ETH = "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE";
+var USDC_DECIMALS = {
+  "0x833589fcd6edb6e08f4c7c32d4f71b54bda02913": 6,
+  // USDC on Base
+  "0xd9aaec86b65d86f6a7b5b1b0c42ffa531710b6ca": 6
+  // USDbC on Base
+};
 var MarketDataService = class {
   rpcUrl;
+  client;
   constructor(rpcUrl) {
     this.rpcUrl = rpcUrl;
+    this.client = (0, import_viem.createPublicClient)({
+      transport: (0, import_viem.http)(rpcUrl)
+    });
   }
   /**
    * Fetch current market data for the agent
@@ -1049,25 +1061,53 @@ var MarketDataService = class {
   async fetchPrices(tokenAddresses) {
     const prices = {};
     const knownPrices = {
+      // Native ETH (sentinel address)
+      [NATIVE_ETH.toLowerCase()]: 3500,
       // WETH
       "0x4200000000000000000000000000000000000006": 3500,
       // USDC
-      "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913": 1,
+      "0x833589fcd6edb6e08f4c7c32d4f71b54bda02913": 1,
       // USDbC (bridged USDC)
-      "0xd9aAEc86B65D86f6A7B5B1b0c42FFA531710b6CA": 1
+      "0xd9aaec86b65d86f6a7b5b1b0c42ffa531710b6ca": 1
     };
+    prices[NATIVE_ETH.toLowerCase()] = knownPrices[NATIVE_ETH.toLowerCase()];
     for (const address of tokenAddresses) {
       prices[address.toLowerCase()] = knownPrices[address.toLowerCase()] || 0;
     }
     return prices;
   }
   /**
-   * Fetch token balances from chain
+   * Fetch real on-chain balances: native ETH + ERC-20 tokens
    */
   async fetchBalances(walletAddress, tokenAddresses) {
     const balances = {};
-    for (const address of tokenAddresses) {
-      balances[address.toLowerCase()] = 0n;
+    const wallet = walletAddress;
+    try {
+      const nativeBalance = await this.client.getBalance({ address: wallet });
+      balances[NATIVE_ETH.toLowerCase()] = nativeBalance;
+      const erc20Promises = tokenAddresses.map(async (tokenAddress) => {
+        try {
+          const balance = await this.client.readContract({
+            address: tokenAddress,
+            abi: import_viem.erc20Abi,
+            functionName: "balanceOf",
+            args: [wallet]
+          });
+          return { address: tokenAddress.toLowerCase(), balance };
+        } catch (error) {
+          return { address: tokenAddress.toLowerCase(), balance: 0n };
+        }
+      });
+      const results = await Promise.all(erc20Promises);
+      for (const { address, balance } of results) {
+        balances[address] = balance;
+      }
+    } catch (error) {
+      console.error("MarketData: Failed to fetch balances:", error instanceof Error ? error.message : error);
+      balances[NATIVE_ETH.toLowerCase()] = 0n;
+      for (const address of tokenAddresses) {
+        balances[address.toLowerCase()] = 0n;
+      }
     }
     return balances;
   }
@@ -1078,7 +1118,8 @@ var MarketDataService = class {
     let total = 0;
     for (const [address, balance] of Object.entries(balances)) {
       const price = prices[address.toLowerCase()] || 0;
-      const amount = Number(balance) / 1e18;
+      const decimals = USDC_DECIMALS[address.toLowerCase()] || 18;
+      const amount = Number(balance) / Math.pow(10, decimals);
       total += amount * price;
     }
     return total;
@@ -1086,7 +1127,7 @@ var MarketDataService = class {
 };
 
 // src/vault/manager.ts
-var import_viem = require("viem");
+var import_viem2 = require("viem");
 var import_accounts = require("viem/accounts");
 var import_chains = require("viem/chains");
 var ADDRESSES = {
@@ -1189,14 +1230,14 @@ var VaultManager = class {
     this.account = (0, import_accounts.privateKeyToAccount)(config.walletKey);
     this.chain = config.network === "mainnet" ? import_chains.base : import_chains.baseSepolia;
     const rpcUrl = config.network === "mainnet" ? "https://mainnet.base.org" : "https://sepolia.base.org";
-    this.publicClient = (0, import_viem.createPublicClient)({
+    this.publicClient = (0, import_viem2.createPublicClient)({
       chain: this.chain,
-      transport: (0, import_viem.http)(rpcUrl)
+      transport: (0, import_viem2.http)(rpcUrl)
     });
-    this.walletClient = (0, import_viem.createWalletClient)({
+    this.walletClient = (0, import_viem2.createWalletClient)({
       account: this.account,
       chain: this.chain,
-      transport: (0, import_viem.http)(rpcUrl)
+      transport: (0, import_viem2.http)(rpcUrl)
     });
   }
   /**
@@ -1885,9 +1926,9 @@ var AgentRuntime = class {
         if (message.includes("insufficient funds") || message.includes("gas") || message.includes("intrinsic gas too low") || message.includes("exceeds the balance")) {
           const ccUrl = `https://exagent.io/agents/${encodeURIComponent(this.config.name)}/command-center`;
           const chain = this.config.network === "mainnet" ? import_chains2.base : import_chains2.baseSepolia;
-          const publicClientInstance = (0, import_viem2.createPublicClient)({
+          const publicClientInstance = (0, import_viem3.createPublicClient)({
             chain,
-            transport: (0, import_viem2.http)(this.getRpcUrl())
+            transport: (0, import_viem3.http)(this.getRpcUrl())
           });
           console.log("");
           console.log("=== ETH NEEDED FOR GAS ===");
@@ -2152,9 +2193,8 @@ var AgentRuntime = class {
     const marketData = await this.marketData.fetchMarketData(this.client.address, tokens);
     console.log(`Portfolio value: $${marketData.portfolioValue.toFixed(2)}`);
     this.lastPortfolioValue = marketData.portfolioValue;
-    const wethAddr = "0x4200000000000000000000000000000000000006";
-    const ethBal = marketData.balances[wethAddr] || BigInt(0);
-    this.lastEthBalance = (Number(ethBal) / 1e18).toFixed(6);
+    const nativeEthBal = marketData.balances[NATIVE_ETH.toLowerCase()] || BigInt(0);
+    this.lastEthBalance = (Number(nativeEthBal) / 1e18).toFixed(6);
     this.checkFundsLow(marketData);
     let signals;
     try {
@@ -2217,8 +2257,7 @@ var AgentRuntime = class {
    */
   checkFundsLow(marketData) {
     if (!this.relay) return;
-    const wethAddress = "0x4200000000000000000000000000000000000006";
-    const ethBalance = marketData.balances[wethAddress] || BigInt(0);
+    const ethBalance = marketData.balances[NATIVE_ETH.toLowerCase()] || BigInt(0);
     const ethAmount = Number(ethBalance) / 1e18;
     if (ethAmount < FUNDS_LOW_THRESHOLD) {
       this.relay.sendMessage(

package/dist/index.mjs

@@ -34,7 +34,7 @@ import {
   loadStrategy,
   validateConfig,
   validateStrategy
-} from "./chunk-CKSCA3O7.mjs";
+} from "./chunk-7GQXIIL2.mjs";
 export {
   AgentConfigSchema,
   AgentRuntime,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exagent/agent",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Autonomous trading agent runtime for Exagent",
   "main": "dist/index.js",
   "module": "dist/index.mjs",