feat: add tool gateway protocol

2026-07-02 09:23:55 +08:00
parent e24e6ca4a7
commit 21e03d86b5
8 changed files with 388 additions and 1 deletions
@@ -0,0 +1,77 @@
+import assert from 'node:assert/strict';
+import test from 'node:test';
+import { createToolGateway, extractToolInstruction } from './tool-gateway.mjs';
+
+test('tool gateway extracts text instructions from H5 message content', () => {
+  assert.equal(
+    extractToolInstruction({
+      content: [
+        { type: 'text', text: '第一步' },
+        { type: 'image_url', image_url: { url: 'https://example.test/a.png' } },
+        { type: 'text', text: '第二步' },
+      ],
+    }),
+    '第一步\n第二步',
+  );
+});
+
+test('tool gateway is disabled by default and reports protocol', () => {
+  const gateway = createToolGateway({ env: {} });
+  assert.deepEqual(gateway.getStatus(), {
+    enabled: false,
+    dryRun: false,
+    protocol: 'agent-run-v1',
+    executors: ['aider', 'openhands'],
+    defaultExecutor: 'aider',
+    openhandsTaskTypes: ['repo_refactor', 'multi_file', 'complex_repo'],
+  });
+});
+
+test('tool gateway selects openhands for configured task types', () => {
+  const gateway = createToolGateway({
+    env: {
+      MEMIND_TOOL_GATEWAY_DEFAULT_EXECUTOR: 'aider',
+      MEMIND_TOOL_GATEWAY_OPENHANDS_TASK_TYPES: 'repo_refactor',
+    },
+  });
+  assert.equal(gateway.selectExecutor({ taskType: 'repo_refactor' }), 'openhands');
+  assert.equal(gateway.selectExecutor({ taskType: 'small_patch' }), 'aider');
+});
+
+test('tool gateway dry run builds executor launch plan without spawning', async () => {
+  const plans = [];
+  const gateway = createToolGateway({
+    env: {
+      MEMIND_TOOL_GATEWAY_ENABLED: '1',
+      MEMIND_TOOL_GATEWAY_DRY_RUN: '1',
+      MEMIND_TOOL_GATEWAY_DEFAULT_EXECUTOR: 'aider',
+    },
+    llmProviderService: {
+      async getExecutorLaunchPlan(executor, options) {
+        plans.push({ executor, options });
+        return {
+          ok: true,
+          executor,
+          cwd: options.cwd,
+          command: '/usr/bin/true',
+          args: ['--noop'],
+        };
+      },
+    },
+  });
+
+  const result = await gateway.executeJob({
+    runId: 'run-1',
+    requestId: 'req-1',
+    userId: 'user-1',
+    cwd: '/tmp/work',
+    taskType: 'small_patch',
+    userMessage: { content: [{ type: 'text', text: 'fix it' }] },
+  });
+
+  assert.equal(result.dryRun, true);
+  assert.equal(result.executor, 'aider');
+  assert.equal(result.command, '/usr/bin/true');
+  assert.equal(plans.length, 1);
+  assert.equal(plans[0].options.instruction, 'fix it');
+});