feat(core,cli): implement task tracker tools and feature flag (Phase 2)

This commit is contained in:
Anjali Sridhar
2026-02-18 12:03:58 -08:00
parent b5db5d300a
commit f7371269cd
9 changed files with 974 additions and 0 deletions
+1
View File
@@ -789,6 +789,7 @@ export async function loadCliConfig(
enableExtensionReloading: settings.experimental?.extensionReloading,
enableAgents: settings.experimental?.enableAgents,
plan: settings.experimental?.plan,
tracker: settings.experimental?.taskTracker,
enableEventDrivenScheduler: true,
skillsSupport: settings.skills?.enabled ?? true,
disabledSkills: settings.skills?.disabled,
@@ -1593,6 +1593,15 @@ const SETTINGS_SCHEMA = {
description: 'Enable planning features (Plan Mode and tools).',
showInDialog: true,
},
taskTracker: {
type: 'boolean',
label: 'Task Tracker',
category: 'Experimental',
requiresRestart: true,
default: false,
description: 'Enable task tracker tools.',
showInDialog: true,
},
},
},
+51
View File
@@ -68,6 +68,15 @@ import { ideContextStore } from '../ide/ideContext.js';
import { WriteTodosTool } from '../tools/write-todos.js';
import type { FileSystemService } from '../services/fileSystemService.js';
import { StandardFileSystemService } from '../services/fileSystemService.js';
import {
TrackerInitTool,
TrackerCreateTaskTool,
TrackerUpdateTaskTool,
TrackerGetTaskTool,
TrackerListTasksTool,
TrackerAddDependencyTool,
TrackerVisualizeTool,
} from '../tools/trackerTools.js';
import { logRipgrepFallback, logFlashFallback } from '../telemetry/loggers.js';
import {
RipgrepFallbackEvent,
@@ -89,6 +98,7 @@ import type {
import { ModelConfigService } from '../services/modelConfigService.js';
import { DEFAULT_MODEL_CONFIGS } from './defaultModelConfigs.js';
import { ContextManager } from '../services/contextManager.js';
import { TrackerService } from '../services/trackerService.js';
import type { GenerateContentParameters } from '@google/genai';
// Re-export OAuth config type
@@ -478,6 +488,7 @@ export interface ConfigParameters {
toolOutputMasking?: Partial<ToolOutputMaskingConfig>;
disableLLMCorrection?: boolean;
plan?: boolean;
tracker?: boolean;
onModelChange?: (model: string) => void;
mcpEnabled?: boolean;
extensionsEnabled?: boolean;
@@ -505,6 +516,7 @@ export class Config {
private sessionId: string;
private clientVersion: string;
private fileSystemService: FileSystemService;
private trackerService?: TrackerService;
private contentGeneratorConfig!: ContentGeneratorConfig;
private contentGenerator!: ContentGenerator;
readonly modelConfigService: ModelConfigService;
@@ -663,6 +675,7 @@ export class Config {
private readonly experimentalJitContext: boolean;
private readonly disableLLMCorrection: boolean;
private readonly planEnabled: boolean;
private readonly trackerEnabled: boolean;
private contextManager?: ContextManager;
private terminalBackground: string | undefined = undefined;
private remoteAdminSettings: AdminControlsSettings | undefined;
@@ -751,6 +764,7 @@ export class Config {
this.agents = params.agents ?? {};
this.disableLLMCorrection = params.disableLLMCorrection ?? true;
this.planEnabled = params.plan ?? false;
this.trackerEnabled = params.tracker ?? false;
this.enableEventDrivenScheduler = params.enableEventDrivenScheduler ?? true;
this.skillsSupport = params.skillsSupport ?? true;
this.disabledSkills = params.disabledSkills ?? [];
@@ -1896,6 +1910,13 @@ export class Config {
return this.bugCommand;
}
getTrackerService(): TrackerService {
if (!this.trackerService) {
this.trackerService = new TrackerService(this.targetDir);
}
return this.trackerService;
}
getFileService(): FileDiscoveryService {
if (!this.fileDiscoveryService) {
this.fileDiscoveryService = new FileDiscoveryService(this.targetDir, {
@@ -1957,6 +1978,10 @@ export class Config {
return this.planEnabled;
}
isTrackerEnabled(): boolean {
return this.trackerEnabled;
}
getApprovedPlanPath(): string | undefined {
return this.approvedPlanPath;
}
@@ -2440,6 +2465,32 @@ export class Config {
);
}
if (this.isTrackerEnabled()) {
maybeRegister(TrackerInitTool, () =>
registry.registerTool(new TrackerInitTool(this, this.messageBus)),
);
maybeRegister(TrackerCreateTaskTool, () =>
registry.registerTool(new TrackerCreateTaskTool(this, this.messageBus)),
);
maybeRegister(TrackerUpdateTaskTool, () =>
registry.registerTool(new TrackerUpdateTaskTool(this, this.messageBus)),
);
maybeRegister(TrackerGetTaskTool, () =>
registry.registerTool(new TrackerGetTaskTool(this, this.messageBus)),
);
maybeRegister(TrackerListTasksTool, () =>
registry.registerTool(new TrackerListTasksTool(this, this.messageBus)),
);
maybeRegister(TrackerAddDependencyTool, () =>
registry.registerTool(
new TrackerAddDependencyTool(this, this.messageBus),
),
);
maybeRegister(TrackerVisualizeTool, () =>
registry.registerTool(new TrackerVisualizeTool(this, this.messageBus)),
);
}
// Register Subagents as Tools
this.registerSubAgentTools(registry);
@@ -0,0 +1,47 @@
/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect } from 'vitest';
import { Config } from './config.js';
import { TRACKER_INIT_TOOL_NAME } from '../tools/tool-names.js';
import * as os from 'node:os';
describe('Config Tracker Feature Flag', () => {
const baseParams = {
sessionId: 'test-session',
targetDir: os.tmpdir(),
cwd: os.tmpdir(),
model: 'gemini-1.5-pro',
debugMode: false,
};
it('should not register tracker tools by default', async () => {
const config = new Config(baseParams);
await config.initialize();
const registry = config.getToolRegistry();
expect(registry.getTool(TRACKER_INIT_TOOL_NAME)).toBeUndefined();
});
it('should register tracker tools when tracker is enabled', async () => {
const config = new Config({
...baseParams,
tracker: true,
});
await config.initialize();
const registry = config.getToolRegistry();
expect(registry.getTool(TRACKER_INIT_TOOL_NAME)).toBeDefined();
});
it('should not register tracker tools when tracker is explicitly disabled', async () => {
const config = new Config({
...baseParams,
tracker: false,
});
await config.initialize();
const registry = config.getToolRegistry();
expect(registry.getTool(TRACKER_INIT_TOOL_NAME)).toBeUndefined();
});
});
+3
View File
@@ -112,6 +112,8 @@ export * from './services/chatRecordingService.js';
export * from './services/fileSystemService.js';
export * from './services/sessionSummaryUtils.js';
export * from './services/contextManager.js';
export * from './services/trackerService.js';
export * from './services/trackerTypes.js';
export * from './skills/skillManager.js';
export * from './skills/skillLoader.js';
@@ -157,6 +159,7 @@ export * from './tools/read-many-files.js';
export * from './tools/mcp-client.js';
export * from './tools/mcp-tool.js';
export * from './tools/write-todos.js';
export * from './tools/trackerTools.js';
// MCP OAuth
export { MCPOAuthProvider } from './mcp/oauth-provider.js';
@@ -0,0 +1,174 @@
/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import type { ToolDefinition } from './types.js';
import {
TRACKER_INIT_TOOL_NAME,
TRACKER_CREATE_TASK_TOOL_NAME,
TRACKER_UPDATE_TASK_TOOL_NAME,
TRACKER_GET_TASK_TOOL_NAME,
TRACKER_LIST_TASKS_TOOL_NAME,
TRACKER_ADD_DEPENDENCY_TOOL_NAME,
TRACKER_VISUALIZE_TOOL_NAME,
} from '../tool-names.js';
export const TRACKER_INIT_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_INIT_TOOL_NAME,
description:
'Initializes the task tracker in the current workspace by creating the .tracker directory.',
parametersJsonSchema: {
type: 'object',
properties: {},
},
},
};
export const TRACKER_CREATE_TASK_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_CREATE_TASK_TOOL_NAME,
description: 'Creates a new task in the tracker.',
parametersJsonSchema: {
type: 'object',
properties: {
title: {
type: 'string',
description: 'Short title of the task.',
},
description: {
type: 'string',
description: 'Detailed description of the task.',
},
type: {
type: 'string',
enum: ['epic', 'task', 'bug'],
description: 'Type of the task.',
},
parentId: {
type: 'string',
description: 'Optional ID of the parent task.',
},
dependencies: {
type: 'array',
items: { type: 'string' },
description: 'Optional list of task IDs that this task depends on.',
},
},
required: ['title', 'description', 'type'],
},
},
};
export const TRACKER_UPDATE_TASK_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_UPDATE_TASK_TOOL_NAME,
description: 'Updates an existing task in the tracker.',
parametersJsonSchema: {
type: 'object',
properties: {
id: {
type: 'string',
description: 'The 6-character hex ID of the task to update.',
},
title: {
type: 'string',
description: 'New title for the task.',
},
description: {
type: 'string',
description: 'New description for the task.',
},
status: {
type: 'string',
enum: ['open', 'in_progress', 'blocked', 'closed'],
description: 'New status for the task.',
},
dependencies: {
type: 'array',
items: { type: 'string' },
description: 'New list of dependency IDs.',
},
},
required: ['id'],
},
},
};
export const TRACKER_GET_TASK_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_GET_TASK_TOOL_NAME,
description: 'Retrieves details for a specific task.',
parametersJsonSchema: {
type: 'object',
properties: {
id: {
type: 'string',
description: 'The 6-character hex ID of the task.',
},
},
required: ['id'],
},
},
};
export const TRACKER_LIST_TASKS_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_LIST_TASKS_TOOL_NAME,
description:
'Lists tasks in the tracker, optionally filtered by status, type, or parent.',
parametersJsonSchema: {
type: 'object',
properties: {
status: {
type: 'string',
enum: ['open', 'in_progress', 'blocked', 'closed'],
description: 'Filter by status.',
},
type: {
type: 'string',
enum: ['epic', 'task', 'bug'],
description: 'Filter by type.',
},
parentId: {
type: 'string',
description: 'Filter by parent task ID.',
},
},
},
},
};
export const TRACKER_ADD_DEPENDENCY_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_ADD_DEPENDENCY_TOOL_NAME,
description: 'Adds a dependency between two tasks.',
parametersJsonSchema: {
type: 'object',
properties: {
taskId: {
type: 'string',
description: 'The ID of the task that has a dependency.',
},
dependencyId: {
type: 'string',
description: 'The ID of the task that is being depended upon.',
},
},
required: ['taskId', 'dependencyId'],
},
},
};
export const TRACKER_VISUALIZE_DEFINITION: ToolDefinition = {
base: {
name: TRACKER_VISUALIZE_TOOL_NAME,
description: 'Renders an ASCII tree visualization of the task graph.',
parametersJsonSchema: {
type: 'object',
properties: {},
},
},
};
+14
View File
@@ -41,6 +41,13 @@ export const ASK_USER_TOOL_NAME = 'ask_user';
export const ASK_USER_DISPLAY_NAME = 'Ask User';
export const EXIT_PLAN_MODE_TOOL_NAME = 'exit_plan_mode';
export const ENTER_PLAN_MODE_TOOL_NAME = 'enter_plan_mode';
export const TRACKER_INIT_TOOL_NAME = 'tracker_init';
export const TRACKER_CREATE_TASK_TOOL_NAME = 'tracker_create_task';
export const TRACKER_UPDATE_TASK_TOOL_NAME = 'tracker_update_task';
export const TRACKER_GET_TASK_TOOL_NAME = 'tracker_get_task';
export const TRACKER_LIST_TASKS_TOOL_NAME = 'tracker_list_tasks';
export const TRACKER_ADD_DEPENDENCY_TOOL_NAME = 'tracker_add_dependency';
export const TRACKER_VISUALIZE_TOOL_NAME = 'tracker_visualize';
/**
* Mapping of legacy tool names to their current names.
@@ -94,6 +101,13 @@ export const ALL_BUILTIN_TOOL_NAMES = [
MEMORY_TOOL_NAME,
ACTIVATE_SKILL_TOOL_NAME,
ASK_USER_TOOL_NAME,
TRACKER_INIT_TOOL_NAME,
TRACKER_CREATE_TASK_TOOL_NAME,
TRACKER_UPDATE_TASK_TOOL_NAME,
TRACKER_GET_TASK_TOOL_NAME,
TRACKER_LIST_TASKS_TOOL_NAME,
TRACKER_ADD_DEPENDENCY_TOOL_NAME,
TRACKER_VISUALIZE_TOOL_NAME,
] as const;
/**
@@ -0,0 +1,112 @@
/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
import { Config } from '../config/config.js';
import { MessageBus } from '../confirmation-bus/message-bus.js';
import type { PolicyEngine } from '../policy/policy-engine.js';
import {
TrackerInitTool,
TrackerCreateTaskTool,
TrackerListTasksTool,
TrackerUpdateTaskTool,
} from './trackerTools.js';
import * as fs from 'node:fs/promises';
import * as path from 'node:path';
import * as os from 'node:os';
describe('Tracker Tools Integration', () => {
let tempDir: string;
let config: Config;
let messageBus: MessageBus;
beforeEach(async () => {
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'tracker-tools-test-'));
config = new Config({
sessionId: 'test-session',
targetDir: tempDir,
cwd: tempDir,
model: 'gemini-2.0-flash',
debugMode: false,
});
messageBus = new MessageBus(null as unknown as PolicyEngine, false);
});
afterEach(async () => {
await fs.rm(tempDir, { recursive: true, force: true });
});
const getSignal = () => new AbortController().signal;
it('runs tracker_init and creates the directory', async () => {
const tool = new TrackerInitTool(config, messageBus);
const result = await tool.buildAndExecute({}, getSignal());
expect(result.llmContent).toContain('Task tracker initialized');
const tasksDir = path.join(tempDir, '.tracker', 'tasks');
const stats = await fs.stat(tasksDir);
expect(stats.isDirectory()).toBe(true);
});
it('creates and lists tasks', async () => {
// Init first
await new TrackerInitTool(config, messageBus).buildAndExecute(
{},
getSignal(),
);
const createTool = new TrackerCreateTaskTool(config, messageBus);
const createResult = await createTool.buildAndExecute(
{
title: 'Test Task',
description: 'Test Description',
type: 'task',
},
getSignal(),
);
expect(createResult.llmContent).toContain('Created task');
const listTool = new TrackerListTasksTool(config, messageBus);
const listResult = await listTool.buildAndExecute({}, getSignal());
expect(listResult.llmContent).toContain('Test Task');
expect(listResult.llmContent).toContain('(open)');
});
it('updates task status', async () => {
await new TrackerInitTool(config, messageBus).buildAndExecute(
{},
getSignal(),
);
const createTool = new TrackerCreateTaskTool(config, messageBus);
await createTool.buildAndExecute(
{
title: 'Update Me',
description: '...',
type: 'task',
},
getSignal(),
);
const tasks = await config.getTrackerService().listTasks();
const taskId = tasks[0].id;
const updateTool = new TrackerUpdateTaskTool(config, messageBus);
const updateResult = await updateTool.buildAndExecute(
{
id: taskId,
status: 'in_progress',
},
getSignal(),
);
expect(updateResult.llmContent).toContain('Status: in_progress');
const task = await config.getTrackerService().getTask(taskId);
expect(task?.status).toBe('in_progress');
});
});
+563
View File
@@ -0,0 +1,563 @@
/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import type { Config } from '../config/config.js';
import type { MessageBus } from '../confirmation-bus/message-bus.js';
import {
TRACKER_ADD_DEPENDENCY_DEFINITION,
TRACKER_CREATE_TASK_DEFINITION,
TRACKER_GET_TASK_DEFINITION,
TRACKER_INIT_DEFINITION,
TRACKER_LIST_TASKS_DEFINITION,
TRACKER_UPDATE_TASK_DEFINITION,
TRACKER_VISUALIZE_DEFINITION,
} from './definitions/trackerTools.js';
import { resolveToolDeclaration } from './definitions/resolver.js';
import {
TRACKER_ADD_DEPENDENCY_TOOL_NAME,
TRACKER_CREATE_TASK_TOOL_NAME,
TRACKER_GET_TASK_TOOL_NAME,
TRACKER_INIT_TOOL_NAME,
TRACKER_LIST_TASKS_TOOL_NAME,
TRACKER_UPDATE_TASK_TOOL_NAME,
TRACKER_VISUALIZE_TOOL_NAME,
} from './tool-names.js';
import type { ToolResult } from './tools.js';
import { BaseDeclarativeTool, BaseToolInvocation, Kind } from './tools.js';
import { ToolErrorType } from './tool-error.js';
import type {
TrackerTask,
TaskStatus,
TaskType,
} from '../services/trackerTypes.js';
// --- Shared Base ---
abstract class BaseTrackerInvocation<
P extends object,
R extends ToolResult,
> extends BaseToolInvocation<P, R> {
constructor(
protected readonly config: Config,
params: P,
messageBus: MessageBus,
toolName: string,
) {
super(params, messageBus, toolName);
}
protected get service() {
return this.config.getTrackerService();
}
abstract override getDescription(): string;
}
// --- tracker_init ---
class TrackerInitInvocation extends BaseTrackerInvocation<
Record<string, never>,
ToolResult
> {
getDescription(): string {
return 'Initializing the task tracker storage.';
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
await this.service.ensureInitialized();
return {
llmContent:
'Task tracker initialized successfully. Storage is ready at .tracker/tasks/',
returnDisplay: 'Tracker initialized.',
};
}
}
export class TrackerInitTool extends BaseDeclarativeTool<
Record<string, never>,
ToolResult
> {
static readonly Name = TRACKER_INIT_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerInitTool.Name,
'Initialize Tracker',
TRACKER_INIT_DEFINITION.base.description!,
Kind.Edit,
TRACKER_INIT_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(
params: Record<string, never>,
messageBus: MessageBus,
) {
return new TrackerInitInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_INIT_DEFINITION, modelId);
}
}
// --- tracker_create_task ---
interface CreateTaskParams {
title: string;
description: string;
type: TaskType;
parentId?: string;
dependencies?: string[];
}
class TrackerCreateTaskInvocation extends BaseTrackerInvocation<
CreateTaskParams,
ToolResult
> {
getDescription(): string {
return `Creating task: ${this.params.title}`;
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
const task = await this.service.createTask({
title: this.params.title,
description: this.params.description,
type: this.params.type,
status: 'open',
parentId: this.params.parentId,
dependencies: this.params.dependencies ?? [],
});
return {
llmContent: `Created task ${task.id}: ${task.title}`,
returnDisplay: `Created task ${task.id}.`,
};
}
}
export class TrackerCreateTaskTool extends BaseDeclarativeTool<
CreateTaskParams,
ToolResult
> {
static readonly Name = TRACKER_CREATE_TASK_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerCreateTaskTool.Name,
'Create Task',
TRACKER_CREATE_TASK_DEFINITION.base.description!,
Kind.Edit,
TRACKER_CREATE_TASK_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(params: CreateTaskParams, messageBus: MessageBus) {
return new TrackerCreateTaskInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_CREATE_TASK_DEFINITION, modelId);
}
}
// --- tracker_update_task ---
interface UpdateTaskParams {
id: string;
title?: string;
description?: string;
status?: TaskStatus;
dependencies?: string[];
}
class TrackerUpdateTaskInvocation extends BaseTrackerInvocation<
UpdateTaskParams,
ToolResult
> {
getDescription(): string {
return `Updating task ${this.params.id}`;
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
const { id, ...updates } = this.params;
try {
const task = await this.service.updateTask(id, updates);
return {
llmContent: `Updated task ${task.id}. Status: ${task.status}`,
returnDisplay: `Updated task ${task.id}.`,
};
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
return {
llmContent: `Error updating task: ${errorMessage}`,
returnDisplay: 'Failed to update task.',
error: {
message: errorMessage,
type: ToolErrorType.EXECUTION_FAILED,
},
};
}
}
}
export class TrackerUpdateTaskTool extends BaseDeclarativeTool<
UpdateTaskParams,
ToolResult
> {
static readonly Name = TRACKER_UPDATE_TASK_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerUpdateTaskTool.Name,
'Update Task',
TRACKER_UPDATE_TASK_DEFINITION.base.description!,
Kind.Edit,
TRACKER_UPDATE_TASK_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(params: UpdateTaskParams, messageBus: MessageBus) {
return new TrackerUpdateTaskInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_UPDATE_TASK_DEFINITION, modelId);
}
}
// --- tracker_get_task ---
interface GetTaskParams {
id: string;
}
class TrackerGetTaskInvocation extends BaseTrackerInvocation<
GetTaskParams,
ToolResult
> {
getDescription(): string {
return `Retrieving task ${this.params.id}`;
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
const task = await this.service.getTask(this.params.id);
if (!task) {
return {
llmContent: `Task ${this.params.id} not found.`,
returnDisplay: 'Task not found.',
};
}
return {
llmContent: JSON.stringify(task, null, 2),
returnDisplay: `Retrieved task ${task.id}.`,
};
}
}
export class TrackerGetTaskTool extends BaseDeclarativeTool<
GetTaskParams,
ToolResult
> {
static readonly Name = TRACKER_GET_TASK_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerGetTaskTool.Name,
'Get Task',
TRACKER_GET_TASK_DEFINITION.base.description!,
Kind.Read,
TRACKER_GET_TASK_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(params: GetTaskParams, messageBus: MessageBus) {
return new TrackerGetTaskInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_GET_TASK_DEFINITION, modelId);
}
}
// --- tracker_list_tasks ---
interface ListTasksParams {
status?: TaskStatus;
type?: TaskType;
parentId?: string;
}
class TrackerListTasksInvocation extends BaseTrackerInvocation<
ListTasksParams,
ToolResult
> {
getDescription(): string {
return 'Listing tasks.';
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
let tasks = await this.service.listTasks();
if (this.params.status) {
tasks = tasks.filter((t) => t.status === this.params.status);
}
if (this.params.type) {
tasks = tasks.filter((t) => t.type === this.params.type);
}
if (this.params.parentId) {
tasks = tasks.filter((t) => t.parentId === this.params.parentId);
}
if (tasks.length === 0) {
return {
llmContent: 'No tasks found matching the criteria.',
returnDisplay: 'No matching tasks.',
};
}
const content = tasks
.map((t) => `- [${t.id}] ${t.title} (${t.status})`)
.join('\n');
return {
llmContent: content,
returnDisplay: `Listed ${tasks.length} tasks.`,
};
}
}
export class TrackerListTasksTool extends BaseDeclarativeTool<
ListTasksParams,
ToolResult
> {
static readonly Name = TRACKER_LIST_TASKS_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerListTasksTool.Name,
'List Tasks',
TRACKER_LIST_TASKS_DEFINITION.base.description!,
Kind.Search,
TRACKER_LIST_TASKS_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(params: ListTasksParams, messageBus: MessageBus) {
return new TrackerListTasksInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_LIST_TASKS_DEFINITION, modelId);
}
}
// --- tracker_add_dependency ---
interface AddDependencyParams {
taskId: string;
dependencyId: string;
}
class TrackerAddDependencyInvocation extends BaseTrackerInvocation<
AddDependencyParams,
ToolResult
> {
getDescription(): string {
return `Adding dependency: ${this.params.taskId} depends on ${this.params.dependencyId}`;
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
const task = await this.service.getTask(this.params.taskId);
if (!task) {
return {
llmContent: `Task ${this.params.taskId} not found.`,
returnDisplay: 'Task not found.',
};
}
const dep = await this.service.getTask(this.params.dependencyId);
if (!dep) {
return {
llmContent: `Dependency task ${this.params.dependencyId} not found.`,
returnDisplay: 'Dependency not found.',
};
}
const newDeps = Array.from(
new Set([...task.dependencies, this.params.dependencyId]),
);
try {
await this.service.updateTask(task.id, { dependencies: newDeps });
return {
llmContent: `Linked ${task.id} -> ${dep.id}.`,
returnDisplay: 'Dependency added.',
};
} catch (error) {
return {
llmContent: `Error adding dependency: ${
error instanceof Error ? error.message : String(error)
}`,
returnDisplay: 'Failed to add dependency.',
};
}
}
}
export class TrackerAddDependencyTool extends BaseDeclarativeTool<
AddDependencyParams,
ToolResult
> {
static readonly Name = TRACKER_ADD_DEPENDENCY_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerAddDependencyTool.Name,
'Add Dependency',
TRACKER_ADD_DEPENDENCY_DEFINITION.base.description!,
Kind.Edit,
TRACKER_ADD_DEPENDENCY_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(
params: AddDependencyParams,
messageBus: MessageBus,
) {
return new TrackerAddDependencyInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_ADD_DEPENDENCY_DEFINITION, modelId);
}
}
// --- tracker_visualize ---
class TrackerVisualizeInvocation extends BaseTrackerInvocation<
Record<string, never>,
ToolResult
> {
getDescription(): string {
return 'Visualizing the task graph.';
}
override async execute(_signal: AbortSignal): Promise<ToolResult> {
const tasks = await this.service.listTasks();
if (tasks.length === 0) {
return {
llmContent: 'No tasks to visualize.',
returnDisplay: 'Empty tracker.',
};
}
const statusEmojis: Record<TaskStatus, string> = {
open: '⭕',
in_progress: '🚧',
blocked: '🚫',
closed: '✅',
};
const typeLabels: Record<TaskType, string> = {
epic: '[EPIC]',
task: '[TASK]',
bug: '[BUG]',
};
// Simple list-based visualization for now (can enhance to tree later if needed)
// We'll organize by epic/parent
const roots = tasks.filter((t) => !t.parentId);
let output = 'Task Tracker Graph:\n';
const renderTask = (task: TrackerTask, depth: number) => {
const indent = ' '.repeat(depth);
output += `${indent}${statusEmojis[task.status]} ${task.id} ${typeLabels[task.type]} ${task.title}\n`;
if (task.dependencies.length > 0) {
output += `${indent} └─ Depends on: ${task.dependencies.join(', ')}\n`;
}
const children = tasks.filter((t) => t.parentId === task.id);
for (const child of children) {
renderTask(child, depth + 1);
}
};
for (const root of roots) {
renderTask(root, 0);
}
return {
llmContent: output,
returnDisplay: 'Graph rendered.',
};
}
}
export class TrackerVisualizeTool extends BaseDeclarativeTool<
Record<string, never>,
ToolResult
> {
static readonly Name = TRACKER_VISUALIZE_TOOL_NAME;
constructor(
private config: Config,
messageBus: MessageBus,
) {
super(
TrackerVisualizeTool.Name,
'Visualize Tracker',
TRACKER_VISUALIZE_DEFINITION.base.description!,
Kind.Read,
TRACKER_VISUALIZE_DEFINITION.base.parametersJsonSchema,
messageBus,
);
}
protected createInvocation(
params: Record<string, never>,
messageBus: MessageBus,
) {
return new TrackerVisualizeInvocation(
this.config,
params,
messageBus,
this.name,
);
}
override getSchema(modelId?: string) {
return resolveToolDeclaration(TRACKER_VISUALIZE_DEFINITION, modelId);
}
}