@wix/ditto-codegen-public 1.0.294 → 1.0.296

package/dist/out.js

@@ -12321,31 +12321,66 @@ var require_category_handler = __commonJS({
     var codeGenerationService_12 = require_codeGenerationService();
     var logger_12 = require_logger();
     var constants_1 = require_constants5();
+    var CategoryKey;
+    (function(CategoryKey2) {
+      CategoryKey2["CodeSearch"] = "code-search";
+      CategoryKey2["DocSearch"] = "doc-search";
+      CategoryKey2["LoadSkill"] = "load-skill";
+      CategoryKey2["RunCommands"] = "run-commands";
+      CategoryKey2["WriteCode"] = "write-code";
+      CategoryKey2["Validate"] = "validate";
+    })(CategoryKey || (CategoryKey = {}));
+    var CATEGORY_DESCRIPTIONS = {
+      [CategoryKey.CodeSearch]: "Searching in code",
+      [CategoryKey.DocSearch]: "Searching in docs",
+      [CategoryKey.LoadSkill]: "Loading agent skills",
+      [CategoryKey.RunCommands]: "Executing commands",
+      [CategoryKey.WriteCode]: "Writing code",
+      [CategoryKey.Validate]: "Validating code"
+    };
     var CategoryTaskHandler = class {
       constructor() {
-        this.codeSearchCreated = false;
-        this.docSearchCreated = false;
+        this.created = /* @__PURE__ */ new Set();
+        this.completed = /* @__PURE__ */ new Set();
       }
-      /** Creates a search sub-task on the first use of code-search or MCP tools. */
-      async handle(tool, ctx) {
-        const jobContext = ctx.getJobContext();
-        if (!jobContext)
+      async handle(tool, ctx, state) {
+        if (!ctx.getJobContext())
           return;
-        if (constants_1.CODE_SEARCH_TOOLS.has(tool) && !this.codeSearchCreated) {
-          this.codeSearchCreated = true;
-          await this.createTask("code-search", "Searching in code", ctx);
+        if ((0, constants_1.isMcpTool)(tool)) {
+          await this.track(CategoryKey.DocSearch, ctx, state);
+        }
+        if (constants_1.CODE_SEARCH_TOOLS.has(tool)) {
+          await this.track(CategoryKey.CodeSearch, ctx, state);
+        }
+        if (tool === "bash") {
+          await this.track(CategoryKey.RunCommands, ctx, state);
+        }
+        if (tool === "validate") {
+          await this.track(CategoryKey.Validate, ctx, state);
+        }
+        if (constants_1.WRITE_TOOLS.has(tool)) {
+          await this.track(CategoryKey.WriteCode, ctx);
+        }
+      }
+      async trackLoadSkill(ctx, state) {
+        await this.track(CategoryKey.LoadSkill, ctx, state);
+      }
+      async track(key, ctx, state) {
+        if (!this.created.has(key)) {
+          await this.createTask(key, ctx);
         }
-        if ((0, constants_1.isMcpTool)(tool) && !this.docSearchCreated) {
-          this.docSearchCreated = true;
-          await this.createTask("doc-search", "Searching in docs", ctx);
+        if (state?.status === "completed" && !this.completed.has(key)) {
+          await this.completeTask(key, ctx);
         }
       }
-      async createTask(suffix, description, ctx) {
+      async createTask(key, ctx) {
+        this.created.add(key);
         const jobContext = ctx.getJobContext();
         if (!jobContext)
           return;
-        const taskId = `${jobContext.taskId}-${suffix}`;
-        ctx.trackedTasks.set(suffix, {
+        const description = CATEGORY_DESCRIPTIONS[key];
+        const taskId = `${jobContext.taskId}-${key}`;
+        ctx.trackedTasks.set(key, {
           taskId,
           description,
           status: ditto_codegen_types_12.Status.RUNNING
@@ -12366,6 +12401,26 @@ var require_category_handler = __commonJS({
           });
         }
       }
+      async completeTask(key, ctx) {
+        this.completed.add(key);
+        const jobContext = ctx.getJobContext();
+        if (!jobContext)
+          return;
+        const tracked = ctx.trackedTasks.get(key);
+        if (!tracked)
+          return;
+        tracked.status = ditto_codegen_types_12.Status.COMPLETED;
+        try {
+          await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, tracked.taskId, ditto_codegen_types_12.Status.COMPLETED, {});
+          logger_12.logger.info("[OpenCode] Completed category task", {
+            description: tracked.description
+          });
+        } catch (error) {
+          logger_12.logger.error("[OpenCode] Failed to mark category task as completed", {
+            error: error instanceof Error ? error.message : String(error)
+          });
+        }
+      }
     };
     exports2.CategoryTaskHandler = CategoryTaskHandler;
   }
@@ -12436,13 +12491,11 @@ var require_task_tracker = __commonJS({
           await this.handleWriteEvent(tool, state);
         }
         if (tool === "skill" && state.status === "completed") {
-          this.handleSkillEvent(state);
-        }
-        if (this.store?.kind === ditto_codegen_types_12.TaskKind.ASK_CODEGEN) {
-          const ctx = this.handlerContext;
-          await this.categoryHandler.handle(tool, ctx);
-          this.taskCounter = ctx.taskCounter;
+          await this.handleSkillEvent(state);
         }
+        const ctx = this.handlerContext;
+        await this.categoryHandler.handle(tool, ctx, state);
+        this.taskCounter = ctx.taskCounter;
       }
       async handleWriteEvent(tool, state) {
         if (state.status !== "completed")
@@ -12477,6 +12530,7 @@ var require_task_tracker = __commonJS({
         const skillName = state.title?.replace("Loaded skill: ", "") || "unknown";
         this.skillsUsed.add(skillName);
         logger_12.logger.info("[OpenCode] Loaded skill", { skillName });
+        await this.categoryHandler.trackLoadSkill(this.handlerContext, state);
       }
       /** Builds the task completion payload — includes file list for extension tasks. */
       buildTaskPayload(tracked) {
@@ -17984,21 +18038,30 @@ var require_opencode_ask = __commonJS({
     var ditto_codegen_types_2 = require_dist4();
     var context_12 = require_context();
     var logger_12 = require_logger();
+    var config_1 = require_config();
+    var biEvents_1 = require_biEvents();
     var runOpencodeAskFlow = async (chatHistory) => {
       const store = job_context_storage_12.jobContextStorage.getStore();
       if (!store?.jobId || !store?.taskId) {
         throw new Error("Job context not available - jobId and taskId are required");
       }
-      const localJobContext = {
-        jobId: store.jobId,
-        taskId: store.taskId
+      const { jobId, taskId, kind } = store;
+      const localJobContext = { jobId, taskId };
+      const biBaseParams = {
+        Job_id: jobId,
+        Task_id: taskId,
+        app_id: context_12.ctx.projectId,
+        Dev_machine_session_id: process.env.CODEGEN_SESSION_ID,
+        run_kind: kind,
+        job_input: JSON.stringify(chatHistory)
       };
-      logger_12.logger.info("[OpenCode Ask] Starting task", {
-        taskId: localJobContext.taskId
-      });
-      await codeGenerationService_12.codeGenerationService.updateTask(localJobContext.jobId, localJobContext.taskId, ditto_codegen_types_12.Status.RUNNING, {});
-      logger_12.logger.info("[OpenCode Ask] Marked task RUNNING", {
-        taskId: localJobContext.taskId
+      logger_12.logger.info("[OpenCode Ask] Starting task", { taskId });
+      await codeGenerationService_12.codeGenerationService.updateTask(jobId, taskId, ditto_codegen_types_12.Status.RUNNING, {});
+      logger_12.logger.info("[OpenCode Ask] Marked task RUNNING", { taskId });
+      (0, biEvents_1.reportSessionStart)({
+        ...biBaseParams,
+        Task_model: config_1.DEFAULT_MODEL,
+        Task_type: kind
       });
       try {
         const outputPath = (0, codegen_flow_helpers_12.getOutputPath)();
@@ -18009,16 +18072,43 @@ var require_opencode_ask = __commonJS({
           projectId: context_12.ctx.projectId,
           chatHistory
         });
-        await codeGenerationService_12.codeGenerationService.updateTask(localJobContext.jobId, localJobContext.taskId, ditto_codegen_types_12.Status.COMPLETED, {
+        await codeGenerationService_12.codeGenerationService.updateTask(jobId, taskId, ditto_codegen_types_12.Status.COMPLETED, {
           taskOutput: {
             answer: result.answer
           }
         });
-        logger_12.logger.info("[OpenCode Ask] Completed task");
+        (0, biEvents_1.reportSessionEnd)({
+          ...biBaseParams,
+          Task_model: config_1.DEFAULT_MODEL,
+          Task_type: kind,
+          Task_cost: Math.round(result.usage.cost * 1e6),
+          Task_status: ditto_codegen_types_12.Status.COMPLETED,
+          Agent_output: JSON.stringify({ answer: result.answer })
+        });
+        logger_12.logger.info("[OpenCode Ask] Completed task", { taskId });
       } catch (error) {
         const codegenError = (0, ditto_codegen_types_2.toCodegenError)(error);
         await (0, codegen_flow_helpers_12.updateParentTaskStatus)(localJobContext, ditto_codegen_types_12.Status.FAILED, codegenError);
+        (0, biEvents_1.reportSessionEnd)({
+          ...biBaseParams,
+          Task_model: config_1.DEFAULT_MODEL,
+          Task_type: kind,
+          Task_status: ditto_codegen_types_12.Status.FAILED
+        });
+        (0, biEvents_1.reportSessionError)({
+          ...biBaseParams,
+          task_model: config_1.DEFAULT_MODEL,
+          task_type: kind,
+          Agent_output: JSON.stringify(codegenError),
+          Error_desc: codegenError.errorType,
+          error_massage: codegenError.message,
+          error_json: JSON.stringify(codegenError),
+          is_retryable: codegenError.retryable,
+          is_expected: codegenError.expected,
+          validation_type: codegenError.validationType
+        });
         logger_12.logger.error("[OpenCode Ask] Failed task", {
+          taskId,
           error: error instanceof Error ? error.message : String(error)
         });
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.294",
+  "version": "1.0.296",
   "description": "AI-powered Wix CLI app generator - standalone executable",
   "scripts": {
     "build": "node build.mjs",
@@ -27,5 +27,5 @@
     "@wix/ditto-codegen": "1.0.0",
     "esbuild": "^0.27.2"
   },
-  "falconPackageHash": "6125c4c39a2c5d06b6cfdaadc601ecddcaab97b50fc7b8524f2102b8"
+  "falconPackageHash": "70e4d2ee1d465a2ed6701d8d2d7f58175b84d7b45e3afe4c4b7cd351"
 }