@oneuptime/common 11.5.4 → 11.5.5

package/Server/API/TelemetryExceptionAPI.ts

@@ -1,15 +1,15 @@
 import TelemetryException from "../../Models/DatabaseModels/TelemetryException";
-import AIAgentTask from "../../Models/DatabaseModels/AIAgentTask";
-import AIAgentTaskTelemetryException from "../../Models/DatabaseModels/AIAgentTaskTelemetryException";
+import AIRun from "../../Models/DatabaseModels/AIRun";
 import BadDataException from "../../Types/Exception/BadDataException";
 import ObjectID from "../../Types/ObjectID";
 import BaseModel from "../../Models/DatabaseModels/DatabaseBaseModel/DatabaseBaseModel";
 import TelemetryExceptionService, {
+  AIFixReadiness,
   DashboardServiceSummary,
   DashboardSummaryResult,
   Service as TelemetryExceptionServiceType,
 } from "../Services/TelemetryExceptionService";
-import AIAgentTaskTelemetryExceptionService from "../Services/AIAgentTaskTelemetryExceptionService";
+import AIRunService from "../Services/AIRunService";
 import UserMiddleware from "../Middleware/UserAuthorization";
 import Response from "../Utils/Response";
 import BaseAPI from "./BaseAPI";
@@ -20,12 +20,52 @@ import {
 } from "../Utils/Express";
 import CommonAPI from "./CommonAPI";
 import DatabaseCommonInteractionProps from "../../Types/BaseDatabase/DatabaseCommonInteractionProps";
-import AIAgentTaskStatus, {
-  AIAgentTaskStatusHelper,
-} from "../../Types/AI/AIAgentTaskStatus";
-import QueryHelper from "../Types/Database/QueryHelper";
+import AIRunStatus, { AIRunStatusHelper } from "../../Types/AI/AIRunStatus";
+import CodeFixTaskType, {
+  CodeFixTaskTypeHelper,
+} from "../../Types/AI/CodeFixTaskType";
 import { JSONArray, JSONObject } from "../../Types/JSON";
 
+/*
+ * User-facing wording for a code-fix run's status on the exception page —
+ * the AIRun analogue of the old AIAgentTaskStatusHelper titles.
+ */
+const CODE_FIX_STATUS_TEXT: {
+  [key in AIRunStatus]: { title: string; description: string };
+} = {
+  [AIRunStatus.Queued]: {
+    title: "Queued",
+    description:
+      "The fix task is queued and waiting to be picked up by an AI agent.",
+  },
+  [AIRunStatus.Running]: {
+    title: "In Progress",
+    description: "An AI agent is working on a fix for this exception.",
+  },
+  [AIRunStatus.WaitingForApproval]: {
+    title: "Waiting for Approval",
+    description: "The run is paused waiting for an approval.",
+  },
+  [AIRunStatus.Completed]: {
+    title: "Completed",
+    description:
+      "The AI agent finished. Review the pull request it opened for the proposed fix.",
+  },
+  [AIRunStatus.Error]: {
+    title: "Error",
+    description: "The AI agent could not complete the fix.",
+  },
+  [AIRunStatus.Cancelled]: {
+    title: "Cancelled",
+    description: "The fix task was cancelled before an agent completed it.",
+  },
+  [AIRunStatus.Stale]: {
+    title: "Stale",
+    description:
+      "The AI agent stopped reporting progress and the run was marked as stale. You can retry the fix.",
+  },
+};
+
 export default class TelemetryExceptionAPI extends BaseAPI<
   TelemetryException,
   TelemetryExceptionServiceType
@@ -63,6 +103,25 @@ export default class TelemetryExceptionAPI extends BaseAPI<
       },
     );
 
