mobbdev 1.0.213 → 1.0.216

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

@@ -96,6 +96,7 @@ type Scalars = {
 type AiBlameInferenceFinalizeInput = {
     aiBlameInferenceId: Scalars['String']['input'];
     aiResponseAt: Scalars['Timestamp']['input'];
+    blameType?: InputMaybe<AiBlameInferenceType>;
     inferenceKey: Scalars['String']['input'];
     metadataJSON?: InputMaybe<Scalars['String']['input']>;
     model?: InputMaybe<Scalars['String']['input']>;
@@ -104,12 +105,18 @@ type AiBlameInferenceFinalizeInput = {
 };
 type AiBlameInferenceInitInput = {
     aiResponseAt: Scalars['Timestamp']['input'];
+    blameType?: InputMaybe<AiBlameInferenceType>;
     inferenceFileName: Scalars['String']['input'];
     metadataJSON?: InputMaybe<Scalars['String']['input']>;
     model?: InputMaybe<Scalars['String']['input']>;
     promptFileName: Scalars['String']['input'];
     toolName?: InputMaybe<Scalars['String']['input']>;
 };
+declare enum AiBlameInferenceType {
+    Chat = "CHAT",
+    HumanEdit = "HUMAN_EDIT",
+    TabAutocomplete = "TAB_AUTOCOMPLETE"
+}
 /** Boolean expression to compare columns of type "Boolean". All fields are combined with logical 'AND'. */
 type Boolean_Comparison_Exp = {
     _eq?: InputMaybe<Scalars['Boolean']['input']>;
@@ -3286,6 +3293,25 @@ type GetLastOrgAndNamedProjectQuery = {
         }>;
     }>;
 };
+type GetLastOrgQueryVariables = Exact<{
+    email: Scalars['String']['input'];
+}>;
+type GetLastOrgQuery = {
+    __typename?: 'query_root';
+    user: Array<{
+        __typename?: 'user';
+        id: any;
+        name?: string | null;
+        userOrganizationsAndUserOrganizationRoles: Array<{
+            __typename?: 'organization_to_user';
+            id: any;
+            organization: {
+                __typename?: 'organization';
+                id: any;
+            };
+        }>;
+    }>;
+};
 type GetEncryptedApiTokenQueryVariables = Exact<{
     loginId: Scalars['uuid']['input'];
 }>;
@@ -4189,6 +4215,7 @@ type SdkFunctionWrapper = <T>(action: (requestHeaders?: Record<string, string>)
 declare function getSdk(client: GraphQLClient, withWrapper?: SdkFunctionWrapper): {
     Me(variables?: MeQueryVariables, requestHeaders?: GraphQLClientRequestHeaders, signal?: RequestInit["signal"]): Promise<MeQuery>;
     getLastOrgAndNamedProject(variables: GetLastOrgAndNamedProjectQueryVariables, requestHeaders?: GraphQLClientRequestHeaders, signal?: RequestInit["signal"]): Promise<GetLastOrgAndNamedProjectQuery>;
+    getLastOrg(variables: GetLastOrgQueryVariables, requestHeaders?: GraphQLClientRequestHeaders, signal?: RequestInit["signal"]): Promise<GetLastOrgQuery>;
     GetEncryptedApiToken(variables: GetEncryptedApiTokenQueryVariables, requestHeaders?: GraphQLClientRequestHeaders, signal?: RequestInit["signal"]): Promise<GetEncryptedApiTokenQuery>;
     FixReportState(variables: FixReportStateQueryVariables, requestHeaders?: GraphQLClientRequestHeaders, signal?: RequestInit["signal"]): Promise<FixReportStateQuery>;
     GetVulnerabilityReportPaths(variables: GetVulnerabilityReportPathsQueryVariables, requestHeaders?: GraphQLClientRequestHeaders, signal?: RequestInit["signal"]): Promise<GetVulnerabilityReportPathsQuery>;
@@ -4272,6 +4299,10 @@ declare class GQLClient {
             isTokenAvailable: boolean;
         } | null>;
     } | null | undefined>;
+    getLastOrg(email: string): Promise<{
+        organizationId: any;
+        userName: string;
+    }>;
     createCliLogin(variables: CreateCliLoginMutationVariables): Promise<string>;
     verifyApiConnection(): Promise<boolean>;
     validateUserToken(): Promise<string | boolean>;
@@ -4431,26 +4462,18 @@ declare const PromptItemZ: z.ZodObject<{
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
     }, {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
     type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
-    tool?: {
-        name: string;
-        parameters: string;
-        result: string;
-        accepted?: boolean | undefined;
-        rawArguments?: string | undefined;
-    } | undefined;
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -4459,17 +4482,17 @@ declare const PromptItemZ: z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
-}, {
-    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
+    text?: string | undefined;
     tool?: {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
     } | undefined;
+}, {
+    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -4478,6 +4501,14 @@ declare const PromptItemZ: z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
+    text?: string | undefined;
+    tool?: {
+        name: string;
+        parameters: string;
+        result: string;
+        rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+    } | undefined;
 }>;
 type PromptItem = z.infer<typeof PromptItemZ>;
 declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
@@ -4514,26 +4545,18 @@ declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
     }, {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
     type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
-    tool?: {
-        name: string;
-        parameters: string;
-        result: string;
-        accepted?: boolean | undefined;
-        rawArguments?: string | undefined;
-    } | undefined;
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -4542,17 +4565,17 @@ declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
-}, {
-    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
+    text?: string | undefined;
     tool?: {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
     } | undefined;
+}, {
+    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -4561,6 +4584,14 @@ declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
+    text?: string | undefined;
+    tool?: {
+        name: string;
+        parameters: string;
+        result: string;
+        rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+    } | undefined;
 }>, "many">;
 type PromptItemArray = z.infer<typeof PromptItemArrayZ>;
 type UploadAiBlameOptions = {
@@ -4569,8 +4600,10 @@ type UploadAiBlameOptions = {
     aiResponseAt?: string[];
     model?: string[];
     toolName?: string[];
+    blameType?: AiBlameInferenceType[];
     'ai-response-at'?: string[];
     'tool-name'?: string[];
+    'blame-type'?: AiBlameInferenceType[];
 };
 declare function uploadAiBlameBuilder(args: Yargs.Argv<unknown>): Yargs.Argv<UploadAiBlameOptions>;
 declare function uploadAiBlameHandlerFromExtension(args: {
@@ -4579,6 +4612,7 @@ declare function uploadAiBlameHandlerFromExtension(args: {
     model: string;
     tool: string;
     responseTime: string;
+    blameType?: AiBlameInferenceType;
 }): Promise<void>;
 /**
  * Initializes and authenticates a GQL client for AI Blame upload operations.

package/dist/args/commands/upload_ai_blame.mjs

@@ -112,6 +112,12 @@ import * as dotenv from "dotenv";
 import { z as z8 } from "zod";
 
 // src/features/analysis/scm/generates/client_generates.ts
+var AiBlameInferenceType = /* @__PURE__ */ ((AiBlameInferenceType2) => {
+  AiBlameInferenceType2["Chat"] = "CHAT";
+  AiBlameInferenceType2["HumanEdit"] = "HUMAN_EDIT";
+  AiBlameInferenceType2["TabAutocomplete"] = "TAB_AUTOCOMPLETE";
+  return AiBlameInferenceType2;
+})(AiBlameInferenceType || {});
 var FixQuestionInputType = /* @__PURE__ */ ((FixQuestionInputType2) => {
   FixQuestionInputType2["Number"] = "NUMBER";
   FixQuestionInputType2["Select"] = "SELECT";
@@ -561,6 +567,20 @@ var GetLastOrgAndNamedProjectDocument = `
   }
 }
     `;
+var GetLastOrgDocument = `
+    query getLastOrg($email: String!) {
+  user(where: {email: {_eq: $email}}, limit: 1) {
+    id
+    name
+    userOrganizationsAndUserOrganizationRoles(order_by: {createdOn: desc}) {
+      id
+      organization {
+        id
+      }
+    }
+  }
+}
+    `;
 var GetEncryptedApiTokenDocument = `
     query GetEncryptedApiToken($loginId: uuid!) {
   cli_login_by_pk(id: $loginId) {
@@ -1056,6 +1076,9 @@ function getSdk(client, withWrapper = defaultWrapper) {
     getLastOrgAndNamedProject(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: GetLastOrgAndNamedProjectDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "getLastOrgAndNamedProject", "query", variables);
     },
+    getLastOrg(variables, requestHeaders, signal) {
+      return withWrapper((wrappedRequestHeaders) => client.request({ document: GetLastOrgDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "getLastOrg", "query", variables);
+    },
     GetEncryptedApiToken(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: GetEncryptedApiTokenDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetEncryptedApiToken", "query", variables);
     },
@@ -4609,6 +4632,13 @@ var GQLClient = class {
     const { me } = await this._clientSdk.Me();
     return me;
   }
+  async getLastOrg(email) {
+    const getLastOrgRes = await this._clientSdk.getLastOrg({ email });
+    return {
+      organizationId: getLastOrgRes?.user?.[0]?.userOrganizationsAndUserOrganizationRoles?.[0]?.organization?.id,
+      userName: getLastOrgRes?.user?.[0]?.name ?? ""
+    };
+  }
   async createCliLogin(variables) {
     const res = await this._clientSdk.CreateCliLogin(variables, {
       // We may have outdated API key in the config storage. Avoid using it for the login request.
@@ -5118,6 +5148,13 @@ function uploadAiBlameBuilder(args) {
     type: "string",
     array: true,
     describe: chalk3.bold("Tool/IDE name(s) (optional, one per session)")
+  }).option("blame-type", {
+    type: "string",
+    array: true,
+    choices: Object.values(AiBlameInferenceType),
+    describe: chalk3.bold(
+      "Blame type(s) (optional, one per session, defaults to CHAT)"
+    )
   }).strict();
 }
 async function uploadAiBlameHandlerFromExtension(args) {
@@ -5126,7 +5163,8 @@ async function uploadAiBlameHandlerFromExtension(args) {
     inference: [],
     model: [],
     toolName: [],
-    aiResponseAt: []
+    aiResponseAt: [],
+    blameType: []
   };
   await withFile(async (promptFile) => {
     await fsPromises2.writeFile(
@@ -5141,6 +5179,7 @@ async function uploadAiBlameHandlerFromExtension(args) {
       uploadArgs.model.push(args.model);
       uploadArgs.toolName.push(args.tool);
       uploadArgs.aiResponseAt.push(args.responseTime);
+      uploadArgs.blameType.push(args.blameType || "CHAT" /* Chat */);
       await uploadAiBlameHandler(uploadArgs, false);
     });
   });
@@ -5163,6 +5202,7 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
   const models = args.model || [];
   const tools = args.toolName || args["tool-name"] || [];
   const responseTimes = args.aiResponseAt || args["ai-response-at"] || [];
+  const blameTypes = args.blameType || args["blame-type"] || [];
   if (prompts.length !== inferences.length) {
     const errorMsg = "prompt and inference must have the same number of entries";
     console.error(chalk3.red(errorMsg));
@@ -5194,7 +5234,8 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
       inferenceFileName: path6.basename(inferencePath),
       aiResponseAt: responseTimes[i] || nowIso,
       model: models[i],
-      toolName: tools[i]
+      toolName: tools[i],
+      blameType: blameTypes[i] || "CHAT" /* Chat */
     });
   }
   const authenticatedClient = await getAuthenticatedGQLClientForIdeExtension();
@@ -5214,18 +5255,22 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
     const us = uploadSessions[i];
     const promptPath = String(prompts[i]);
     const inferencePath = String(inferences[i]);
-    await uploadFile({
-      file: promptPath,
-      url: us.prompt.url,
-      uploadFields: JSON.parse(us.prompt.uploadFieldsJSON),
-      uploadKey: us.prompt.uploadKey
-    });
-    await uploadFile({
-      file: inferencePath,
-      url: us.inference.url,
-      uploadFields: JSON.parse(us.inference.uploadFieldsJSON),
-      uploadKey: us.inference.uploadKey
-    });
+    await Promise.all([
+      // Prompt
+      uploadFile({
+        file: promptPath,
+        url: us.prompt.url,
+        uploadFields: JSON.parse(us.prompt.uploadFieldsJSON),
+        uploadKey: us.prompt.uploadKey
+      }),
+      // Inference
+      uploadFile({
+        file: inferencePath,
+        url: us.inference.url,
+        uploadFields: JSON.parse(us.inference.uploadFieldsJSON),
+        uploadKey: us.inference.uploadKey
+      })
+    ]);
   }
   const finalizeSessions = uploadSessions.map((us, i) => {
     const s = sessions[i];
@@ -5235,7 +5280,8 @@ async function uploadAiBlameHandler(args, exitOnError = true) {
       inferenceKey: us.inference.uploadKey,
       aiResponseAt: s.aiResponseAt,
       model: s.model,
-      toolName: s.toolName
+      toolName: s.toolName,
+      blameType: s.blameType
     };
   });
   const finRes = await authenticatedClient.finalizeAIBlameInferencesUploadRaw({