refactor(evm-wallet-experiment): extract wallet capabilities into capabilities.ts

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: grypez

packages/evm-wallet-experiment/test/e2e/docker/helpers/capabilities.ts

@@ -0,0 +1,95 @@
+import { capability } from '@ocap/kernel-agents/capabilities/capability';
+import type { CapabilityRecord } from '@ocap/kernel-agents/types';
+
+type CallAway = (
+  kref: string,
+  method: string,
+  args?: unknown[],
+) => Promise<unknown>;
+
+/**
+ * Build the wallet capability set for an agent operating against the away coordinator.
+ *
+ * @param callAway - Function to invoke a method on a vat in the away container.
+ * @param awayCoordKref - Kref of the away coordinator vat.
+ * @returns A capability record suitable for passing to {@link makeChatAgent}.
+ */
+export function makeWalletCapabilities(
+  callAway: CallAway,
+  awayCoordKref: string,
+): CapabilityRecord {
+  const balance = capability(
+    async () => {
+      const accounts = (await callAway(
+        awayCoordKref,
+        'getAccounts',
+      )) as string[];
+      const raw = (await callAway(awayCoordKref, 'request', [
+        'eth_getBalance',
+        [accounts[0], 'latest'],
+      ])) as string;
+      return `Balance: ${(parseInt(raw, 16) / 1e18).toFixed(4)} ETH`;
+    },
+    {
+      description: 'Get the ETH balance of the wallet',
+      args: {},
+      returns: { type: 'string' },
+    },
+  );
+
+  const accounts = capability(
+    async () => {
+      const addrs = (await callAway(awayCoordKref, 'getAccounts')) as string[];
+      return `Accounts: ${addrs.join(', ')}`;
+    },
+    {
+      description: 'Get wallet accounts/addresses',
+      args: {},
+      returns: { type: 'string' },
+    },
+  );
+
+  const send = capability(
+    async ({ to, value }: { to: string; value: string }) => {
+      const addrs = (await callAway(awayCoordKref, 'getAccounts')) as string[];
+      const weiValue = `0x${(parseFloat(value || '0.01') * 1e18).toString(16)}`;
+      const txHash = await callAway(awayCoordKref, 'request', [
+        'eth_sendTransaction',
+        [
+          {
+            from: addrs[0],
+            to: to || '0x70997970c51812dc3a010c7d01b50e0d17dc79c8',
+            value: weiValue,
+          },
+        ],
+      ]);
+      return `Transaction sent: ${txHash as string}`;
+    },
+    {
+      description: 'Send ETH to an address',
+      args: { to: { type: 'string' }, value: { type: 'string' } },
+      returns: { type: 'string' },
+    },
+  );
+
+  const sign = capability(
+    async ({ message }: { message: string }) => {
+      const signature = await callAway(awayCoordKref, 'signMessage', [
+        message || 'test',
+      ]);
+      return `Signature: ${signature as string}`;
+    },
+    {
+      description: 'Sign a message with the wallet',
+      args: { message: { type: 'string' } },
+      returns: { type: 'string' },
+    },
+  );
+
+  return {
+    wallet_balance: balance,
+    wallet_accounts: accounts,
+    wallet_send: send,
+    wallet_sign: sign,
+  };
+}

packages/evm-wallet-experiment/test/e2e/docker/run-agent-loop-tests.mjs

@@ -11,9 +11,10 @@
  * as a bonus check rather than a hard requirement.
  */
 
-import { capability } from '@ocap/kernel-agents/capabilities/capability';
 import { makeChatAgent } from '@ocap/kernel-agents/chat';
 
+import { makeWalletCapabilities } from './helpers/capabilities.ts';
+
 const LLM_BASE_URL = process.env.LLM_BASE_URL ?? 'http://llm:11434';
 const LLM_MODEL = process.env.LLM_MODEL ?? 'qwen2.5:0.5b';
 const LLM_API = process.env.LLM_API ?? 'ollama';
@@ -50,71 +51,6 @@ export async function runAgentLoopTests(ctx) {
 
   const awayCoordKref = awayInfo.coordinatorKref;
 
-  const walletBalance = capability(
-    async () => {
-      const accounts = await callAway(awayCoordKref, 'getAccounts');
-      const balance = await callAway(awayCoordKref, 'request', [
-        'eth_getBalance',
-        [accounts[0], 'latest'],
-      ]);
-      return `Balance: ${(parseInt(balance, 16) / 1e18).toFixed(4)} ETH`;
-    },
-    {
-      description: 'Get the ETH balance of the wallet',
-      args: {},
-      returns: { type: 'string' },
-    },
-  );
-
-  const walletAccounts = capability(
-    async () => {
-      const accounts = await callAway(awayCoordKref, 'getAccounts');
-      return `Accounts: ${accounts.join(', ')}`;
-    },
-    {
-      description: 'Get wallet accounts/addresses',
-      args: {},
-      returns: { type: 'string' },
-    },
-  );
-
-  const walletSend = capability(
-    async ({ to, value }) => {
-      const accounts = await callAway(awayCoordKref, 'getAccounts');
-      const weiValue = `0x${(parseFloat(value || '0.01') * 1e18).toString(16)}`;
-      const txHash = await callAway(awayCoordKref, 'request', [
-        'eth_sendTransaction',
-        [
-          {
-            from: accounts[0],
-            to: to || '0x70997970c51812dc3a010c7d01b50e0d17dc79c8',
-            value: weiValue,
-          },
-        ],
-      ]);
-      return `Transaction sent: ${txHash}`;
-    },
-    {
-      description: 'Send ETH to an address',
-      args: { to: { type: 'string' }, value: { type: 'string' } },
-      returns: { type: 'string' },
-    },
-  );
-
-  const walletSign = capability(
-    async ({ message }) => {
-      const signature = await callAway(awayCoordKref, 'signMessage', [
-        message || 'test',
-      ]);
-      return `Signature: ${signature}`;
-    },
-    {
-      description: 'Sign a message with the wallet',
-      args: { message: { type: 'string' } },
-      returns: { type: 'string' },
-    },
-  );
-
   const chat = async (messages) => {
     const response = await fetch(CHAT_URL, {
       method: 'POST',
@@ -126,12 +62,10 @@ export async function runAgentLoopTests(ctx) {
 
   const agent = makeChatAgent({
     chat,
-    capabilities: {
-      wallet_balance: walletBalance,
-      wallet_accounts: walletAccounts,
-      wallet_send: walletSend,
-      wallet_sign: walletSign,
-    },
+    capabilities: makeWalletCapabilities(
+      (_kref, method, args) => callAway(method, args),
+      awayCoordKref,
+    ),
   });
 
   // -- Test 1: Balance query --