refactor: Enable lazy loading for images in Copilot/Utils/CopilotActi…

…on.ts, Common/Utils/CronTime.ts, Common/Server/Services/BillingInvoiceService.ts, and Dashboard/src/Pages/Settings/Invoices.tsx
OneUptime · Sep 5, 2024 · e3c68fd · e3c68fd
1 parent 1682a25
commit e3c68fd
Show file tree

Hide file tree

Showing 6 changed files with 40 additions and 11 deletions.
diff --git a/App/FeatureSet/Workers/Index.ts b/App/FeatureSet/Workers/Index.ts
@@ -67,6 +67,9 @@ import logger from "Common/Server/Utils/Logger";
 import "./Jobs/Probe/SendOwnerAddedNotification";
 import "./Jobs/Probe/UpdateConnectionStatus";
 
+// Copilot Actions.
+import "./Jobs/CopilotActions/MoveThemBackToQueueIfProcessingForLongtime";
+
 // Telemetry Monitors.
 import "./Jobs/TelemetryMonitor/MonitorTelemetryMonitor";
 import Express, { ExpressApplication } from "Common/Server/Utils/Express";

diff --git a/App/FeatureSet/Workers/Jobs/CopilotActions/MoveThemBackToQueueIfProcessingForLongtime.ts b/App/FeatureSet/Workers/Jobs/CopilotActions/MoveThemBackToQueueIfProcessingForLongtime.ts
@@ -22,7 +22,7 @@ RunCron(
       await CopilotActionService.findBy({
         query: {
           copilotActionStatus: CopilotActionStatus.PROCESSING,
-          statusChangedAt: QueryHelper.lessThanEqualTo(lastHour),
+          statusChangedAt: QueryHelper.lessThanEqualToOrNull(lastHour),
         },
         select: {
           _id: true,

diff --git a/Copilot/Init.ts b/Copilot/Init.ts
@@ -28,8 +28,6 @@ let currentFixCount: number = 1;
 const init: PromiseVoidFunction = async (): Promise<void> => {
   // check if copilot is disabled.
 
-  debugger;
-
   if (GetIsCopilotDisabled()) {
     logger.info("Copilot is disabled. Exiting.");
     ProcessUtil.haltProcessWithSuccess();

diff --git a/Copilot/Service/CopilotActions/ImproveComments.ts b/Copilot/Service/CopilotActions/ImproveComments.ts
@@ -16,6 +16,7 @@ import ServiceRepositoryUtil from "../../Utils/ServiceRepository";
 import Dictionary from "Common/Types/Dictionary";
 import ArrayUtil from "Common/Utils/Array";
 import CopilotActionProp from "Common/Types/Copilot/CopilotActionProps/Index";
+import BadDataException from "Common/Types/Exception/BadDataException";
 
 export default class ImproveComments extends CopilotActionBase {
   public isRequirementsMet: boolean = false;
@@ -54,9 +55,6 @@ export default class ImproveComments extends CopilotActionBase {
     maxActionsToQueue: number;
   }): Promise<Array<CopilotActionProp>> {
     // get files in the repo.
-
-    debugger;
-
     logger.debug(
       `${this.copilotActionType} - Getting files to queue for improve comments.`,
     );
@@ -113,17 +111,50 @@ export default class ImproveComments extends CopilotActionBase {
     return actionsPropsQueued;
   }
 
+  public override async getCommitMessage(
+    data: CopilotProcess,
+  ): Promise<string> {
+    return (
+      "Improved comments on " + (data.actionProp as FileActionProp).filePath
+    );
+  }
+
+  public override async getPullRequestTitle(
+    data: CopilotProcess,
+  ): Promise<string> {
+    return (
+      "Improved comments on " + (data.actionProp as FileActionProp).filePath
+    );
+  }
+
+  public override async getPullRequestBody(
+    data: CopilotProcess,
+  ): Promise<string> {
+    return `Improved comments on ${(data.actionProp as FileActionProp).filePath}
+    
+    ${await this.getDefaultPullRequestBody()}
+    `;
+  }
+
   public override isActionComplete(_data: CopilotProcess): Promise<boolean> {
     return Promise.resolve(this.isRequirementsMet);
   }
 
   public override async onExecutionStep(
     data: CopilotProcess,
   ): Promise<CopilotProcess> {
-    // Action Prompt
+    const filePath: string = (data.actionProp as FileActionProp).filePath;
+
+    if (!filePath) {
+      throw new BadDataException("File Path is not set in the action prop.");
+    }
+
+    const fileContent: string = await ServiceRepositoryUtil.getFileContent({
+      filePath: filePath,
+    });
 
     const codeParts: string[] = await this.splitInputCode({
-      code: "",
+      code: fileContent,
       itemSize: 500,
     });
 

diff --git a/Copilot/Service/CopilotActions/Index.ts b/Copilot/Service/CopilotActions/Index.ts
@@ -39,7 +39,6 @@ export default class CopilotActionService {
       ActionDictionary[data.copilotAction.copilotActionType!];
 
     if (!ActionType) {
-      debugger;
       throw new BadDataException("Invalid CopilotActionType");
     }
 

diff --git a/Copilot/Utils/CopilotAction.ts b/Copilot/Utils/CopilotAction.ts
@@ -285,8 +285,6 @@ export default class CopilotActionUtil {
       throw new BadDataException("Repository Secret Key is required");
     }
 
-    debugger;
-
     const url: URL = URL.fromString(
       GetOneUptimeURL().toString() + "/api",
     ).addRoute(