@coinbase/agentkit 0.9.1 → 0.10.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (151) hide show
  1. package/README.md +181 -54
  2. package/dist/action-providers/across/acrossActionProvider.js +3 -3
  3. package/dist/action-providers/cdp/cdpApiActionProvider.d.ts +3 -12
  4. package/dist/action-providers/cdp/cdpApiActionProvider.js +2 -79
  5. package/dist/action-providers/cdp/cdpApiActionProvider.test.js +0 -125
  6. package/dist/action-providers/cdp/cdpEvmWalletActionProvider.d.ts +58 -0
  7. package/dist/action-providers/cdp/cdpEvmWalletActionProvider.js +351 -0
  8. package/dist/action-providers/cdp/cdpEvmWalletActionProvider.test.d.ts +1 -0
  9. package/dist/action-providers/cdp/cdpEvmWalletActionProvider.test.js +520 -0
  10. package/dist/action-providers/cdp/cdpSmartWalletActionProvider.d.ts +57 -0
  11. package/dist/action-providers/cdp/cdpSmartWalletActionProvider.js +337 -0
  12. package/dist/action-providers/cdp/cdpSmartWalletActionProvider.test.d.ts +1 -0
  13. package/dist/action-providers/cdp/cdpSmartWalletActionProvider.test.js +463 -0
  14. package/dist/action-providers/cdp/index.d.ts +3 -0
  15. package/dist/action-providers/cdp/index.js +3 -0
  16. package/dist/action-providers/cdp/schemas.d.ts +38 -9
  17. package/dist/action-providers/cdp/schemas.js +49 -6
  18. package/dist/action-providers/cdp/spendPermissionUtils.d.ts +24 -0
  19. package/dist/action-providers/cdp/spendPermissionUtils.js +66 -0
  20. package/dist/action-providers/cdp/swapUtils.d.ts +32 -0
  21. package/dist/action-providers/cdp/swapUtils.js +142 -0
  22. package/dist/action-providers/clanker/clankerActionProvider.d.ts +43 -0
  23. package/dist/action-providers/clanker/clankerActionProvider.js +130 -0
  24. package/dist/action-providers/clanker/clankerActionProvider.test.d.ts +4 -0
  25. package/dist/action-providers/clanker/clankerActionProvider.test.js +119 -0
  26. package/dist/action-providers/clanker/index.d.ts +2 -0
  27. package/dist/action-providers/clanker/index.js +18 -0
  28. package/dist/action-providers/clanker/schemas.d.ts +56 -0
  29. package/dist/action-providers/clanker/schemas.js +47 -0
  30. package/dist/action-providers/clanker/utils.d.ts +9 -0
  31. package/dist/action-providers/clanker/utils.js +23 -0
  32. package/dist/action-providers/compound/constants.d.ts +1 -1
  33. package/dist/action-providers/compound/constants.js +2 -2
  34. package/dist/action-providers/erc20/constants.d.ts +35 -135
  35. package/dist/action-providers/erc20/constants.js +37 -189
  36. package/dist/action-providers/erc20/erc20ActionProvider.d.ts +9 -1
  37. package/dist/action-providers/erc20/erc20ActionProvider.js +87 -35
  38. package/dist/action-providers/erc20/erc20ActionProvider.test.js +115 -52
  39. package/dist/action-providers/erc20/schemas.d.ts +25 -12
  40. package/dist/action-providers/erc20/schemas.js +34 -6
  41. package/dist/action-providers/erc20/utils.d.ts +19 -0
  42. package/dist/action-providers/erc20/utils.js +54 -0
  43. package/dist/action-providers/flaunch/constants.d.ts +1 -1
  44. package/dist/action-providers/flaunch/constants.js +2 -2
  45. package/dist/action-providers/flaunch/flaunchActionProvider.js +3 -11
  46. package/dist/action-providers/flaunch/flaunchActionProvider.test.js +5 -0
  47. package/dist/action-providers/index.d.ts +4 -0
  48. package/dist/action-providers/index.js +4 -0
  49. package/dist/action-providers/jupiter/schemas.d.ts +1 -1
  50. package/dist/action-providers/moonwell/schemas.d.ts +2 -2
  51. package/dist/action-providers/morpho/morphoActionProvider.js +5 -5
  52. package/dist/action-providers/morpho/schemas.d.ts +2 -2
  53. package/dist/action-providers/pyth/pythActionProvider.js +5 -0
  54. package/dist/action-providers/pyth/pythActionProvider.test.js +5 -1
  55. package/dist/action-providers/superfluid/constants.d.ts +814 -0
  56. package/dist/action-providers/superfluid/constants.js +2826 -0
  57. package/dist/action-providers/superfluid/graphQueries/endpoints.d.ts +2 -0
  58. package/dist/action-providers/superfluid/graphQueries/endpoints.js +5 -0
  59. package/dist/action-providers/superfluid/graphQueries/queries.d.ts +1 -0
  60. package/dist/action-providers/superfluid/graphQueries/queries.js +35 -0
  61. package/dist/action-providers/superfluid/graphQueries/superfluidGraphQueries.d.ts +8 -0
  62. package/dist/action-providers/superfluid/graphQueries/superfluidGraphQueries.js +24 -0
  63. package/dist/action-providers/superfluid/graphQueries/types.d.ts +27 -0
  64. package/dist/action-providers/superfluid/graphQueries/types.js +2 -0
  65. package/dist/action-providers/superfluid/index.d.ts +7 -0
  66. package/dist/action-providers/superfluid/index.js +23 -0
  67. package/dist/action-providers/superfluid/schemas.d.ts +86 -0
  68. package/dist/action-providers/superfluid/schemas.js +103 -0
  69. package/dist/action-providers/superfluid/superfluidActionProvider.d.ts +20 -0
  70. package/dist/action-providers/superfluid/superfluidActionProvider.js +36 -0
  71. package/dist/action-providers/superfluid/superfluidPoolActionProvider.d.ts +46 -0
  72. package/dist/action-providers/superfluid/superfluidPoolActionProvider.js +143 -0
  73. package/dist/action-providers/superfluid/superfluidPoolActionProvider.test.d.ts +1 -0
  74. package/dist/action-providers/superfluid/superfluidPoolActionProvider.test.js +92 -0
  75. package/dist/action-providers/superfluid/superfluidQueryActionProvider.d.ts +27 -0
  76. package/dist/action-providers/superfluid/superfluidQueryActionProvider.js +71 -0
  77. package/dist/action-providers/superfluid/superfluidQueryActionProvider.test.d.ts +1 -0
  78. package/dist/action-providers/superfluid/superfluidQueryActionProvider.test.js +57 -0
  79. package/dist/action-providers/superfluid/superfluidStreamActionProvider.d.ts +56 -0
  80. package/dist/action-providers/superfluid/superfluidStreamActionProvider.js +191 -0
  81. package/dist/action-providers/superfluid/superfluidStreamActionProvider.test.d.ts +1 -0
  82. package/dist/action-providers/superfluid/superfluidStreamActionProvider.test.js +80 -0
  83. package/dist/action-providers/superfluid/superfluidSuperTokenCreatorActionProvider.d.ts +30 -0
  84. package/dist/action-providers/superfluid/superfluidSuperTokenCreatorActionProvider.js +108 -0
  85. package/dist/action-providers/superfluid/superfluidSuperTokenCreatorActionProvider.test.d.ts +1 -0
  86. package/dist/action-providers/superfluid/superfluidSuperTokenCreatorActionProvider.test.js +75 -0
  87. package/dist/action-providers/superfluid/superfluidWrapperActionProvider.d.ts +32 -0
  88. package/dist/action-providers/superfluid/superfluidWrapperActionProvider.js +101 -0
  89. package/dist/action-providers/superfluid/superfluidWrapperActionProvider.test.d.ts +1 -0
  90. package/dist/action-providers/superfluid/superfluidWrapperActionProvider.test.js +85 -0
  91. package/dist/action-providers/superfluid/utils/parseLogs.d.ts +18 -0
  92. package/dist/action-providers/superfluid/utils/parseLogs.js +78 -0
  93. package/dist/action-providers/truemarkets/truemarketsActionProvider.d.ts +4 -16
  94. package/dist/action-providers/truemarkets/truemarketsActionProvider.js +20 -41
  95. package/dist/action-providers/truemarkets/truemarketsActionProvider.test.js +11 -33
  96. package/dist/action-providers/wallet/walletActionProvider.js +21 -10
  97. package/dist/action-providers/wallet/walletActionProvider.test.js +6 -2
  98. package/dist/action-providers/zerion/constants.d.ts +1 -0
  99. package/dist/action-providers/zerion/constants.js +4 -0
  100. package/dist/action-providers/zerion/index.d.ts +2 -0
  101. package/dist/action-providers/zerion/index.js +18 -0
  102. package/dist/action-providers/zerion/schemas.d.ts +11 -0
  103. package/dist/action-providers/zerion/schemas.js +15 -0
  104. package/dist/action-providers/zerion/types.d.ts +125 -0
  105. package/dist/action-providers/zerion/types.js +16 -0
  106. package/dist/action-providers/zerion/utils.d.ts +3 -0
  107. package/dist/action-providers/zerion/utils.js +45 -0
  108. package/dist/action-providers/zerion/zerionActionProvider.d.ts +57 -0
  109. package/dist/action-providers/zerion/zerionActionProvider.js +159 -0
  110. package/dist/action-providers/zerion/zerionActionProvider.test.d.ts +1 -0
  111. package/dist/action-providers/zerion/zerionActionProvider.test.js +213 -0
  112. package/dist/action-providers/zeroX/index.d.ts +1 -0
  113. package/dist/action-providers/zeroX/index.js +17 -0
  114. package/dist/action-providers/zeroX/schemas.d.ts +51 -0
  115. package/dist/action-providers/zeroX/schemas.js +82 -0
  116. package/dist/action-providers/zeroX/utils.d.ts +23 -0
  117. package/dist/action-providers/zeroX/utils.js +106 -0
  118. package/dist/action-providers/zeroX/zeroXActionProvider.d.ts +57 -0
  119. package/dist/action-providers/zeroX/zeroXActionProvider.js +407 -0
  120. package/dist/action-providers/zeroX/zeroXActionProvider.test.d.ts +1 -0
  121. package/dist/action-providers/zeroX/zeroXActionProvider.test.js +445 -0
  122. package/dist/wallet-providers/cdpEvmWalletProvider.d.ts +20 -2
  123. package/dist/wallet-providers/cdpEvmWalletProvider.js +40 -15
  124. package/dist/wallet-providers/cdpShared.d.ts +9 -0
  125. package/dist/wallet-providers/cdpSmartWalletProvider.d.ts +29 -3
  126. package/dist/wallet-providers/cdpSmartWalletProvider.js +64 -28
  127. package/dist/wallet-providers/cdpSolanaWalletProvider.d.ts +1 -1
  128. package/dist/wallet-providers/cdpSolanaWalletProvider.js +7 -7
  129. package/dist/wallet-providers/cdpSolanaWalletProvider.test.js +15 -12
  130. package/dist/wallet-providers/evmWalletProvider.d.ts +5 -1
  131. package/dist/wallet-providers/legacyCdpSmartWalletProvider.d.ts +9 -2
  132. package/dist/wallet-providers/legacyCdpSmartWalletProvider.js +12 -2
  133. package/dist/wallet-providers/legacyCdpWalletProvider.d.ts +12 -2
  134. package/dist/wallet-providers/legacyCdpWalletProvider.js +11 -2
  135. package/dist/wallet-providers/privyEvmDelegatedEmbeddedWalletProvider.d.ts +10 -2
  136. package/dist/wallet-providers/privyEvmDelegatedEmbeddedWalletProvider.js +12 -3
  137. package/dist/wallet-providers/privyEvmDelegatedEmbeddedWalletProvider.test.js +1 -1
  138. package/dist/wallet-providers/privyEvmWalletProvider.d.ts +2 -0
  139. package/dist/wallet-providers/privyEvmWalletProvider.js +2 -1
  140. package/dist/wallet-providers/privyEvmWalletProvider.test.js +1 -1
  141. package/dist/wallet-providers/solanaKeypairWalletProvider.d.ts +1 -1
  142. package/dist/wallet-providers/solanaKeypairWalletProvider.js +3 -4
  143. package/dist/wallet-providers/solanaKeypairWalletProvider.test.js +4 -2
  144. package/dist/wallet-providers/viemWalletProvider.d.ts +12 -2
  145. package/dist/wallet-providers/viemWalletProvider.js +12 -3
  146. package/dist/wallet-providers/viemWalletProvider.test.js +6 -5
  147. package/dist/wallet-providers/walletProvider.d.ts +1 -1
  148. package/dist/wallet-providers/zeroDevWalletProvider.d.ts +10 -2
  149. package/dist/wallet-providers/zeroDevWalletProvider.js +14 -5
  150. package/dist/wallet-providers/zeroDevWalletProvider.test.js +2 -2
  151. package/package.json +4 -2
