@oneuptime/common 9.2.16 → 9.2.18

package/build/dist/Server/Services/AIBillingService.js

@@ -0,0 +1,187 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+import { IsBillingEnabled, NotificationSlackWebhookOnSubscriptionUpdate, } from "../EnvironmentConfig";
+import logger from "../Utils/Logger";
+import BaseService from "./BaseService";
+import BillingService from "./BillingService";
+import ProjectService from "./ProjectService";
+import BadDataException from "../../Types/Exception/BadDataException";
+import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import SlackUtil from "../Utils/Workspace/Slack/Slack";
+import URL from "../../Types/API/URL";
+export class AIBillingService extends BaseService {
+    constructor() {
+        super();
+    }
+    async rechargeBalance(projectId, amountInUSD) {
+        if (!IsBillingEnabled) {
+            throw new BadDataException("Billing is not enabled");
+        }
+        const project = await ProjectService.findOneById({
+            id: projectId,
+            select: {
+                aiCurrentBalanceInUSDCents: true,
+                enableAutoRechargeAiBalance: true,
+                enableAi: true,
+                autoAiRechargeByBalanceInUSD: true,
+                autoRechargeAiWhenCurrentBalanceFallsInUSD: true,
+                paymentProviderCustomerId: true,
+                name: true,
+                failedAiBalanceChargeNotificationSentToOwners: true,
+            },
+            props: {
+                isRoot: true,
+            },
+        });
+        if (!project) {
+            return 0;
+        }
+        try {
+            if (!(await BillingService.hasPaymentMethods(project.paymentProviderCustomerId))) {
+                if (!project.failedAiBalanceChargeNotificationSentToOwners) {
+                    await ProjectService.updateOneById({
+                        data: {
+                            failedAiBalanceChargeNotificationSentToOwners: true,
+                        },
+                        id: project.id,
+                        props: {
+                            isRoot: true,
+                        },
+                    });
+                    await ProjectService.sendEmailToProjectOwners(project.id, "ACTION REQUIRED: AI Balance Recharge Failed for project - " +
+                        (project.name || ""), `We have tried to recharge your AI balance for project - ${project.name || ""} and failed. We could not find a payment method for the project. Please add a payment method in Project Settings.`);
+                }
+                throw new BadDataException("No payment methods found for the project. Please add a payment method in Project Settings to continue.");
+            }
+            // recharge balance
+            const updatedAmount = Math.floor((project.aiCurrentBalanceInUSDCents || 0) + amountInUSD * 100);
+            // If the recharge is successful, then update the project balance.
+            await BillingService.generateInvoiceAndChargeCustomer(project.paymentProviderCustomerId, "AI Balance Recharge", amountInUSD);
+            await ProjectService.updateOneById({
+                data: {
+                    aiCurrentBalanceInUSDCents: updatedAmount,
+                    failedAiBalanceChargeNotificationSentToOwners: false, // reset this flag
+                    lowAiBalanceNotificationSentToOwners: false, // reset this flag
+                    notEnabledAiNotificationSentToOwners: false,
+                },
+                id: project.id,
+                props: {
+                    isRoot: true,
+                },
+            });
+            await ProjectService.sendEmailToProjectOwners(project.id, "AI Balance Recharge Successful for project - " + (project.name || ""), `We have successfully recharged your AI balance for project - ${project.name || ""} by ${amountInUSD} USD. Your current balance is ${updatedAmount / 100} USD.`);
+            // Send Slack notification for balance refill
+            this.sendBalanceRefillSlackNotification({
+                project: project,
+                amountInUSD: amountInUSD,
+                currentBalanceInUSD: updatedAmount / 100,
+            }).catch((error) => {
+                logger.error("Error sending slack message for AI balance refill: " + error);
+            });
+            project.aiCurrentBalanceInUSDCents = updatedAmount;
+            return updatedAmount;
+        }
+        catch (err) {
+            await ProjectService.updateOneById({
+                data: {
+                    failedAiBalanceChargeNotificationSentToOwners: true,
+                },
+                id: project.id,
+                props: {
+                    isRoot: true,
+                },
+            });
+            await ProjectService.sendEmailToProjectOwners(project.id, "ACTION REQUIRED: AI Balance Recharge Failed for project - " +
+                (project.name || ""), `We have tried to recharge your AI balance for project - ${project.name || ""} and failed. Please make sure your payment method is up to date and has sufficient balance. You can add new payment methods in Project Settings.`);
+            logger.error(err);
+            throw err;
+        }
+    }
+    async rechargeIfBalanceIsLow(projectId, options) {
+        let project = null;
+        if (projectId && IsBillingEnabled) {
+            // check payment methods.
+            project = await ProjectService.findOneById({
+                id: projectId,
+                select: {
+                    aiCurrentBalanceInUSDCents: true,
+                    enableAutoRechargeAiBalance: true,
+                    autoAiRechargeByBalanceInUSD: true,
+                    autoRechargeAiWhenCurrentBalanceFallsInUSD: true,
+                },
+                props: {
+                    isRoot: true,
+                },
+            });
+            const autoRechargeAiWhenCurrentBalanceFallsInUSD = (options === null || options === void 0 ? void 0 : options.autoRechargeAiWhenCurrentBalanceFallsInUSD) ||
+                (project === null || project === void 0 ? void 0 : project.autoRechargeAiWhenCurrentBalanceFallsInUSD) ||
+                0;
+            const autoAiRechargeByBalanceInUSD = (options === null || options === void 0 ? void 0 : options.autoAiRechargeByBalanceInUSD) ||
+                (project === null || project === void 0 ? void 0 : project.autoAiRechargeByBalanceInUSD) ||
+                0;
+            const enableAutoRechargeAiBalance = options
+                ? options.enableAutoRechargeAiBalance
+                : (project === null || project === void 0 ? void 0 : project.enableAutoRechargeAiBalance) || false;
+            if (!project) {
+                return 0;
+            }
+            if (enableAutoRechargeAiBalance &&
+                autoAiRechargeByBalanceInUSD &&
+                autoRechargeAiWhenCurrentBalanceFallsInUSD) {
+                if ((project.aiCurrentBalanceInUSDCents || 0) / 100 <
+                    autoRechargeAiWhenCurrentBalanceFallsInUSD) {
+                    const updatedAmount = await this.rechargeBalance(projectId, autoAiRechargeByBalanceInUSD);
+                    project.aiCurrentBalanceInUSDCents = updatedAmount;
+                }
+            }
+        }
+        return (project === null || project === void 0 ? void 0 : project.aiCurrentBalanceInUSDCents) || 0;
+    }
+    async sendBalanceRefillSlackNotification(data) {
+        var _a, _b;
+        const { project, amountInUSD, currentBalanceInUSD } = data;
+        if (NotificationSlackWebhookOnSubscriptionUpdate) {
+            const slackMessage = `*AI Balance Refilled:*
+*Project Name:* ${((_a = project.name) === null || _a === void 0 ? void 0 : _a.toString()) || "N/A"}
+*Project ID:* ${((_b = project.id) === null || _b === void 0 ? void 0 : _b.toString()) || "N/A"}
+*Refill Amount:* $${amountInUSD} USD
+*Current Balance:* $${currentBalanceInUSD} USD
+
+${project.createdOwnerName && project.createdOwnerEmail ? `*Project Created By:* ${project.createdOwnerName.toString()} (${project.createdOwnerEmail.toString()})` : ""}`;
+            SlackUtil.sendMessageToChannelViaIncomingWebhook({
+                url: URL.fromString(NotificationSlackWebhookOnSubscriptionUpdate),
+                text: slackMessage,
+            }).catch((error) => {
+                logger.error("Error sending slack message for AI balance refill: " + error);
+            });
+        }
+    }
+}
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [ObjectID, Number]),
+    __metadata("design:returntype", Promise)
+], AIBillingService.prototype, "rechargeBalance", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [ObjectID, Object]),
+    __metadata("design:returntype", Promise)
+], AIBillingService.prototype, "rechargeIfBalanceIsLow", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], AIBillingService.prototype, "sendBalanceRefillSlackNotification", null);
+export default new AIBillingService();
+//# sourceMappingURL=AIBillingService.js.map

