task-o-matic 0.0.14 → 0.0.16

package/dist/lib/ai-service/prd-operations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"prd-operations.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/prd-operations.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,QAAQ,EAER,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EAIZ,MAAM,aAAa,CAAC;AAYrB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,qBAAa,aAAc,SAAQ,cAAc;IACzC,QAAQ,CACZ,UAAU,EAAE,MAAM,EAClB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,gBAAgB,CAAC;IAyKtB,SAAS,CACb,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,CAAC;IAuGZ,oBAAoB,CACxB,UAAU,EAAE,MAAM,EAClB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,EAAE,CAAC;IAyGd,kBAAkB,CACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EAAE,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,WAAW,CAAC,EAAE;QACZ,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,EACD,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IA6E5B,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IAqBZ,WAAW,CACf,IAAI,EAAE,MAAM,EAAE,EACd,mBAAmB,EAAE,MAAM,EAC3B,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;CA0BnB"}
+{"version":3,"file":"prd-operations.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/prd-operations.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,QAAQ,EAER,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EAIZ,MAAM,aAAa,CAAC;AAYrB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAMnD,qBAAa,aAAc,SAAQ,cAAc;IACzC,QAAQ,CACZ,UAAU,EAAE,MAAM,EAClB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,gBAAgB,CAAC;IA2LtB,SAAS,CACb,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,CAAC;IA+GZ,oBAAoB,CACxB,UAAU,EAAE,MAAM,EAClB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,EAAE,CAAC;IA2Hd,kBAAkB,CACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EAAE,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,WAAW,CAAC,EAAE;QACZ,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,EACD,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IA6F5B,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IAqBZ,WAAW,CACf,IAAI,EAAE,MAAM,EAAE,EACd,mBAAmB,EAAE,MAAM,EAC3B,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;CA0BnB"}

package/dist/lib/ai-service/prd-operations.js

@@ -39,6 +39,7 @@ const prompt_builder_1 = require("../prompt-builder");
 const prompts_1 = require("../../prompts");
 const filesystem_tools_1 = require("./filesystem-tools");
 const base_operations_1 = require("./base-operations");
+const task_o_matic_error_1 = require("../../utils/task-o-matic-error");
 class PRDOperations extends base_operations_1.BaseOperations {
     async parsePRD(prdContent, config, promptOverride, userMessage, streamingOptions, retryConfig, workingDirectory, enableFilesystemTools) {
         return this.retryHandler.executeWithRetry(async () => {
@@ -69,7 +70,13 @@ class PRDOperations extends base_operations_1.BaseOperations {
                     variables,
                 });
                 if (!promptResult.success) {
-                    throw new Error(`Failed to build PRD parsing prompt: ${promptResult.error}`);
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_PARSING_ERROR, `Failed to build PRD parsing prompt: ${promptResult.error}`, {
+                        context: "Prompt building failed during PRD parsing",
+                        suggestions: [
+                            "Verify prompt template exists",
+                            "Check variable substitution",
+                        ],
+                    });
                 }
                 enhancedPrompt = promptResult.prompt;
             }
@@ -125,7 +132,13 @@ Use these tools to understand the project structure, existing code patterns, and
             }
             const parseResult = this.jsonParser.parseJSONFromResponse(response);
             if (!parseResult.success) {
-                throw new Error(parseResult.error || "Failed to parse PRD response");
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_PARSING_ERROR, parseResult.error || "Failed to parse PRD response", {
+                    context: "AI response parsing failed during PRD parsing",
+                    suggestions: [
+                        "Check AI response format",
+                        "Verify JSON structure",
+                    ],
+                });
             }
             const parsed = parseResult.data;
             const tasks = (parsed?.tasks || []).map((task, index) => {
@@ -195,7 +208,13 @@ Use these tools to understand the project structure, existing code patterns, and
                     variables,
                 });
                 if (!promptResult.success) {
-                    throw new Error(`Failed to build PRD rework prompt: ${promptResult.error}`);
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_GENERATION_ERROR, `Failed to build PRD rework prompt: ${promptResult.error}`, {
+                        context: "Prompt building failed during PRD rework",
+                        suggestions: [
+                            "Verify prompt template exists",
+                            "Check variable substitution",
+                        ],
+                    });
                 }
                 prompt = promptResult.prompt;
             }
