@apart-tech/intelligence-core 1.21.1 → 1.22.0

package/dist/services/chat-cost-limit-service.test.js

@@ -0,0 +1,184 @@
+import { describe, expect, it, vi } from "vitest";
+import { ChatCostLimitService } from "./chat-cost-limit-service.js";
+const ORG_ID = "org-1";
+const USER_ID = "user-1";
+function makeMockChatUsageSvc(totalCostCents = 0) {
+    return {
+        getUserUsage: vi.fn(async () => ({
+            totalMessages: 10,
+            totalTokensIn: 1000,
+            totalTokensOut: 500,
+            totalCostCents,
+        })),
+    };
+}
+function makeMockDb(opts) {
+    return {
+        chatUserCostLimit: {
+            findUnique: vi.fn(async () => opts?.userLimit ?? null),
+            upsert: vi.fn(async ({ create }) => ({
+                id: "limit-1",
+                ...create,
+                updatedAt: new Date(),
+            })),
+            deleteMany: vi.fn(async () => ({ count: 1 })),
+            findMany: vi.fn(async () => opts?.userLimit
+                ? [{ id: "limit-1", organizationId: ORG_ID, userId: USER_ID, ...opts.userLimit, updatedAt: new Date(), updatedBy: "admin" }]
+                : []),
+        },
+        organization: {
+            findUnique: vi.fn(async () => ({
+                metadata: opts?.orgMetadata ?? {},
+            })),
+        },
+    };
+}
+describe("ChatCostLimitService", () => {
+    describe("getEffectiveLimit", () => {
+        it("returns null when no config exists (unlimited)", async () => {
+            const db = makeMockDb();
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const limit = await svc.getEffectiveLimit(ORG_ID, USER_ID);
+            expect(limit).toBeNull();
+        });
+        it("returns org default when no user override exists", async () => {
+            const db = makeMockDb({
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const limit = await svc.getEffectiveLimit(ORG_ID, USER_ID);
+            expect(limit).toBe(500);
+        });
+        it("returns user override when it exists (takes precedence over org default)", async () => {
+            const db = makeMockDb({
+                userLimit: { costLimitCents: 1000 },
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const limit = await svc.getEffectiveLimit(ORG_ID, USER_ID);
+            expect(limit).toBe(1000);
+        });
+    });
+    describe("checkChatCostQuota", () => {
+        it("allows when no limit is set", async () => {
+            const db = makeMockDb();
+            const usageSvc = makeMockChatUsageSvc(200);
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
+            expect(result.allowed).toBe(true);
+            expect(result.warning).toBe(false);
+            expect(result.limitCents).toBeNull();
+            expect(result.currentCostCents).toBe(200);
+        });
+        it("allows when under limit with no warning", async () => {
+            const db = makeMockDb({
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc(200); // 200/500 = 40%
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
+            expect(result.allowed).toBe(true);
+            expect(result.warning).toBe(false);
+            expect(result.limitCents).toBe(500);
+        });
+        it("warns at 80% threshold", async () => {
+            const db = makeMockDb({
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc(400); // 400/500 = 80%
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
+            expect(result.allowed).toBe(true);
+            expect(result.warning).toBe(true);
+            expect(result.currentCostCents).toBe(400);
+        });
+        it("blocks at 100%", async () => {
+            const db = makeMockDb({
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc(500); // 500/500 = 100%
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
+            expect(result.allowed).toBe(false);
+            expect(result.warning).toBe(true);
+            expect(result.limitCents).toBe(500);
+        });
+        it("blocks when over limit", async () => {
+            const db = makeMockDb({
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc(600); // 600/500 = 120%
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
+            expect(result.allowed).toBe(false);
+            expect(result.warning).toBe(true);
+        });
+        it("uses user override for limit check", async () => {
+            const db = makeMockDb({
+                userLimit: { costLimitCents: 1000 },
+                orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
+            });
+            const usageSvc = makeMockChatUsageSvc(600); // 600/1000 = 60%, under 80%
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
+            expect(result.allowed).toBe(true);
+            expect(result.warning).toBe(false);
+            expect(result.limitCents).toBe(1000);
+        });
+    });
+    describe("setUserLimit", () => {
+        it("upserts a per-user limit", async () => {
+            const db = makeMockDb();
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const result = await svc.setUserLimit(ORG_ID, USER_ID, 1000, "admin@test.com");
+            expect(result.costLimitCents).toBe(1000);
+            expect(db.chatUserCostLimit.upsert).toHaveBeenCalled();
+        });
+    });
+    describe("removeUserLimit", () => {
+        it("deletes the per-user limit", async () => {
+            const db = makeMockDb({ userLimit: { costLimitCents: 1000 } });
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            await svc.removeUserLimit(ORG_ID, USER_ID);
+            expect(db.chatUserCostLimit.deleteMany).toHaveBeenCalledWith({
+                where: { organizationId: ORG_ID, userId: USER_ID },
+            });
+        });
+    });
+    describe("listUserLimits", () => {
+        it("returns all overrides for org", async () => {
+            const db = makeMockDb({ userLimit: { costLimitCents: 1000 } });
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const limits = await svc.listUserLimits(ORG_ID);
+            expect(limits).toHaveLength(1);
+            expect(limits[0].costLimitCents).toBe(1000);
+        });
+        it("returns empty array when no overrides", async () => {
+            const db = makeMockDb();
+            const usageSvc = makeMockChatUsageSvc();
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const svc = new ChatCostLimitService(db, usageSvc);
+            const limits = await svc.listUserLimits(ORG_ID);
+            expect(limits).toHaveLength(0);
+        });
+    });
+});
+//# sourceMappingURL=chat-cost-limit-service.test.js.map

package/dist/services/chat-cost-limit-service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-cost-limit-service.test.js","sourceRoot":"","sources":["../../src/services/chat-cost-limit-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAc,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AAGpE,MAAM,MAAM,GAAG,OAAO,CAAC;AACvB,MAAM,OAAO,GAAG,QAAQ,CAAC;AAEzB,SAAS,oBAAoB,CAAC,cAAc,GAAG,CAAC;IAC9C,OAAO;QACL,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAC/B,aAAa,EAAE,EAAE;YACjB,aAAa,EAAE,IAAI;YACnB,cAAc,EAAE,GAAG;YACnB,cAAc;SACf,CAAC,CAAC;KAC2B,CAAC;AACnC,CAAC;AAED,SAAS,UAAU,CAAC,IAGnB;IACC,OAAO;QACL,iBAAiB,EAAE;YACjB,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC;YACtD,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,MAAM,EAAuC,EAAE,EAAE,CAAC,CAAC;gBACxE,EAAE,EAAE,SAAS;gBACb,GAAG,MAAM;gBACT,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC,CAAC;YACH,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7C,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CACzB,IAAI,EAAE,SAAS;gBACb,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;gBAC5H,CAAC,CAAC,EAAE,CACP;SACF;QACD,YAAY,EAAE;YACZ,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC7B,QAAQ,EAAE,IAAI,EAAE,WAAW,IAAI,EAAE;aAClC,CAAC,CAAC;SACJ;KACF,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,SAAS,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE;gBACnC,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;YAC3C,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB;YAC5D,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB;YAC5D,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;YAC9B,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,iBAAiB;YAC7D,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,iBAAiB;YAC7D,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,EAAE,GAAG,UAAU,CAAC;gBACpB,SAAS,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE;gBACnC,WAAW,EAAE,EAAE,UAAU,EAAE,EAAE,4BAA4B,EAAE,GAAG,EAAE,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,4BAA4B;YACxE,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;YAC/E,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAC,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,EAAE,GAAG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/D,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,GAAG,CAAC,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC3C,MAAM,CAAC,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;gBAC3D,KAAK,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE;aACnD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,EAAE,GAAG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/D,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,8DAA8D;YAC9D,MAAM,GAAG,GAAG,IAAI,oBAAoB,CAAC,EAAS,EAAE,QAAQ,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/chat-usage-service.d.ts

@@ -11,6 +11,13 @@ export interface ChatUsageSummary {
     totalTokensOut: number;
     totalCostCents: number;
 }
+export interface ChatModelUsage {
+    model: string;
+    totalMessages: number;
+    totalTokensIn: number;
+    totalTokensOut: number;
+    totalCostCents: number;
+}
 /**
  * Extends UsageService with chat-specific metrics. Uses the existing
  * incrementMetric() pattern from UsageService to track:
@@ -32,5 +39,13 @@ export declare class ChatUsageService {
         start: Date;
         end: Date;
     }): Promise<ChatUsageSummary>;
+    getUserUsageByModel(organizationId: string, userId: string, period?: {
+        start: Date;
+        end: Date;
+    }): Promise<ChatModelUsage[]>;
+    getOrgUsageByModel(organizationId: string, period?: {
+        start: Date;
+        end: Date;
+    }): Promise<ChatModelUsage[]>;
 }
 //# sourceMappingURL=chat-usage-service.d.ts.map

package/dist/services/chat-usage-service.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-usage-service.d.ts","sourceRoot":"","sources":["../../src/services/chat-usage-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;;GAOG;AACH,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,YAAY;IACpB,OAAO,CAAC,EAAE;gBADF,YAAY,EAAE,YAAY,EAC1B,EAAE,EAAE,OAAO,gBAAgB,EAAE,YAAY;IAG7C,kBAAkB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IAS/D,YAAY,CAChB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,EACd,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,GAAG,EAAE,IAAI,CAAA;KAAE,GAClC,OAAO,CAAC,gBAAgB,CAAC;IAqCtB,WAAW,CACf,cAAc,EAAE,MAAM,EACtB,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,GAAG,EAAE,IAAI,CAAA;KAAE,GAClC,OAAO,CAAC,gBAAgB,CAAC;CAgC7B"}
+{"version":3,"file":"chat-usage-service.d.ts","sourceRoot":"","sources":["../../src/services/chat-usage-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;;GAOG;AACH,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,YAAY;IACpB,OAAO,CAAC,EAAE;gBADF,YAAY,EAAE,YAAY,EAC1B,EAAE,EAAE,OAAO,gBAAgB,EAAE,YAAY;IAG7C,kBAAkB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IAS/D,YAAY,CAChB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,EACd,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,GAAG,EAAE,IAAI,CAAA;KAAE,GAClC,OAAO,CAAC,gBAAgB,CAAC;IAqCtB,WAAW,CACf,cAAc,EAAE,MAAM,EACtB,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,GAAG,EAAE,IAAI,CAAA;KAAE,GAClC,OAAO,CAAC,gBAAgB,CAAC;IAiCtB,mBAAmB,CACvB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,EACd,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,GAAG,EAAE,IAAI,CAAA;KAAE,GAClC,OAAO,CAAC,cAAc,EAAE,CAAC;IAoCtB,kBAAkB,CACtB,cAAc,EAAE,MAAM,EACtB,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,GAAG,EAAE,IAAI,CAAA;KAAE,GAClC,OAAO,CAAC,cAAc,EAAE,CAAC;CAkC7B"}

package/dist/services/chat-usage-service.js

@@ -85,5 +85,70 @@ export class ChatUsageService {
             totalCostCents: result._sum.costCents ?? 0,
         };
     }
+    async getUserUsageByModel(organizationId, userId, period) {
+        const where = {
+            organizationId,
+            session: { userId },
+            model: { not: null },
+        };
+        if (period) {
+            where.createdAt = { gte: period.start, lte: period.end };
+        }
+        else {
+            const now = new Date();
+            where.createdAt = {
+                gte: new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1)),
+            };
+        }
+        const groups = await this.db.chatMessage.groupBy({
+            by: ["model"],
+            where,
+            _count: { id: true },
+            _sum: {
+                tokensIn: true,
+                tokensOut: true,
+                costCents: true,
+            },
+        });
+        return groups.map((g) => ({
+            model: g.model,
+            totalMessages: g._count.id,
+            totalTokensIn: g._sum.tokensIn ?? 0,
+            totalTokensOut: g._sum.tokensOut ?? 0,
+            totalCostCents: g._sum.costCents ?? 0,
+        }));
+    }
+    async getOrgUsageByModel(organizationId, period) {
+        const where = {
+            organizationId,
+            model: { not: null },
+        };
+        if (period) {
+            where.createdAt = { gte: period.start, lte: period.end };
+        }
+        else {
+            const now = new Date();
+            where.createdAt = {
+                gte: new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1)),
+            };
+        }
+        const groups = await this.db.chatMessage.groupBy({
+            by: ["model"],
+            where,
+            _count: { id: true },
+            _sum: {
+                tokensIn: true,
+                tokensOut: true,
+                costCents: true,
+            },
+        });
+        return groups.map((g) => ({
+            model: g.model,
+            totalMessages: g._count.id,
+            totalTokensIn: g._sum.tokensIn ?? 0,
+            totalTokensOut: g._sum.tokensOut ?? 0,
+            totalCostCents: g._sum.costCents ?? 0,
+        }));
+    }
 }
 //# sourceMappingURL=chat-usage-service.js.map

package/dist/services/chat-usage-service.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-usage-service.js","sourceRoot":"","sources":["../../src/services/chat-usage-service.ts"],"names":[],"mappings":"AAgBA;;;;;;;GAOG;AACH,MAAM,OAAO,gBAAgB;IAEjB;IACA;IAFV,YACU,YAA0B,EAC1B,EAAyC;QADzC,iBAAY,GAAZ,YAAY,CAAc;QAC1B,OAAE,GAAF,EAAE,CAAuC;IAChD,CAAC;IAEJ,KAAK,CAAC,kBAAkB,CAAC,KAA4B;QACnD,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC;YAC5D,qEAAqE;YACrE,kEAAkE;YAClE,iCAAiC;SAClC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,cAAsB,EACtB,MAAc,EACd,MAAmC;QAEnC,MAAM,KAAK,GAAQ;YACjB,cAAc;YACd,OAAO,EAAE,EAAE,MAAM,EAAE;SACpB,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,MAAM,CAAC,KAAK;gBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;aAChB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,2BAA2B;YAC3B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;aACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,SAAS,CAAC;YACjD,KAAK;YACL,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAC;QAEH,OAAO;YACL,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE;YAC/B,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC;YACxC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;YAC1C,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;SAC3C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,cAAsB,EACtB,MAAmC;QAEnC,MAAM,KAAK,GAAQ,EAAE,cAAc,EAAE,CAAC;QAEtC,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,MAAM,CAAC,KAAK;gBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;aAChB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;aACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,SAAS,CAAC;YACjD,KAAK;YACL,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAC;QAEH,OAAO;YACL,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE;YAC/B,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC;YACxC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;YAC1C,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;SAC3C,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"chat-usage-service.js","sourceRoot":"","sources":["../../src/services/chat-usage-service.ts"],"names":[],"mappings":"AAwBA;;;;;;;GAOG;AACH,MAAM,OAAO,gBAAgB;IAEjB;IACA;IAFV,YACU,YAA0B,EAC1B,EAAyC;QADzC,iBAAY,GAAZ,YAAY,CAAc;QAC1B,OAAE,GAAF,EAAE,CAAuC;IAChD,CAAC;IAEJ,KAAK,CAAC,kBAAkB,CAAC,KAA4B;QACnD,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC;YAC5D,qEAAqE;YACrE,kEAAkE;YAClE,iCAAiC;SAClC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,cAAsB,EACtB,MAAc,EACd,MAAmC;QAEnC,MAAM,KAAK,GAAQ;YACjB,cAAc;YACd,OAAO,EAAE,EAAE,MAAM,EAAE;SACpB,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,MAAM,CAAC,KAAK;gBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;aAChB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,2BAA2B;YAC3B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;aACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,SAAS,CAAC;YACjD,KAAK;YACL,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAC;QAEH,OAAO;YACL,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE;YAC/B,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC;YACxC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;YAC1C,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;SAC3C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,cAAsB,EACtB,MAAmC;QAEnC,MAAM,KAAK,GAAQ,EAAE,cAAc,EAAE,CAAC;QAEtC,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,MAAM,CAAC,KAAK;gBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;aAChB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;aACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,SAAS,CAAC;YACjD,KAAK;YACL,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAC;QAEH,OAAO;YACL,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE;YAC/B,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC;YACxC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;YAC1C,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;SAC3C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,cAAsB,EACtB,MAAc,EACd,MAAmC;QAEnC,MAAM,KAAK,GAAQ;YACjB,cAAc;YACd,OAAO,EAAE,EAAE,MAAM,EAAE;YACnB,KAAK,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE;SACrB,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;aACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC;YAC/C,EAAE,EAAE,CAAC,OAAO,CAAC;YACb,KAAK;YACL,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACxB,KAAK,EAAE,CAAC,CAAC,KAAM;YACf,aAAa,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE;YAC1B,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC;YACnC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;YACrC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;SACtC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,cAAsB,EACtB,MAAmC;QAEnC,MAAM,KAAK,GAAQ;YACjB,cAAc;YACd,KAAK,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE;SACrB,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG;gBAChB,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;aACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC;YAC/C,EAAE,EAAE,CAAC,OAAO,CAAC;YACb,KAAK;YACL,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;YACpB,IAAI,EAAE;gBACJ,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACxB,KAAK,EAAE,CAAC,CAAC,KAAM;YACf,aAAa,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE;YAC1B,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC;YACnC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;YACrC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC;SACtC,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}

package/dist/services/dedup-service.d.ts

@@ -0,0 +1,132 @@
+import type { PrismaClient } from "@prisma/client";
+import { type TenantContext } from "../db/tenant.js";
+import type { NodeService } from "./node-service.js";
+import type { EdgeService } from "./edge-service.js";
+import type { TagService } from "./tag-service.js";
+export interface SimilarityPair {
+    nodeAId: string;
+    nodeBId: string;
+    similarity: number;
+}
+export interface MergeCluster {
+    nodeIds: string[];
+    minSimilarity: number;
+    avgSimilarity: number;
+    topicLabel: string;
+}
+export interface ScanResult {
+    proposalIds: string[];
+    clustersFound: number;
+    nodesScanned: number;
+}
+export interface MergeResult {
+    mergedNodeId: string;
+    archivedNodeIds: string[];
+    edgesTransferred: number;
+}
+export declare class DedupService {
+    private db;
+    private tenantCtx;
+    constructor(db: PrismaClient, tenantCtx?: TenantContext);
+    /**
+     * Scan chat nodes for similarity clusters and create merge proposals.
+     */
+    scanChatNodes(opts: {
+        threshold?: number;
+        limit?: number;
+        createdBy: string;
+        llmLabeler?: (titles: string[]) => Promise<string>;
+    }): Promise<ScanResult>;
+    /**
+     * Union-find clustering of similarity pairs.
+     * Returns clusters with min/avg similarity.
+     */
+    computeClusters(pairs: SimilarityPair[]): MergeCluster[];
+    /**
+     * List merge proposals with optional status filter and pagination.
+     */
+    listProposals(opts?: {
+        status?: string;
+        limit?: number;
+        offset?: number;
+    }): Promise<{
+        proposals: any[];
+        total: number;
+    }>;
+    /**
+     * Get a single proposal with full source node data.
+     */
+    getProposal(id: string): Promise<{
+        sourceNodes: {
+            id: string;
+            metadata: import("@prisma/client/runtime/library").JsonValue;
+            createdAt: Date;
+            type: string;
+            title: string;
+            content: string;
+            status: string;
+            createdBy: string;
+            updatedAt: Date;
+            version: number;
+            hasPii: boolean;
+            embeddingPiiMode: string | null;
+            domainId: string | null;
+            organizationId: string;
+        }[];
+        id: string;
+        metadata: import("@prisma/client/runtime/library").JsonValue;
+        createdAt: Date;
+        status: string;
+        createdBy: string;
+        organizationId: string;
+        topicLabel: string;
+        sourceNodeIds: string[];
+        mergedNodeId: string | null;
+        similarity: number;
+        resolvedBy: string | null;
+        resolvedAt: Date | null;
+    } | null>;
+    /**
+     * Update proposal status with transition validation.
+     * Allowed: pending→approved, pending→rejected, approved→merged.
+     */
+    updateProposalStatus(id: string, status: string, resolvedBy: string): Promise<{
+        id: string;
+        metadata: import("@prisma/client/runtime/library").JsonValue;
+        createdAt: Date;
+        status: string;
+        createdBy: string;
+        organizationId: string;
+        topicLabel: string;
+        sourceNodeIds: string[];
+        mergedNodeId: string | null;
+        similarity: number;
+        resolvedBy: string | null;
+        resolvedAt: Date | null;
+    }>;
+    /**
+     * Execute a merge: create merged node, transfer edges, archive sources.
+     */
+    executeMerge(proposalId: string, opts: {
+        mergedBy: string;
+        nodeService: NodeService;
+        edgeService: EdgeService;
+        title?: string;
+        content?: string;
+        llmSummarizer?: (contents: string[]) => Promise<string>;
+    }): Promise<MergeResult>;
+    /**
+     * Mark a node as canonical by applying the `canonical` tag.
+     */
+    markCanonical(nodeId: string, tagService: TagService, taggedBy: string): Promise<void>;
+    /**
+     * Remove canonical tag from a node.
+     */
+    removeCanonical(nodeId: string): Promise<void>;
+    /**
+     * Find similarity pairs among chat nodes using the embedding self-join.
+     * Scoped to nodes with capturedFromSession metadata.
+     */
+    private findChatSimilarityPairs;
+}
+//# sourceMappingURL=dedup-service.d.ts.map

package/dist/services/dedup-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dedup-service.d.ts","sourceRoot":"","sources":["../../src/services/dedup-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAwB,KAAK,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAC3E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAInD,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,UAAU;IACzB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAID,qBAAa,YAAY;IAIrB,OAAO,CAAC,EAAE;IAHZ,OAAO,CAAC,SAAS,CAAgB;gBAGvB,EAAE,EAAE,YAAY,EACxB,SAAS,CAAC,EAAE,aAAa;IAK3B;;OAEG;IACG,aAAa,CAAC,IAAI,EAAE;QACxB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;KACpD,GAAG,OAAO,CAAC,UAAU,CAAC;IAiFvB;;;OAGG;IACH,eAAe,CAAC,KAAK,EAAE,cAAc,EAAE,GAAG,YAAY,EAAE;IAwExD;;OAEG;IACG,aAAa,CAAC,IAAI,CAAC,EAAE;QACzB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,GAAG,EAAE,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAqChD;;OAEG;IACG,WAAW,CAAC,EAAE,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAW5B;;;OAGG;IACG,oBAAoB,CACxB,EAAE,EAAE,MAAM,EACV,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM;;;;;;;;;;;;;;IA0BpB;;OAEG;IACG,YAAY,CAChB,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,EAAE,WAAW,CAAC;QACzB,WAAW,EAAE,WAAW,CAAC;QACzB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;KACzD,GACA,OAAO,CAAC,WAAW,CAAC;IA8IvB;;OAEG;IACG,aAAa,CACjB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,UAAU,EACtB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAyBhB;;OAEG;IACG,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAWpD;;;OAGG;YACW,uBAAuB;CAkCtC"}