@locusai/cli 0.14.4 → 0.15.0

package/bin/agent/worker.js

@@ -14343,6 +14343,32 @@ var init_auth = __esm(() => {
   });
 });
 
+// ../shared/src/models/autonomy.ts
+var RiskLevel, ChangeCategory, AutonomyRuleSchema;
+var init_autonomy = __esm(() => {
+  init_zod();
+  ((RiskLevel2) => {
+    RiskLevel2["LOW"] = "LOW";
+    RiskLevel2["HIGH"] = "HIGH";
+  })(RiskLevel ||= {});
+  ((ChangeCategory2) => {
+    ChangeCategory2["FIX"] = "FIX";
+    ChangeCategory2["REFACTOR"] = "REFACTOR";
+    ChangeCategory2["STYLE"] = "STYLE";
+    ChangeCategory2["DEPENDENCY"] = "DEPENDENCY";
+    ChangeCategory2["FEATURE"] = "FEATURE";
+    ChangeCategory2["ARCHITECTURE"] = "ARCHITECTURE";
+    ChangeCategory2["DATABASE"] = "DATABASE";
+    ChangeCategory2["AUTH"] = "AUTH";
+    ChangeCategory2["API"] = "API";
+  })(ChangeCategory ||= {});
+  AutonomyRuleSchema = exports_external.object({
+    category: exports_external.enum(ChangeCategory),
+    riskLevel: exports_external.enum(RiskLevel),
+    autoExecute: exports_external.boolean()
+  });
+});
+
 // ../shared/src/models/aws-instance.ts
 var InstanceAction, AwsCredentialsSchema, IntegrationSchema, AwsInstanceSchema, CreateAwsInstanceSchema, UpdateAwsInstanceSchema, SaveAwsCredentialsSchema, ProvisionAwsInstanceSchema, InstanceActionBodySchema, InstanceIdParamSchema, CIDR_REGEX, UpdateSecurityRulesSchema;
 var init_aws_instance = __esm(() => {
@@ -14564,6 +14590,71 @@ var init_invitation = __esm(() => {
   });
 });
 
+// ../shared/src/models/job.ts
+var JobType, JobStatus, JobSeverity, JobScheduleSchema, JobConfigSchema, JobRunResultSchema, JobRunSchema, CreateJobRunSchema, UpdateJobRunSchema;
+var init_job = __esm(() => {
+  init_zod();
+  ((JobType2) => {
+    JobType2["LINT_SCAN"] = "LINT_SCAN";
+    JobType2["DEPENDENCY_CHECK"] = "DEPENDENCY_CHECK";
+    JobType2["TODO_CLEANUP"] = "TODO_CLEANUP";
+    JobType2["FLAKY_TEST_DETECTION"] = "FLAKY_TEST_DETECTION";
+    JobType2["CUSTOM"] = "CUSTOM";
+  })(JobType ||= {});
+  ((JobStatus2) => {
+    JobStatus2["IDLE"] = "IDLE";
+    JobStatus2["RUNNING"] = "RUNNING";
+    JobStatus2["COMPLETED"] = "COMPLETED";
+    JobStatus2["FAILED"] = "FAILED";
+    JobStatus2["CANCELLED"] = "CANCELLED";
+  })(JobStatus ||= {});
+  ((JobSeverity2) => {
+    JobSeverity2["AUTO_EXECUTE"] = "AUTO_EXECUTE";
+    JobSeverity2["REQUIRE_APPROVAL"] = "REQUIRE_APPROVAL";
+  })(JobSeverity ||= {});
+  JobScheduleSchema = exports_external.object({
+    cronExpression: exports_external.string(),
+    enabled: exports_external.boolean(),
+    lastRunAt: exports_external.string().optional(),
+    nextRunAt: exports_external.string().optional()
+  });
+  JobConfigSchema = exports_external.object({
+    type: exports_external.enum(JobType),
+    schedule: JobScheduleSchema,
+    severity: exports_external.enum(JobSeverity),
+    enabled: exports_external.boolean(),
+    options: exports_external.record(exports_external.string(), exports_external.any())
+  });
+  JobRunResultSchema = exports_external.object({
+    summary: exports_external.string(),
+    filesChanged: exports_external.number(),
+    prUrl: exports_external.string().optional(),
+    errors: exports_external.array(exports_external.string()).optional()
+  });
+  JobRunSchema = exports_external.object({
+    id: exports_external.string(),
+    jobType: exports_external.enum(JobType),
+    status: exports_external.enum(JobStatus),
+    startedAt: exports_external.string(),
+    completedAt: exports_external.string().optional(),
+    result: JobRunResultSchema.optional(),
+    workspaceId: exports_external.string()
+  });
+  CreateJobRunSchema = exports_external.object({
+    jobType: exports_external.enum(JobType),
+    status: exports_external.enum(JobStatus).optional().default("RUNNING" /* RUNNING */),
+    startedAt: exports_external.string().optional(),
+    error: exports_external.string().optional(),
+    result: JobRunResultSchema.optional()
+  });
+  UpdateJobRunSchema = exports_external.object({
+    status: exports_external.enum(JobStatus).optional(),
+    result: JobRunResultSchema.optional(),
+    error: exports_external.string().nullable().optional(),
+    completedAt: exports_external.string().optional()
+  });
+});
+
 // ../shared/src/models/organization.ts
 var OrganizationSchema, CreateOrganizationSchema, UpdateOrganizationSchema, AddMemberSchema, MembershipWithUserSchema, OrgIdParamSchema, MembershipIdParamSchema, OrganizationResponseSchema, OrganizationsResponseSchema, MembersResponseSchema, MembershipResponseSchema;
 var init_organization = __esm(() => {
@@ -14661,6 +14752,49 @@ var init_sprint = __esm(() => {
   });
 });
 
