@antseed/cli 0.1.149 → 0.1.151

package/dist/proxy/buyer-proxy.test.js

@@ -8,7 +8,7 @@ import test from 'node:test';
 import { ANTSEED_BUYER_FAULT_ERROR_CODE, ANTSEED_FAULT_ATTRIBUTION_HEADER, CONNECTION_CAPABILITY_COOPERATIVE_CLOSE_V1, CONNECTION_CAPABILITY_RELAYS_SWEEPS_V1, buyerFault, computeOnChainReputationScore, } from '@antseed/node';
 import { DEFAULT_BUYER_PEER_REFRESH_INTERVAL_MS } from '../config/defaults.js';
 import { BuyerProxy, isModelNotFoundResponse, makeVerifierReach, mergeJsonStateFile, parsePeerPinnedService, parsePersistedPeers, rewritePeerPinnedServiceInBody, selectCandidatePeersForRouting, substituteRoutedModelAlias, sweepStaleStateTmpFiles, } from './buyer-proxy.js';
-import { overrideRoutedModelInBody, SYSTEM_ROUTED_MODEL_HEADER } from './request-utils.js';
+import { extractRequestedService, overrideRoutedModelInBody, SYSTEM_ROUTED_MODEL_HEADER } from './request-utils.js';
 function makePeer(seed, providers) {
     const repeated = (seed.repeat(40) + 'a'.repeat(40)).slice(0, 40);
     return {
@@ -423,7 +423,7 @@ test('selectCandidatePeersForRouting can still include peers without service pro
     assert.equal(result.candidatePeers.length, 1);
     assert.equal(result.candidatePeers[0]?.peerId, peerWithoutMetadata.peerId);
 });
-test('model-only request routes to the highest-reputation canonical service match', async () => {
+test('model-only request routes to the highest-ranked canonical service match', async () => {
     const lower = makePeer('a', ['anthropic']);
     lower.reputationScore = 40;
     lower.providerServiceApiProtocols = {
@@ -471,6 +471,7 @@ test('conversation routing keeps the actual peer as a soft preference and fails
     let preferredReachable = true;
     proxy._node.sendRequest = async (peer, request) => {
         attempts.push(peer.peerId);
+        assert.equal(request.headers['x-vpr-session-id'], undefined);
         assert.equal(request.headers['x-antstation-session-id'], undefined);
         assert.equal(request.headers['x-antseed-prefer-peer'], undefined);
         return {
@@ -481,7 +482,7 @@ test('conversation routing keeps the actual peer as a soft preference and fails
         };
     };
     const conversationHeaders = {
-        'x-antstation-session-id': 'conversation-soft-affinity',
+        'x-vpr-session-id': 'conversation-soft-affinity',
         'x-antseed-prefer-peer': preferred.peerId,
     };
     await invokeProxy(proxy, makeProxyRequest({
@@ -492,22 +493,22 @@ test('conversation routing keeps the actual peer as a soft preference and fails
     preferredReachable = false;
     attempts.length = 0;
     await invokeProxy(proxy, makeProxyRequest({
-        headers: { 'x-antstation-session-id': 'conversation-soft-affinity' },
+        headers: { 'x-vpr-session-id': 'conversation-soft-affinity' },
         body: { model: 'kimi-k3', messages: [{ role: 'user', content: 'again' }] },
     }));
     assert.deepEqual(attempts, [preferred.peerId, rankedFirst.peerId]);
-    const stored = proxy._conversations.get('antstation:conversation-soft-affinity');
+    const stored = proxy._conversations.get('vpr:conversation-soft-affinity');
     assert.equal(stored?.pinnedModel, `${rankedFirst.peerId}@Kimi K3`);
     assert.equal(stored?.lastModel, `${rankedFirst.peerId}@Kimi K3`);
     attempts.length = 0;
     await invokeProxy(proxy, makeProxyRequest({
-        headers: { 'x-antstation-session-id': 'conversation-soft-affinity' },
+        headers: { 'x-vpr-session-id': 'conversation-soft-affinity' },
         body: { model: 'kimi-k3', messages: [{ role: 'user', content: 'third' }] },
     }));
     assert.equal(attempts[0], rankedFirst.peerId);
     const route = await invokeProxy(proxy, makeProxyRequest({
         method: 'GET',
-        path: `/_antseed/conversations/${encodeURIComponent('antstation:conversation-soft-affinity')}`,
+        path: `/_antseed/conversations/${encodeURIComponent('vpr:conversation-soft-affinity')}`,
     }));
     assert.equal(route.statusCode, 200);
     assert.equal(JSON.parse(route.body).conversation.lastModel, `${rankedFirst.peerId}@Kimi K3`);
@@ -1958,6 +1959,29 @@ function parseJsonBody(body) {
     return JSON.parse(new TextDecoder().decode(body));
 }
 const jsonHeaders = { 'content-type': 'application/json' };
+test('extractRequestedService reads the model from multipart image edits', () => {
+    const boundary = 'image-edit-boundary';
+    const body = new TextEncoder().encode([
+        `--${boundary}`,
+        'Content-Disposition: form-data; name="model"',
+        '',
+        'gpt-image-2',
+        `--${boundary}`,
+        'Content-Disposition: form-data; name="image"; filename="source.png"',
+        'Content-Type: image/png',
+        '',
+        'image-bytes',
+        `--${boundary}--`,
+        '',
+    ].join('\r\n'));
+    assert.equal(extractRequestedService({
+        requestId: 'request-1',
+        method: 'POST',
+        path: '/v1/images/edits',
+        headers: { 'content-type': `multipart/form-data; boundary=${boundary}` },
+        body,
+    }), 'gpt-image-2');
+});
 test('parsePeerPinnedService parses 40-char hex peer prefixes', () => {
     assert.deepEqual(parsePeerPinnedService(`${validPeerId}@claude-sonnet-4-5`), {
         peerId: validPeerId,
@@ -2511,4 +2535,69 @@ test('sweepStaleStateTmpFiles: removes aged temp files, keeps fresh and unrelate
 test('sweepStaleStateTmpFiles: tolerates a missing directory', async () => {
     await sweepStaleStateTmpFiles(join(tmpdir(), 'antseed-does-not-exist', randomUUID()));
 });
+// ---------- Deposit watcher control plane ----------
+test('deposits/status reports no watcher until one is attached', async () => {
+    const proxy = new BuyerProxy({
+        port: 0,
+        dataDir: '/tmp/antseed-test',
+        node: { router: null },
+    });
+    const res = await invokeProxy(proxy, makeProxyRequest({ method: 'GET', path: '/_antseed/deposits/status' }));
+    assert.equal(res.statusCode, 200);
+    assert.deepEqual(JSON.parse(res.body), { ok: true, watcher: false, status: null });
+});
+test('deposits/watch returns 503 when no watcher is attached', async () => {
+    const proxy = new BuyerProxy({
+        port: 0,
+        dataDir: '/tmp/antseed-test',
+        node: { router: null },
+    });
+    const res = await invokeProxy(proxy, makeProxyRequest({ path: '/_antseed/deposits/watch', body: { mode: 'active' } }));
+    assert.equal(res.statusCode, 503);
+    assert.equal(JSON.parse(res.body).ok, false);
+});
+test('deposits/watch promotes and demotes an attached watcher and returns its status', async () => {
+    const proxy = new BuyerProxy({
+        port: 0,
+        dataDir: '/tmp/antseed-test',
+        node: { router: null },
+    });
+    const calls = [];
+    const fakeStatus = {
+        mode: 'idle',
+        address: '0x' + 'ab'.repeat(20),
+        usdcBalance: '0',
+        sweepInFlight: false,
+        lastEvent: null,
+    };
+    proxy.setDepositWatcher({
+        promote: () => { calls.push('promote'); },
+        demote: () => { calls.push('demote'); },
+        status: () => fakeStatus,
+    });
+    const active = await invokeProxy(proxy, makeProxyRequest({ path: '/_antseed/deposits/watch', body: { mode: 'active' } }));
+    assert.equal(active.statusCode, 200);
+    assert.deepEqual(JSON.parse(active.body), { ok: true, status: fakeStatus });
+    const background = await invokeProxy(proxy, makeProxyRequest({ path: '/_antseed/deposits/watch', body: { mode: 'background' } }));
+    assert.equal(background.statusCode, 200);
+    assert.deepEqual(calls, ['promote', 'demote']);
+    const invalid = await invokeProxy(proxy, makeProxyRequest({ path: '/_antseed/deposits/watch', body: { mode: 'nonsense' } }));
+    assert.equal(invalid.statusCode, 400);
+    assert.deepEqual(calls, ['promote', 'demote']);
+    const status = await invokeProxy(proxy, makeProxyRequest({ method: 'GET', path: '/_antseed/deposits/status' }));
+    assert.deepEqual(JSON.parse(status.body), { ok: true, watcher: true, status: fakeStatus });
+});
+test('getSweepReceipt returns cached relayer receipts case-insensitively', () => {
+    const proxy = new BuyerProxy({
+        port: 0,
+        dataDir: '/tmp/antseed-test',
+        node: { router: null },
+    });
+    const nonce = '0x' + 'AB'.repeat(32);
+    const receipt = { authNonce: nonce.toLowerCase(), status: 'confirmed', txHash: '0x' + '12'.repeat(32) };
+    proxy._sweepReceipts.set(nonce.toLowerCase(), receipt);
+    assert.equal(proxy.getSweepReceipt(nonce), receipt);
+    assert.equal(proxy.getSweepReceipt(nonce.toLowerCase()), receipt);
+    assert.equal(proxy.getSweepReceipt('0x' + '00'.repeat(32)), null);
+});
 //# sourceMappingURL=buyer-proxy.test.js.map