mobbdev 1.2.53 → 1.2.54

package/dist/args/commands/upload_ai_blame.d.mts

@@ -239,6 +239,7 @@ type UploadAiBlameResult = {
     inferenceCounts: SanitizationCounts;
     promptsUUID?: string;
     inferenceUUID?: string;
+    sanitizationDurationMs?: number;
 };
 declare function uploadAiBlameHandlerFromExtension(args: {
     prompts: PromptItemArray;
@@ -251,6 +252,7 @@ declare function uploadAiBlameHandlerFromExtension(args: {
     apiUrl?: string;
     webAppUrl?: string;
     repositoryUrl?: string | null;
+    sanitize?: boolean;
 }): Promise<UploadAiBlameResult>;
 type UploadAiBlameHandlerOptions = {
     args: UploadAiBlameOptions;

package/dist/args/commands/upload_ai_blame.mjs

@@ -7938,6 +7938,7 @@ function uploadAiBlameBuilder(args) {
   }).strict();
 }
 async function uploadAiBlameHandlerFromExtension(args) {
+  const shouldSanitize = args.sanitize ?? true;
   const uploadArgs = {
     prompt: [],
     inference: [],
@@ -7951,28 +7952,50 @@ async function uploadAiBlameHandlerFromExtension(args) {
   let inferenceCounts;
   let promptsUUID;
   let inferenceUUID;
+  const zeroCounts = {
+    detections: { total: 0, high: 0, medium: 0, low: 0 }
+  };
+  let sanitizationDurationMs;
   await withFile(async (promptFile) => {
-    const promptsResult = await sanitizeDataWithCounts(args.prompts);
-    promptsCounts = promptsResult.counts;
     promptsUUID = path6.basename(promptFile.path, path6.extname(promptFile.path));
-    await fsPromises2.writeFile(
-      promptFile.path,
-      JSON.stringify(promptsResult.sanitizedData, null, 2),
-      "utf-8"
-    );
+    if (shouldSanitize) {
+      const sanitizeStart = performance.now();
+      const promptsResult = await sanitizeDataWithCounts(args.prompts);
+      promptsCounts = promptsResult.counts;
+      await fsPromises2.writeFile(
+        promptFile.path,
+        JSON.stringify(promptsResult.sanitizedData, null, 2),
+        "utf-8"
+      );
+      sanitizationDurationMs = performance.now() - sanitizeStart;
+    } else {
+      promptsCounts = zeroCounts;
+      await fsPromises2.writeFile(
+        promptFile.path,
+        JSON.stringify(args.prompts, null, 2),
+        "utf-8"
+      );
+    }
     uploadArgs.prompt.push(promptFile.path);
     await withFile(async (inferenceFile) => {
-      const inferenceResult = await sanitizeDataWithCounts(args.inference);
-      inferenceCounts = inferenceResult.counts;
       inferenceUUID = path6.basename(
         inferenceFile.path,
         path6.extname(inferenceFile.path)
       );
-      await fsPromises2.writeFile(
-        inferenceFile.path,
-        inferenceResult.sanitizedData,
-        "utf-8"
-      );
+      if (shouldSanitize) {
+        const inferenceStart = performance.now();
+        const inferenceResult = await sanitizeDataWithCounts(args.inference);
+        inferenceCounts = inferenceResult.counts;
+        await fsPromises2.writeFile(
+          inferenceFile.path,
+          inferenceResult.sanitizedData,
+          "utf-8"
+        );
+        sanitizationDurationMs = (sanitizationDurationMs ?? 0) + (performance.now() - inferenceStart);
+      } else {
+        inferenceCounts = zeroCounts;
+        await fsPromises2.writeFile(inferenceFile.path, args.inference, "utf-8");
+      }
       uploadArgs.inference.push(inferenceFile.path);
       uploadArgs.model.push(args.model);
       uploadArgs.toolName.push(args.tool);
@@ -7994,7 +8017,8 @@ async function uploadAiBlameHandlerFromExtension(args) {
     promptsCounts,
     inferenceCounts,
     promptsUUID,
-    inferenceUUID
+    inferenceUUID,
+    sanitizationDurationMs
   };
 }
 async function uploadAiBlameHandler(options) {

package/dist/index.mjs

@@ -13421,6 +13421,7 @@ function uploadAiBlameBuilder(args) {
   }).strict();
 }
 async function uploadAiBlameHandlerFromExtension(args) {
+  const shouldSanitize = args.sanitize ?? true;
   const uploadArgs = {
     prompt: [],
     inference: [],
@@ -13434,28 +13435,50 @@ async function uploadAiBlameHandlerFromExtension(args) {
   let inferenceCounts;
   let promptsUUID;
   let inferenceUUID;
+  const zeroCounts = {
+    detections: { total: 0, high: 0, medium: 0, low: 0 }
+  };
+  let sanitizationDurationMs;
   await withFile(async (promptFile) => {
-    const promptsResult = await sanitizeDataWithCounts(args.prompts);
-    promptsCounts = promptsResult.counts;
     promptsUUID = path7.basename(promptFile.path, path7.extname(promptFile.path));
-    await fsPromises2.writeFile(
-      promptFile.path,
-      JSON.stringify(promptsResult.sanitizedData, null, 2),
-      "utf-8"
-    );
+    if (shouldSanitize) {
+      const sanitizeStart = performance.now();
+      const promptsResult = await sanitizeDataWithCounts(args.prompts);
+      promptsCounts = promptsResult.counts;
+      await fsPromises2.writeFile(
+        promptFile.path,
+        JSON.stringify(promptsResult.sanitizedData, null, 2),
+        "utf-8"
+      );
+      sanitizationDurationMs = performance.now() - sanitizeStart;
+    } else {
+      promptsCounts = zeroCounts;
+      await fsPromises2.writeFile(
+        promptFile.path,
+        JSON.stringify(args.prompts, null, 2),
+        "utf-8"
+      );
+    }
     uploadArgs.prompt.push(promptFile.path);
     await withFile(async (inferenceFile) => {
-      const inferenceResult = await sanitizeDataWithCounts(args.inference);
-      inferenceCounts = inferenceResult.counts;
       inferenceUUID = path7.basename(
         inferenceFile.path,
         path7.extname(inferenceFile.path)
       );
-      await fsPromises2.writeFile(
-        inferenceFile.path,
-        inferenceResult.sanitizedData,
-        "utf-8"
-      );
+      if (shouldSanitize) {
+        const inferenceStart = performance.now();
+        const inferenceResult = await sanitizeDataWithCounts(args.inference);
+        inferenceCounts = inferenceResult.counts;
+        await fsPromises2.writeFile(
+          inferenceFile.path,
+          inferenceResult.sanitizedData,
+          "utf-8"
+        );
+        sanitizationDurationMs = (sanitizationDurationMs ?? 0) + (performance.now() - inferenceStart);
+      } else {
+        inferenceCounts = zeroCounts;
+        await fsPromises2.writeFile(inferenceFile.path, args.inference, "utf-8");
+      }
       uploadArgs.inference.push(inferenceFile.path);
       uploadArgs.model.push(args.model);
       uploadArgs.toolName.push(args.tool);
@@ -13477,7 +13500,8 @@ async function uploadAiBlameHandlerFromExtension(args) {
     promptsCounts,
     inferenceCounts,
     promptsUUID,
-    inferenceUUID
+    inferenceUUID,
+    sanitizationDurationMs
   };
 }
 async function uploadAiBlameHandler(options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mobbdev",
-  "version": "1.2.53",
+  "version": "1.2.54",
   "description": "Automated secure code remediation tool",
   "repository": "git+https://github.com/mobb-dev/bugsy.git",
   "main": "dist/index.mjs",