feat(ai): context tool — resolve the active view entity
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
20
lib/ai/agent/tools/context.js
Normal file
20
lib/ai/agent/tools/context.js
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
import { pool } from '../../../db/pool.js';
|
||||||
|
|
||||||
|
const TABLE = { page: 'pages', ref: 'refs', task: 'tasks', project: 'projects', space: 'spaces' };
|
||||||
|
|
||||||
|
export const contextTool = {
|
||||||
|
name: 'context',
|
||||||
|
description: "Resolve what the owner is currently looking at (the active view). Call this first to ground your answer in the right entity.",
|
||||||
|
input_schema: { type: 'object', properties: {} },
|
||||||
|
async handler(_args, ctx) {
|
||||||
|
const view = ctx.view;
|
||||||
|
if (!view?.entityType || !view?.entityId) {
|
||||||
|
return { note: 'No specific entity is active; only the Space context is available.', space_id: ctx.space_id };
|
||||||
|
}
|
||||||
|
const table = TABLE[view.entityType];
|
||||||
|
if (!table) return { entityType: view.entityType, entityId: view.entityId, note: 'unrecognised entity type' };
|
||||||
|
const { rows: [row] } = await pool.query(`SELECT * FROM ${table} WHERE id=$1`, [view.entityId]);
|
||||||
|
if (!row) return { entityType: view.entityType, entityId: view.entityId, error: 'not found' };
|
||||||
|
return { entityType: view.entityType, ...row };
|
||||||
|
}
|
||||||
|
};
|
||||||
26
tests/ai/agent/tools/context.test.js
Normal file
26
tests/ai/agent/tools/context.test.js
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
import { describe, it, expect, beforeAll } from 'vitest';
|
||||||
|
import { pool } from '../../../../lib/db/pool.js';
|
||||||
|
import { resetDb } from '../../../helpers/db.js';
|
||||||
|
import { migrateUp } from '../../../../lib/db/migrate.js';
|
||||||
|
import { contextTool } from '../../../../lib/ai/agent/tools/context.js';
|
||||||
|
|
||||||
|
let spaceId, taskId;
|
||||||
|
beforeAll(async () => {
|
||||||
|
await resetDb(); await migrateUp();
|
||||||
|
({ rows: [{ id: spaceId }] } = await pool.query(
|
||||||
|
`INSERT INTO spaces(slug,name) VALUES('s','S') RETURNING id`));
|
||||||
|
({ rows: [{ id: taskId }] } = await pool.query(
|
||||||
|
`INSERT INTO tasks(space_id,title) VALUES($1,'Wire telemetry') RETURNING id`, [spaceId]));
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('context tool', () => {
|
||||||
|
it('summarises the current view entity', async () => {
|
||||||
|
const out = await contextTool.handler({}, { space_id: spaceId, view: { entityType: 'task', entityId: taskId } });
|
||||||
|
expect(out.entityType).toBe('task');
|
||||||
|
expect(out.title).toBe('Wire telemetry');
|
||||||
|
});
|
||||||
|
it('handles no active view', async () => {
|
||||||
|
const out = await contextTool.handler({}, { space_id: spaceId, view: null });
|
||||||
|
expect(out.note).toMatch(/no specific entity/i);
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user