mobbdev 1.1.27 → 1.1.29

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

@@ -16,6 +16,10 @@ type SanitizationCounts = {
     };
 };
 
+type Logger = {
+    info: (msg: string, data?: unknown) => void;
+    error: (msg: string, data?: unknown) => void;
+};
 declare const PromptItemZ: z.ZodObject<{
     type: z.ZodEnum<["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING"]>;
     attachedFiles: z.ZodOptional<z.ZodArray<z.ZodObject<{
@@ -210,7 +214,17 @@ declare function uploadAiBlameHandlerFromExtension(args: {
     responseTime: string;
     blameType?: AiBlameInferenceType;
     sessionId?: string;
+    apiUrl?: string;
+    webAppUrl?: string;
 }): Promise<UploadAiBlameResult>;
-declare function uploadAiBlameHandler(args: UploadAiBlameOptions, exitOnError?: boolean): Promise<void>;
+type UploadAiBlameHandlerOptions = {
+    args: UploadAiBlameOptions;
+    exitOnError?: boolean;
+    apiUrl?: string;
+    webAppUrl?: string;
+    logger?: Logger;
+};
+declare function uploadAiBlameHandler(options: UploadAiBlameHandlerOptions): Promise<void>;
+declare function uploadAiBlameCommandHandler(args: UploadAiBlameOptions): Promise<void>;
 
-export { type PromptItem, type PromptItemArray, type UploadAiBlameOptions, type UploadAiBlameResult, uploadAiBlameBuilder, uploadAiBlameHandler, uploadAiBlameHandlerFromExtension };
+export { type PromptItem, type PromptItemArray, type UploadAiBlameOptions, type UploadAiBlameResult, uploadAiBlameBuilder, uploadAiBlameCommandHandler, uploadAiBlameHandler, uploadAiBlameHandlerFromExtension };

package/dist/args/commands/upload_ai_blame.mjs

@@ -80,11 +80,13 @@ import fs4 from "fs";
 import ignore from "ignore";
 import * as path5 from "path";
 import { simpleGit as simpleGit2 } from "simple-git";
+var MAX_COMMIT_DIFF_SIZE_BYTES;
 var init_GitService = __esm({
   "src/features/analysis/scm/services/GitService.ts"() {
     "use strict";
     init_configs();
     init_FileUtils();
+    MAX_COMMIT_DIFF_SIZE_BYTES = 3 * 1024 * 1024;
   }
 });
 
@@ -817,11 +819,14 @@ var UploadS3BucketInfoDocument = `
 }
     `;
 var AnalyzeCommitForExtensionAiBlameDocument = `
-    mutation AnalyzeCommitForExtensionAIBlame($repositoryURL: String!, $commitSha: String!, $organizationId: String!) {
+    mutation AnalyzeCommitForExtensionAIBlame($repositoryURL: String!, $commitSha: String!, $organizationId: String!, $commitDiff: String, $commitTimestamp: Timestamp, $parentCommits: [ParentCommitInput!]) {
   analyzeCommitForAIBlame(
     repositoryURL: $repositoryURL
     commitSha: $commitSha
     organizationId: $organizationId
+    commitDiff: $commitDiff
+    commitTimestamp: $commitTimestamp
+    parentCommits: $parentCommits
   ) {
     __typename
     ... on ProcessAIBlameFinalResult {
@@ -867,6 +872,30 @@ var GetAiBlameAttributionPromptDocument = `
   }
 }
     `;
+var GetPromptSummaryDocument = `
+    query GetPromptSummary($aiBlameAttributionId: String!) {
+  getPromptSummary(aiBlameAttributionId: $aiBlameAttributionId) {
+    __typename
+    ... on PromptSummarySuccess {
+      status
+      summary {
+        goal
+        developersPlan
+        aiImplementationDetails
+        importantInstructionsAndPushbacks
+        frictionScore {
+          score
+          justification
+        }
+      }
+    }
+    ... on PromptSummaryError {
+      status
+      error
+    }
+  }
+}
+    `;
 var UploadAiBlameInferencesInitDocument = `
     mutation UploadAIBlameInferencesInit($sessions: [AIBlameInferenceInitInput!]!) {
   uploadAIBlameInferencesInit(sessions: $sessions) {
@@ -1194,6 +1223,9 @@ function getSdk(client, withWrapper = defaultWrapper) {
     GetAIBlameAttributionPrompt(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: GetAiBlameAttributionPromptDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetAIBlameAttributionPrompt", "query", variables);
     },
+    GetPromptSummary(variables, requestHeaders, signal) {
+      return withWrapper((wrappedRequestHeaders) => client.request({ document: GetPromptSummaryDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetPromptSummary", "query", variables);
+    },
     UploadAIBlameInferencesInit(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: UploadAiBlameInferencesInitDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "UploadAIBlameInferencesInit", "mutation", variables);
     },
@@ -4734,6 +4766,11 @@ var GQLClient = class {
     debug6(`init with  ${args}`);
     this._auth = args;
     this._apiUrl = args.apiUrl || API_URL;
+    debug6(
+      "GQLClient constructor: resolved apiUrl=%s (from param: %s)",
+      this._apiUrl,
+      args.apiUrl || "fallback to API_URL constant"
+    );
     this._client = new GraphQLClient(this._apiUrl, {
       headers: args.type === "apiKey" ? { [API_KEY_HEADER_NAME]: args.apiKey || "" } : {
         Authorization: `Bearer ${args.token}`
@@ -5098,6 +5135,9 @@ var GQLClient = class {
   async getAIBlameAttributionPrompt(variables) {
     return await this._clientSdk.GetAIBlameAttributionPrompt(variables);
   }
+  async getAIBlameAttributionPromptSummary(variables) {
+    return await this._clientSdk.GetPromptSummary(variables);
+  }
 };
 
 // src/mcp/services/types.ts
@@ -5144,6 +5184,11 @@ async function getAuthenticatedGQLClient({
   apiUrl,
   webAppUrl
 }) {
+  debug7(
+    "getAuthenticatedGQLClient called with: apiUrl=%s, webAppUrl=%s",
+    apiUrl || "undefined",
+    webAppUrl || "undefined"
+  );
   let gqlClient = new GQLClient({
     apiKey: inputApiKey || configStore.get("apiToken") || "",
     type: "apiKey",
@@ -5167,6 +5212,13 @@ async function handleMobbLogin({
 }) {
   const resolvedWebAppUrl = webAppUrl || WEB_APP_URL;
   const resolvedApiUrl = apiUrl || API_URL;
+  debug7(
+    "handleMobbLogin: resolved URLs - apiUrl=%s (from param: %s), webAppUrl=%s (from param: %s)",
+    resolvedApiUrl,
+    apiUrl || "fallback",
+    resolvedWebAppUrl,
+    webAppUrl || "fallback"
+  );
   const { createSpinner } = Spinner({ ci: skipPrompts });
   const isConnected = await inGqlClient.verifyApiConnection();
   if (!isConnected) {
@@ -5447,6 +5499,22 @@ async function sanitizeDataWithCounts(obj) {
 }
 
 // src/args/commands/upload_ai_blame.ts
+var defaultLogger = {
+  info: (msg, data) => {
+    if (data !== void 0) {
+      console.log(msg, data);
+    } else {
+      console.log(msg);
+    }
+  },
+  error: (msg, data) => {
+    if (data !== void 0) {
+      console.error(msg, data);
+    } else {
+      console.error(msg);
+    }
+  }
+};
 var PromptItemZ = z26.object({
   type: z26.enum(["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING"]),
   attachedFiles: z26.array(
@@ -5562,7 +5630,12 @@ async function uploadAiBlameHandlerFromExtension(args) {
       if (args.sessionId) {
         uploadArgs.sessionId.push(args.sessionId);
       }
-      await uploadAiBlameHandler(uploadArgs, false);
+      await uploadAiBlameHandler({
+        args: uploadArgs,
+        exitOnError: false,
+        apiUrl: args.apiUrl,
+        webAppUrl: args.webAppUrl
+      });
     });
   });
   return {
@@ -5572,7 +5645,14 @@ async function uploadAiBlameHandlerFromExtension(args) {
     inferenceUUID
   };
 }
-async function uploadAiBlameHandler(args, exitOnError = true) {
+async function uploadAiBlameHandler(options) {
+  const {
+    args,
+    exitOnError = true,
+    apiUrl,
+    webAppUrl,
+    logger = defaultLogger
+  } = options;
   const prompts = args.prompt || [];
   const inferences = args.inference || [];
   const models = args.model || [];
@@ -5582,7 +5662,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
   const sessionIds = args.sessionId || args["session-id"] || [];
   if (prompts.length !== inferences.length) {
     const errorMsg = "prompt and inference must have the same number of entries";
-    console.error(chalk3.red(errorMsg));
+    logger.error(chalk3.red(errorMsg));
     if (exitOnError) {
       process.exit(1);
     }
@@ -5601,7 +5681,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
       ]);
     } catch {
       const errorMsg = `File not found for session ${i + 1}`;
-      console.error(chalk3.red(errorMsg));
+      logger.error(chalk3.red(errorMsg));
       if (exitOnError) {
         process.exit(1);
       }
@@ -5620,7 +5700,9 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
     });
   }
   const authenticatedClient = await getAuthenticatedGQLClient({
-    isSkipPrompts: true
+    isSkipPrompts: true,
+    apiUrl,
+    webAppUrl
   });
   const initSessions = sessions.map(
     ({ sessionId: _sessionId, ...rest }) => rest
@@ -5631,7 +5713,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
   const uploadSessions = initRes.uploadAIBlameInferencesInit?.uploadSessions ?? [];
   if (uploadSessions.length !== sessions.length) {
     const errorMsg = "Init failed to return expected number of sessions";
-    console.error(chalk3.red(errorMsg));
+    logger.error(chalk3.red(errorMsg));
     if (exitOnError) {
       process.exit(1);
     }
@@ -5673,22 +5755,41 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
       sessionId: s.sessionId
     };
   });
-  const finRes = await authenticatedClient.finalizeAIBlameInferencesUploadRaw({
-    sessions: finalizeSessions
-  });
-  const status = finRes?.finalizeAIBlameInferencesUpload?.status;
-  if (status !== "OK") {
-    const errorMsg = finRes?.finalizeAIBlameInferencesUpload?.error || "unknown error";
-    console.error(chalk3.red(errorMsg));
-    if (exitOnError) {
-      process.exit(1);
+  try {
+    logger.info(
+      `[UPLOAD] Calling finalizeAIBlameInferencesUploadRaw with ${finalizeSessions.length} sessions`
+    );
+    const finRes = await authenticatedClient.finalizeAIBlameInferencesUploadRaw(
+      {
+        sessions: finalizeSessions
+      }
+    );
+    logger.info("[UPLOAD] Finalize response:", JSON.stringify(finRes, null, 2));
+    const status = finRes?.finalizeAIBlameInferencesUpload?.status;
+    if (status !== "OK") {
+      const errorMsg = finRes?.finalizeAIBlameInferencesUpload?.error || "unknown error";
+      logger.error(
+        chalk3.red(
+          `[UPLOAD] Finalize failed with status: ${status}, error: ${errorMsg}`
+        )
+      );
+      if (exitOnError) {
+        process.exit(1);
+      }
+      throw new Error(errorMsg);
     }
-    throw new Error(errorMsg);
+    logger.info(chalk3.green("[UPLOAD] AI Blame uploads finalized successfully"));
+  } catch (error) {
+    logger.error("[UPLOAD] Finalize threw error:", error);
+    throw error;
   }
-  console.log(chalk3.green("AI Blame uploads finalized successfully"));
+}
+async function uploadAiBlameCommandHandler(args) {
+  await uploadAiBlameHandler({ args });
 }
 export {
   uploadAiBlameBuilder,
+  uploadAiBlameCommandHandler,
   uploadAiBlameHandler,
   uploadAiBlameHandlerFromExtension
 };

package/dist/index.mjs

@@ -723,12 +723,13 @@ import fs2 from "fs";
 import ignore from "ignore";
 import * as path2 from "path";
 import { simpleGit } from "simple-git";
-var GitService;
+var MAX_COMMIT_DIFF_SIZE_BYTES, GitService;
 var init_GitService = __esm({
   "src/features/analysis/scm/services/GitService.ts"() {
     "use strict";
     init_configs();
     init_FileUtils();
+    MAX_COMMIT_DIFF_SIZE_BYTES = 3 * 1024 * 1024;
     GitService = class {
       constructor(repositoryPath, log2) {
         __publicField(this, "git");
@@ -1307,6 +1308,97 @@ ${rootContent}`;
           throw new Error(errorMessage);
         }
       }
+      /**
+       * Gets timestamps for parent commits in a single git call.
+       * @param parentShas Array of parent commit SHAs
+       * @returns Array of parent commits with timestamps, or undefined if unavailable
+       */
+      async getParentCommitTimestamps(parentShas) {
+        if (parentShas.length === 0) {
+          return void 0;
+        }
+        try {
+          const output = await this.git.raw([
+            "log",
+            "--format=%H %cI",
+            "--no-walk",
+            ...parentShas
+          ]);
+          const parentCommits = output.trim().split("\n").filter(Boolean).map((line) => {
+            const [sha, ts] = line.split(" ");
+            return { sha: sha ?? "", timestamp: new Date(ts ?? "") };
+          }).filter((p) => p.sha !== "");
+          return parentCommits.length > 0 ? parentCommits : void 0;
+        } catch {
+          this.log("[GitService] Could not get parent commit timestamps", "debug", {
+            parentShas
+          });
+          return void 0;
+        }
+      }
+      /**
+       * Gets local commit data including diff, timestamp, and parent commits.
+       * Used by Tracy extension to send commit data directly without requiring SCM token.
+       * @param commitSha The commit SHA to get data for
+       * @param maxDiffSizeBytes Maximum diff size in bytes (default 3MB). Returns null if exceeded.
+       * @returns Commit data or null if unavailable/too large
+       */
+      async getLocalCommitData(commitSha, maxDiffSizeBytes = MAX_COMMIT_DIFF_SIZE_BYTES) {
+        this.log("[GitService] Getting local commit data", "debug", { commitSha });
+        try {
+          const DIFF_DELIMITER = "---MOBB_DIFF_START---";
+          const output = await this.git.show([
+            commitSha,
+            `--format=%cI%n%P%n${DIFF_DELIMITER}`,
+            "--patch"
+          ]);
+          const delimiterIndex = output.indexOf(DIFF_DELIMITER);
+          if (delimiterIndex === -1) {
+            this.log("[GitService] Could not parse git show output", "warning", {
+              commitSha
+            });
+            return null;
+          }
+          const metadataOutput = output.substring(0, delimiterIndex);
+          const diff = output.substring(delimiterIndex + DIFF_DELIMITER.length + 1);
+          const diffSizeBytes = Buffer.byteLength(diff, "utf8");
+          if (diffSizeBytes > maxDiffSizeBytes) {
+            this.log("[GitService] Commit diff exceeds size limit", "warning", {
+              commitSha,
+              diffSizeBytes,
+              maxDiffSizeBytes
+            });
+            return null;
+          }
+          const metadataLines = metadataOutput.trim().split("\n");
+          if (metadataLines.length < 1 || !metadataLines[0]) {
+            this.log("[GitService] Unexpected metadata format", "warning", {
+              commitSha,
+              metadataLines
+            });
+            return null;
+          }
+          const timestampStr = metadataLines[0];
+          const timestamp = new Date(timestampStr);
+          const parentShas = (metadataLines[1] ?? "").trim().split(/\s+/).filter(Boolean);
+          const parentCommits = await this.getParentCommitTimestamps(parentShas);
+          this.log("[GitService] Local commit data retrieved", "debug", {
+            commitSha,
+            diffSizeBytes,
+            timestamp: timestamp.toISOString(),
+            parentCommitCount: parentCommits?.length ?? 0
+          });
+          return {
+            diff,
+            timestamp,
+            parentCommits
+          };
+        } catch (error) {
+          const errorMessage = `Failed to get local commit data: ${error.message}`;
+          this.log(`[GitService] ${errorMessage}`, "debug", { error, commitSha });
+          return null;
+        }
+      }
     };
   }
 });
@@ -2070,11 +2162,14 @@ var UploadS3BucketInfoDocument = `
 }
     `;
 var AnalyzeCommitForExtensionAiBlameDocument = `
-    mutation AnalyzeCommitForExtensionAIBlame($repositoryURL: String!, $commitSha: String!, $organizationId: String!) {
+    mutation AnalyzeCommitForExtensionAIBlame($repositoryURL: String!, $commitSha: String!, $organizationId: String!, $commitDiff: String, $commitTimestamp: Timestamp, $parentCommits: [ParentCommitInput!]) {
   analyzeCommitForAIBlame(
     repositoryURL: $repositoryURL
     commitSha: $commitSha
     organizationId: $organizationId
+    commitDiff: $commitDiff
+    commitTimestamp: $commitTimestamp
+    parentCommits: $parentCommits
   ) {
     __typename
     ... on ProcessAIBlameFinalResult {
@@ -2120,6 +2215,30 @@ var GetAiBlameAttributionPromptDocument = `
   }
 }
     `;
+var GetPromptSummaryDocument = `
+    query GetPromptSummary($aiBlameAttributionId: String!) {
+  getPromptSummary(aiBlameAttributionId: $aiBlameAttributionId) {
+    __typename
+    ... on PromptSummarySuccess {
+      status
+      summary {
+        goal
+        developersPlan
+        aiImplementationDetails
+        importantInstructionsAndPushbacks
+        frictionScore {
+          score
+          justification
+        }
+      }
+    }
+    ... on PromptSummaryError {
+      status
+      error
+    }
+  }
+}
+    `;
 var UploadAiBlameInferencesInitDocument = `
     mutation UploadAIBlameInferencesInit($sessions: [AIBlameInferenceInitInput!]!) {
   uploadAIBlameInferencesInit(sessions: $sessions) {
@@ -2447,6 +2566,9 @@ function getSdk(client, withWrapper = defaultWrapper) {
     GetAIBlameAttributionPrompt(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: GetAiBlameAttributionPromptDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetAIBlameAttributionPrompt", "query", variables);
     },
+    GetPromptSummary(variables, requestHeaders, signal) {
+      return withWrapper((wrappedRequestHeaders) => client.request({ document: GetPromptSummaryDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetPromptSummary", "query", variables);
+    },
     UploadAIBlameInferencesInit(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: UploadAiBlameInferencesInitDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "UploadAIBlameInferencesInit", "mutation", variables);
     },
@@ -11034,6 +11156,11 @@ var GQLClient = class {
     debug6(`init with  ${args}`);
     this._auth = args;
     this._apiUrl = args.apiUrl || API_URL;
+    debug6(
+      "GQLClient constructor: resolved apiUrl=%s (from param: %s)",
+      this._apiUrl,
+      args.apiUrl || "fallback to API_URL constant"
+    );
     this._client = new GraphQLClient(this._apiUrl, {
       headers: args.type === "apiKey" ? { [API_KEY_HEADER_NAME]: args.apiKey || "" } : {
         Authorization: `Bearer ${args.token}`
@@ -11398,6 +11525,9 @@ var GQLClient = class {
   async getAIBlameAttributionPrompt(variables) {
     return await this._clientSdk.GetAIBlameAttributionPrompt(variables);
   }
+  async getAIBlameAttributionPromptSummary(variables) {
+    return await this._clientSdk.GetPromptSummary(variables);
+  }
 };
 
 // src/mcp/services/types.ts
@@ -11464,6 +11594,11 @@ async function getAuthenticatedGQLClient({
   apiUrl,
   webAppUrl
 }) {
+  debug7(
+    "getAuthenticatedGQLClient called with: apiUrl=%s, webAppUrl=%s",
+    apiUrl || "undefined",
+    webAppUrl || "undefined"
+  );
   let gqlClient = new GQLClient({
     apiKey: inputApiKey || configStore.get("apiToken") || "",
     type: "apiKey",
@@ -11487,6 +11622,13 @@ async function handleMobbLogin({
 }) {
   const resolvedWebAppUrl = webAppUrl || WEB_APP_URL;
   const resolvedApiUrl = apiUrl || API_URL;
+  debug7(
+    "handleMobbLogin: resolved URLs - apiUrl=%s (from param: %s), webAppUrl=%s (from param: %s)",
+    resolvedApiUrl,
+    apiUrl || "fallback",
+    resolvedWebAppUrl,
+    webAppUrl || "fallback"
+  );
   const { createSpinner: createSpinner5 } = Spinner({ ci: skipPrompts });
   const isConnected = await inGqlClient.verifyApiConnection();
   if (!isConnected) {
@@ -13801,6 +13943,22 @@ async function sanitizeDataWithCounts(obj) {
 }
 
 // src/args/commands/upload_ai_blame.ts
+var defaultLogger2 = {
+  info: (msg, data) => {
+    if (data !== void 0) {
+      console.log(msg, data);
+    } else {
+      console.log(msg);
+    }
+  },
+  error: (msg, data) => {
+    if (data !== void 0) {
+      console.error(msg, data);
+    } else {
+      console.error(msg);
+    }
+  }
+};
 var PromptItemZ = z31.object({
   type: z31.enum(["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING"]),
   attachedFiles: z31.array(
@@ -13916,7 +14074,12 @@ async function uploadAiBlameHandlerFromExtension(args) {
       if (args.sessionId) {
         uploadArgs.sessionId.push(args.sessionId);
       }
-      await uploadAiBlameHandler(uploadArgs, false);
+      await uploadAiBlameHandler({
+        args: uploadArgs,
+        exitOnError: false,
+        apiUrl: args.apiUrl,
+        webAppUrl: args.webAppUrl
+      });
     });
   });
   return {
@@ -13926,7 +14089,14 @@ async function uploadAiBlameHandlerFromExtension(args) {
     inferenceUUID
   };
 }
-async function uploadAiBlameHandler(args, exitOnError = true) {
+async function uploadAiBlameHandler(options) {
+  const {
+    args,
+    exitOnError = true,
+    apiUrl,
+    webAppUrl,
+    logger: logger2 = defaultLogger2
+  } = options;
   const prompts = args.prompt || [];
   const inferences = args.inference || [];
   const models = args.model || [];
@@ -13936,7 +14106,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
   const sessionIds = args.sessionId || args["session-id"] || [];
   if (prompts.length !== inferences.length) {
     const errorMsg = "prompt and inference must have the same number of entries";
-    console.error(chalk9.red(errorMsg));
+    logger2.error(chalk9.red(errorMsg));
     if (exitOnError) {
       process.exit(1);
     }
@@ -13955,7 +14125,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
       ]);
     } catch {
       const errorMsg = `File not found for session ${i + 1}`;
-      console.error(chalk9.red(errorMsg));
+      logger2.error(chalk9.red(errorMsg));
       if (exitOnError) {
         process.exit(1);
       }
@@ -13974,7 +14144,9 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
     });
   }
   const authenticatedClient = await getAuthenticatedGQLClient({
-    isSkipPrompts: true
+    isSkipPrompts: true,
+    apiUrl,
+    webAppUrl
   });
   const initSessions = sessions.map(
     ({ sessionId: _sessionId, ...rest }) => rest
@@ -13985,7 +14157,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
   const uploadSessions = initRes.uploadAIBlameInferencesInit?.uploadSessions ?? [];
   if (uploadSessions.length !== sessions.length) {
     const errorMsg = "Init failed to return expected number of sessions";
-    console.error(chalk9.red(errorMsg));
+    logger2.error(chalk9.red(errorMsg));
     if (exitOnError) {
       process.exit(1);
     }
@@ -14027,19 +14199,37 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
       sessionId: s.sessionId
     };
   });
-  const finRes = await authenticatedClient.finalizeAIBlameInferencesUploadRaw({
-    sessions: finalizeSessions
-  });
-  const status = finRes?.finalizeAIBlameInferencesUpload?.status;
-  if (status !== "OK") {
-    const errorMsg = finRes?.finalizeAIBlameInferencesUpload?.error || "unknown error";
-    console.error(chalk9.red(errorMsg));
-    if (exitOnError) {
-      process.exit(1);
+  try {
+    logger2.info(
+      `[UPLOAD] Calling finalizeAIBlameInferencesUploadRaw with ${finalizeSessions.length} sessions`
+    );
+    const finRes = await authenticatedClient.finalizeAIBlameInferencesUploadRaw(
+      {
+        sessions: finalizeSessions
+      }
+    );
+    logger2.info("[UPLOAD] Finalize response:", JSON.stringify(finRes, null, 2));
+    const status = finRes?.finalizeAIBlameInferencesUpload?.status;
+    if (status !== "OK") {
+      const errorMsg = finRes?.finalizeAIBlameInferencesUpload?.error || "unknown error";
+      logger2.error(
+        chalk9.red(
+          `[UPLOAD] Finalize failed with status: ${status}, error: ${errorMsg}`
+        )
+      );
+      if (exitOnError) {
+        process.exit(1);
+      }
+      throw new Error(errorMsg);
     }
-    throw new Error(errorMsg);
+    logger2.info(chalk9.green("[UPLOAD] AI Blame uploads finalized successfully"));
+  } catch (error) {
+    logger2.error("[UPLOAD] Finalize threw error:", error);
+    throw error;
   }
-  console.log(chalk9.green("AI Blame uploads finalized successfully"));
+}
+async function uploadAiBlameCommandHandler(args) {
+  await uploadAiBlameHandler({ args });
 }
 
 // src/features/claude_code/transcript_parser.ts
@@ -22792,7 +22982,7 @@ var parseArgs = async (args) => {
       "Upload AI Blame inference artifacts (prompt + inference) and finalize them."
     ),
     uploadAiBlameBuilder,
-    uploadAiBlameHandler
+    uploadAiBlameCommandHandler
   ).command(
     mobbCliCommand.claudeCodeInstallHook,
     chalk12.bold("Install Claude Code hooks for data collection."),

package/package.json

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