John
78 lines
2.3 KiB
JavaScript
78 lines
2.3 KiB
JavaScript
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');
|
|
});
|