task-o-matic 0.0.14 → 0.0.15

package/dist/services/workflow-ai-assistant.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"workflow-ai-assistant.d.ts","sourceRoot":"","sources":["../../src/services/workflow-ai-assistant.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAEtD,OAAO,EAAiB,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAItE;;;GAGG;AACH,qBAAa,mBAAmB;IAC9B;;OAEG;IACG,gBAAgB,CAAC,KAAK,EAAE;QAC5B,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,IAAI,CAAC,EAAE,OAAO,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IAqDF;;OAEG;IACG,iBAAiB,CAAC,KAAK,EAAE;QAC7B,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC;IAYnB;;OAEG;IACG,mBAAmB,CAAC,KAAK,EAAE;QAC/B,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC;IAmCnB;;OAEG;IACG,wBAAwB,CAAC,KAAK,EAAE;QACpC,KAAK,EAAE,KAAK,CAAC;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAC;YAAC,WAAW,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAClE,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC;QACV,gBAAgB,EAAE,KAAK,CAAC;YACtB,EAAE,EAAE,MAAM,CAAC;YACX,QAAQ,EAAE,MAAM,CAAC;YACjB,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC,CAAC;QACH,eAAe,EAAE,MAAM,CAAC;KACzB,CAAC;IA6DF;;OAEG;IACG,mBAAmB,CAAC,KAAK,EAAE;QAC/B,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC;CAuCpB;AAGD,eAAO,MAAM,mBAAmB,qBAA4B,CAAC"}
+{"version":3,"file":"workflow-ai-assistant.d.ts","sourceRoot":"","sources":["../../src/services/workflow-ai-assistant.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAEtD,OAAO,EAAiB,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAQtE;;;GAGG;AACH,qBAAa,mBAAmB;IAC9B;;OAEG;IACG,gBAAgB,CAAC,KAAK,EAAE;QAC5B,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,IAAI,CAAC,EAAE,OAAO,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IAgEF;;OAEG;IACG,iBAAiB,CAAC,KAAK,EAAE;QAC7B,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC;IAYnB;;OAEG;IACG,mBAAmB,CAAC,KAAK,EAAE;QAC/B,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC;IAoCnB;;OAEG;IACG,wBAAwB,CAAC,KAAK,EAAE;QACpC,KAAK,EAAE,KAAK,CAAC;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAC;YAAC,WAAW,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAClE,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC;QACV,gBAAgB,EAAE,KAAK,CAAC;YACtB,EAAE,EAAE,MAAM,CAAC;YACX,QAAQ,EAAE,MAAM,CAAC;YACjB,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC,CAAC;QACH,eAAe,EAAE,MAAM,CAAC;KACzB,CAAC;IAwEF;;OAEG;IACG,mBAAmB,CAAC,KAAK,EAAE;QAC/B,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC;CAwCpB;AAGD,eAAO,MAAM,mBAAmB,qBAA4B,CAAC"}

package/dist/services/workflow-ai-assistant.js

@@ -4,6 +4,7 @@ exports.workflowAIAssistant = exports.WorkflowAIAssistant = void 0;
 const ai_service_factory_1 = require("../utils/ai-service-factory");
 const ai_config_builder_1 = require("../utils/ai-config-builder");
 const prompt_builder_1 = require("../lib/prompt-builder");
+const task_o_matic_error_1 = require("../utils/task-o-matic-error");
 /**
  * WorkflowAIAssistant - AI-powered decision making for workflow steps
  * Helps users make configuration choices using natural language
@@ -22,7 +23,7 @@ class WorkflowAIAssistant {
             },
         });
         if (!promptResult.success) {
-            throw new Error(`Failed to build project init prompt: ${promptResult.error}`);
+            throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.CONFIGURATION_ERROR, `Failed to build project init prompt: ${promptResult.error}`);
         }
         const systemPromptResult = prompt_builder_1.PromptBuilder.buildPrompt({
             name: "project-init-suggestion",
@@ -34,7 +35,13 @@ class WorkflowAIAssistant {
         try {
             const jsonMatch = result.match(/\{[\s\S]*\}/);
             if (!jsonMatch) {
-                throw new Error("No JSON found in AI response");
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_OPERATION_FAILED, "No JSON found in AI response for init config", {
+                    context: "The AI did not return a valid JSON object.",
+                    suggestions: [
+                        "Try a different model or provider.",
+                        "Check the prompt for clarity.",
+                    ],
+                });
             }
             return JSON.parse(jsonMatch[0]);
         }
@@ -73,7 +80,7 @@ class WorkflowAIAssistant {
             },
         });
         if (!promptResult.success) {
-            throw new Error(`Failed to build PRD improvement prompt: ${promptResult.error}`);
+            throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.CONFIGURATION_ERROR, `Failed to build PRD improvement prompt: ${promptResult.error}`);
         }
         const systemPromptResult = prompt_builder_1.PromptBuilder.buildPrompt({
             name: "prd-improvement",
@@ -100,7 +107,7 @@ class WorkflowAIAssistant {
             },
         });
         if (!promptResult.success) {
-            throw new Error(`Failed to build task prioritization prompt: ${promptResult.error}`);
+            throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.CONFIGURATION_ERROR, `Failed to build task prioritization prompt: ${promptResult.error}`);
         }
         const systemPromptResult = prompt_builder_1.PromptBuilder.buildPrompt({
             name: "task-prioritization",
@@ -112,7 +119,13 @@ class WorkflowAIAssistant {
         try {
             const jsonMatch = result.match(/\{[\s\S]*\}/);
             if (!jsonMatch) {
-                throw new Error("No JSON found in AI response");
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_OPERATION_FAILED, "No JSON found in AI response for task prioritization", {
+                    context: "The AI did not return a valid JSON object.",
+                    suggestions: [
+                        "Try a different model or provider.",
+                        "Check the prompt for clarity.",
+                    ],
+                });
             }
             return JSON.parse(jsonMatch[0]);
         }
@@ -146,7 +159,7 @@ class WorkflowAIAssistant {
             },
         });
         if (!promptResult.success) {
-            throw new Error(`Failed to build task splitting prompt: ${promptResult.error}`);
+            throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.CONFIGURATION_ERROR, `Failed to build task splitting prompt: ${promptResult.error}`);
         }
         const systemPromptResult = prompt_builder_1.PromptBuilder.buildPrompt({
             name: "task-splitting-assistance",

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

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=task-operations.test.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"task-operations.test.d.ts","sourceRoot":"","sources":["../../../../src/test/lib/ai-service/task-operations.test.ts"],"names":[],"mappings":""}

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

@@ -0,0 +1,362 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = __importStar(require("assert"));
+const task_operations_1 = require("../../../lib/ai-service/task-operations");
+const task_o_matic_error_1 = require("../../../utils/task-o-matic-error");
+/**
+ * ⚠️ CRITICAL: These integration tests use 100% MOCKS - ZERO real AI calls
+ * No API calls are made, no costs incurred, tests run fast
+ *
+ * These tests verify that TaskOperations properly integrates with AIOperationUtility
+ * and that errors propagate correctly through the stack.
+ */
+describe("TaskOperations Integration Tests", () => {
+    let taskOps;
+    let mockAIOperationUtility;
+    beforeEach(() => {
+        taskOps = new task_operations_1.TaskOperations();
+        // Create mock AIOperationUtility - NO REAL AI CALLS
+        mockAIOperationUtility = {
+            executeAIOperation: async (operationName, operation, options) => {
+                // Execute the operation and return mock metrics
+                const result = await operation();
+                return {
+                    result,
+                    metrics: {
+                        duration: 100,
+                        tokenUsage: {
+                            prompt: 50,
+                            completion: 25,
+                            total: 75,
+                        },
+                    },
+                };
+            },
+            streamTextWithTools: async (systemPrompt, userMessage, config, streamingOptions, tools) => {
+                // Return mock JSON response
+                return JSON.stringify({
+                    subtasks: [
+                        {
+                            title: "Mock subtask 1",
+                            description: "Mock description 1",
+                            effort: "small",
+                        },
+                        {
+                            title: "Mock subtask 2",
+                            description: "Mock description 2",
+                            effort: "medium",
+                        },
+                    ],
+                });
+            },
+            streamText: async (_prompt, _config, _systemPrompt, _userMessage, _streamingOptions) => {
+                return "Mock enhanced content";
+            },
+        };
+        // Inject mock into TaskOperations
+        taskOps.aiOperationUtility = mockAIOperationUtility;
+    });
+    describe("breakdownTask Integration", () => {
+        it("should successfully break down a task using AIOperationUtility", async () => {
+            const mockTask = {
+                id: "test-task-1",
+                title: "Test Task",
+                description: "A task to break down",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            const result = await taskOps.breakdownTask(mockTask);
+            // Verify result structure
+            assert.ok(Array.isArray(result));
+            assert.strictEqual(result.length, 2);
+            assert.strictEqual(result[0].title, "Mock subtask 1");
+            assert.strictEqual(result[0].content, "Mock description 1");
+            assert.strictEqual(result[0].estimatedEffort, "small");
+        });
+        it("should handle filesystem tools when enabled", async () => {
+            let toolsPassed = false;
+            // Override streamTextWithTools to capture tools parameter
+            mockAIOperationUtility.streamTextWithTools = async (_systemPrompt, _userMessage, _config, _streamingOptions, tools) => {
+                if (tools && Object.keys(tools).length > 0) {
+                    toolsPassed = true;
+                }
+                return JSON.stringify({
+                    subtasks: [
+                        {
+                            title: "Subtask with tools",
+                            description: "Used filesystem tools",
+                            effort: "small",
+                        },
+                    ],
+                });
+            };
+            const mockTask = {
+                id: "test-task-2",
+                title: "Task with tools",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            const result = await taskOps.breakdownTask(mockTask, undefined, undefined, undefined, undefined, undefined, undefined, undefined, undefined, true // enableFilesystemTools
+            );
+            assert.ok(toolsPassed, "Filesystem tools should be passed when enabled");
+            assert.strictEqual(result.length, 1);
+        });
+        it("should propagate TaskOMaticError from AIOperationUtility", async () => {
+            // Mock AIOperationUtility to throw error
+            taskOps.aiOperationUtility.executeAIOperation = async () => {
+                throw new task_o_matic_error_1.TaskOMaticError("AI operation failed: Task breakdown", {
+                    code: "AI_OPERATION_FAILED",
+                    context: "Mock error context",
+                    suggestions: ["Check AI configuration"],
+                });
+            };
+            const mockTask = {
+                id: "test-task-error",
+                title: "Task that will fail",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            try {
+                await taskOps.breakdownTask(mockTask);
+                assert.fail("Should have thrown TaskOMaticError");
+            }
+            catch (error) {
+                assert.ok(error instanceof task_o_matic_error_1.TaskOMaticError);
+                assert.strictEqual(error.code, "AI_OPERATION_FAILED");
+                assert.ok(error.message.includes("Task breakdown"));
+            }
+        });
+    });
+    describe("enhanceTask Integration", () => {
+        it("should successfully enhance a task using AIOperationUtility", async () => {
+            const result = await taskOps.enhanceTask("Test Task", "Basic description");
+            // Verify result
+            assert.strictEqual(typeof result, "string");
+            assert.strictEqual(result, "Mock enhanced content");
+        });
+        it("should propagate errors from AIOperationUtility", async () => {
+            // Mock AIOperationUtility to throw error
+            taskOps.aiOperationUtility.executeAIOperation = async () => {
+                throw new task_o_matic_error_1.TaskOMaticError("AI operation failed: Task enhancement", {
+                    code: "AI_OPERATION_FAILED",
+                    context: "Enhancement failed",
+                    suggestions: ["Retry with different parameters"],
+                });
+            };
+            try {
+                await taskOps.enhanceTask("Test", "Description");
+                assert.fail("Should have thrown TaskOMaticError");
+            }
+            catch (error) {
+                assert.ok(error instanceof task_o_matic_error_1.TaskOMaticError);
+                assert.ok(error.message.includes("Task enhancement"));
+            }
+        });
+    });
+    describe("planTask Integration", () => {
+        it("should successfully plan a task using AIOperationUtility", async () => {
+            // Mock streamTextWithTools to return plan
+            mockAIOperationUtility.streamTextWithTools = async () => {
+                return "Mock implementation plan:\n1. Step 1\n2. Step 2\n3. Step 3";
+            };
+            const result = await taskOps.planTask("Project context", "Task details");
+            // Verify result
+            assert.strictEqual(typeof result, "string");
+            assert.ok(result.includes("Mock implementation plan"));
+            assert.ok(result.includes("Step 1"));
+        });
+        it("should handle MCP tools and filesystem tools", async () => {
+            let toolsReceived = false;
+            mockAIOperationUtility.streamTextWithTools = async (_systemPrompt, _userMessage, _config, _streamingOptions, tools) => {
+                if (tools && Object.keys(tools).length > 0) {
+                    toolsReceived = true;
+                }
+                return "Plan with tools";
+            };
+            // Mock Context7Client to return MCP tools
+            const mockContext7Client = {
+                getMCPTools: async () => ({
+                    mockTool: {
+                        description: "Mock MCP tool",
+                        parameters: {},
+                        execute: async () => ({}),
+                    },
+                }),
+                saveContext7Documentation: () => { },
+            };
+            taskOps.context7Client = mockContext7Client;
+            const result = await taskOps.planTask("Context", "Details");
+            assert.ok(toolsReceived, "Tools should be passed to streamTextWithTools");
+            assert.strictEqual(result, "Plan with tools");
+        });
+        it("should propagate errors from AIOperationUtility", async () => {
+            // Mock AIOperationUtility to throw error
+            taskOps.aiOperationUtility.executeAIOperation = async () => {
+                throw new task_o_matic_error_1.TaskOMaticError("AI operation failed: Task planning", {
+                    code: "AI_OPERATION_FAILED",
+                    context: "Planning failed",
+                    suggestions: ["Check task context"],
+                });
+            };
+            try {
+                await taskOps.planTask("Context", "Details");
+                assert.fail("Should have thrown TaskOMaticError");
+            }
+            catch (error) {
+                assert.ok(error instanceof task_o_matic_error_1.TaskOMaticError);
+                assert.ok(error.message.includes("Task planning"));
+            }
+        });
+    });
+    describe("Metrics Flow", () => {
+        it("should return results from AIOperationUtility without exposing metrics", async () => {
+            // breakdownTask returns result.result directly, not the full AIOperationResult
+            const mockTask = {
+                id: "metrics-test",
+                title: "Metrics test task",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            const result = await taskOps.breakdownTask(mockTask);
+            // Verify we got the result, not the AIOperationResult wrapper
+            assert.ok(Array.isArray(result));
+            assert.ok(!("metrics" in result), "Should not expose metrics in result");
+            assert.ok(!("result" in result), "Should not expose AIOperationResult structure");
+        });
+    });
+    describe("Error Handling Integration", () => {
+        it("should wrap string errors in TaskOMaticError", async () => {
+            // Mock operation that throws string error
+            taskOps.aiOperationUtility.executeAIOperation = async () => {
+                const error = new Error("String error was converted");
+                throw new task_o_matic_error_1.TaskOMaticError("AI operation failed: Task breakdown", {
+                    code: "AI_OPERATION_FAILED",
+                    cause: error,
+                    context: JSON.stringify({
+                        operation: "Task breakdown",
+                        error: "String error was converted",
+                    }),
+                    suggestions: ["Check AI configuration"],
+                });
+            };
+            const mockTask = {
+                id: "string-error-test",
+                title: "Test",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            try {
+                await taskOps.breakdownTask(mockTask);
+                assert.fail("Should have thrown");
+            }
+            catch (error) {
+                assert.ok(error instanceof task_o_matic_error_1.TaskOMaticError);
+                const taskError = error;
+                assert.ok(taskError.cause instanceof Error);
+            }
+        });
+        it("should preserve error context and suggestions", async () => {
+            const mockError = new task_o_matic_error_1.TaskOMaticError("Original error", {
+                code: "TEST_ERROR",
+                context: "Original context",
+                suggestions: ["Original suggestion 1", "Original suggestion 2"],
+                metadata: { testData: "test value" },
+            });
+            taskOps.aiOperationUtility.executeAIOperation = async () => {
+                throw mockError;
+            };
+            const mockTask = {
+                id: "context-test",
+                title: "Test",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            try {
+                await taskOps.breakdownTask(mockTask);
+                assert.fail("Should have thrown");
+            }
+            catch (error) {
+                assert.ok(error instanceof task_o_matic_error_1.TaskOMaticError);
+                const taskError = error;
+                assert.strictEqual(taskError.code, "TEST_ERROR");
+                assert.ok(taskError.context?.includes("Original context"));
+                assert.ok(Array.isArray(taskError.suggestions));
+                assert.ok(taskError.metadata?.testData === "test value");
+            }
+        });
+    });
+    describe("Streaming Options Integration", () => {
+        it("should pass streaming options through to AIOperationUtility", async () => {
+            let onChunkCalled = false;
+            let onFinishCalled = false;
+            const streamingOptions = {
+                onChunk: (chunk) => {
+                    onChunkCalled = true;
+                },
+                onFinish: (result) => {
+                    onFinishCalled = true;
+                },
+            };
+            // Mock to verify streaming options are passed
+            let receivedOptions;
+            taskOps.aiOperationUtility.executeAIOperation = async (_operationName, operation, options) => {
+                receivedOptions = options;
+                const result = await operation();
+                return {
+                    result,
+                    metrics: { duration: 100 },
+                };
+            };
+            const mockTask = {
+                id: "streaming-test",
+                title: "Test",
+                status: "todo",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            };
+            await taskOps.breakdownTask(mockTask, undefined, undefined, undefined, streamingOptions);
+            // Verify streaming options were passed
+            assert.ok(receivedOptions);
+            assert.ok(receivedOptions.streamingOptions);
+        });
+    });
+});

package/dist/test/mocks/mock-ai-operations.d.ts

@@ -0,0 +1,15 @@
+import { Task, StreamingOptions, AIConfig } from "../../types";
+export declare class MockAIOperations {
+    enhanceTaskWithDocumentation(taskId: string, title: string, description: string, stackInfo: string, streamingOptions?: StreamingOptions, retryConfig?: any, aiConfig?: AIConfig, existingResearch?: any): Promise<string>;
+    breakdownTask(task: Task, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions, retryConfig?: any, fullContent?: string, stackInfo?: string, existingSubtasks?: Task[], enableFilesystemTools?: boolean): Promise<Task[]>;
+    planTask(taskContext: string, taskDetails: string, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions): Promise<string>;
+    analyzeDocumentationNeeds(taskId: string, title: string, content: string, stackInfo: string, streamingOptions?: StreamingOptions, retryConfig?: any, aiConfig?: AIConfig, existingDocumentations?: string[]): Promise<any>;
+    generateDocumentationRecap(libraries: any[], toolResults: any[], streamingOptions?: StreamingOptions): Promise<string>;
+    parsePRD(prdContent: string, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions, retryConfig?: any, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<any>;
+    generatePRDQuestions(prdContent: string, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions, retryConfig?: any, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<string[]>;
+    reworkPRD(prdContent: string, feedback: string, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions, retryConfig?: any, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<string>;
+    answerPRDQuestions(prdContent: string, questions: string[], aiConfig: AIConfig, context: any, streamingOptions?: StreamingOptions): Promise<Record<string, string>>;
+    generatePRD(description: string, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions): Promise<string>;
+    combinePRDs(prds: string[], originalDescription: string, aiConfig: AIConfig, promptOverride?: string, messageOverride?: string, streamingOptions?: StreamingOptions): Promise<string>;
+}
+//# sourceMappingURL=mock-ai-operations.d.ts.map

package/dist/test/mocks/mock-ai-operations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-ai-operations.d.ts","sourceRoot":"","sources":["../../../src/test/mocks/mock-ai-operations.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,gBAAgB,EAAkB,QAAQ,EAAE,MAAM,aAAa,CAAC;AAG/E,qBAAa,gBAAgB;IACrB,4BAA4B,CAChC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,EACjB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,GAAG,EACjB,QAAQ,CAAC,EAAE,QAAQ,EACnB,gBAAgB,CAAC,EAAE,GAAG,GACrB,OAAO,CAAC,MAAM,CAAC;IAIZ,aAAa,CACjB,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,GAAG,EACjB,WAAW,CAAC,EAAE,MAAM,EACpB,SAAS,CAAC,EAAE,MAAM,EAClB,gBAAgB,CAAC,EAAE,IAAI,EAAE,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,IAAI,EAAE,CAAC;IAyBZ,QAAQ,CACZ,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,GAClC,OAAO,CAAC,MAAM,CAAC;IAMZ,yBAAyB,CAC7B,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,MAAM,EACjB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,GAAG,EACjB,QAAQ,CAAC,EAAE,QAAQ,EACnB,sBAAsB,CAAC,EAAE,MAAM,EAAE,GAChC,OAAO,CAAC,GAAG,CAAC;IAeT,0BAA0B,CAC9B,SAAS,EAAE,GAAG,EAAE,EAChB,WAAW,EAAE,GAAG,EAAE,EAClB,gBAAgB,CAAC,EAAE,gBAAgB,GAClC,OAAO,CAAC,MAAM,CAAC;IAMZ,QAAQ,CACZ,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,GAAG,EACjB,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,GAAG,CAAC;IA4BT,oBAAoB,CACxB,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,GAAG,EACjB,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,EAAE,CAAC;IAQd,SAAS,CACb,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,GAAG,EACjB,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,CAAC;IAIZ,kBAAkB,CACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EAAE,EACnB,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,GAAG,EACZ,gBAAgB,CAAC,EAAE,gBAAgB,GAClC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAQ5B,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,GAClC,OAAO,CAAC,MAAM,CAAC;IAIZ,WAAW,CACf,IAAI,EAAE,MAAM,EAAE,EACd,mBAAmB,EAAE,MAAM,EAC3B,QAAQ,EAAE,QAAQ,EAClB,cAAc,CAAC,EAAE,MAAM,EACvB,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,gBAAgB,GAClC,OAAO,CAAC,MAAM,CAAC;CAKnB"}

package/dist/test/mocks/mock-ai-operations.js

@@ -0,0 +1,107 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MockAIOperations = void 0;
+class MockAIOperations {
+    async enhanceTaskWithDocumentation(taskId, title, description, stackInfo, streamingOptions, retryConfig, aiConfig, existingResearch) {
+        return `${description}\n\n🤖 Enhanced with AI documentation for ${title}`;
+    }
+    async breakdownTask(task, aiConfig, promptOverride, messageOverride, streamingOptions, retryConfig, fullContent, stackInfo, existingSubtasks, enableFilesystemTools) {
+        return [
+            {
+                id: `${task.id}.1`,
+                title: `Subtask 1: ${task.title}`,
+                description: `First subtask for ${task.title}`,
+                content: `Content for subtask 1`,
+                status: "todo",
+                estimatedEffort: "small",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            },
+            {
+                id: `${task.id}.2`,
+                title: `Subtask 2: ${task.title}`,
+                description: `Second subtask for ${task.title}`,
+                content: `Content for subtask 2`,
+                status: "todo",
+                estimatedEffort: "medium",
+                createdAt: Date.now(),
+                updatedAt: Date.now(),
+            },
+        ];
+    }
+    async planTask(taskContext, taskDetails, aiConfig, promptOverride, messageOverride, streamingOptions) {
+        return `# Implementation Plan for ${taskContext.split("\n")[0]}\n\n1. Analyze requirements\n2. Implement core functionality\n3. Write tests\n4. Review and refactor`;
+    }
+    async analyzeDocumentationNeeds(taskId, title, content, stackInfo, streamingOptions, retryConfig, aiConfig, existingDocumentations) {
+        return {
+            libraries: [
+                {
+                    name: "react",
+                    context7Id: "/facebook/react",
+                    reason: "Task involves React development",
+                    searchQuery: "React hooks best practices",
+                },
+            ],
+            confidence: 0.9,
+            files: ["src/components/TaskComponent.tsx"],
+        };
+    }
+    async generateDocumentationRecap(libraries, toolResults, streamingOptions) {
+        return `## Documentation Recap\n\n${libraries
+            .map((lib) => `- ${lib.name}: ${lib.reason}`)
+            .join("\n")}`;
+    }
+    async parsePRD(prdContent, aiConfig, promptOverride, messageOverride, streamingOptions, retryConfig, workingDirectory, enableFilesystemTools) {
+        return {
+            tasks: [
+                {
+                    id: "1",
+                    title: "Implement user authentication",
+                    description: "Add login/logout functionality",
+                    content: "Full authentication implementation",
+                    estimatedEffort: "medium",
+                    dependencies: [],
+                    tags: ["auth", "backend"],
+                },
+                {
+                    id: "2",
+                    title: "Create task management UI",
+                    description: "Build React components for task management",
+                    content: "Task list, create, edit, delete components",
+                    estimatedEffort: "large",
+                    dependencies: ["1"],
+                    tags: ["ui", "frontend"],
+                },
+            ],
+            summary: "PRD for task management application",
+            estimatedDuration: "2 weeks",
+            confidence: 0.95,
+        };
+    }
+    async generatePRDQuestions(prdContent, aiConfig, promptOverride, messageOverride, streamingOptions, retryConfig, workingDirectory, enableFilesystemTools) {
+        return [
+            "What authentication providers should be supported?",
+            "Should the task management support real-time collaboration?",
+            "What are the expected performance requirements?",
+        ];
+    }
+    async reworkPRD(prdContent, feedback, aiConfig, promptOverride, messageOverride, streamingOptions, retryConfig, workingDirectory, enableFilesystemTools) {
+        return `${prdContent}\n\n## Improvements Based on Feedback\n\n${feedback}`;
+    }
+    async answerPRDQuestions(prdContent, questions, aiConfig, context, streamingOptions) {
+        const answers = {};
+        questions.forEach((question) => {
+            answers[question] = `AI-generated answer for: ${question}`;
+        });
+        return answers;
+    }
+    async generatePRD(description, aiConfig, promptOverride, messageOverride, streamingOptions) {
+        return `# Product Requirements Document\n\n## Overview\n${description}\n\n## Features\n- User authentication\n- Task management\n- Real-time collaboration\n\n## Technical Requirements\n- React frontend\n- Node.js backend\n- PostgreSQL database`;
+    }
+    async combinePRDs(prds, originalDescription, aiConfig, promptOverride, messageOverride, streamingOptions) {
+        return `# Master PRD\n\n${prds
+            .map((prd, index) => `## PRD ${index + 1}\n\n${prd}`)
+            .join("\n\n")}`;
+    }
+}
+exports.MockAIOperations = MockAIOperations;

package/dist/test/mocks/mock-context-builder.d.ts

@@ -0,0 +1,10 @@
+import { TaskRepository } from "../../lib/storage/types";
+import { TaskContext } from "../../types";
+export declare class MockContextBuilder {
+    private storage;
+    constructor(storage: TaskRepository);
+    buildContextForNewTask(title: string, content?: string): Promise<TaskContext>;
+    buildContext(taskId: string): Promise<TaskContext>;
+    isDocumentationFresh(documentation: any): boolean;
+}
+//# sourceMappingURL=mock-context-builder.d.ts.map

package/dist/test/mocks/mock-context-builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-context-builder.d.ts","sourceRoot":"","sources":["../../../src/test/mocks/mock-context-builder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,qBAAa,kBAAkB;IACjB,OAAO,CAAC,OAAO;gBAAP,OAAO,EAAE,cAAc;IAErC,sBAAsB,CAC1B,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,WAAW,CAAC;IAiCjB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAsCxD,oBAAoB,CAAC,aAAa,EAAE,GAAG,GAAG,OAAO;CAGlD"}