task-o-matic 0.0.13 → 0.0.15

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

@@ -11,8 +11,36 @@ class BaseOperations {
     context7Client = new mcp_client_1.Context7Client();
     retryHandler = new retry_handler_1.RetryHandler();
     modelProvider = new model_provider_1.ModelProvider();
+    /**
+     * Merges AI configuration with proper precedence.
+     *
+     * Configuration precedence (highest to lowest):
+     * 1. Method parameter `config` (operation-specific overrides)
+     * 2. ConfigManager global config (project-level settings)
+     * 3. Environment variables (OPENAI_API_KEY, etc.)
+     * 4. Provider defaults (defined in config.ts)
+     *
+     * @param config - Optional operation-specific config overrides
+     * @returns Merged AIConfig with all precedence levels applied
+     *
+     * @example
+     * ```typescript
+     * // Override just the model for this operation
+     * const finalConfig = this.mergeAIConfig({ model: "gpt-4o" });
+     *
+     * // Use default config (from ConfigManager + env vars)
+     * const finalConfig = this.mergeAIConfig();
+     * ```
+     */
+    mergeAIConfig(config) {
+        // Get base config (includes ConfigManager + env vars + defaults)
+        const baseConfig = this.modelProvider.getAIConfig();
+        // Apply operation-specific overrides (highest priority)
+        return { ...baseConfig, ...config };
+    }
     async streamText(prompt, config, systemPrompt, userMessage, streamingOptions, retryConfig) {
-        const aiConfig = { ...this.modelProvider.getAIConfig(), ...config };
+        // Merge config with proper precedence (Bug fix 2.9)
+        const aiConfig = this.mergeAIConfig(config);
         return this.retryHandler.executeWithRetry(async () => {
             const model = this.modelProvider.getModel(aiConfig);
             const result = (0, ai_1.streamText)({

package/dist/lib/ai-service/model-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model-provider.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/model-provider.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAGvC,qBAAa,aAAa;IACjB,WAAW,IAAI,QAAQ;IAY9B,OAAO,CAAC,YAAY;IAsBpB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,eAAe;CA+B9C"}
+{"version":3,"file":"model-provider.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/model-provider.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAOvC,qBAAa,aAAa;IACjB,WAAW,IAAI,QAAQ;IAY9B,OAAO,CAAC,YAAY;IAsBpB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,eAAe;CA0F9C"}

package/dist/lib/ai-service/model-provider.js

@@ -6,6 +6,7 @@ const anthropic_1 = require("@ai-sdk/anthropic");
 const openai_compatible_1 = require("@ai-sdk/openai-compatible");
 const ai_sdk_provider_1 = require("@openrouter/ai-sdk-provider");
 const config_1 = require("../config");
+const task_o_matic_error_1 = require("../../utils/task-o-matic-error");
 class ModelProvider {
     getAIConfig() {
         const config = config_1.configManager.getAIConfig();
@@ -41,22 +42,47 @@ class ModelProvider {
         switch (provider) {
             case "openai":
                 if (!apiKey)
-                    throw new Error("OpenAI API key is required");
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "OpenAI API key is required", {
+                        suggestions: [
+                            "Set the OPENAI_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
                 return (0, openai_1.openai)(model);
             case "anthropic":
                 if (!apiKey)
-                    throw new Error("Anthropic API key is required");
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Anthropic API key is required", {
+                        suggestions: [
+                            "Set the ANTHROPIC_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
                 return (0, anthropic_1.anthropic)(model);
             case "openrouter":
                 if (!apiKey)
-                    throw new Error("OpenRouter API key is required");
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "OpenRouter API key is required", {
+                        suggestions: [
+                            "Set the OPENROUTER_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
                 const openRouterProvider = (0, ai_sdk_provider_1.createOpenRouter)({ apiKey });
                 return openRouterProvider(model);
             case "custom":
                 if (!apiKey)
-                    throw new Error("Custom API key is required");
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Custom API key is required for custom provider", {
+                        suggestions: [
+                            "Set the CUSTOM_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
                 if (!baseURL)
-                    throw new Error("Custom provider requires baseURL");
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Custom provider requires baseURL", {
+                        suggestions: [
+                            "Set the CUSTOM_API_URL environment variable.",
+                            "Run `task-o-matic config set-ai-provider-url <url>`",
+                        ],
+                    });
                 const customProvider = (0, openai_compatible_1.createOpenAICompatible)({
                     name: "custom",
                     apiKey,
@@ -64,7 +90,12 @@ class ModelProvider {
                 });
                 return customProvider(model);
             default:
-                throw new Error(`Unsupported provider: ${provider}`);
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, `Unsupported provider: ${provider}`, {
+                    suggestions: [
+                        "Use one of the supported providers: 'openai', 'anthropic', 'openrouter', 'custom'.",
+                        "Run `task-o-matic config set-ai-provider <provider>`",
+                    ],
+                });
         }
     }
 }

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

@@ -1,12 +1,13 @@
 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;
         estimatedEffort?: string;
     }>>;
     enhanceTask(title: string, description?: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, taskId?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>): Promise<string>;
-    planTask(taskContext: string, taskDetails: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>): Promise<any>;
+    planTask(taskContext: string, taskDetails: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>): Promise<string>;
 }
 //# sourceMappingURL=task-operations.d.ts.map

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,GAAG,CAAC;CA6EhB"}
+{"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;AAGnD,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;IAmGK,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;IAoFZ,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;CA4DnB"}

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

@@ -1,95 +1,62 @@
 "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");
 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;
+            }
+            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}`);
             }
-            let response;
-            if (enableFilesystemTools) {
-                const model = this.modelProvider.getModel({
-                    ...this.modelProvider.getAIConfig(),
-                    ...config,
-                });
-                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");
@@ -100,126 +67,121 @@ 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",
-                    },
-                });
-                if (!promptResult.success) {
-                    throw new Error(`Failed to build task enhancement prompt: ${promptResult.error}`);
-                }
-                prompt = promptResult.prompt;
+        }
+        // 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 new Error(`Failed to build task enhancement prompt: ${promptResult.error}`);
             }
-            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,
-                    },
-                });
-                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,
+        // 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 new Error(`Failed to build task planning prompt: ${promptResult.error}`);
+            }
+            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

package/dist/lib/better-t-stack-cli.d.ts

@@ -1,4 +1,4 @@
-import { BTSConfig } from "../types";
+import { BTSConfig, InitOptions } from "../types";
 export declare class BetterTStackService {
     createProject(name: string, config: BTSConfig, workingDirectory?: string): Promise<{
         success: boolean;
@@ -10,26 +10,41 @@ export declare class BetterTStackService {
     private addCheckTypesScript;
     private copyDocumentation;
 }
-export interface InitOptions {
-    projectName?: string;
-    name?: string;
-    frontend: string;
-    backend: string;
-    database: string;
-    noAuth?: boolean;
-    addons?: string[];
-    runtime?: string;
-    api?: string;
-    payment?: string;
-    orm?: string;
-    dbSetup?: string;
-    packageManager?: string;
-    noGit?: boolean;
-    webDeploy?: string;
-    serverDeploy?: string;
-    noInstall?: boolean;
-    examples?: string[];
-    includeDocs?: boolean;
+export declare class BetterTStackIntegration {
+    private btsService;
+    constructor();
+    /**
+     * Parse frontend option into array of frontends
+     */
+    private parseFrontends;
+    /**
+     * Split frontends into Better-T-Stack frontends vs custom frontends
+     */
+    private splitFrontends;
+    /**
+     * Create project with support for multiple frontends
+     */
+    createProject(name: string, options: InitOptions, workingDirectory?: string): Promise<{
+        success: boolean;
+        message: string;
+        projectPath?: string;
+    }>;
+    /**
+     * Bootstrap Better-T-Stack project with one or more frontends
+     */
+    private bootstrapBetterTStackProject;
+    /**
+     * Add CLI app to project (standalone or monorepo)
+     */
+    private addCliToProject;
+    /**
+     * Add TUI app to project (standalone or monorepo)
+     */
+    private addTuiToProject;
+    /**
+     * Add MedusaJS app to project (standalone or monorepo)
+     */
+    private addMedusaToProject;
 }
 export declare function runBetterTStackCLI(options: InitOptions, workingDirectory?: string): Promise<{
     success: boolean;