package/build/dist/Server/Services/AIBillingService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AIBillingService.js","sourceRoot":"","sources":["../../../../Server/Services/AIBillingService.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EACL,gBAAgB,EAChB,4CAA4C,GAC7C,MAAM,sBAAsB,CAAC;AAC9B,OAAO,MAAM,MAAM,iBAAiB,CAAC;AACrC,OAAO,WAAW,MAAM,eAAe,CAAC;AACxC,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,gBAAgB,MAAM,wCAAwC,CAAC;AACtE,OAAO,QAAQ,MAAM,sBAAsB,CAAC;AAE5C,OAAO,WAAW,MAAM,gCAAgC,CAAC;AACzD,OAAO,SAAS,MAAM,gCAAgC,CAAC;AACvD,OAAO,GAAG,MAAM,qBAAqB,CAAC;AAGtC,MAAM,OAAO,gBAAiB,SAAQ,WAAW;IAC/C;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAGY,AAAN,KAAK,CAAC,eAAe,CAC1B,SAAmB,EACnB,WAAmB;QAEnB,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,MAAM,IAAI,gBAAgB,CAAC,wBAAwB,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,OAAO,GAAmB,MAAM,cAAc,CAAC,WAAW,CAAC;YAC/D,EAAE,EAAE,SAAS;YACb,MAAM,EAAE;gBACN,0BAA0B,EAAE,IAAI;gBAChC,2BAA2B,EAAE,IAAI;gBACjC,QAAQ,EAAE,IAAI;gBACd,4BAA4B,EAAE,IAAI;gBAClC,0CAA0C,EAAE,IAAI;gBAChD,yBAAyB,EAAE,IAAI;gBAC/B,IAAI,EAAE,IAAI;gBACV,6CAA6C,EAAE,IAAI;aACpD;YACD,KAAK,EAAE;gBACL,MAAM,EAAE,IAAI;aACb;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,CAAC;QACX,CAAC;QAED,IAAI,CAAC;YACH,IACE,CAAC,CAAC,MAAM,cAAc,CAAC,iBAAiB,CACtC,OAAO,CAAC,yBAA0B,CACnC,CAAC,EACF,CAAC;gBACD,IAAI,CAAC,OAAO,CAAC,6CAA6C,EAAE,CAAC;oBAC3D,MAAM,cAAc,CAAC,aAAa,CAAC;wBACjC,IAAI,EAAE;4BACJ,6CAA6C,EAAE,IAAI;yBACpD;wBACD,EAAE,EAAE,OAAO,CAAC,EAAG;wBACf,KAAK,EAAE;4BACL,MAAM,EAAE,IAAI;yBACb;qBACF,CAAC,CAAC;oBACH,MAAM,cAAc,CAAC,wBAAwB,CAC3C,OAAO,CAAC,EAAG,EACX,4DAA4D;wBAC1D,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,EACtB,2DACE,OAAO,CAAC,IAAI,IAAI,EAClB,mHAAmH,CACpH,CAAC;gBACJ,CAAC;gBACD,MAAM,IAAI,gBAAgB,CACxB,wGAAwG,CACzG,CAAC;YACJ,CAAC;YAED,mBAAmB;YACnB,MAAM,aAAa,GAAW,IAAI,CAAC,KAAK,CACtC,CAAC,OAAO,CAAC,0BAA0B,IAAI,CAAC,CAAC,GAAG,WAAW,GAAG,GAAG,CAC9D,CAAC;YAEF,kEAAkE;YAClE,MAAM,cAAc,CAAC,gCAAgC,CACnD,OAAO,CAAC,yBAA0B,EAClC,qBAAqB,EACrB,WAAW,CACZ,CAAC;YAEF,MAAM,cAAc,CAAC,aAAa,CAAC;gBACjC,IAAI,EAAE;oBACJ,0BAA0B,EAAE,aAAa;oBACzC,6CAA6C,EAAE,KAAK,EAAE,kBAAkB;oBACxE,oCAAoC,EAAE,KAAK,EAAE,kBAAkB;oBAC/D,oCAAoC,EAAE,KAAK;iBAC5C;gBACD,EAAE,EAAE,OAAO,CAAC,EAAG;gBACf,KAAK,EAAE;oBACL,MAAM,EAAE,IAAI;iBACb;aACF,CAAC,CAAC;YAEH,MAAM,cAAc,CAAC,wBAAwB,CAC3C,OAAO,CAAC,EAAG,EACX,+CAA+C,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,EACtE,gEACE,OAAO,CAAC,IAAI,IAAI,EAClB,OAAO,WAAW,iCAChB,aAAa,GAAG,GAClB,OAAO,CACR,CAAC;YAEF,6CAA6C;YAC7C,IAAI,CAAC,kCAAkC,CAAC;gBACtC,OAAO,EAAE,OAAO;gBAChB,WAAW,EAAE,WAAW;gBACxB,mBAAmB,EAAE,aAAa,GAAG,GAAG;aACzC,CAAC,CAAC,KAAK,CAAC,CAAC,KAAgB,EAAE,EAAE;gBAC5B,MAAM,CAAC,KAAK,CACV,qDAAqD,GAAG,KAAK,CAC9D,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,0BAA0B,GAAG,aAAa,CAAC;YAEnD,OAAO,aAAa,CAAC;QACvB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAc,CAAC,aAAa,CAAC;gBACjC,IAAI,EAAE;oBACJ,6CAA6C,EAAE,IAAI;iBACpD;gBACD,EAAE,EAAE,OAAO,CAAC,EAAG;gBACf,KAAK,EAAE;oBACL,MAAM,EAAE,IAAI;iBACb;aACF,CAAC,CAAC;YACH,MAAM,cAAc,CAAC,wBAAwB,CAC3C,OAAO,CAAC,EAAG,EACX,4DAA4D;gBAC1D,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,EACtB,2DACE,OAAO,CAAC,IAAI,IAAI,EAClB,kJAAkJ,CACnJ,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAClB,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAGY,AAAN,KAAK,CAAC,sBAAsB,CACjC,SAAmB,EACnB,OAIC;QAED,IAAI,OAAO,GAAmB,IAAI,CAAC;QACnC,IAAI,SAAS,IAAI,gBAAgB,EAAE,CAAC;YAClC,yBAAyB;YAEzB,OAAO,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC;gBACzC,EAAE,EAAE,SAAS;gBACb,MAAM,EAAE;oBACN,0BAA0B,EAAE,IAAI;oBAChC,2BAA2B,EAAE,IAAI;oBACjC,4BAA4B,EAAE,IAAI;oBAClC,0CAA0C,EAAE,IAAI;iBACjD;gBACD,KAAK,EAAE;oBACL,MAAM,EAAE,IAAI;iBACb;aACF,CAAC,CAAC;YAEH,MAAM,0CAA0C,GAC9C,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,0CAA0C;iBACnD,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,0CAA0C,CAAA;gBACnD,CAAC,CAAC;YACJ,MAAM,4BAA4B,GAChC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,4BAA4B;iBACrC,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,4BAA4B,CAAA;gBACrC,CAAC,CAAC;YAEJ,MAAM,2BAA2B,GAAY,OAAO;gBAClD,CAAC,CAAC,OAAO,CAAC,2BAA2B;gBACrC,CAAC,CAAC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,2BAA2B,KAAI,KAAK,CAAC;YAElD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,CAAC,CAAC;YACX,CAAC;YAED,IACE,2BAA2B;gBAC3B,4BAA4B;gBAC5B,0CAA0C,EAC1C,CAAC;gBACD,IACE,CAAC,OAAO,CAAC,0BAA0B,IAAI,CAAC,CAAC,GAAG,GAAG;oBAC/C,0CAA0C,EAC1C,CAAC;oBACD,MAAM,aAAa,GAAW,MAAM,IAAI,CAAC,eAAe,CACtD,SAAS,EACT,4BAA4B,CAC7B,CAAC;oBACF,OAAO,CAAC,0BAA0B,GAAG,aAAa,CAAC;gBACrD,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,0BAA0B,KAAI,CAAC,CAAC;IAClD,CAAC;IAGa,AAAN,KAAK,CAAC,kCAAkC,CAAC,IAIhD;;QACC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAE,GAAG,IAAI,CAAC;QAE3D,IAAI,4CAA4C,EAAE,CAAC;YACjD,MAAM,YAAY,GAAW;kBACjB,CAAA,MAAA,OAAO,CAAC,IAAI,0CAAE,QAAQ,EAAE,KAAI,KAAK;gBACnC,CAAA,MAAA,OAAO,CAAC,EAAE,0CAAE,QAAQ,EAAE,KAAI,KAAK;oBAC3B,WAAW;sBACT,mBAAmB;;EAEvC,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,yBAAyB,OAAO,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAC,iBAAiB,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;YAEpK,SAAS,CAAC,sCAAsC,CAAC;gBAC/C,GAAG,EAAE,GAAG,CAAC,UAAU,CAAC,4CAA4C,CAAC;gBACjE,IAAI,EAAE,YAAY;aACnB,CAAC,CAAC,KAAK,CAAC,CAAC,KAAgB,EAAE,EAAE;gBAC5B,MAAM,CAAC,KAAK,CACV,qDAAqD,GAAG,KAAK,CAC9D,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF;AA9Nc;IADZ,WAAW,EAAE;;qCAED,QAAQ;;uDAgIpB;AAGY;IADZ,WAAW,EAAE;;qCAED,QAAQ;;8DA4DpB;AAGa;IADb,WAAW,EAAE;;;;0EA0Bb;AAGH,eAAe,IAAI,gBAAgB,EAAE,CAAC"}