+    /*
+     * Everything "Fix with AI Agent" needs, checked up front so the
+     * dashboard can render a setup checklist instead of a button that
+     * fails minutes later inside the agent container.
+     */
+    this.router.get(
+      `${new this.entityType()
+        .getCrudApiPath()
+        ?.toString()}/ai-fix-readiness/:telemetryExceptionId`,
+      UserMiddleware.getUserMiddleware,
+      async (req: ExpressRequest, res: ExpressResponse, next: NextFunction) => {
+        try {
+          await this.getAIFixReadiness(req, res);
+        } catch (err) {
+          next(err);
+        }
+      },
+    );
+
     /*
      * Aggregated dashboard summary for the Exceptions overview page.
      * Returns counts, top/recent exceptions, and per-service summaries
@@ -103,15 +162,47 @@ export default class TelemetryExceptionAPI extends BaseAPI<
     const props: DatabaseCommonInteractionProps =
       await CommonAPI.getDatabaseCommonInteractionProps(req);
 
-    // Create the AI Agent Task using the service
-    const createdTask: AIAgentTask =
-      await this.service.createAIAgentTaskForException({
-        telemetryExceptionId,
-        props,
-      });
+    /*
+     * Optional body: { taskType?: "FixException" | "WriteRegressionTest" }.
+     * Unknown strings are rejected here; known-but-not-yet-triggerable
+     * recipes (ImproveInstrumentation, ...) are rejected by the service.
+     */
+    const body: JSONObject = (req.body as JSONObject) || {};
+
+    let taskType: CodeFixTaskType = CodeFixTaskType.FixException;
+
+    if (body["taskType"] !== undefined && body["taskType"] !== null) {
+      const rawTaskType: unknown = body["taskType"];
+
+      if (
+        typeof rawTaskType !== "string" ||
+        !CodeFixTaskTypeHelper.isValidTaskType(rawTaskType)
+      ) {
+        throw new BadDataException(
+          `Invalid taskType. Valid values: ${Object.values(
+            CodeFixTaskType,
+          ).join(", ")}.`,
+        );
+      }
+
+      taskType = rawTaskType;
+    }
+
+    // Create the code-fix AIRun using the service
+    const createdRun: AIRun = await this.service.createCodeFixRunForException({
+      telemetryExceptionId,
+      props,
+      taskType,
+    });
 
+    /*
+     * `aiAgentTaskId` is the legacy JSON key the dashboard still navigates
+     * with — it now carries the AIRun id. `aiRunId` is the honest name.
+     */
     return Response.sendJsonObjectResponse(req, res, {
-      aiAgentTaskId: createdTask.id!.toString(),
+      aiAgentTaskId: createdRun.id!.toString(),
+      aiRunId: createdRun.id!.toString(),
+      taskType: taskType,
     });
   }
 
@@ -137,54 +228,108 @@ export default class TelemetryExceptionAPI extends BaseAPI<
     const props: DatabaseCommonInteractionProps =
       await CommonAPI.getDatabaseCommonInteractionProps(req);
 