@@ -8,18 +8,6 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
8
8
  var __metadata = (this && this.__metadata) || function (k, v) {
9
9
  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
10
10
  };
11
- var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
12
- if (kind === "m") throw new TypeError("Private method is not writable");
13
- if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
14
- if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
15
- return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
16
- };
17
- var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
18
- if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
19
- if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
20
- return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
21
- };
22
- var _TrueMarketsActionProvider_publicClient;
23
11
  Object.defineProperty(exports, "__esModule", { value: true });
24
12
  exports.truemarketsActionProvider = exports.TrueMarketsActionProvider = void 0;
25
13
  const zod_1 = require("zod");
@@ -27,22 +15,18 @@ const actionProvider_1 = require("../actionProvider");
27
15
  const actionDecorator_1 = require("../actionDecorator");
28
16
  const schemas_1 = require("./schemas");
29
17
  const constants_1 = require("./constants");
30
- const constants_2 = require("../erc20/constants");
31
- const wallet_providers_1 = require("../../wallet-providers");
32
18
  const viem_1 = require("viem");
33
- const chains_1 = require("viem/chains");
19
+ const wallet_providers_1 = require("../../wallet-providers");
20
+ const viem_2 = require("viem");
34
21
  /**
35
- * TrueMarketsActionProvider provides actions to interact with TrueMarkets contracts.
22
+ * Action provider for TrueMarkets interactions.
36
23
  */