package/build/dist/Server/Services/AIService.js

@@ -0,0 +1,184 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+import { IsBillingEnabled } from "../EnvironmentConfig";
+import BaseService from "./BaseService";
+import LlmProviderService from "./LlmProviderService";
+import LlmLogService from "./LlmLogService";
+import ProjectService from "./ProjectService";
+import AIBillingService from "./AIBillingService";
+import LLMService from "../Utils/LLM/LLMService";
+import LlmLog from "../../Models/DatabaseModels/LlmLog";
+import LlmLogStatus from "../../Types/LlmLogStatus";
+import BadDataException from "../../Types/Exception/BadDataException";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import logger from "../Utils/Logger";
+export class Service extends BaseService {
+    constructor() {
+        super();
+    }
+    async executeWithLogging(request) {
+        var _a, _b;
+        const startTime = new Date();
+        // Get LLM provider for the project
+        const llmProvider = await LlmProviderService.getLLMProviderForProject(request.projectId);
+        if (!llmProvider) {
+            throw new BadDataException("No LLM provider configured for this project. Please configure an LLM provider in Settings > AI > LLM Providers.");
+        }
+        if (!llmProvider.llmType) {
+            throw new BadDataException("LLM provider type is not configured properly.");
+        }
+        // Create log entry (will be updated after completion)
+        const logEntry = new LlmLog();
+        logEntry.projectId = request.projectId;
+        logEntry.isGlobalProvider = llmProvider.isGlobalLlm || false;
+        logEntry.feature = request.feature;
+        logEntry.requestPrompt = request.messages
+            .map((m) => {
+            return m.content;
+        })
+            .join("\n")
+            .substring(0, 5000); // Store first 5000 chars
+        logEntry.requestStartedAt = startTime;
+        // Set optional fields only if they have values
+        if (llmProvider.id) {
+            logEntry.llmProviderId = llmProvider.id;
+        }
+        if (llmProvider.name) {
+            logEntry.llmProviderName = llmProvider.name;
+        }
+        if (llmProvider.llmType) {
+            logEntry.llmType = llmProvider.llmType;
+        }
+        if (llmProvider.modelName) {
+            logEntry.modelName = llmProvider.modelName;
+        }
+        if (request.userId) {
+            logEntry.userId = request.userId;
+        }
+        if (request.incidentId) {
+            logEntry.incidentId = request.incidentId;
+        }
+        if (request.alertId) {
+            logEntry.alertId = request.alertId;
+        }
+        if (request.scheduledMaintenanceId) {
+            logEntry.scheduledMaintenanceId = request.scheduledMaintenanceId;
+        }
+        // Check if billing should apply
+        const shouldBill = IsBillingEnabled && (llmProvider.isGlobalLlm || false);
+        // Check balance if billing enabled and using global provider
+        if (shouldBill) {
+            const project = await ProjectService.findOneById({
+                id: request.projectId,
+                select: { aiCurrentBalanceInUSDCents: true },
+                props: { isRoot: true },
+            });
+            if (!project || (project.aiCurrentBalanceInUSDCents || 0) <= 0) {
+                logEntry.status = LlmLogStatus.InsufficientBalance;
+                logEntry.statusMessage = "Insufficient AI balance";
+                logEntry.requestCompletedAt = new Date();
+                logEntry.durationMs = new Date().getTime() - startTime.getTime();
+                await LlmLogService.create({
+                    data: logEntry,
+                    props: { isRoot: true },
+                });
+                throw new BadDataException("Insufficient AI balance. Please recharge your AI balance in Project Settings > Billing.");
+            }
+        }
+        try {
+            // Build LLM config
+            const llmConfig = {
+                llmType: llmProvider.llmType,
+            };
+            if (llmProvider.apiKey) {
+                llmConfig.apiKey = llmProvider.apiKey;
+            }
+            if (llmProvider.baseUrl) {
+                llmConfig.baseUrl = llmProvider.baseUrl.toString();
+            }
+            if (llmProvider.modelName) {
+                llmConfig.modelName = llmProvider.modelName;
+            }
+            // Execute LLM call
+            const response = await LLMService.getCompletion({
+                llmProviderConfig: llmConfig,
+                messages: request.messages,
+                temperature: (_a = request.temperature) !== null && _a !== void 0 ? _a : 0.7,
+            });
+            const endTime = new Date();
+            // Update log with success info
+            logEntry.status = LlmLogStatus.Success;
+            logEntry.totalTokens = ((_b = response.usage) === null || _b === void 0 ? void 0 : _b.totalTokens) || 0;
+            logEntry.responsePreview = response.content.substring(0, 2000); // Store first 2000 chars
+            logEntry.requestCompletedAt = endTime;
+            logEntry.durationMs = endTime.getTime() - startTime.getTime();
+            // Calculate and apply costs if using global provider with billing enabled
+            if (shouldBill && response.usage) {
+                const totalCost = Math.ceil((response.usage.totalTokens / 1000000) *
+                    (llmProvider.costPerMillionTokensInUSDCents || 0));
+                logEntry.costInUSDCents = totalCost;
+                logEntry.wasBilled = true;
+                // Deduct from project balance
+                if (totalCost > 0) {
+                    const project = await ProjectService.findOneById({
+                        id: request.projectId,
+                        select: { aiCurrentBalanceInUSDCents: true },
+                        props: { isRoot: true },
+                    });
+                    if (project) {
+                        const newBalance = Math.max(0, (project.aiCurrentBalanceInUSDCents || 0) - totalCost);
+                        await ProjectService.updateOneById({
+                            id: request.projectId,
+                            data: {
+                                aiCurrentBalanceInUSDCents: newBalance,
+                            },
+                            props: { isRoot: true },
+                        });
+                    }
+                    // Check if auto-recharge is needed (do this async, don't wait)
+                    AIBillingService.rechargeIfBalanceIsLow(request.projectId).catch((err) => {
+                        logger.error("Error during AI balance auto-recharge check:");
+                        logger.error(err);
+                    });
+                }
+            }
+            // Save log entry
+            const savedLog = await LlmLogService.create({
+                data: logEntry,
+                props: { isRoot: true },
+            });
+            return {
+                content: response.content,
+                llmLog: savedLog,
+            };
+        }
+        catch (error) {
+            // Log the error
+            logEntry.status = LlmLogStatus.Error;
+            logEntry.statusMessage =
+                error instanceof Error ? error.message : String(error);
+            logEntry.requestCompletedAt = new Date();
+            logEntry.durationMs = new Date().getTime() - startTime.getTime();
+            await LlmLogService.create({
+                data: logEntry,
+                props: { isRoot: true },
+            });
+            throw error;
+        }
+    }
+}
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "executeWithLogging", null);
+export default new Service();
+//# sourceMappingURL=AIService.js.map