-    // Find the most recent AI agent task for this exception that is not completed or errored
-    const taskLink: AIAgentTaskTelemetryException | null =
-      await AIAgentTaskTelemetryExceptionService.findOneBy({
-        query: {
-          telemetryExceptionId: telemetryExceptionId,
-          aiAgentTask: {
-            status: QueryHelper.notIn([
-              AIAgentTaskStatus.Completed,
-              AIAgentTaskStatus.Error,
-            ]),
-          },
-        },
+    /*
+     * Access check under the USER's permissions: code-fix runs are
+     * system-written AIRuns (hidden from the caller by the per-user privacy
+     * pin on the generic CRUD), so confirm the caller can read the
+     * exception first and then read the run as root — the same pattern as
+     * AIInvestigationAPI.
+     */
+    const telemetryException: TelemetryException | null =
+      await this.service.findOneById({
+        id: telemetryExceptionId,
         select: {
           _id: true,
-          aiAgentTaskId: true,
-          aiAgentTask: {
-            _id: true,
-            status: true,
-            statusMessage: true,
-            createdAt: true,
-          },
         },
         props,
       });
 
-    if (!taskLink || !taskLink.aiAgentTask) {
+    if (!telemetryException) {
+      throw new BadDataException(
+        "Telemetry Exception not found (or you do not have access to it).",
+      );
+    }
+
+    /*
+     * Return the LATEST run PER TASK TYPE (at most one per CodeFixTaskType,
+     * newest first) so the exception page can show a FixException run and a
+     * WriteRegressionTest run side by side. Runs are returned regardless of
+     * status: errored runs must stay visible (a failed fix must not silently
+     * vanish with the button just reappearing).
+     *
+     * Back-compat: `hasActiveTask` / `aiAgentTask` still reflect the latest
+     * run OVERALL, for clients that predate task recipes.
+     */
+    const runs: Array<AIRun> =
+      await AIRunService.getLatestCodeFixRunPerTaskType({
+        telemetryExceptionId,
+      });
+
+    if (runs.length === 0) {
       return Response.sendJsonObjectResponse(req, res, {
         hasActiveTask: false,
         aiAgentTask: null,
+        aiAgentTasks: [],
       });
     }
 
-    const task: AIAgentTask = taskLink.aiAgentTask;
+    const toTaskJson: (run: AIRun) => JSONObject = (run: AIRun): JSONObject => {
+      const status: AIRunStatus = run.status || AIRunStatus.Queued;
+
+      return {
+        _id: run.id?.toString(),
+        status: status,
+        statusMessage: run.errorMessage,
+        statusTitle: CODE_FIX_STATUS_TEXT[status]?.title,
+        statusDescription: CODE_FIX_STATUS_TEXT[status]?.description,
+        createdAt: run.createdAt,
+        // Normalized by the service: legacy null rows read as FixException.
+        taskType: CodeFixTaskTypeHelper.fromDatabaseValue(run.codeFixTaskType),
+      };
+    };
+
+    // Newest first — the first element is the latest run overall.
+    const latestRunOverall: AIRun = runs[0]!;
+    const latestStatus: AIRunStatus =
+      latestRunOverall.status || AIRunStatus.Queued;
 
     return Response.sendJsonObjectResponse(req, res, {
-      hasActiveTask: true,
-      aiAgentTask: {
-        _id: task.id?.toString(),
-        status: task.status,
-        statusMessage: task.statusMessage,
-        statusTitle: task.status
-          ? AIAgentTaskStatusHelper.getTitle(task.status)
-          : undefined,
-        statusDescription: task.status
-          ? AIAgentTaskStatusHelper.getDescription(task.status)
-          : undefined,
-        createdAt: task.createdAt,
-      },
+      hasActiveTask: !AIRunStatusHelper.isTerminalStatus(latestStatus),
+      aiAgentTask: toTaskJson(latestRunOverall),
+      aiAgentTasks: runs.map(toTaskJson),
+    });
+  }
+
+  private async getAIFixReadiness(
+    req: ExpressRequest,
+    res: ExpressResponse,
+  ): Promise<void> {
+    const telemetryExceptionIdParam: string | undefined =
+      req.params["telemetryExceptionId"];
+
+    if (!telemetryExceptionIdParam) {
+      throw new BadDataException("Telemetry Exception ID is required");
+    }
+
+    let telemetryExceptionId: ObjectID;
+
+    try {
+      telemetryExceptionId = new ObjectID(telemetryExceptionIdParam);
+    } catch {
+      throw new BadDataException("Invalid Telemetry Exception ID");
+    }
+
+    const props: DatabaseCommonInteractionProps =
+      await CommonAPI.getDatabaseCommonInteractionProps(req);
+
+    const readiness: AIFixReadiness = await this.service.getAIFixReadiness({
+      telemetryExceptionId,
+      props,
+    });
+
+    return Response.sendJsonObjectResponse(req, res, {
+      ready: readiness.ready,
+      checks: readiness.checks as unknown as JSONArray,
     });
   }
 

package/Server/EnvironmentConfig.ts

@@ -480,7 +480,7 @@ export const WorkflowScriptTimeoutInMS: number = process.env[
 
 export const WorkflowTimeoutInMs: number = process.env["WORKFLOW_TIMEOUT_IN_MS"]
   ? parseInt(process.env["WORKFLOW_TIMEOUT_IN_MS"].toString())
-  : 5000;
+  : 120000;
 
 export const AllowedActiveMonitorCountInFreePlan: number = process.env[
   "ALLOWED_ACTIVE_MONITOR_COUNT_IN_FREE_PLAN"

package/Server/Infrastructure/Postgres/SchemaMigrations/1783943300000-DropServiceCodeRepository.ts

@@ -0,0 +1,26 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+/*
+ * The ServiceCodeRepository mapping table is removed: service → repository
+ * resolution now happens at runtime (stack-trace path matching over the
+ * project's connected repositories, with name-match and only-repository
+ * fallbacks — see Common/Server/Utils/CodeRepository/StackTraceRepoResolver).
+ * Hand-written because typeorm migration:generate does not emit drops for
+ * tables whose entity was deleted.
+ */
+export class DropServiceCodeRepository1783943300000
+  implements MigrationInterface
+{
+  public name = "DropServiceCodeRepository1783943300000";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(`DROP TABLE IF EXISTS "ServiceCodeRepository"`);
+  }
+
+  public async down(_queryRunner: QueryRunner): Promise<void> {
+    /*
+     * Intentionally irreversible: the feature this table served (manual
+     * service → repository links) no longer exists in the application.
+     */
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783947444597-CodeFixRunsOnAIRun.ts

@@ -0,0 +1,73 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class CodeFixRunsOnAIRun1783947444597 implements MigrationInterface {
+  public name = "CodeFixRunsOnAIRun1783947444597";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ADD "aiRunId" uuid`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "triggeredByTelemetryExceptionId" uuid`,
+    );
+    await queryRunner.query(`ALTER TABLE "AIRun" ADD "aiAgentId" uuid`);
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" DROP CONSTRAINT "FK_e1de7edf3e00ae0d00617086063"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ALTER COLUMN "aiAgentTaskId" DROP NOT NULL`,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_f7155e112ddac1d0c60767baf0" ON "AIAgentTaskPullRequest" ("aiRunId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_1b64c644c8247682fdedb3e92d" ON "AIRun" ("triggeredByTelemetryExceptionId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_ed7f37e218ed8ba39c5bc79056" ON "AIRun" ("aiAgentId") `,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ADD CONSTRAINT "FK_e1de7edf3e00ae0d00617086063" FOREIGN KEY ("aiAgentTaskId") REFERENCES "AIAgentTask"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" DROP CONSTRAINT "FK_e1de7edf3e00ae0d00617086063"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_ed7f37e218ed8ba39c5bc79056"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_1b64c644c8247682fdedb3e92d"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_f7155e112ddac1d0c60767baf0"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ALTER COLUMN "aiAgentTaskId" SET NOT NULL`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ADD CONSTRAINT "FK_e1de7edf3e00ae0d00617086063" FOREIGN KEY ("aiAgentTaskId") REFERENCES "AIAgentTask"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(`ALTER TABLE "AIRun" DROP COLUMN "aiAgentId"`);
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" DROP COLUMN "triggeredByTelemetryExceptionId"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" DROP COLUMN "aiRunId"`,
+    );
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783950962813-AddCodeFixTaskType.ts

@@ -0,0 +1,29 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class AddCodeFixTaskType1783950962813 implements MigrationInterface {
+  public name = "AddCodeFixTaskType1783950962813";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "codeFixTaskType" character varying(100)`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" DROP COLUMN "codeFixTaskType"`,
+    );
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783954040984-AddInstrumentationFixTasksFlag.ts

@@ -0,0 +1,31 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class AddInstrumentationFixTasksFlag1783954040984
+  implements MigrationInterface
+{
+  public name = "AddInstrumentationFixTasksFlag1783954040984";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "enableInstrumentationFixTasks" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "enableInstrumentationFixTasks"`,
+    );
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783957160597-MigrationName.ts

@@ -0,0 +1,51 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class MigrationName1783957160597 implements MigrationInterface {
+  public name: string = "MigrationName1783957160597";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `CREATE TABLE "InstanceHealthLog" ("_id" uuid NOT NULL DEFAULT uuid_generate_v4(), "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), "updatedAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), "deletedAt" TIMESTAMP WITH TIME ZONE, "version" integer NOT NULL, "eventType" character varying(100) NOT NULL, "status" character varying(100) NOT NULL, "message" text NOT NULL, "completedAt" TIMESTAMP WITH TIME ZONE, "nextCheckAt" TIMESTAMP WITH TIME ZONE, "capacityBeforePercent" numeric, "capacityAfterPercent" numeric, "thresholdPercent" integer, "targetPercent" integer, "estimatedFreedBytes" bigint, "metadata" jsonb, CONSTRAINT "PK_625f9ce409f0648439e9940c0c6" PRIMARY KEY ("_id"))`,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_3c2586a18f586d8f02dbec4a46" ON "InstanceHealthLog" ("eventType", "createdAt") `,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" ADD "clickhouseCapacityNotificationEnabled" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" ADD "clickhouseCapacityNotificationThresholdPercent" integer NOT NULL DEFAULT '80'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" ADD "clickhouseDataPruningEnabled" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" ADD "clickhouseDataPruningThresholdPercent" integer NOT NULL DEFAULT '90'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" ADD "clickhouseDataPruningTargetPercent" integer NOT NULL DEFAULT '80'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" DROP COLUMN "clickhouseDataPruningTargetPercent"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" DROP COLUMN "clickhouseDataPruningThresholdPercent"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" DROP COLUMN "clickhouseDataPruningEnabled"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" DROP COLUMN "clickhouseCapacityNotificationThresholdPercent"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "GlobalConfig" DROP COLUMN "clickhouseCapacityNotificationEnabled"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_3c2586a18f586d8f02dbec4a46"`,
+    );
+    await queryRunner.query(`DROP TABLE "InstanceHealthLog"`);
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783958542237-AddTaskContextToAIRun.ts

@@ -0,0 +1,25 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class AddTaskContextToAIRun1783958542237 implements MigrationInterface {
+  public name = "AddTaskContextToAIRun1783958542237";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(`ALTER TABLE "AIRun" ADD "taskContext" jsonb`);
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(`ALTER TABLE "AIRun" DROP COLUMN "taskContext"`);
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783965945957-AddInvestigationVerdictAndGrade.ts

@@ -0,0 +1,47 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class AddInvestigationVerdictAndGrade1783965945957
+  implements MigrationInterface
+{
+  public name = "AddInvestigationVerdictAndGrade1783965945957";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "humanVerdict" character varying(100)`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "humanVerdictAt" TIMESTAMP WITH TIME ZONE`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "humanVerdictByUserId" uuid`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "autoGrade" character varying(100)`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" ADD "autoGradeAt" TIMESTAMP WITH TIME ZONE`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(`ALTER TABLE "AIRun" DROP COLUMN "autoGradeAt"`);
+    await queryRunner.query(`ALTER TABLE "AIRun" DROP COLUMN "autoGrade"`);
+    await queryRunner.query(
+      `ALTER TABLE "AIRun" DROP COLUMN "humanVerdictByUserId"`,
+    );
+    await queryRunner.query(`ALTER TABLE "AIRun" DROP COLUMN "humanVerdictAt"`);
+    await queryRunner.query(`ALTER TABLE "AIRun" DROP COLUMN "humanVerdict"`);
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1783970619301-AddFixGuardrailColumns.ts

@@ -0,0 +1,53 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class AddFixGuardrailColumns1783970619301 implements MigrationInterface {
+  public name = "AddFixGuardrailColumns1783970619301";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" DROP CONSTRAINT "FK_e1de7edf3e00ae0d00617086063"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_e1de7edf3e00ae0d0061708606"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" DROP COLUMN "aiAgentTaskId"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "aiDailyFixTaskLimit" integer`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "CodeRepository" ADD "maxOpenFixPullRequests" integer`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "CodeRepository" DROP COLUMN "maxOpenFixPullRequests"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "aiDailyFixTaskLimit"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ADD "aiAgentTaskId" uuid`,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_e1de7edf3e00ae0d0061708606" ON "AIAgentTaskPullRequest" ("aiAgentTaskId") `,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "AIAgentTaskPullRequest" ADD CONSTRAINT "FK_e1de7edf3e00ae0d00617086063" FOREIGN KEY ("aiAgentTaskId") REFERENCES "AIAgentTask"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+  }
+}