@@ -277,7 +296,13 @@ Use these tools to understand the current project structure, existing code patte
                     variables,
                 });
                 if (!promptResult.success) {
-                    throw new Error(`Failed to build PRD question prompt: ${promptResult.error}`);
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_GENERATION_ERROR, `Failed to build PRD question prompt: ${promptResult.error}`, {
+                        context: "Prompt building failed during PRD question generation",
+                        suggestions: [
+                            "Verify prompt template exists",
+                            "Check variable substitution",
+                        ],
+                    });
                 }
                 prompt = promptResult.prompt;
             }
@@ -324,7 +349,13 @@ Use these tools to understand the current project structure, existing code patte
             }
             const parseResult = this.jsonParser.parseJSONFromResponse(response);
             if (!parseResult.success) {
-                throw new Error(parseResult.error || "Failed to parse PRD questions");
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_GENERATION_ERROR, parseResult.error || "Failed to parse PRD questions", {
+                    context: "AI response parsing failed during PRD question generation",
+                    suggestions: [
+                        "Check AI response format",
+                        "Verify JSON structure",
+                    ],
+                });
             }
             return parseResult.data?.questions || [];
         }, retryConfig, "PRD questioning");
@@ -351,7 +382,13 @@ Use these tools to understand the current project structure, existing code patte
                 },
             });
             if (!promptResult.success) {
-                throw new Error(`Failed to build PRD question answer prompt: ${promptResult.error}`);
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_GENERATION_ERROR, `Failed to build PRD question answer prompt: ${promptResult.error}`, {
+                    context: "Prompt building failed during PRD answer generation",
+                    suggestions: [
+                        "Verify prompt template exists",
+                        "Check variable substitution",
+                    ],
+                });
             }
             const systemPromptResult = prompt_builder_1.PromptBuilder.buildPrompt({
                 name: "prd-question-answer",
@@ -361,7 +398,13 @@ Use these tools to understand the current project structure, existing code patte
             const response = await this.streamText("", config, systemPromptResult.prompt, promptResult.prompt, streamingOptions, { maxAttempts: 1 });
             const parseResult = this.jsonParser.parseJSONFromResponse(response);
             if (!parseResult.success) {
-                throw new Error(parseResult.error || "Failed to parse PRD answers response");
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.PRD_GENERATION_ERROR, parseResult.error || "Failed to parse PRD answers response", {
+                    context: "AI response parsing failed during PRD answer generation",
+                    suggestions: [
+                        "Check AI response format",
+                        "Verify JSON structure",
+                    ],
+                });
             }
             const answers = {};
             const numberedAnswers = parseResult.data?.answers || {};

package/dist/lib/ai-service/task-operations.d.ts

@@ -1,6 +1,7 @@
 import { AIConfig, Task, StreamingOptions, RetryConfig } from "../../types";
 import { BaseOperations } from "./base-operations";
 export declare class TaskOperations extends BaseOperations {
+    private aiOperationUtility;
     breakdownTask(task: Task, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>, fullContent?: string, stackInfo?: string, existingSubtasks?: Task[], enableFilesystemTools?: boolean): Promise<Array<{
         title: string;
         content: string;

package/dist/lib/ai-service/task-operations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"task-operations.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/task-operations.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,QAAQ,EACR,IAAI,EACJ,gBAAgB,EAChB,WAAW,EAEZ,MAAM,aAAa,CAAC;AAUrB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,qBAAa,cAAe,SAAQ,cAAc;IAC1C,aAAa,CACjB,IAAI,EAAE,IAAI,EACV,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,WAAW,CAAC,EAAE,MAAM,EACpB,SAAS,CAAC,EAAE,MAAM,EAClB,gBAAgB,CAAC,EAAE,IAAI,EAAE,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CACR,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CACpE;IAgIK,WAAW,CACf,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,MAAM,CAAC,EAAE,MAAM,EACf,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IAyEZ,QAAQ,CACZ,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;CA6EnB"}
+{"version":3,"file":"task-operations.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/task-operations.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,QAAQ,EACR,IAAI,EACJ,gBAAgB,EAChB,WAAW,EAEZ,MAAM,aAAa,CAAC;AAUrB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAOnD,qBAAa,cAAe,SAAQ,cAAc;IAChD,OAAO,CAAC,kBAAkB,CAA4B;IAChD,aAAa,CACjB,IAAI,EAAE,IAAI,EACV,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,WAAW,CAAC,EAAE,MAAM,EACpB,SAAS,CAAC,EAAE,MAAM,EAClB,gBAAgB,CAAC,EAAE,IAAI,EAAE,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CACR,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CACpE;IAmHK,WAAW,CACf,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,MAAM,CAAC,EAAE,MAAM,EACf,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IA4FZ,QAAQ,CACZ,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;CAoEnB"}

package/dist/lib/ai-service/task-operations.js

@@ -1,98 +1,78 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TaskOperations = void 0;
-const ai_1 = require("ai");
 const prompt_builder_1 = require("../prompt-builder");
 const stack_formatter_1 = require("../../utils/stack-formatter");
 const prompts_1 = require("../../prompts");
 const ai_service_factory_1 = require("../../utils/ai-service-factory");
 const filesystem_tools_1 = require("./filesystem-tools");
 const base_operations_1 = require("./base-operations");
+const ai_operation_utility_1 = require("../../utils/ai-operation-utility");
+const task_o_matic_error_1 = require("../../utils/task-o-matic-error");
 class TaskOperations extends base_operations_1.BaseOperations {
+    aiOperationUtility = new ai_operation_utility_1.AIOperationUtility();
     async breakdownTask(task, config, promptOverride, userMessage, streamingOptions, retryConfig, fullContent, stackInfo, existingSubtasks, enableFilesystemTools) {
-        return this.retryHandler.executeWithRetry(async () => {
-            let prompt;
-            if (promptOverride) {
-                prompt = promptOverride;
+        // Build prompt
+        let prompt;
+        if (promptOverride) {
+            prompt = promptOverride;
+        }
+        else {
+            const variables = {
+                TASK_TITLE: task.title,
+                TASK_DESCRIPTION: task.description || "No description",
+            };
+            if (fullContent) {
+                variables.TASK_CONTENT = fullContent;
             }
-            else {
-                const variables = {
-                    TASK_TITLE: task.title,
-                    TASK_DESCRIPTION: task.description || "No description",
-                };
-                if (fullContent) {
-                    variables.TASK_CONTENT = fullContent;
-                }
-                if (existingSubtasks && existingSubtasks.length > 0) {
-                    const existingSubtasksText = existingSubtasks
-                        .map((subtask, index) => `${index + 1}. ${subtask.title}: ${subtask.description || "No description"}`)
-                        .join("\n");
-                    variables.EXISTING_SUBTASKS = existingSubtasksText;
-                }
-                if (stackInfo) {
-                    variables.STACK_INFO = stackInfo;
-                }
-                const promptResult = prompt_builder_1.PromptBuilder.buildPrompt({
-                    name: "task-breakdown",
-                    type: "user",
-                    variables,
-                });
-                if (!promptResult.success) {
-                    throw new Error(`Failed to build task breakdown prompt: ${promptResult.error}`);
-                }
-                prompt = promptResult.prompt;
+            if (existingSubtasks && existingSubtasks.length > 0) {
+                const existingSubtasksText = existingSubtasks
+                    .map((subtask, index) => `${index + 1}. ${subtask.title}: ${subtask.description || "No description"}`)
+                    .join("\n");
+                variables.EXISTING_SUBTASKS = existingSubtasksText;
             }
-            let response;
-            if (enableFilesystemTools) {
-                const model = this.modelProvider.getModel({
-                    ...this.modelProvider.getAIConfig(),
-                    ...config,
+            if (stackInfo) {
+                variables.STACK_INFO = stackInfo;
+            }
+            const promptResult = prompt_builder_1.PromptBuilder.buildPrompt({
+                name: "task-breakdown",
+                type: "user",
+                variables,
+            });
+            if (!promptResult.success) {
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_OPERATION_FAILED, `Failed to build task breakdown prompt: ${promptResult.error}`, {
+                    context: "Prompt building failed during task breakdown operation",
+                    suggestions: [
+                        "Verify prompt template exists",
+                        "Check variable substitution",
+                    ],
                 });
-                const allTools = {
-                    ...filesystem_tools_1.filesystemTools,
-                };
-                const result = await (0, ai_1.streamText)({
-                    model,
-                    tools: allTools,
-                    system: prompts_1.TASK_BREAKDOWN_SYSTEM_PROMPT +
-                        `
+            }
+            prompt = promptResult.prompt;
+        }
+        // Execute AI operation with proper error handling
+        const result = await this.aiOperationUtility.executeAIOperation("Task breakdown", async () => {
+            // Prepare tools if filesystem tools are enabled
+            const tools = enableFilesystemTools ? filesystem_tools_1.filesystemTools : undefined;
+            const response = await this.aiOperationUtility.streamTextWithTools(prompts_1.TASK_BREAKDOWN_SYSTEM_PROMPT +
+                (enableFilesystemTools
+                    ? `
 
 You have access to filesystem tools that allow you to:
 - readFile: Read the contents of any file in the project
 - listDirectory: List contents of directories
 
-Use these tools to understand the project structure, existing code, and dependencies when breaking down tasks into subtasks.`,
-                    messages: [{ role: "user", content: userMessage || prompt }],
-                    maxRetries: 0,
-                    onChunk: streamingOptions?.onChunk
-                        ? ({ chunk }) => {
-                            if (chunk.type === "text-delta") {
-                                streamingOptions.onChunk(chunk.text);
-                            }
-                            else if (chunk.type === "reasoning-delta") {
-                                streamingOptions.onReasoning?.(chunk.text);
-                            }
-                        }
-                        : undefined,
-                    onFinish: streamingOptions?.onFinish
-                        ? ({ text, finishReason, usage }) => {
-                            streamingOptions.onFinish({
-                                text,
-                                finishReason,
-                                usage,
-                                isAborted: false,
-                            });
-                        }
-                        : undefined,
-                });
-                response = await result.text;
-            }
-            else {
-                response = await this.streamText("", config, prompts_1.TASK_BREAKDOWN_SYSTEM_PROMPT, userMessage || prompt, streamingOptions, { maxAttempts: 1 });
-            }
+Use these tools to understand the project structure, existing code, and dependencies when breaking down tasks into subtasks.`
+                    : ""), userMessage || prompt, config, streamingOptions, tools);
             const parseResult = this.jsonParser.parseJSONFromResponse(response);
             if (!parseResult.success) {
-                throw new Error(parseResult.error || "Failed to parse task breakdown response");
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_OPERATION_FAILED, parseResult.error || "Failed to parse task breakdown response", {
+                    context: "AI response parsing failed during task breakdown",
+                    suggestions: [
+                        "Check AI response format",
+                        "Verify JSON structure",
+                    ],
+                });
             }
             const parsed = parseResult.data;
             return (parsed?.subtasks || []).map((subtask) => ({
@@ -100,126 +80,133 @@ Use these tools to understand the project structure, existing code, and dependen
                 content: subtask.description || "",
                 estimatedEffort: subtask.effort,
             }));
-        }, retryConfig, "Task breakdown");
+        }, {
+            streamingOptions,
+            retryConfig,
+            aiConfig: config,
+            maxRetries: retryConfig?.maxAttempts || 2,
+        });
+        // Return the result directly (errors are thrown, not returned)
+        return result.result;
     }
     async enhanceTask(title, description, config, promptOverride, userMessage, taskId, streamingOptions, retryConfig) {
-        return this.retryHandler.executeWithRetry(async () => {
-            let contextInfo = "";
-            let prdContent = "";
-            if (taskId) {
-                const contextBuilder = (0, ai_service_factory_1.getContextBuilder)();
-                try {
-                    const context = await contextBuilder.buildContext(taskId);
-                    if (context.documentation || context.stack || context.prdContent) {
-                        contextInfo = "\n\nAvailable Context:\n";
-                        if (context.stack) {
-                            contextInfo += (0, stack_formatter_1.formatStackForContext)(context.stack) + "\n";
-                        }
-                        if (context.documentation) {
-                            contextInfo += `Documentation Available: ${context.documentation.recap}\n`;
-                            if (context.documentation.files.length > 0) {
-                                contextInfo += `Documentation Files: ${context.documentation.files
-                                    .map((f) => f.path)
-                                    .join(", ")}\n`;
-                            }
-                        }
-                        if (context.prdContent) {
-                            prdContent = context.prdContent;
+        // Build context
+        let contextInfo = "";
+        let prdContent = "";
+        if (taskId) {
+            const contextBuilder = (0, ai_service_factory_1.getContextBuilder)();
+            try {
+                const context = await contextBuilder.buildContext(taskId);
+                if (context.documentation || context.stack || context.prdContent) {
+                    contextInfo = "\n\nAvailable Context:\n";
+                    if (context.stack) {
+                        contextInfo += (0, stack_formatter_1.formatStackForContext)(context.stack) + "\n";
+                    }
+                    if (context.documentation) {
+                        contextInfo += `Documentation Available: ${context.documentation.recap}\n`;
+                        if (context.documentation.files.length > 0) {
+                            contextInfo += `Documentation Files: ${context.documentation.files
+                                .map((f) => f.path)
+                                .join(", ")}\n`;
                         }
                     }
-                }
-                catch (error) {
-                    throw error;
+                    if (context.prdContent) {
+                        prdContent = context.prdContent;
+                    }
                 }
             }
-            let prompt;
-            if (promptOverride) {
-                prompt = promptOverride;
+            catch (error) {
+                throw error;
             }
-            else {
-                const promptResult = prompt_builder_1.PromptBuilder.buildPrompt({
-                    name: "task-enhancement",
-                    type: "user",
-                    variables: {
-                        TASK_TITLE: title,
-                        TASK_DESCRIPTION: description || "None",
-                        CONTEXT_INFO: contextInfo,
-                        PRD_CONTENT: prdContent || "No PRD content available",
-                    },
+        }
+        // Build prompt
+        let prompt;
+        if (promptOverride) {
+            prompt = promptOverride;
+        }
+        else {
+            const promptResult = prompt_builder_1.PromptBuilder.buildPrompt({
+                name: "task-enhancement",
+                type: "user",
+                variables: {
+                    TASK_TITLE: title,
+                    TASK_DESCRIPTION: description || "None",
+                    CONTEXT_INFO: contextInfo,
+                    PRD_CONTENT: prdContent || "No PRD content available",
+                },
+            });
+            if (!promptResult.success) {
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_OPERATION_FAILED, `Failed to build task enhancement prompt: ${promptResult.error}`, {
+                    context: "Prompt building failed during task enhancement operation",
+                    suggestions: [
+                        "Verify prompt template exists",
+                        "Check variable substitution",
+                    ],
                 });
-                if (!promptResult.success) {
-                    throw new Error(`Failed to build task enhancement prompt: ${promptResult.error}`);
-                }
-                prompt = promptResult.prompt;
             }
-            return this.streamText("", config, prompts_1.TASK_ENHANCEMENT_SYSTEM_PROMPT, userMessage || prompt, streamingOptions, { maxAttempts: 1 });
-        }, retryConfig, "Task enhancement");
+            prompt = promptResult.prompt;
+        }
+        // Execute AI operation with proper error handling
+        const result = await this.aiOperationUtility.executeAIOperation("Task enhancement", async () => {
+            return await this.aiOperationUtility.streamText("", config, prompts_1.TASK_ENHANCEMENT_SYSTEM_PROMPT, userMessage || prompt, streamingOptions, { maxAttempts: 1 });
+        }, {
+            streamingOptions,
+            retryConfig,
+            aiConfig: config,
+            maxRetries: retryConfig?.maxAttempts || 2,
+        });
+        // Return the result directly (errors are thrown, not returned)
+        return result.result;
     }
     async planTask(taskContext, taskDetails, config, promptOverride, userMessage, streamingOptions, retryConfig) {
-        return this.retryHandler.executeWithRetry(async () => {
-            let prompt;
-            if (promptOverride) {
-                prompt = promptOverride;
-            }
-            else {
-                const promptResult = prompt_builder_1.PromptBuilder.buildPrompt({
-                    name: "task-planning",
-                    type: "user",
-                    variables: {
-                        TASK_CONTEXT: taskContext,
-                        TASK_DETAILS: taskDetails,
-                    },
+        // Build prompt
+        let prompt;
+        if (promptOverride) {
+            prompt = promptOverride;
+        }
+        else {
+            const promptResult = prompt_builder_1.PromptBuilder.buildPrompt({
+                name: "task-planning",
+                type: "user",
+                variables: {
+                    TASK_CONTEXT: taskContext,
+                    TASK_DETAILS: taskDetails,
+                },
+            });
+            if (!promptResult.success) {
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_OPERATION_FAILED, `Failed to build task planning prompt: ${promptResult.error}`, {
+                    context: "Prompt building failed during task planning operation",
+                    suggestions: [
+                        "Verify prompt template exists",
+                        "Check variable substitution",
+                    ],
                 });
-                if (!promptResult.success) {
-                    throw new Error(`Failed to build task planning prompt: ${promptResult.error}`);
-                }
-                prompt = promptResult.prompt;
             }
-            const model = this.modelProvider.getModel({
-                ...this.modelProvider.getAIConfig(),
-                ...config,
-            });
+            prompt = promptResult.prompt;
+        }
+        // Execute AI operation with proper error handling
+        const result = await this.aiOperationUtility.executeAIOperation("Task planning", async () => {
             const mcpTools = await this.context7Client.getMCPTools();
             const allTools = {
                 ...mcpTools,
                 ...filesystem_tools_1.filesystemTools,
             };
-            const result = (0, ai_1.streamText)({
-                model,
-                tools: allTools,
-                system: prompts_1.TASK_PLANNING_SYSTEM_PROMPT +
-                    `
+            return await this.aiOperationUtility.streamTextWithTools(prompts_1.TASK_PLANNING_SYSTEM_PROMPT +
+                `
 
 You have access to filesystem tools that allow you to:
 - readFile: Read the contents of any file in the project
 - listDirectory: List contents of directories
 
-Use these tools to understand the project structure, existing code, and dependencies when creating implementation plans.`,
-                messages: [{ role: "user", content: userMessage || prompt }],
-                maxRetries: 0,
-                onChunk: streamingOptions?.onChunk
-                    ? ({ chunk }) => {
-                        if (chunk.type === "text-delta") {
-                            streamingOptions.onChunk(chunk.text);
-                        }
-                        else if (chunk.type === "reasoning-delta") {
-                            streamingOptions.onReasoning?.(chunk.text);
-                        }
-                    }
-                    : undefined,
-                onFinish: streamingOptions?.onFinish
-                    ? ({ text, finishReason, usage }) => {
-                        streamingOptions.onFinish({
-                            text,
-                            finishReason,
-                            usage,
-                            isAborted: false,
-                        });
-                    }
-                    : undefined,
-            });
-            return (await result).text;
-        }, retryConfig, "Task planning");
+Use these tools to understand the project structure, existing code, and dependencies when creating implementation plans.`, userMessage || prompt, config, streamingOptions, allTools);
+        }, {
+            streamingOptions,
+            retryConfig,
+            aiConfig: config,
+            maxRetries: retryConfig?.maxAttempts || 2,
+        });
+        // Return the result directly (errors are thrown, not returned)
+        return result.result;
     }
 }
 exports.TaskOperations = TaskOperations;

package/dist/lib/benchmark/registry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../src/lib/benchmark/registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAA0B,MAAM,SAAS,CAAC;AAOzE,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,UAAU,CAAkD;;IAMpE,QAAQ,CAAC,EAAE,EAAE,sBAAsB;IAInC,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,sBAAsB,GAAG,SAAS;IAInD,IAAI,IAAI,sBAAsB,EAAE;IAIhC,OAAO,CAAC,gBAAgB;CA2RzB;AAED,eAAO,MAAM,iBAAiB,mBAA0B,CAAC"}
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../src/lib/benchmark/registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAA0B,MAAM,SAAS,CAAC;AAOzE,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,UAAU,CAAkD;;IAMpE,QAAQ,CAAC,EAAE,EAAE,sBAAsB;IAInC,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,sBAAsB,GAAG,SAAS;IAInD,IAAI,IAAI,sBAAsB,EAAE;IAIhC,OAAO,CAAC,gBAAgB;CAsRzB;AAED,eAAO,MAAM,iBAAiB,mBAA0B,CAAC"}

package/dist/lib/benchmark/registry.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.benchmarkRegistry = exports.BenchmarkRegistry = void 0;
 const prd_1 = require("../../services/prd");
-const tasks_1 = require("../../services/tasks"); // We'll need to instantiate this or use a singleton if available
+const tasks_1 = require("../../services/tasks");
 const workflow_benchmark_1 = require("../../services/workflow-benchmark");
 class BenchmarkRegistry {
     operations = new Map();
@@ -59,17 +59,13 @@ class BenchmarkRegistry {
             },
         });
         // Task Breakdown Adapter
-        // Note: TaskService is a class, we need an instance.
-        // In a real app we should use dependency injection or a singleton.
-        // For now, we'll create a new instance or assume one is available.
-        const taskService = new tasks_1.TaskService();
         this.register({
             id: "task-breakdown",
             name: "Task Breakdown",
             description: "Break down a task into subtasks",
             validateInput: (input) => typeof input.taskId === "string",
             execute: async (input, aiOptions, streamingOptions) => {
-                return await taskService.splitTask(input.taskId, aiOptions, input.prompt, input.message, streamingOptions, // streaming options
+                return await tasks_1.taskService.splitTask(input.taskId, aiOptions, input.prompt, input.message, streamingOptions, // streaming options
                 input.tools);
             },
         });
@@ -90,7 +86,7 @@ class BenchmarkRegistry {
             description: "Create a new task with AI enhancement using Context7 documentation",
             validateInput: (input) => typeof input.title === "string" && input.title.length > 0,
             execute: async (input, aiOptions, streamingOptions) => {
-                return await taskService.createTask({
+                return await tasks_1.taskService.createTask({
                     title: input.title,
                     content: input.content,
                     parentId: input.parentId,
@@ -108,7 +104,7 @@ class BenchmarkRegistry {
             description: "Enhance an existing task with AI using Context7 documentation",
             validateInput: (input) => typeof input.taskId === "string",
             execute: async (input, aiOptions, streamingOptions) => {
-                return await taskService.enhanceTask(input.taskId, aiOptions, streamingOptions);
+                return await tasks_1.taskService.enhanceTask(input.taskId, aiOptions, streamingOptions);
             },
         });
         // Task Planning
@@ -118,7 +114,7 @@ class BenchmarkRegistry {
             description: "Create a detailed implementation plan for a task",
             validateInput: (input) => typeof input.taskId === "string",
             execute: async (input, aiOptions, streamingOptions) => {
-                return await taskService.planTask(input.taskId, aiOptions, streamingOptions);
+                return await tasks_1.taskService.planTask(input.taskId, aiOptions, streamingOptions);
             },
         });
         // Task Documentation
@@ -128,7 +124,7 @@ class BenchmarkRegistry {
             description: "Analyze and generate documentation for a task",
             validateInput: (input) => typeof input.taskId === "string",
             execute: async (input, aiOptions, streamingOptions) => {
-                return await taskService.documentTask(input.taskId, input.force || false, aiOptions, streamingOptions);
+                return await tasks_1.taskService.documentTask(input.taskId, input.force || false, aiOptions, streamingOptions);
             },
         });
         // PRD Creation