package/build/dist/Server/Services/AIService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AIService.js","sourceRoot":"","sources":["../../../../Server/Services/AIService.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,WAAW,MAAM,eAAe,CAAC;AACxC,OAAO,kBAAkB,MAAM,sBAAsB,CAAC;AACtD,OAAO,aAAa,MAAM,iBAAiB,CAAC;AAC5C,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAE9C,OAAO,gBAAgB,MAAM,oBAAoB,CAAC;AAClD,OAAO,UAIN,MAAM,yBAAyB,CAAC;AAEjC,OAAO,MAAM,MAAM,oCAAoC,CAAC;AACxD,OAAO,YAAY,MAAM,0BAA0B,CAAC;AAEpD,OAAO,gBAAgB,MAAM,wCAAwC,CAAC;AACtE,OAAO,WAAW,MAAM,gCAAgC,CAAC;AACzD,OAAO,MAAM,MAAM,iBAAiB,CAAC;AAmBrC,MAAM,OAAO,OAAQ,SAAQ,WAAW;IACtC;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAGY,AAAN,KAAK,CAAC,kBAAkB,CAC7B,OAAqB;;QAErB,MAAM,SAAS,GAAS,IAAI,IAAI,EAAE,CAAC;QAEnC,mCAAmC;QACnC,MAAM,WAAW,GACf,MAAM,kBAAkB,CAAC,wBAAwB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEvE,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,gBAAgB,CACxB,iHAAiH,CAClH,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,MAAM,IAAI,gBAAgB,CACxB,+CAA+C,CAChD,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,MAAM,QAAQ,GAAW,IAAI,MAAM,EAAE,CAAC;QACtC,QAAQ,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACvC,QAAQ,CAAC,gBAAgB,GAAG,WAAW,CAAC,WAAW,IAAI,KAAK,CAAC;QAC7D,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACnC,QAAQ,CAAC,aAAa,GAAG,OAAO,CAAC,QAAQ;aACtC,GAAG,CAAC,CAAC,CAAa,EAAE,EAAE;YACrB,OAAO,CAAC,CAAC,OAAO,CAAC;QACnB,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;aACV,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,yBAAyB;QAChD,QAAQ,CAAC,gBAAgB,GAAG,SAAS,CAAC;QAEtC,+CAA+C;QAC/C,IAAI,WAAW,CAAC,EAAE,EAAE,CAAC;YACnB,QAAQ,CAAC,aAAa,GAAG,WAAW,CAAC,EAAE,CAAC;QAC1C,CAAC;QACD,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC;YACrB,QAAQ,CAAC,eAAe,GAAG,WAAW,CAAC,IAAI,CAAC;QAC9C,CAAC;QACD,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;YACxB,QAAQ,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC;QACzC,CAAC;QACD,IAAI,WAAW,CAAC,SAAS,EAAE,CAAC;YAC1B,QAAQ,CAAC,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC;QAC7C,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QACnC,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QAC3C,CAAC;QACD,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACrC,CAAC;QACD,IAAI,OAAO,CAAC,sBAAsB,EAAE,CAAC;YACnC,QAAQ,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,CAAC;QACnE,CAAC;QAED,gCAAgC;QAChC,MAAM,UAAU,GACd,gBAAgB,IAAI,CAAC,WAAW,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;QAEzD,6DAA6D;QAC7D,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,OAAO,GAAmB,MAAM,cAAc,CAAC,WAAW,CAAC;gBAC/D,EAAE,EAAE,OAAO,CAAC,SAAS;gBACrB,MAAM,EAAE,EAAE,0BAA0B,EAAE,IAAI,EAAE;gBAC5C,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aACxB,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,0BAA0B,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/D,QAAQ,CAAC,MAAM,GAAG,YAAY,CAAC,mBAAmB,CAAC;gBACnD,QAAQ,CAAC,aAAa,GAAG,yBAAyB,CAAC;gBACnD,QAAQ,CAAC,kBAAkB,GAAG,IAAI,IAAI,EAAE,CAAC;gBACzC,QAAQ,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;gBAEjE,MAAM,aAAa,CAAC,MAAM,CAAC;oBACzB,IAAI,EAAE,QAAQ;oBACd,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;iBACxB,CAAC,CAAC;gBAEH,MAAM,IAAI,gBAAgB,CACxB,yFAAyF,CAC1F,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,mBAAmB;YACnB,MAAM,SAAS,GAAsB;gBACnC,OAAO,EAAE,WAAW,CAAC,OAAO;aAC7B,CAAC;YAEF,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;gBACvB,SAAS,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;YACxC,CAAC;YAED,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACxB,SAAS,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrD,CAAC;YAED,IAAI,WAAW,CAAC,SAAS,EAAE,CAAC;gBAC1B,SAAS,CAAC,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC;YAC9C,CAAC;YAED,mBAAmB;YACnB,MAAM,QAAQ,GAA0B,MAAM,UAAU,CAAC,aAAa,CAAC;gBACrE,iBAAiB,EAAE,SAAS;gBAC5B,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,WAAW,EAAE,MAAA,OAAO,CAAC,WAAW,mCAAI,GAAG;aACxC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAS,IAAI,IAAI,EAAE,CAAC;YAEjC,+BAA+B;YAC/B,QAAQ,CAAC,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC;YACvC,QAAQ,CAAC,WAAW,GAAG,CAAA,MAAA,QAAQ,CAAC,KAAK,0CAAE,WAAW,KAAI,CAAC,CAAC;YACxD,QAAQ,CAAC,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,yBAAyB;YACzF,QAAQ,CAAC,kBAAkB,GAAG,OAAO,CAAC;YACtC,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YAE9D,0EAA0E;YAC1E,IAAI,UAAU,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACjC,MAAM,SAAS,GAAW,IAAI,CAAC,IAAI,CACjC,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,GAAG,OAAS,CAAC;oBACtC,CAAC,WAAW,CAAC,8BAA8B,IAAI,CAAC,CAAC,CACpD,CAAC;gBAEF,QAAQ,CAAC,cAAc,GAAG,SAAS,CAAC;gBACpC,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC;gBAE1B,8BAA8B;gBAC9B,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;oBAClB,MAAM,OAAO,GAAmB,MAAM,cAAc,CAAC,WAAW,CAAC;wBAC/D,EAAE,EAAE,OAAO,CAAC,SAAS;wBACrB,MAAM,EAAE,EAAE,0BAA0B,EAAE,IAAI,EAAE;wBAC5C,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;qBACxB,CAAC,CAAC;oBAEH,IAAI,OAAO,EAAE,CAAC;wBACZ,MAAM,UAAU,GAAW,IAAI,CAAC,GAAG,CACjC,CAAC,EACD,CAAC,OAAO,CAAC,0BAA0B,IAAI,CAAC,CAAC,GAAG,SAAS,CACtD,CAAC;wBAEF,MAAM,cAAc,CAAC,aAAa,CAAC;4BACjC,EAAE,EAAE,OAAO,CAAC,SAAS;4BACrB,IAAI,EAAE;gCACJ,0BAA0B,EAAE,UAAU;6BACvC;4BACD,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;yBACxB,CAAC,CAAC;oBACL,CAAC;oBAED,+DAA+D;oBAC/D,gBAAgB,CAAC,sBAAsB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,CAC9D,CAAC,GAAU,EAAE,EAAE;wBACb,MAAM,CAAC,KAAK,CAAC,8CAA8C,CAAC,CAAC;wBAC7D,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACpB,CAAC,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,iBAAiB;YACjB,MAAM,QAAQ,GAAW,MAAM,aAAa,CAAC,MAAM,CAAC;gBAClD,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aACxB,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,MAAM,EAAE,QAAQ;aACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;YAChB,QAAQ,CAAC,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC;YACrC,QAAQ,CAAC,aAAa;gBACpB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,QAAQ,CAAC,kBAAkB,GAAG,IAAI,IAAI,EAAE,CAAC;YACzC,QAAQ,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YAEjE,MAAM,aAAa,CAAC,MAAM,CAAC;gBACzB,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aACxB,CAAC,CAAC;YAEH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAhMc;IADZ,WAAW,EAAE;;;;iDAgMb;AAGH,eAAe,IAAI,OAAO,EAAE,CAAC"}