37
24
  class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
38
25
  /**
39
- * Constructor for the TrueMarketsActionProvider.
40
- *
41
- * @param config - The configuration options for the TrueMarketsActionProvider.
26
+ * Creates a new TrueMarkets action provider.
42
27
  */
43
- constructor(config) {
28
+ constructor() {
44
29
  super("truemarkets", []);
45
- _TrueMarketsActionProvider_publicClient.set(this, void 0);
46
30
  /**
47
31
  * Checks if the TrueMarkets action provider supports the given network.
48
32
  * Currently only supports Base mainnet.
@@ -51,10 +35,6 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
51
35
  * @returns True if the TrueMarkets action provider supports the network, false otherwise.
52
36
  */
53
37
  this.supportsNetwork = (network) => network.networkId === "base-mainnet";
54
- __classPrivateFieldSet(this, _TrueMarketsActionProvider_publicClient, (0, viem_1.createPublicClient)({
55
- chain: chains_1.base,
56
- transport: config?.RPC_URL ? (0, viem_1.http)(config.RPC_URL) : (0, viem_1.http)(),
57
- }), "f");
58
38
  }
59
39
  /**
60
40
  * Gets active markets from the TruthMarketManager contract.
@@ -105,7 +85,7 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
105
85
  functionName: "getActiveMarketAddress",
106
86
  args: [BigInt(index)],
107
87
  }));
108
- const marketAddresses = await __classPrivateFieldGet(this, _TrueMarketsActionProvider_publicClient, "f").multicall({
88
+ const marketAddresses = await walletProvider.getPublicClient().multicall({
109
89
  contracts: addressCalls,
110
90
  });
111
91
  // Filter out errors and extract results
@@ -124,7 +104,7 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
124
104
  abi: constants_1.TruthMarketABI,
125
105
  functionName: "marketQuestion",
126
106
  }));
127
- const marketQuestionsResult = await __classPrivateFieldGet(this, _TrueMarketsActionProvider_publicClient, "f").multicall({
107
+ const marketQuestionsResult = await walletProvider.getPublicClient().multicall({
128
108
  contracts: questionCalls,
129
109
  });
130
110
  // Create market objects mapping indices to addresses and questions
@@ -175,7 +155,7 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
175
155
  });
176
156
  }
177
157
  try {
178
- marketAddress = (await __classPrivateFieldGet(this, _TrueMarketsActionProvider_publicClient, "f").readContract({
158
+ marketAddress = (await walletProvider.getPublicClient().readContract({
179
159
  address: constants_1.TruthMarketManager_ADDRESS,
180
160
  abi: constants_1.TruthMarketManagerABI,
181
161
  functionName: "getActiveMarketAddress",
@@ -227,7 +207,7 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
227
207
  functionName: "winningPosition",
228
208
  },
229
209
  ];
230
- const basicInfoResults = await __classPrivateFieldGet(this, _TrueMarketsActionProvider_publicClient, "f").multicall({
210
+ const basicInfoResults = await walletProvider.getPublicClient().multicall({
231
211
  contracts: basicInfoCalls,
232
212
  });
233
213
  // Extract results, handling potential errors
@@ -279,7 +259,7 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
279
259
  functionName: "slot0",
280
260
  },
281
261
  ];
282
- const poolInfoResults = await __classPrivateFieldGet(this, _TrueMarketsActionProvider_publicClient, "f").multicall({
262
+ const poolInfoResults = await walletProvider.getPublicClient().multicall({
283
263
  contracts: poolInfoCalls,
284
264
  });
285
265
  if (poolInfoResults.some(result => result.status === "failure")) {
@@ -308,30 +288,30 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
308
288
  const balanceCalls = [
309
289
  {
310
290
  address: payToken,
311
- abi: constants_2.abi,
291
+ abi: viem_1.erc20Abi,
312
292
  functionName: "balanceOf",
313
293
  args: [yesPool],
314
294
  },
315
295
  {
316
296
  address: yesToken,
317
- abi: constants_2.abi,
297
+ abi: viem_1.erc20Abi,
318
298
  functionName: "balanceOf",
319
299
  args: [yesPool],
320
300
  },
321
301
  {
322
302
  address: payToken,
323
- abi: constants_2.abi,
303
+ abi: viem_1.erc20Abi,
324
304
  functionName: "balanceOf",
325
305
  args: [noPool],
326
306
  },
327
307
  {
328
308
  address: noToken,
329
- abi: constants_2.abi,
309
+ abi: viem_1.erc20Abi,
330
310
  functionName: "balanceOf",
331
311
  args: [noPool],
332
312
  },
333
313
  ];
334
- const balanceResults = await __classPrivateFieldGet(this, _TrueMarketsActionProvider_publicClient, "f").multicall({
314
+ const balanceResults = await walletProvider.getPublicClient().multicall({
335
315
  contracts: balanceCalls,
336
316
  });
337
317
  if (balanceResults.some(result => result.status === "failure")) {
@@ -366,10 +346,10 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
366
346
  const yesPrice = calculatePrice(yesSqrtPriceX96, isYesToken0, payDecimals, yesNoTokenDecimals_);
367
347
  const noPrice = calculatePrice(noSqrtPriceX96, isNoToken0, payDecimals, yesNoTokenDecimals_);
368
348
  // Calculate TVL using token balances
369
- const yesPoolStableValue = Number((0, viem_1.formatUnits)(yesPoolStableBalance || 0n, payDecimals));
370
- const yesPoolTokenValue = Number((0, viem_1.formatUnits)(yesPoolTokenBalance || 0n, yesNoTokenDecimals_)) * yesPrice;
371
- const noPoolStableValue = Number((0, viem_1.formatUnits)(noPoolStableBalance || 0n, payDecimals));
372
- const noPoolTokenValue = Number((0, viem_1.formatUnits)(noPoolTokenBalance || 0n, yesNoTokenDecimals_)) * noPrice;
349
+ const yesPoolStableValue = Number((0, viem_2.formatUnits)(yesPoolStableBalance || 0n, payDecimals));
350
+ const yesPoolTokenValue = Number((0, viem_2.formatUnits)(yesPoolTokenBalance || 0n, yesNoTokenDecimals_)) * yesPrice;
351
+ const noPoolStableValue = Number((0, viem_2.formatUnits)(noPoolStableBalance || 0n, payDecimals));
352
+ const noPoolTokenValue = Number((0, viem_2.formatUnits)(noPoolTokenBalance || 0n, yesNoTokenDecimals_)) * noPrice;
373
353
  const yesTVL = yesPoolStableValue + yesPoolTokenValue;
374
354
  const noTVL = noPoolStableValue + noPoolTokenValue;
375
355
  const totalTVL = yesTVL + noTVL;
@@ -430,7 +410,6 @@ class TrueMarketsActionProvider extends actionProvider_1.ActionProvider {
430
410
  }
431
411
  }
432
412
  exports.TrueMarketsActionProvider = TrueMarketsActionProvider;
433
- _TrueMarketsActionProvider_publicClient = new WeakMap();
434
413
  __decorate([
435
414
  (0, actionDecorator_1.CreateAction)({
436
415
  name: "get_prediction_markets",
@@ -465,5 +444,5 @@ __decorate([
465
444
  __metadata("design:paramtypes", [wallet_providers_1.EvmWalletProvider, void 0]),
466
445
  __metadata("design:returntype", Promise)
467
446
  ], TrueMarketsActionProvider.prototype, "getPredictionMarketDetails", null);
468
- const truemarketsActionProvider = (config) => new TrueMarketsActionProvider(config);
447
+ const truemarketsActionProvider = () => new TrueMarketsActionProvider();
469
448
  exports.truemarketsActionProvider = truemarketsActionProvider;
@@ -2,29 +2,10 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  const truemarketsActionProvider_1 = require("./truemarketsActionProvider");
4
4
  const constants_1 = require("./constants");
5
- const viem_1 = require("viem");
6
- // Mock viem's createPublicClient
7
- jest.mock("viem", () => {
8
- const originalModule = jest.requireActual("viem");
9
- return {
10
- ...originalModule,
11
- createPublicClient: jest.fn().mockImplementation(() => ({
12
- // Mock public client methods as needed
13
- multicall: jest.fn().mockImplementation(({ contracts }) => {
14
- // Create mock responses with success status
15
- return contracts.map(() => ({
16
- status: "success",
17
- result: "mock result",
18
- }));
19
- }),
20
- readContract: jest.fn(),
21
- })),
22
- http: jest.fn().mockImplementation(url => ({ url })),
23
- };
24
- });
25
5
  describe("TrueMarketsActionProvider", () => {
26
6
  let provider;
27
7
  let mockWallet;
8
+ let publicClientMock;
28
9
  // Mock addresses and data for tests
29
10
  const MOCK_MARKET_ADDRESS = "0x1234567890123456789012345678901234567890";
30
11
  const MOCK_YES_POOL_ADDRESS = "0x2345678901234567890123456789012345678901";
@@ -36,26 +17,22 @@ describe("TrueMarketsActionProvider", () => {
36
17
  const MOCK_MARKET_SOURCE = "Test source";
37
18
  const MOCK_STATUS_NUM = 0n; // Created status
38
19
  const MOCK_END_OF_TRADING = 1717171717n; // Unix timestamp
39
- describe("constructor", () => {
40
- it("should use the provided RPC_URL for the public client", () => {
41
- const customRpcUrl = "https://custom-rpc.example.com";
42
- (0, truemarketsActionProvider_1.truemarketsActionProvider)({ RPC_URL: customRpcUrl });
43
- // Verify createPublicClient was called with the correct URL
44
- expect(viem_1.createPublicClient).toHaveBeenCalledWith(expect.objectContaining({
45
- chain: expect.anything(),
46
- transport: expect.objectContaining({ url: customRpcUrl }),
47
- }));
48
- });
49
- });
50
20
  beforeEach(() => {
51
21
  jest.clearAllMocks();
52
22
  provider = (0, truemarketsActionProvider_1.truemarketsActionProvider)();
23
+ publicClientMock = {
24
+ multicall: jest
25
+ .fn()
26
+ .mockImplementation(({ contracts }) => contracts.map(() => ({ status: "success", result: "mock result" }))),
27
+ readContract: jest.fn(),
28
+ };
53
29
  mockWallet = {
54
30
  readContract: jest.fn(),
55
31
  getName: jest.fn().mockReturnValue("evm_wallet_provider"),
56
32
  getNetwork: jest.fn().mockReturnValue({
57
33
  networkId: "base-mainnet",
58
34
  }),
35
+ getPublicClient: jest.fn().mockReturnValue(publicClientMock),
59
36
  };
60
37
  });
61
38
  afterEach(() => {
@@ -134,8 +111,9 @@ describe("TrueMarketsActionProvider", () => {
134
111
  describe("getPredictionMarketDetails", () => {
135
112
  let mockPublicClient;
136
113
  beforeEach(() => {
137
- // Access the mocked public client
138
- mockPublicClient = viem_1.createPublicClient.mock.results[0].value;
114
+ // Use the shared mocked public client
115
+ mockPublicClient = publicClientMock;
116
+ mockPublicClient.multicall.mockReset();
139
117
  // Setup multicall mock responses
140
118
  mockPublicClient.multicall
141
119
  // Basic info calls
@@ -14,12 +14,25 @@ const zod_1 = require("zod");
14
14
  const actionDecorator_1 = require("../actionDecorator");
15
15
  const actionProvider_1 = require("../actionProvider");
16
16
  const wallet_providers_1 = require("../../wallet-providers");
17
+ const viem_1 = require("viem");
17
18
  const schemas_1 = require("./schemas");
18
19
  const PROTOCOL_FAMILY_TO_TERMINOLOGY = {
19
- evm: { unit: "WEI", displayUnit: "ETH", type: "Transaction hash", verb: "transaction" },
20
- svm: { unit: "LAMPORTS", displayUnit: "SOL", type: "Signature", verb: "transfer" },
20
+ evm: {
21
+ unit: "WEI",
22
+ displayUnit: "ETH",
23
+ decimals: 18,
24
+ type: "Transaction hash",
25
+ verb: "transaction",
26
+ },
27
+ svm: { unit: "LAMPORTS", displayUnit: "SOL", decimals: 9, type: "Signature", verb: "transfer" },
28
+ };
29
+ const DEFAULT_TERMINOLOGY = {
30
+ unit: "",
31
+ displayUnit: "",
32
+ decimals: 0,
33
+ type: "Hash",
34
+ verb: "transfer",
21
35
  };
22
- const DEFAULT_TERMINOLOGY = { unit: "", displayUnit: "", type: "Hash", verb: "transfer" };
23
36
  /**
24
37
  * WalletActionProvider provides actions for getting basic wallet information.
25
38
  */
@@ -61,6 +74,7 @@ class WalletActionProvider extends actionProvider_1.ActionProvider {
61
74
  ` * Network ID: ${network.networkId || "N/A"}`,
62
75
  ` * Chain ID: ${network.chainId || "N/A"}`,
63
76
  `- Native Balance: ${balance.toString()} ${terminology.unit}`,
77
+ `- Native Balance: ${(0, viem_1.formatUnits)(balance, terminology.decimals)} ${terminology.displayUnit}`,
64
78
  ].join("\n");
65
79
  }
