BREAKING CHANGE(api): Migrate public API to ai-sdk v6 and refactor core agent architecture: replace class-based DualAgent/Driver/Guardian with a single runAgent function; introduce ts_tools factories for tools, a compactMessages compaction subpath, and truncateOutput utility; simplify ToolRegistry to return ToolSet and remove legacy BaseToolWrapper/tool classes; update package exports and dependencies and bump major version.
This commit is contained in:
294
test/test.ts
294
test/test.ts
@@ -1,150 +1,188 @@
|
||||
import { expect, tap } from '@git.zone/tstest/tapbundle';
|
||||
import * as smartagent from '../ts/index.js';
|
||||
import { filesystemTool, shellTool, httpTool, jsonTool, truncateOutput } from '../ts_tools/index.js';
|
||||
import { compactMessages } from '../ts_compaction/index.js';
|
||||
|
||||
// Test exports
|
||||
tap.test('should export DualAgentOrchestrator class', async () => {
|
||||
expect(smartagent.DualAgentOrchestrator).toBeTypeOf('function');
|
||||
// ============================================================
|
||||
// Core exports
|
||||
// ============================================================
|
||||
|
||||
tap.test('should export runAgent function', async () => {
|
||||
expect(smartagent.runAgent).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
tap.test('should export DriverAgent class', async () => {
|
||||
expect(smartagent.DriverAgent).toBeTypeOf('function');
|
||||
tap.test('should export ToolRegistry class', async () => {
|
||||
expect(smartagent.ToolRegistry).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
tap.test('should export GuardianAgent class', async () => {
|
||||
expect(smartagent.GuardianAgent).toBeTypeOf('function');
|
||||
tap.test('should export ContextOverflowError class', async () => {
|
||||
expect(smartagent.ContextOverflowError).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
tap.test('should export BaseToolWrapper class', async () => {
|
||||
expect(smartagent.BaseToolWrapper).toBeTypeOf('function');
|
||||
tap.test('should export truncateOutput function', async () => {
|
||||
expect(smartagent.truncateOutput).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
// Test standard tools exports
|
||||
tap.test('should export FilesystemTool class', async () => {
|
||||
expect(smartagent.FilesystemTool).toBeTypeOf('function');
|
||||
tap.test('should re-export tool helper', async () => {
|
||||
expect(smartagent.tool).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
tap.test('should export HttpTool class', async () => {
|
||||
expect(smartagent.HttpTool).toBeTypeOf('function');
|
||||
tap.test('should re-export z (zod)', async () => {
|
||||
expect(smartagent.z).toBeTruthy();
|
||||
});
|
||||
|
||||
tap.test('should export ShellTool class', async () => {
|
||||
expect(smartagent.ShellTool).toBeTypeOf('function');
|
||||
tap.test('should re-export stepCountIs', async () => {
|
||||
expect(smartagent.stepCountIs).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
tap.test('should export BrowserTool class', async () => {
|
||||
expect(smartagent.BrowserTool).toBeTypeOf('function');
|
||||
});
|
||||
// ============================================================
|
||||
// ToolRegistry
|
||||
// ============================================================
|
||||
|
||||
tap.test('should export DenoTool class', async () => {
|
||||
expect(smartagent.DenoTool).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
// Test tool instantiation
|
||||
tap.test('should be able to instantiate FilesystemTool', async () => {
|
||||
const fsTool = new smartagent.FilesystemTool();
|
||||
expect(fsTool.name).toEqual('filesystem');
|
||||
expect(fsTool.actions).toBeTypeOf('object');
|
||||
expect(fsTool.actions.length).toBeGreaterThan(0);
|
||||
});
|
||||
|
||||
tap.test('should be able to instantiate HttpTool', async () => {
|
||||
const httpTool = new smartagent.HttpTool();
|
||||
expect(httpTool.name).toEqual('http');
|
||||
expect(httpTool.actions).toBeTypeOf('object');
|
||||
});
|
||||
|
||||
tap.test('should be able to instantiate ShellTool', async () => {
|
||||
const shellTool = new smartagent.ShellTool();
|
||||
expect(shellTool.name).toEqual('shell');
|
||||
expect(shellTool.actions).toBeTypeOf('object');
|
||||
});
|
||||
|
||||
tap.test('should be able to instantiate BrowserTool', async () => {
|
||||
const browserTool = new smartagent.BrowserTool();
|
||||
expect(browserTool.name).toEqual('browser');
|
||||
expect(browserTool.actions).toBeTypeOf('object');
|
||||
});
|
||||
|
||||
tap.test('should be able to instantiate DenoTool', async () => {
|
||||
const denoTool = new smartagent.DenoTool();
|
||||
expect(denoTool.name).toEqual('deno');
|
||||
expect(denoTool.actions).toBeTypeOf('object');
|
||||
});
|
||||
|
||||
// Test tool descriptions
|
||||
tap.test('FilesystemTool should have required actions', async () => {
|
||||
const fsTool = new smartagent.FilesystemTool();
|
||||
const actionNames = fsTool.actions.map((a) => a.name);
|
||||
expect(actionNames).toContain('read');
|
||||
expect(actionNames).toContain('write');
|
||||
expect(actionNames).toContain('list');
|
||||
expect(actionNames).toContain('delete');
|
||||
expect(actionNames).toContain('exists');
|
||||
});
|
||||
|
||||
tap.test('HttpTool should have required actions', async () => {
|
||||
const httpTool = new smartagent.HttpTool();
|
||||
const actionNames = httpTool.actions.map((a) => a.name);
|
||||
expect(actionNames).toContain('get');
|
||||
expect(actionNames).toContain('post');
|
||||
expect(actionNames).toContain('put');
|
||||
expect(actionNames).toContain('delete');
|
||||
});
|
||||
|
||||
tap.test('ShellTool should have required actions', async () => {
|
||||
const shellTool = new smartagent.ShellTool();
|
||||
const actionNames = shellTool.actions.map((a) => a.name);
|
||||
expect(actionNames).toContain('execute');
|
||||
expect(actionNames).toContain('which');
|
||||
});
|
||||
|
||||
tap.test('BrowserTool should have required actions', async () => {
|
||||
const browserTool = new smartagent.BrowserTool();
|
||||
const actionNames = browserTool.actions.map((a) => a.name);
|
||||
expect(actionNames).toContain('screenshot');
|
||||
expect(actionNames).toContain('pdf');
|
||||
expect(actionNames).toContain('evaluate');
|
||||
expect(actionNames).toContain('getPageContent');
|
||||
});
|
||||
|
||||
tap.test('DenoTool should have required actions', async () => {
|
||||
const denoTool = new smartagent.DenoTool();
|
||||
const actionNames = denoTool.actions.map((a) => a.name);
|
||||
expect(actionNames).toContain('execute');
|
||||
expect(actionNames).toContain('executeWithResult');
|
||||
});
|
||||
|
||||
// Test getCallSummary
|
||||
tap.test('FilesystemTool should generate call summaries', async () => {
|
||||
const fsTool = new smartagent.FilesystemTool();
|
||||
const summary = fsTool.getCallSummary('read', { path: '/tmp/test.txt' });
|
||||
expect(summary).toBeTypeOf('string');
|
||||
expect(summary).toInclude('/tmp/test.txt');
|
||||
});
|
||||
|
||||
tap.test('HttpTool should generate call summaries', async () => {
|
||||
const httpTool = new smartagent.HttpTool();
|
||||
const summary = httpTool.getCallSummary('get', { url: 'https://example.com' });
|
||||
expect(summary).toBeTypeOf('string');
|
||||
expect(summary).toInclude('example.com');
|
||||
});
|
||||
|
||||
tap.test('DenoTool should generate call summaries', async () => {
|
||||
const denoTool = new smartagent.DenoTool();
|
||||
const summary = denoTool.getCallSummary('execute', { code: 'console.log("hello");' });
|
||||
expect(summary).toBeTypeOf('string');
|
||||
expect(summary).toInclude('sandboxed');
|
||||
});
|
||||
|
||||
tap.test('DenoTool should show permissions in call summary', async () => {
|
||||
const denoTool = new smartagent.DenoTool();
|
||||
const summary = denoTool.getCallSummary('execute', {
|
||||
code: 'console.log("hello");',
|
||||
permissions: ['net', 'read']
|
||||
tap.test('ToolRegistry should register and return tools', async () => {
|
||||
const registry = new smartagent.ToolRegistry();
|
||||
const echoTool = smartagent.tool({
|
||||
description: 'Echo tool',
|
||||
inputSchema: smartagent.z.object({ text: smartagent.z.string() }),
|
||||
execute: async ({ text }: { text: string }) => text,
|
||||
});
|
||||
expect(summary).toBeTypeOf('string');
|
||||
expect(summary).toInclude('permissions');
|
||||
expect(summary).toInclude('net');
|
||||
registry.register('echo', echoTool);
|
||||
const tools = registry.getTools();
|
||||
expect(Object.keys(tools)).toContain('echo');
|
||||
});
|
||||
|
||||
// ============================================================
|
||||
// Truncation
|
||||
// ============================================================
|
||||
|
||||
tap.test('truncateOutput should not truncate short strings', async () => {
|
||||
const result = truncateOutput('hello world');
|
||||
expect(result.truncated).toBeFalse();
|
||||
expect(result.content).toEqual('hello world');
|
||||
});
|
||||
|
||||
tap.test('truncateOutput should truncate strings over maxLines', async () => {
|
||||
const lines = Array.from({ length: 3000 }, (_, i) => `line ${i}`).join('\n');
|
||||
const result = truncateOutput(lines, { maxLines: 100 });
|
||||
expect(result.truncated).toBeTrue();
|
||||
expect(result.notice).toBeTruthy();
|
||||
expect(result.content).toInclude('[Output truncated');
|
||||
});
|
||||
|
||||
tap.test('truncateOutput should truncate strings over maxBytes', async () => {
|
||||
const big = 'x'.repeat(100_000);
|
||||
const result = truncateOutput(big, { maxBytes: 1000 });
|
||||
expect(result.truncated).toBeTrue();
|
||||
});
|
||||
|
||||
// ============================================================
|
||||
// Tool factories
|
||||
// ============================================================
|
||||
|
||||
tap.test('filesystemTool returns expected tool names', async () => {
|
||||
const tools = filesystemTool();
|
||||
const names = Object.keys(tools);
|
||||
expect(names).toContain('read_file');
|
||||
expect(names).toContain('write_file');
|
||||
expect(names).toContain('list_directory');
|
||||
expect(names).toContain('delete_file');
|
||||
});
|
||||
|
||||
tap.test('shellTool returns expected tool names', async () => {
|
||||
const tools = shellTool();
|
||||
const names = Object.keys(tools);
|
||||
expect(names).toContain('run_command');
|
||||
});
|
||||
|
||||
tap.test('httpTool returns expected tool names', async () => {
|
||||
const tools = httpTool();
|
||||
const names = Object.keys(tools);
|
||||
expect(names).toContain('http_get');
|
||||
expect(names).toContain('http_post');
|
||||
});
|
||||
|
||||
tap.test('jsonTool returns expected tool names', async () => {
|
||||
const tools = jsonTool();
|
||||
const names = Object.keys(tools);
|
||||
expect(names).toContain('json_validate');
|
||||
expect(names).toContain('json_transform');
|
||||
});
|
||||
|
||||
tap.test('json_validate tool should validate valid JSON', async () => {
|
||||
const tools = jsonTool();
|
||||
const result = await (tools.json_validate as any).execute({
|
||||
jsonString: '{"name":"test","value":42}',
|
||||
});
|
||||
expect(result).toInclude('Valid JSON');
|
||||
});
|
||||
|
||||
tap.test('json_validate tool should detect invalid JSON', async () => {
|
||||
const tools = jsonTool();
|
||||
const result = await (tools.json_validate as any).execute({
|
||||
jsonString: '{invalid json',
|
||||
});
|
||||
expect(result).toInclude('Invalid JSON');
|
||||
});
|
||||
|
||||
tap.test('json_validate tool should check required fields', async () => {
|
||||
const tools = jsonTool();
|
||||
const result = await (tools.json_validate as any).execute({
|
||||
jsonString: '{"name":"test"}',
|
||||
requiredFields: ['name', 'missing_field'],
|
||||
});
|
||||
expect(result).toInclude('missing_field');
|
||||
});
|
||||
|
||||
tap.test('json_transform tool should pretty-print JSON', async () => {
|
||||
const tools = jsonTool();
|
||||
const result = await (tools.json_transform as any).execute({
|
||||
jsonString: '{"a":1,"b":2}',
|
||||
});
|
||||
expect(result).toInclude(' "a": 1');
|
||||
});
|
||||
|
||||
// ============================================================
|
||||
// Compaction export
|
||||
// ============================================================
|
||||
|
||||
tap.test('compactMessages should be a function', async () => {
|
||||
expect(compactMessages).toBeTypeOf('function');
|
||||
});
|
||||
|
||||
// ============================================================
|
||||
// Filesystem tool read/write round-trip
|
||||
// ============================================================
|
||||
|
||||
tap.test('filesystem tool should write and read a file', async () => {
|
||||
const tmpDir = '/tmp/smartagent-test-' + Date.now();
|
||||
const tools = filesystemTool({ rootDir: tmpDir });
|
||||
|
||||
await (tools.write_file as any).execute({
|
||||
path: tmpDir + '/hello.txt',
|
||||
content: 'Hello, world!',
|
||||
});
|
||||
|
||||
const content = await (tools.read_file as any).execute({
|
||||
path: tmpDir + '/hello.txt',
|
||||
});
|
||||
expect(content).toInclude('Hello, world!');
|
||||
|
||||
// Cleanup
|
||||
await (tools.delete_file as any).execute({
|
||||
path: tmpDir + '/hello.txt',
|
||||
});
|
||||
});
|
||||
|
||||
tap.test('filesystem tool should enforce rootDir restriction', async () => {
|
||||
const tools = filesystemTool({ rootDir: '/tmp/restricted' });
|
||||
let threw = false;
|
||||
try {
|
||||
await (tools.read_file as any).execute({ path: '/etc/passwd' });
|
||||
} catch (e) {
|
||||
threw = true;
|
||||
expect((e as Error).message).toInclude('Access denied');
|
||||
}
|
||||
expect(threw).toBeTrue();
|
||||
});
|
||||
|
||||
export default tap.start();
|
||||
|
||||
Reference in New Issue
Block a user