package/build/dist/Server/Services/CodeRepositoryService.js

@@ -0,0 +1,9 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/CodeRepository";
+export class Service extends DatabaseService {
+    constructor() {
+        super(Model);
+    }
+}
+export default new Service();
+//# sourceMappingURL=CodeRepositoryService.js.map

package/build/dist/Server/Services/CodeRepositoryService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeRepositoryService.js","sourceRoot":"","sources":["../../../../Server/Services/CodeRepositoryService.ts"],"names":[],"mappings":"AAAA,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,MAAM,4CAA4C,CAAC;AAE/D,MAAM,OAAO,OAAQ,SAAQ,eAAsB;IACjD;QACE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;CACF;AAED,eAAe,IAAI,OAAO,EAAE,CAAC"}

package/build/dist/Server/Services/IncidentService.js

@@ -50,6 +50,9 @@ import IncidentWorkspaceMessages from "../Utils/Workspace/WorkspaceMessages/Inci
 import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
 import MetricType from "../../Models/DatabaseModels/MetricType";
 import IncidentTemplateService from "./IncidentTemplateService";
+import LLMService from "../Utils/LLM/LLMService";
+import LlmProviderService from "./LlmProviderService";
+import IncidentAIContextBuilder from "../Utils/AI/IncidentAIContextBuilder";
 export class Service extends DatabaseService {
     constructor() {
         super(Model);
@@ -1758,6 +1761,57 @@ ${incidentSeverity.name}
             logger.info(`Updated Incident ${incident.id} current state to ${latestTimeline.incidentStateId}`);
         }
     }
