diff --git a/lib/ai/agent/tools/context.js b/lib/ai/agent/tools/context.js new file mode 100644 index 0000000..4a5f9f0 --- /dev/null +++ b/lib/ai/agent/tools/context.js @@ -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 }; + } +}; diff --git a/tests/ai/agent/tools/context.test.js b/tests/ai/agent/tools/context.test.js new file mode 100644 index 0000000..3d78827 --- /dev/null +++ b/tests/ai/agent/tools/context.test.js @@ -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); + }); +});