+// ../shared/src/models/suggestion.ts
+var SuggestionStatus, SuggestionType, SuggestionSchema, CreateSuggestionSchema, UpdateSuggestionStatusSchema;
+var init_suggestion = __esm(() => {
+  init_zod();
+  ((SuggestionStatus2) => {
+    SuggestionStatus2["NEW"] = "NEW";
+    SuggestionStatus2["NOTIFIED"] = "NOTIFIED";
+    SuggestionStatus2["ACTED_ON"] = "ACTED_ON";
+    SuggestionStatus2["SKIPPED"] = "SKIPPED";
+    SuggestionStatus2["EXPIRED"] = "EXPIRED";
+  })(SuggestionStatus ||= {});
+  ((SuggestionType2) => {
+    SuggestionType2["CODE_FIX"] = "CODE_FIX";
+    SuggestionType2["DEPENDENCY_UPDATE"] = "DEPENDENCY_UPDATE";
+    SuggestionType2["NEXT_STEP"] = "NEXT_STEP";
+    SuggestionType2["REFACTOR"] = "REFACTOR";
+    SuggestionType2["TEST_FIX"] = "TEST_FIX";
+  })(SuggestionType ||= {});
+  SuggestionSchema = exports_external.object({
+    id: exports_external.string(),
+    type: exports_external.enum(SuggestionType),
+    status: exports_external.enum(SuggestionStatus),
+    title: exports_external.string(),
+    description: exports_external.string(),
+    jobRunId: exports_external.string().optional(),
+    workspaceId: exports_external.string(),
+    createdAt: exports_external.string(),
+    expiresAt: exports_external.string(),
+    metadata: exports_external.record(exports_external.string(), exports_external.any()).optional()
+  });
+  CreateSuggestionSchema = exports_external.object({
+    type: exports_external.enum(SuggestionType),
+    title: exports_external.string().min(1, "Title is required"),
+    description: exports_external.string().min(1, "Description is required"),
+    jobRunId: exports_external.string().uuid().optional(),
+    metadata: exports_external.record(exports_external.string(), exports_external.any()).optional(),
+    expiresAt: exports_external.string().optional()
+  });
+  UpdateSuggestionStatusSchema = exports_external.object({
+    status: exports_external.enum(SuggestionStatus)
+  });
+});
+
 // ../shared/src/models/task.ts
 var AcceptanceItemSchema, TaskSchema, CreateTaskSchema, UpdateTaskSchema, AddCommentSchema, DispatchTaskSchema, TaskIdParamSchema, TaskQuerySchema, TaskResponseSchema, TasksResponseSchema;
 var init_task = __esm(() => {
@@ -14800,13 +14934,16 @@ var init_models = __esm(() => {
   init_activity();
   init_agent();
   init_auth();
+  init_autonomy();
   init_aws_instance();
   init_ci();
   init_doc();
   init_doc_group();
   init_invitation();
+  init_job();
   init_organization();
   init_sprint();
+  init_suggestion();
   init_task();
   init_user();
   init_workspace();
@@ -15572,6 +15709,7 @@ class ClaudeRunner {
   currentToolName;
   activeTools = new Map;
   activeProcess = null;
+  aborted = false;
   timeoutMs;
   constructor(projectPath, model = DEFAULT_MODEL[PROVIDER.CLAUDE], log, timeoutMs) {
     this.model = model;
@@ -15584,6 +15722,7 @@ class ClaudeRunner {
   }
   abort() {
     if (this.activeProcess && !this.activeProcess.killed) {
+      this.aborted = true;
       this.activeProcess.kill("SIGTERM");
       this.activeProcess = null;
     }
@@ -15642,6 +15781,7 @@ class ClaudeRunner {
     return args;
   }
   async* runStream(prompt) {
+    this.aborted = false;
     const args = this.buildCliArgs();
     const env = getAugmentedEnv({
       FORCE_COLOR: "1",
@@ -15736,7 +15876,7 @@ class ClaudeRunner {
         process.stderr.write(`${stderrBuffer}
 `);
       }
-      if (code !== 0 && !errorMessage) {
+      if (code !== 0 && !errorMessage && !this.aborted) {
         const detail = stderrFull.trim() || lastResultContent.trim();
         errorMessage = this.createExecutionError(code, detail).message;
         this.eventEmitter?.emitErrorOccurred(errorMessage, `EXIT_${code}`);
@@ -15881,6 +16021,7 @@ class ClaudeRunner {
     return null;
   }
   executeRun(prompt) {
+    this.aborted = false;
     return new Promise((resolve2, reject) => {
       const args = this.buildCliArgs();
       const env = getAugmentedEnv({
@@ -15933,7 +16074,7 @@ class ClaudeRunner {
         }
         process.stdout.write(`
 `);
-        if (code === 0) {
+        if (code === 0 || this.aborted) {
           resolve2(finalResult);
         } else {
           const detail = errorOutput.trim() || finalResult.trim();
@@ -16004,6 +16145,7 @@ class CodexRunner {
   log;
   reasoningEffort;
   activeProcess = null;
+  aborted = false;
   eventEmitter;
   currentToolName;
   timeoutMs;
@@ -16019,11 +16161,13 @@ class CodexRunner {
   }
   abort() {
     if (this.activeProcess && !this.activeProcess.killed) {
+      this.aborted = true;
       this.activeProcess.kill("SIGTERM");
       this.activeProcess = null;
     }
   }
   async run(prompt) {
+    this.aborted = false;
     const maxRetries = 3;
     let lastError = null;
     for (let attempt = 1;attempt <= maxRetries; attempt++) {
@@ -16139,7 +16283,7 @@ class CodexRunner {
     });
     codex.on("close", (code) => {
       this.activeProcess = null;
-      if (code === 0) {
+      if (code === 0 || this.aborted) {
         const result = this.readOutput(outputPath, finalOutput);
         this.cleanupTempFile(outputPath);
         if (result && finalContent.trim().length === 0) {
@@ -16252,7 +16396,7 @@ class CodexRunner {
       });
       codex.on("close", (code) => {
         this.activeProcess = null;
-        if (code === 0) {
+        if (code === 0 || this.aborted) {
           const result = this.readOutput(outputPath, output);
           this.cleanupTempFile(outputPath);
           resolve2(result);
@@ -31616,6 +31760,29 @@ var init_invitations = __esm(() => {
   };
 });
 
+// ../sdk/src/modules/jobs.ts
+var JobsModule;
+var init_jobs = __esm(() => {
+  JobsModule = class JobsModule extends BaseModule {
+    async create(workspaceId, data) {
+      const { data: res } = await this.api.post(`/workspaces/${workspaceId}/job-runs`, data);
+      return res.jobRun;
+    }
+    async list(workspaceId, params) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/job-runs`, { params });
+      return data.jobRuns;
+    }
+    async get(workspaceId, id) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/job-runs/${id}`);
+      return data.jobRun;
+    }
+    async update(workspaceId, id, data) {
+      const { data: res } = await this.api.patch(`/workspaces/${workspaceId}/job-runs/${id}`, data);
+      return res.jobRun;
+    }
+  };
+});
+
 // ../sdk/src/modules/organizations.ts
 var OrganizationsModule;
 var init_organizations = __esm(() => {
@@ -31698,6 +31865,29 @@ var init_sprints = __esm(() => {
   };
 });
 
+// ../sdk/src/modules/suggestions.ts
+var SuggestionsModule;
+var init_suggestions = __esm(() => {
+  SuggestionsModule = class SuggestionsModule extends BaseModule {
+    async create(workspaceId, data) {
+      const { data: res } = await this.api.post(`/workspaces/${workspaceId}/suggestions`, data);
+      return res.suggestion;
+    }
+    async list(workspaceId, params) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/suggestions`, { params });
+      return data.suggestions;
+    }
+    async get(workspaceId, id) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/suggestions/${id}`);
+      return data.suggestion;
+    }
+    async updateStatus(workspaceId, id, status) {
+      const { data } = await this.api.patch(`/workspaces/${workspaceId}/suggestions/${id}/status`, status);
+      return data.suggestion;
+    }
+  };
+});
+
 // ../sdk/src/modules/tasks.ts
 var TasksModule;
 var init_tasks = __esm(() => {
@@ -31882,6 +32072,8 @@ class LocusClient {
   docs;
   ci;
   instances;
+  jobs;
+  suggestions;
   constructor(config2) {
     this.emitter = new LocusEmitter;
     this.api = axios_default.create({
@@ -31902,6 +32094,8 @@ class LocusClient {
     this.docs = new DocsModule(this.api, this.emitter);
     this.ci = new CiModule(this.api, this.emitter);
     this.instances = new InstancesModule(this.api, this.emitter);
+    this.jobs = new JobsModule(this.api, this.emitter);
+    this.suggestions = new SuggestionsModule(this.api, this.emitter);
     if (config2.retryOptions) {
       this.setupRetryInterceptor(config2.retryOptions);
     }
@@ -31969,8 +32163,10 @@ var init_src2 = __esm(() => {
   init_docs();
   init_instances();
   init_invitations();
+  init_jobs();
   init_organizations();
   init_sprints();
+  init_suggestions();
   init_tasks();
   init_workspaces();
   init_discussion_types();
@@ -31980,8 +32176,10 @@ var init_src2 = __esm(() => {
   init_docs();
   init_instances();
   init_invitations();
+  init_jobs();
   init_organizations();
   init_sprints();
+  init_suggestions();
   init_tasks();
   init_workspaces();
 });