+    async generatePostmortemFromAI(data) {
+        // Get the incident to verify it exists and get the project ID
+        const incident = await this.findOneById({
+            id: data.incidentId,
+            select: {
+                _id: true,
+                projectId: true,
+            },
+            props: {
+                isRoot: true,
+            },
+        });
+        if (!incident || !incident.projectId) {
+            throw new BadDataException("Incident not found");
+        }
+        // Get LLM provider for the project
+        const llmProvider = await LlmProviderService.getLLMProviderForProject(incident.projectId);
+        if (!llmProvider) {
+            throw new BadDataException("No LLM provider configured for this project. Please configure an LLM provider in Settings > AI > LLM Providers.");
+        }
+        if (!llmProvider.llmType) {
+            throw new BadDataException("LLM provider type is not configured properly.");
+        }
+        // Build incident context - always include workspace messages
+        const contextData = await IncidentAIContextBuilder.buildIncidentContext({
+            incidentId: data.incidentId,
+            includeWorkspaceMessages: true,
+            workspaceMessageLimit: 500,
+        });
+        // Format context for postmortem generation
+        const aiContext = IncidentAIContextBuilder.formatIncidentContextForPostmortem(contextData, data.template);
+        // Generate postmortem using LLM
+        const llmConfig = {
+            llmType: llmProvider.llmType,
+        };
+        if (llmProvider.apiKey) {
+            llmConfig.apiKey = llmProvider.apiKey;
+        }
+        if (llmProvider.baseUrl) {
+            llmConfig.baseUrl = llmProvider.baseUrl.toString();
+        }
+        if (llmProvider.modelName) {
+            llmConfig.modelName = llmProvider.modelName;
+        }
+        const response = await LLMService.getCompletion({
+            llmProviderConfig: llmConfig,
+            messages: aiContext.messages,
+            temperature: 0.7,
+        });
+        return response.content;
+    }
 }
 __decorate([
     CaptureSpan(),
@@ -1929,5 +1983,11 @@ __decorate([
     __metadata("design:paramtypes", [Object]),
     __metadata("design:returntype", Promise)
 ], Service.prototype, "getIncidentNumber", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "generatePostmortemFromAI", null);
 export default new Service();
 //# sourceMappingURL=IncidentService.js.map