gencode-ai 0.1.1 → 0.1.3

package/src/tools/builtin/glob.ts

@@ -5,12 +5,13 @@
 import fastGlob from 'fast-glob';
 import type { Tool, ToolResult } from '../types.js';
 import { GlobInputSchema, type GlobInput, resolvePath, getErrorMessage } from '../types.js';
+import { loadToolDescription } from '../../prompts/index.js';
 
 const MAX_RESULTS = 100;
 
 export const globTool: Tool<GlobInput> = {
   name: 'Glob',
-  description: 'Find files matching a glob pattern. Returns a list of matching file paths.',
+  description: loadToolDescription('glob'),
   parameters: GlobInputSchema,
 
   async execute(input, context): Promise<ToolResult> {

package/src/tools/builtin/grep.ts

@@ -7,12 +7,13 @@ import * as path from 'path';
 import fastGlob from 'fast-glob';
 import type { Tool, ToolResult } from '../types.js';
 import { GrepInputSchema, type GrepInput, resolvePath, getErrorMessage } from '../types.js';
+import { loadToolDescription } from '../../prompts/index.js';
 
 const MAX_MATCHES = 50;
 
 export const grepTool: Tool<GrepInput> = {
   name: 'Grep',
-  description: 'Search for a regex pattern in files. Returns matching lines with file paths and line numbers.',
+  description: loadToolDescription('grep'),
   parameters: GrepInputSchema,
 
   async execute(input, context): Promise<ToolResult> {

package/src/tools/builtin/read.ts

@@ -5,10 +5,11 @@
 import * as fs from 'fs/promises';
 import type { Tool, ToolResult } from '../types.js';
 import { ReadInputSchema, type ReadInput, resolvePath, getErrorMessage } from '../types.js';
+import { loadToolDescription } from '../../prompts/index.js';
 
 export const readTool: Tool<ReadInput> = {
   name: 'Read',
-  description: 'Read the contents of a file. Returns the file content with line numbers.',
+  description: loadToolDescription('read'),
   parameters: ReadInputSchema,
 
   async execute(input, context): Promise<ToolResult> {

package/src/tools/builtin/todowrite.ts

@@ -0,0 +1,102 @@
+/**
+ * TodoWrite Tool - Manage task list for tracking progress
+ */
+
+import type { Tool, ToolResult, TodoItem } from '../types.js';
+import { TodoWriteInputSchema, type TodoWriteInput } from '../types.js';
+import { loadToolDescription } from '../../prompts/index.js';
+
+// Global todo state - shared across tool invocations
+let currentTodos: TodoItem[] = [];
+
+/**
+ * Get the current todo list
+ */
+export function getTodos(): TodoItem[] {
+  return [...currentTodos];
+}
+
+/**
+ * Clear all todos
+ */
+export function clearTodos(): void {
+  currentTodos = [];
+}
+
+/**
+ * Format todos for display
+ */
+function formatTodos(todos: TodoItem[]): string {
+  if (todos.length === 0) {
+    return 'Todo list is empty.';
+  }
+
+  const statusIcons: Record<string, string> = {
+    pending: '[ ]',
+    in_progress: '[>]',
+    completed: '[x]',
+  };
+
+  const lines = todos.map((todo, index) => {
+    const icon = statusIcons[todo.status] || '[ ]';
+    return `${index + 1}. ${icon} ${todo.content}`;
+  });
+
+  return lines.join('\n');
+}
+
+/**
+ * Validate todo list rules
+ */
+function validateTodos(todos: TodoItem[]): string | null {
+  const inProgress = todos.filter((t) => t.status === 'in_progress');
+  if (inProgress.length > 1) {
+    return `Only one task should be in_progress at a time. Found ${inProgress.length} tasks in progress.`;
+  }
+  return null;
+}
+
+export const todowriteTool: Tool<TodoWriteInput> = {
+  name: 'TodoWrite',
+  description: loadToolDescription('todowrite'),
+  parameters: TodoWriteInputSchema,
+
+  async execute(input): Promise<ToolResult> {
+    try {
+      // Validate the todo list
+      const validationError = validateTodos(input.todos);
+      if (validationError) {
+        return {
+          success: false,
+          output: '',
+          error: validationError,
+        };
+      }
+
+      // Update the global todo state (auto-generate id if missing)
+      currentTodos = input.todos.map((todo, index) => ({
+        ...todo,
+        id: todo.id || `todo-${index + 1}`,
+      }));
+
+      // Count statistics
+      const pending = currentTodos.filter((t) => t.status === 'pending').length;
+      const inProgress = currentTodos.filter((t) => t.status === 'in_progress').length;
+      const completed = currentTodos.filter((t) => t.status === 'completed').length;
+
+      const summary = `Todos updated: ${completed} completed, ${inProgress} in progress, ${pending} pending`;
+      const formatted = formatTodos(currentTodos);
+
+      return {
+        success: true,
+        output: `${summary}\n\n${formatted}`,
+      };
+    } catch (error) {
+      return {
+        success: false,
+        output: '',
+        error: `Failed to update todos: ${error instanceof Error ? error.message : String(error)}`,
+      };
+    }
+  },
+};

package/src/tools/builtin/webfetch.ts

@@ -7,6 +7,7 @@ import { z } from 'zod';
 import type { Tool, ToolContext, ToolResult } from '../types.js';
 import { getErrorMessage } from '../types.js';
 import { validateUrl } from '../utils/ssrf.js';
+import { loadToolDescription } from '../../prompts/index.js';
 
 // Constants
 const MAX_RESPONSE_SIZE = 5 * 1024 * 1024; // 5MB
@@ -148,11 +149,7 @@ function truncateOutput(output: string): string {
  */
 export const webfetchTool: Tool<WebFetchInput> = {
   name: 'WebFetch',
-  description: `Fetch content from a URL and return it in the specified format.
-- Converts HTML to Markdown by default for easier reading
-- Supports text, markdown, and html output formats
-- Maximum response size: 5MB
-- Timeout: 30 seconds (configurable up to 120 seconds)`,
+  description: loadToolDescription('webfetch'),
   parameters: WebFetchInputSchema,
 
   async execute(input: WebFetchInput, context: ToolContext): Promise<ToolResult> {

package/src/tools/builtin/websearch.ts

@@ -10,6 +10,7 @@ import {
   getCurrentSearchProviderName,
   type SearchResult,
 } from '../../providers/search/index.js';
+import { loadToolDescription } from '../../prompts/index.js';
 
 // Constants
 const DEFAULT_NUM_RESULTS = 10;
@@ -62,22 +63,7 @@ function formatResults(results: SearchResult[], query: string): string {
  */
 export const websearchTool: Tool<WebSearchInput> = {
   name: 'WebSearch',
-  description: `Search the web for current information.
-
-Use this tool when you need:
-- Up-to-date information beyond your knowledge cutoff
-- Current documentation or release notes
-- Recent solutions to technical problems
-- Current best practices
-
-IMPORTANT: After answering, include a "Sources:" section with all relevant URLs as markdown hyperlinks.
-
-Example:
-  [Your answer]
-
-  Sources:
-  - [Title 1](https://url1)
-  - [Title 2](https://url2)`,
+  description: loadToolDescription('websearch'),
   parameters: WebSearchInputSchema,
 
   async execute(input: WebSearchInput, context: ToolContext): Promise<ToolResult> {

package/src/tools/builtin/write.ts

@@ -6,10 +6,11 @@ import * as fs from 'fs/promises';
 import * as path from 'path';
 import type { Tool, ToolResult } from '../types.js';
 import { WriteInputSchema, type WriteInput, resolvePath, getErrorMessage } from '../types.js';
+import { loadToolDescription } from '../../prompts/index.js';
 
 export const writeTool: Tool<WriteInput> = {
   name: 'Write',
-  description: 'Write content to a file. Creates the file if it does not exist, overwrites if it does.',
+  description: loadToolDescription('write'),
   parameters: WriteInputSchema,
 
   async execute(input, context): Promise<ToolResult> {

package/src/tools/index.ts

@@ -14,6 +14,19 @@ export { globTool } from './builtin/glob.js';
 export { grepTool } from './builtin/grep.js';
 export { webfetchTool } from './builtin/webfetch.js';
 export { websearchTool } from './builtin/websearch.js';
+export { todowriteTool, getTodos, clearTodos } from './builtin/todowrite.js';
+export {
+  askUserQuestionTool,
+  formatAnswersForAgent,
+  formatAnswersForDisplay,
+} from './builtin/ask-user.js';
+export type {
+  Question as AskUserQuestion,
+  QuestionOption as AskUserQuestionOption,
+  QuestionAnswer as AskUserQuestionAnswer,
+  AskUserQuestionInput,
+  AskUserQuestionResult,
+} from './builtin/ask-user.js';
 
 import { ToolRegistry } from './registry.js';
 import { readTool } from './builtin/read.js';
@@ -24,6 +37,8 @@ import { globTool } from './builtin/glob.js';
 import { grepTool } from './builtin/grep.js';
 import { webfetchTool } from './builtin/webfetch.js';
 import { websearchTool } from './builtin/websearch.js';
+import { todowriteTool } from './builtin/todowrite.js';
+import { askUserQuestionTool } from './builtin/ask-user.js';
 
 /**
  * Create a registry with all built-in tools
@@ -39,6 +54,8 @@ export function createDefaultRegistry(): ToolRegistry {
     grepTool,
     webfetchTool,
     websearchTool,
+    todowriteTool,
+    askUserQuestionTool,
   ]);
   return registry;
 }
@@ -55,4 +72,6 @@ export const builtinTools = [
   grepTool,
   webfetchTool,
   websearchTool,
+  todowriteTool,
+  askUserQuestionTool,
 ];

package/src/tools/types.ts

@@ -9,9 +9,27 @@ import { z } from 'zod';
 // Tool Definition Types
 // ============================================================================
 
+// Forward declaration for Question type (used by AskUserQuestion tool)
+export interface Question {
+  question: string;
+  header: string;
+  options: Array<{ label: string; description: string }>;
+  multiSelect: boolean;
+}
+
+// Forward declaration for QuestionAnswer type (used by AskUserQuestion tool)
+export interface QuestionAnswer {
+  question: string;
+  header: string;
+  selectedOptions: string[];
+  customInput?: string;
+}
+
 export interface ToolContext {
   cwd: string;
   abortSignal?: AbortSignal;
+  /** Callback for AskUserQuestion tool to interact with user */
+  askUser?: (questions: Question[]) => Promise<QuestionAnswer[]>;
 }
 
 export interface ToolResultMetadata {
@@ -108,6 +126,18 @@ export const WebFetchInputSchema = z.object({
 });
 export type WebFetchInput = z.infer<typeof WebFetchInputSchema>;
 
+export const TodoItemSchema = z.object({
+  content: z.string().min(1).describe('The task description'),
+  status: z.enum(['pending', 'in_progress', 'completed']).describe('Current status of the task'),
+  id: z.string().optional().describe('Unique task identifier'),
+});
+export type TodoItem = z.infer<typeof TodoItemSchema>;
+
+export const TodoWriteInputSchema = z.object({
+  todos: z.array(TodoItemSchema).describe('The complete todo list to write'),
+});
+export type TodoWriteInput = z.infer<typeof TodoWriteInputSchema>;
+
 // ============================================================================
 // JSON Schema Conversion
 // ============================================================================

package/tsconfig.json

@@ -17,5 +17,5 @@
     "jsx": "react-jsx"
   },
   "include": ["src/**/*"],
-  "exclude": ["node_modules", "dist", "examples"]
+  "exclude": ["node_modules", "dist", "examples", "**/*.test.ts"]
 }