66
80
  catch (error) {
@@ -81,7 +95,8 @@ class WalletActionProvider extends actionProvider_1.ActionProvider {
81
95
  if (protocolFamily === "evm" && !args.to.startsWith("0x")) {
82
96
  args.to = `0x${args.to}`;
83
97
  }
84
- const result = await walletProvider.nativeTransfer(args.to, args.value);
98
+ const amountInAtomicUnits = (0, viem_1.parseUnits)(args.value, terminology.decimals);
99
+ const result = await walletProvider.nativeTransfer(args.to, amountInAtomicUnits.toString());
85
100
  return [
86
101
  `Transferred ${args.value} ${terminology.displayUnit} to ${args.to}`,
87
102
  `${terminology.type}: ${result}`,
@@ -115,15 +130,11 @@ __decorate([
115
130
  (0, actionDecorator_1.CreateAction)({
116
131
  name: "native_transfer",
117
132
  description: `
118
- This tool will transfer native tokens from the wallet to another onchain address.
133
+ This tool will transfer (send) native tokens from the wallet to another onchain address.
119
134
 
120
135
  It takes the following inputs:
121
- - amount: The amount to transfer in whole units (e.g. 1 ETH, 0.1 SOL)
136
+ - amount: The amount to transfer in whole units (e.g. 4.2 ETH, 0.1 SOL)
122
137
  - destination: The address to receive the funds
123
-
124
- Important notes:
125
- - Ensure sufficient balance of the input asset before transferring
126
- - Ensure there is sufficient native token balance for gas fees
127
138
  `,
128
139
  schema: schemas_1.NativeTransferSchema,
129
140
  }),
@@ -2,6 +2,7 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  const walletActionProvider_1 = require("./walletActionProvider");
4
4
  const schemas_1 = require("./schemas");
5
+ const viem_1 = require("viem");
5
6
  describe("Wallet Action Provider", () => {
6
7
  const MOCK_ADDRESS = "0xe6b2af36b3bb8d47206a129ff11d5a2de2a63c83";
7
8
  const MOCK_ETH_BALANCE = 1000000000000000000n;
@@ -47,6 +48,7 @@ describe("Wallet Action Provider", () => {
47
48
  ` * Network ID: ${MOCK_EVM_NETWORK.networkId}`,
48
49
  ` * Chain ID: ${MOCK_EVM_NETWORK.chainId}`,
49
50
  `- Native Balance: ${MOCK_ETH_BALANCE.toString()} WEI`,
51
+ `- Native Balance: ${(0, viem_1.formatUnits)(MOCK_ETH_BALANCE, 18)} ETH`,
50
52
  ].join("\n");
51
53
  expect(response).toBe(expectedResponse);
52
54
  });
@@ -63,6 +65,7 @@ describe("Wallet Action Provider", () => {
63
65
  ` * Network ID: ${MOCK_SOLANA_NETWORK.networkId}`,
64
66
  ` * Chain ID: N/A`,
65
67
  `- Native Balance: ${MOCK_SOL_BALANCE.toString()} LAMPORTS`,
68
+ `- Native Balance: ${(0, viem_1.formatUnits)(MOCK_SOL_BALANCE, 9)} SOL`,
66
69
  ].join("\n");
67
70
  expect(response).toBe(expectedResponse);
68
71
  });
@@ -79,6 +82,7 @@ describe("Wallet Action Provider", () => {
79
82
  ` * Network ID: ${MOCK_UNKNOWN_NETWORK.networkId}`,
80
83
  ` * Chain ID: N/A`,
81
84
  `- Native Balance: ${MOCK_ETH_BALANCE.toString()} `,
85
+ `- Native Balance: ${MOCK_ETH_BALANCE.toString()} `,
82
86
  ].join("\n");
83
87
  expect(response).toBe(expectedResponse);
84
88
  });
@@ -125,7 +129,7 @@ describe("Wallet Action Provider", () => {
125
129
  value: MOCK_AMOUNT,
126
130
  };
127
131
  const response = await actionProvider.nativeTransfer(mockWallet, args);
128
- expect(mockWallet.nativeTransfer).toHaveBeenCalledWith(MOCK_DESTINATION, MOCK_AMOUNT);
132
+ expect(mockWallet.nativeTransfer).toHaveBeenCalledWith(MOCK_DESTINATION, (0, viem_1.parseUnits)(MOCK_AMOUNT, 18).toString());
129
133
  expect(response).toBe(`Transferred ${MOCK_AMOUNT} ETH to ${MOCK_DESTINATION}\nTransaction hash: ${MOCK_TRANSACTION_HASH}`);
130
134
  });
131
135
  it("should successfully transfer SOL", async () => {
@@ -136,7 +140,7 @@ describe("Wallet Action Provider", () => {
136
140
  value: MOCK_AMOUNT,
137
141
  };
138
142
  const response = await actionProvider.nativeTransfer(mockWallet, args);
139
- expect(mockWallet.nativeTransfer).toHaveBeenCalledWith(MOCK_DESTINATION, MOCK_AMOUNT);
143
+ expect(mockWallet.nativeTransfer).toHaveBeenCalledWith(MOCK_DESTINATION, (0, viem_1.parseUnits)(MOCK_AMOUNT, 9).toString());
140
144
  expect(response).toBe(`Transferred ${MOCK_AMOUNT} SOL to ${MOCK_DESTINATION}\nSignature: ${MOCK_SIGNATURE}`);
141
145
  });
142
146
  it("should handle ETH transfer errors", async () => {
@@ -0,0 +1 @@
1
+ export declare const ZERION_V1_BASE_URL = "https://api.zerion.io/v1";
@@ -0,0 +1,4 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.ZERION_V1_BASE_URL = void 0;
4
+ exports.ZERION_V1_BASE_URL = "https://api.zerion.io/v1";
@@ -0,0 +1,2 @@
1
+ export * from "./zerionActionProvider";
2
+ export * from "./schemas";
@@ -0,0 +1,18 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
16
+ Object.defineProperty(exports, "__esModule", { value: true });
17
+ __exportStar(require("./zerionActionProvider"), exports);
18
+ __exportStar(require("./schemas"), exports);
@@ -0,0 +1,11 @@
1
+ import { z } from "zod";
2
+ /**
3
+ * Input schema for getting wallet portfolio.
4
+ */
5
+ export declare const GetWalletPortfolioSchema: z.ZodObject<{
6
+ walletAddress: z.ZodString;
7
+ }, "strip", z.ZodTypeAny, {
8
+ walletAddress: string;
9
+ }, {
10
+ walletAddress: string;
11
+ }>;
@@ -0,0 +1,15 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.GetWalletPortfolioSchema = void 0;
4
+ const zod_1 = require("zod");
5
+ /**
6
+ * Input schema for getting wallet portfolio.
7
+ */
8
+ exports.GetWalletPortfolioSchema = zod_1.z
9
+ .object({
10
+ walletAddress: zod_1.z
11
+ .string()
12
+ .describe("The wallet address to fetch portfolio for. Defaults is empty string, which will raise an error later if not provided."),
13
+ })
14
+ .strip()
15
+ .describe("Input schema for fetching wallet portfolio");
@@ -0,0 +1,125 @@
1
+ export interface ZerionFungiblePositionsResponse {
2
+ links: {
3
+ self: string;
4
+ };
5
+ data: ZerionFungiblePosition[];
6
+ }
7
+ export interface ZerionPortfolioResponse {
8
+ links: {
9
+ self: string;
10
+ };
11
+ data: ZerionPortfolio;
12
+ }
13
+ export interface ZerionPortfolio {
14
+ type: string;
15
+ id: string;
16
+ attributes: PortfolioAttributes;
17
+ }
18
+ export interface PortfolioAttributes {
19
+ positions_distribution_by_type: {
20
+ wallet: number;
21
+ deposited: number;
22
+ borrowed: number;
23
+ locked: number;
24
+ staked: number;
25
+ };
26
+ positions_distribution_by_chain: {
27
+ [key: string]: number;
28
+ };
29
+ total: {
30
+ positions: number;
31
+ };
32
+ changes: {
33
+ absolute_1d: number;
34
+ percent_1d: number;
35
+ };
36
+ }
37
+ export interface ZerionFungiblePosition {
38
+ type: string;
39
+ id: string;
40
+ attributes: FungibleAttributes;
41
+ relationships: Relationships;
42
+ }
43
+ export interface FungibleAttributes {
44
+ parent: null | string;
45
+ protocol: null | string;
46
+ pool_address?: string;
47
+ group_id?: string;
48
+ name: string;
49
+ position_type: PositionType;
50
+ quantity: Quantity;
51
+ value: number | null;
52
+ price: number;
53
+ changes: Changes | null;
54
+ fungible_info: FungibleInfo;
55
+ flags: AttributesFlags;
56
+ updated_at: Date;
57
+ updated_at_block: number | null;
58
+ application_metadata?: ApplicationMetadata;
59
+ }
60
+ export interface ApplicationMetadata {
61
+ name: string;
62
+ icon: Icon;
63
+ url: string;
64
+ }
65
+ export interface Icon {
66
+ url: string;
67
+ }
68
+ export interface Changes {
69
+ absolute_1d: number;
70
+ percent_1d: number;
71
+ }
72
+ export interface AttributesFlags {
73
+ displayable: boolean;
74
+ is_trash: boolean;
75
+ }
76
+ export interface FungibleInfo {
77
+ name: string;
78
+ symbol: string;
79
+ icon: Icon | null;
80
+ flags: FungibleInfoFlags;
81
+ implementations: Implementation[];
82
+ }
83
+ export interface FungibleInfoFlags {
84
+ verified: boolean;
85
+ }
86
+ export interface Implementation {
87
+ chain_id: string;
88
+ address: null | string;
89
+ decimals: number;
90
+ }
91
+ export declare enum PositionType {
92
+ Deposit = "deposit",
93
+ Reward = "reward",
94
+ Staked = "staked",
95
+ Wallet = "wallet"
96
+ }
97
+ export interface Quantity {
98
+ int: string;
99
+ decimals: number;
100
+ float: number;
101
+ numeric: string;
102
+ }
103
+ export interface Relationships {
104
+ chain: Chain;
105
+ dapp?: Dapp;
106
+ fungible: Chain;
107
+ }
108
+ export interface Dapp {
109
+ data: Data;
110
+ }
111
+ export interface Chain {
112
+ links: {
113
+ related: string;
114
+ };
115
+ data: Data;
116
+ }
117
+ export interface Data {
118
+ type: DataType;
119
+ id: string;
120
+ }
121
+ export declare enum DataType {
122
+ Chains = "chains",
123
+ Dapps = "dapps",
124
+ Fungibles = "fungibles"
125
+ }
@@ -0,0 +1,16 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.DataType = exports.PositionType = void 0;
4
+ var PositionType;
5
+ (function (PositionType) {
6
+ PositionType["Deposit"] = "deposit";
7
+ PositionType["Reward"] = "reward";
8
+ PositionType["Staked"] = "staked";
9
+ PositionType["Wallet"] = "wallet";
10
+ })(PositionType || (exports.PositionType = PositionType = {}));
11
+ var DataType;
12
+ (function (DataType) {
13
+ DataType["Chains"] = "chains";
14
+ DataType["Dapps"] = "dapps";
15
+ DataType["Fungibles"] = "fungibles";
16
+ })(DataType || (exports.DataType = DataType = {}));
@@ -0,0 +1,3 @@
1
+ import { ZerionFungiblePosition, ZerionPortfolio } from "./types";
2
+ export declare const formatPortfolioData: (data: ZerionPortfolio) => string;
3
+ export declare const formatPositionsData: (data: ZerionFungiblePosition[]) => string;
@@ -0,0 +1,45 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.formatPositionsData = exports.formatPortfolioData = void 0;
4
+ const formatPortfolioData = (data) => {
5
+ // Total value
6
+ const totalValue = data.attributes.total.positions;
7
+ const totalValueStr = `$${totalValue.toFixed(2)}`;
8
+ // 24h change
9
+ const changePercent = data.attributes.changes.percent_1d;
10
+ const changeStr = `${changePercent.toFixed(2)}%`;
11
+ // Positions by type (filter out 0)
12
+ const types = Object.entries(data.attributes.positions_distribution_by_type)
13
+ .filter(([_, v]) => v > 0)
14
+ .map(([type, value]) => `${type}: $${value.toFixed(2)}`)
15
+ .join(", ");
16
+ // Positions by chain (top 5 by value)
17
+ const topChains = Object.entries(data.attributes.positions_distribution_by_chain)
18
+ .sort(([, a], [, b]) => b - a)
19
+ .slice(0, 5)
20
+ .map(([chain, value]) => `${chain}: $${value.toFixed(2)}`)
21
+ .join(", ");
22
+ // Final summary string
23
+ return `Wallet Portfolio Overview:
24
+ - Total Value: ${totalValueStr}
25
+ - 24h Change: ${changeStr}
26
+ - Position Types: ${types}
27
+ - Top Chains: ${topChains}`;
28
+ };
29
+ exports.formatPortfolioData = formatPortfolioData;
30
+ const formatPositionsData = (data) => {
31
+ const filtered = data.filter(pos => pos.attributes.value !== null);
32
+ // Sort by value (descending)
33
+ const sorted = filtered.sort((a, b) => b.attributes.value - a.attributes.value);
34
+ const lines = [];
35
+ let totalValue = 0;
36
+ for (const pos of sorted) {
37
+ const { value, position_type, fungible_info, application_metadata } = pos.attributes;
38
+ const chain = pos.relationships.chain.data.id;
39
+ const protocolLine = application_metadata?.name ? `via ${application_metadata.name}` : "";
40
+ lines.push(`- ${fungible_info.symbol} (${fungible_info.name}) on ${chain} — $${value.toFixed(2)} [${position_type}${protocolLine ? " " + protocolLine : ""}]`);
41
+ totalValue += value;
42
+ }
43
+ return `Total Value: $${totalValue.toFixed(2)}\n\nToken Positions (>$1):\n${lines.join("\n")}`;
44
+ };
45
+ exports.formatPositionsData = formatPositionsData;