@apart-tech/intelligence-core 1.21.2 → 1.22.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.ts +5 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -0
- package/dist/index.js.map +1 -1
- package/dist/services/chat-cost-limit-service.d.ts +44 -0
- package/dist/services/chat-cost-limit-service.d.ts.map +1 -0
- package/dist/services/chat-cost-limit-service.js +95 -0
- package/dist/services/chat-cost-limit-service.js.map +1 -0
- package/dist/services/chat-cost-limit-service.test.d.ts +2 -0
- package/dist/services/chat-cost-limit-service.test.d.ts.map +1 -0
- package/dist/services/chat-cost-limit-service.test.js +184 -0
- package/dist/services/chat-cost-limit-service.test.js.map +1 -0
- package/dist/services/chat-usage-service.d.ts +15 -0
- package/dist/services/chat-usage-service.d.ts.map +1 -1
- package/dist/services/chat-usage-service.js +65 -0
- package/dist/services/chat-usage-service.js.map +1 -1
- package/dist/services/model-pricing-service.d.ts +39 -0
- package/dist/services/model-pricing-service.d.ts.map +1 -0
- package/dist/services/model-pricing-service.js +96 -0
- package/dist/services/model-pricing-service.js.map +1 -0
- package/dist/services/model-pricing-service.test.d.ts +2 -0
- package/dist/services/model-pricing-service.test.d.ts.map +1 -0
- package/dist/services/model-pricing-service.test.js +134 -0
- package/dist/services/model-pricing-service.test.js.map +1 -0
- package/dist/services/org-llm-provider-service.d.ts +1 -1
- package/dist/services/org-llm-provider-service.d.ts.map +1 -1
- package/dist/types/index.d.ts +7 -0
- package/dist/types/index.d.ts.map +1 -1
- package/package.json +1 -1
- package/prisma/schema.prisma +37 -4
package/dist/index.d.ts
CHANGED
|
@@ -91,7 +91,11 @@ export type { ChatSessionRecord, ChatMessageRecord, CreateSessionInput, CreateMe
|
|
|
91
91
|
export { ChatContextService } from "./services/chat-context-service.js";
|
|
92
92
|
export type { ChatContextOptions, ChatContextPackage } from "./services/chat-context-service.js";
|
|
93
93
|
export { ChatUsageService } from "./services/chat-usage-service.js";
|
|
94
|
-
export type { ChatMessageUsageInput, ChatUsageSummary } from "./services/chat-usage-service.js";
|
|
94
|
+
export type { ChatMessageUsageInput, ChatUsageSummary, ChatModelUsage } from "./services/chat-usage-service.js";
|
|
95
|
+
export { ModelPricingService } from "./services/model-pricing-service.js";
|
|
96
|
+
export type { ModelPricing, ModelPricingDefaultRecord } from "./services/model-pricing-service.js";
|
|
97
|
+
export { ChatCostLimitService } from "./services/chat-cost-limit-service.js";
|
|
98
|
+
export type { ChatCostQuotaResult, ChatUserCostLimitRecord } from "./services/chat-cost-limit-service.js";
|
|
95
99
|
export { ChatCaptureService } from "./services/chat-capture-service.js";
|
|
96
100
|
export type { CaptureResult } from "./services/chat-capture-service.js";
|
|
97
101
|
export { ContentPolicyService, DEFAULT_THRESHOLDS } from "./services/content-policy-service.js";
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACjE,YAAY,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACvF,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAGpD,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAGrG,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAGjE,YAAY,EACV,IAAI,EACJ,aAAa,EACb,iBAAiB,EACjB,uBAAuB,EACvB,SAAS,GACV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,yBAAyB,GAC1B,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACvF,OAAO,EACL,YAAY,EACZ,SAAS,EACT,4BAA4B,GAC7B,MAAM,mBAAmB,CAAC;AAG3B,YAAY,EACV,sBAAsB,EACtB,uBAAuB,EACvB,0BAA0B,EAC1B,4BAA4B,GAC7B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,wBAAwB,EACxB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,wBAAwB,EACxB,mBAAmB,EACnB,UAAU,EACV,qBAAqB,GACtB,MAAM,0BAA0B,CAAC;AAGlC,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAG/C,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,YAAY,EAAE,eAAe,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AACnH,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,yBAAyB,EAAE,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,YAAY,EAAE,QAAQ,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACvF,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAC5E,YAAY,EAAE,oBAAoB,EAAE,SAAS,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AACjH,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,YAAY,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,sCAAsC,CAAC;AACvG,YAAY,EAAE,kBAAkB,EAAE,yBAAyB,EAAE,MAAM,4CAA4C,CAAC;AAChH,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAC/E,YAAY,EAAE,cAAc,EAAE,sBAAsB,EAAE,MAAM,wCAAwC,CAAC;AACrG,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,YAAY,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AACjI,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,YAAY,EAAE,kBAAkB,EAAE,eAAe,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AAClJ,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,YAAY,EAAE,QAAQ,EAAE,mBAAmB,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAC7I,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,YAAY,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AAC3K,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,YAAY,EACV,WAAW,EACX,WAAW,EACX,qBAAqB,GACtB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,MAAM,0CAA0C,CAAC;AACnF,YAAY,EACV,mBAAmB,EACnB,6BAA6B,GAC9B,MAAM,0CAA0C,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAC5E,YAAY,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,MAAM,sCAAsC,CAAC;AACpI,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,YAAY,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,eAAe,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACrP,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AACzG,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAC5G,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AACrE,YAAY,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,YAAY,EAAE,MAAM,EAAE,MAAM,8BAA8B,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC1F,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,YAAY,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AAC5E,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzG,YAAY,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACvF,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,YAAY,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAClI,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAC9E,YAAY,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,YAAY,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,qCAAqC,CAAC;AACjG,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,EACpB,eAAe,EACf,kBAAkB,EAClB,uBAAuB,GACxB,MAAM,qCAAqC,CAAC;AAC7C,YAAY,EACV,qBAAqB,EACrB,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,gBAAgB,GACjB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AACrF,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,2CAA2C,CAAC;AACjG,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,YAAY,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAC9E,YAAY,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,uCAAuC,CAAC;AAGlG,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAC/E,YAAY,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,wCAAwC,CAAC;AACnJ,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,YAAY,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AACzF,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACpJ,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,YAAY,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACjG,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,YAAY,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACjE,YAAY,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACvF,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAGpD,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAGrG,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAGjE,YAAY,EACV,IAAI,EACJ,aAAa,EACb,iBAAiB,EACjB,uBAAuB,EACvB,SAAS,GACV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,yBAAyB,GAC1B,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACvF,OAAO,EACL,YAAY,EACZ,SAAS,EACT,4BAA4B,GAC7B,MAAM,mBAAmB,CAAC;AAG3B,YAAY,EACV,sBAAsB,EACtB,uBAAuB,EACvB,0BAA0B,EAC1B,4BAA4B,GAC7B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,wBAAwB,EACxB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,wBAAwB,EACxB,mBAAmB,EACnB,UAAU,EACV,qBAAqB,GACtB,MAAM,0BAA0B,CAAC;AAGlC,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAG/C,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,YAAY,EAAE,eAAe,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AACnH,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,yBAAyB,EAAE,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,YAAY,EAAE,QAAQ,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACvF,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAC5E,YAAY,EAAE,oBAAoB,EAAE,SAAS,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AACjH,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,YAAY,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,sCAAsC,CAAC;AACvG,YAAY,EAAE,kBAAkB,EAAE,yBAAyB,EAAE,MAAM,4CAA4C,CAAC;AAChH,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAC/E,YAAY,EAAE,cAAc,EAAE,sBAAsB,EAAE,MAAM,wCAAwC,CAAC;AACrG,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,YAAY,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AACjI,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,YAAY,EAAE,kBAAkB,EAAE,eAAe,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AAClJ,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,YAAY,EAAE,QAAQ,EAAE,mBAAmB,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAC7I,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,YAAY,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AAC3K,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,YAAY,EACV,WAAW,EACX,WAAW,EACX,qBAAqB,GACtB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,MAAM,0CAA0C,CAAC;AACnF,YAAY,EACV,mBAAmB,EACnB,6BAA6B,GAC9B,MAAM,0CAA0C,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAC5E,YAAY,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,MAAM,sCAAsC,CAAC;AACpI,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,YAAY,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,eAAe,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACrP,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AACzG,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAC5G,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AACrE,YAAY,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,YAAY,EAAE,MAAM,EAAE,MAAM,8BAA8B,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC1F,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,YAAY,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AAC5E,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzG,YAAY,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACvF,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,YAAY,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAClI,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAC9E,YAAY,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,YAAY,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,qCAAqC,CAAC;AACjG,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,EACpB,eAAe,EACf,kBAAkB,EAClB,uBAAuB,GACxB,MAAM,qCAAqC,CAAC;AAC7C,YAAY,EACV,qBAAqB,EACrB,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,gBAAgB,GACjB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AACrF,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,2CAA2C,CAAC;AACjG,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,YAAY,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAC9E,YAAY,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,uCAAuC,CAAC;AAGlG,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAC/E,YAAY,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,wCAAwC,CAAC;AACnJ,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,YAAY,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AACzF,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACpJ,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,YAAY,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACjG,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,YAAY,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAChH,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,YAAY,EAAE,YAAY,EAAE,yBAAyB,EAAE,MAAM,qCAAqC,CAAC;AACnG,OAAO,EAAE,oBAAoB,EAAE,MAAM,uCAAuC,CAAC;AAC7E,YAAY,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,MAAM,uCAAuC,CAAC;AAC1G,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,YAAY,EAAE,aAAa,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAChG,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,eAAe,EAAE,iBAAiB,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,sCAAsC,CAAC;AAC/K,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAChK,YAAY,EAAE,8BAA8B,EAAE,MAAM,sBAAsB,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,YAAY,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACpG,YAAY,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,YAAY,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -59,6 +59,8 @@ export { LlmRouterService } from "./services/llm-router-service.js";
|
|
|
59
59
|
export { ChatService } from "./services/chat-service.js";
|
|
60
60
|
export { ChatContextService } from "./services/chat-context-service.js";
|
|
61
61
|
export { ChatUsageService } from "./services/chat-usage-service.js";
|
|
62
|
+
export { ModelPricingService } from "./services/model-pricing-service.js";
|
|
63
|
+
export { ChatCostLimitService } from "./services/chat-cost-limit-service.js";
|
|
62
64
|
export { ChatCaptureService } from "./services/chat-capture-service.js";
|
|
63
65
|
export { ContentPolicyService, DEFAULT_THRESHOLDS } from "./services/content-policy-service.js";
|
|
64
66
|
export { cosineSimilarity } from "./lib/vector.js";
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,WAAW;AACX,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAGvF,SAAS;AACT,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAErG,MAAM;AACN,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAW9C,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,yBAAyB,GAC1B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,YAAY,EACZ,SAAS,EACT,4BAA4B,GAC7B,MAAM,mBAAmB,CAAC;AAS3B,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,wBAAwB,EACxB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,wBAAwB,EACxB,mBAAmB,EACnB,UAAU,EACV,qBAAqB,GACtB,MAAM,0BAA0B,CAAC;AAElC,QAAQ;AACR,cAAc,kBAAkB,CAAC;AAEjC,SAAS;AACT,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,WAAW;AACX,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,yBAAyB,EAAE,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAE5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAG3E,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAE/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAE3E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAE3E,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAElF,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AAMtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,0CAA0C,CAAC;AAKnF,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAE5E,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAElE,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAE3D,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAErE,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAE7D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAE3D,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzG,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAE/D,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AAEjE,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAE9E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,EACpB,eAAe,EACf,kBAAkB,EAClB,uBAAuB,GACxB,MAAM,qCAAqC,CAAC;AAQ7C,OAAO,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AAErF,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAEvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAG9E,eAAe;AACf,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAE/E,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AAEpE,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AAEpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAEhG,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAEnD,YAAY;AACZ,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAEhK,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAEpE,OAAO,EAAE,wBAAwB,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEpG,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,WAAW;AACX,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAGvF,SAAS;AACT,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAErG,MAAM;AACN,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAW9C,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,yBAAyB,GAC1B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,YAAY,EACZ,SAAS,EACT,4BAA4B,GAC7B,MAAM,mBAAmB,CAAC;AAS3B,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,wBAAwB,EACxB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,wBAAwB,EACxB,mBAAmB,EACnB,UAAU,EACV,qBAAqB,GACtB,MAAM,0BAA0B,CAAC;AAElC,QAAQ;AACR,cAAc,kBAAkB,CAAC;AAEjC,SAAS;AACT,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,WAAW;AACX,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,yBAAyB,EAAE,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAE5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAG3E,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAE/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAE3E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAE3E,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAElF,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AAMtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,0CAA0C,CAAC;AAKnF,OAAO,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AAE5E,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAElE,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAE3D,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAErE,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAE7D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAE3D,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzG,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAE/D,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AAEjE,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAE9E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,EACpB,eAAe,EACf,kBAAkB,EAClB,uBAAuB,GACxB,MAAM,qCAAqC,CAAC;AAQ7C,OAAO,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AAErF,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAEvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAG9E,eAAe;AACf,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAE/E,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AAEpE,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AAEpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,OAAO,EAAE,oBAAoB,EAAE,MAAM,uCAAuC,CAAC;AAE7E,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAEhG,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAEnD,YAAY;AACZ,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAEhK,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAEpE,OAAO,EAAE,wBAAwB,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEpG,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import type { PrismaClient } from "@prisma/client";
|
|
2
|
+
import type { ChatUsageService } from "./chat-usage-service.js";
|
|
3
|
+
export interface ChatCostQuotaResult {
|
|
4
|
+
allowed: boolean;
|
|
5
|
+
warning: boolean;
|
|
6
|
+
currentCostCents: number;
|
|
7
|
+
limitCents: number | null;
|
|
8
|
+
}
|
|
9
|
+
export interface ChatUserCostLimitRecord {
|
|
10
|
+
id: string;
|
|
11
|
+
organizationId: string;
|
|
12
|
+
userId: string;
|
|
13
|
+
costLimitCents: number;
|
|
14
|
+
updatedAt: Date;
|
|
15
|
+
updatedBy: string;
|
|
16
|
+
}
|
|
17
|
+
export declare class ChatCostLimitService {
|
|
18
|
+
private db;
|
|
19
|
+
private chatUsageSvc;
|
|
20
|
+
constructor(db: PrismaClient, chatUsageSvc: ChatUsageService);
|
|
21
|
+
/**
|
|
22
|
+
* Resolve the effective cost limit for a user.
|
|
23
|
+
* Priority: per-user override > org default > null (unlimited).
|
|
24
|
+
*/
|
|
25
|
+
getEffectiveLimit(organizationId: string, userId: string): Promise<number | null>;
|
|
26
|
+
/**
|
|
27
|
+
* Check whether a user is allowed to send a chat message
|
|
28
|
+
* based on their current monthly spend vs. their effective limit.
|
|
29
|
+
*/
|
|
30
|
+
checkChatCostQuota(organizationId: string, userId: string): Promise<ChatCostQuotaResult>;
|
|
31
|
+
/**
|
|
32
|
+
* Set (upsert) a per-user cost limit override.
|
|
33
|
+
*/
|
|
34
|
+
setUserLimit(organizationId: string, userId: string, costLimitCents: number, updatedBy: string): Promise<ChatUserCostLimitRecord>;
|
|
35
|
+
/**
|
|
36
|
+
* Remove a per-user cost limit override (reverts to org default).
|
|
37
|
+
*/
|
|
38
|
+
removeUserLimit(organizationId: string, userId: string): Promise<void>;
|
|
39
|
+
/**
|
|
40
|
+
* List all per-user cost limit overrides for an org.
|
|
41
|
+
*/
|
|
42
|
+
listUserLimits(organizationId: string): Promise<ChatUserCostLimitRecord[]>;
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=chat-cost-limit-service.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-cost-limit-service.d.ts","sourceRoot":"","sources":["../../src/services/chat-cost-limit-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAGhE,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,gBAAgB,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B;AAED,MAAM,WAAW,uBAAuB;IACtC,EAAE,EAAE,MAAM,CAAC;IACX,cAAc,EAAE,MAAM,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAID,qBAAa,oBAAoB;IAE7B,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,YAAY;gBADZ,EAAE,EAAE,YAAY,EAChB,YAAY,EAAE,gBAAgB;IAGxC;;;OAGG;IACG,iBAAiB,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAsBzB;;;OAGG;IACG,kBAAkB,CACtB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,mBAAmB,CAAC;IAyB/B;;OAEG;IACG,YAAY,CAChB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,EACd,cAAc,EAAE,MAAM,EACtB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,uBAAuB,CAAC;IAmBnC;;OAEG;IACG,eAAe,CACnB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAMhB;;OAEG;IACG,cAAc,CAClB,cAAc,EAAE,MAAM,GACrB,OAAO,CAAC,uBAAuB,EAAE,CAAC;CAMtC"}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
const WARNING_THRESHOLD = 0.8;
|
|
2
|
+
export class ChatCostLimitService {
|
|
3
|
+
db;
|
|
4
|
+
chatUsageSvc;
|
|
5
|
+
constructor(db, chatUsageSvc) {
|
|
6
|
+
this.db = db;
|
|
7
|
+
this.chatUsageSvc = chatUsageSvc;
|
|
8
|
+
}
|
|
9
|
+
/**
|
|
10
|
+
* Resolve the effective cost limit for a user.
|
|
11
|
+
* Priority: per-user override > org default > null (unlimited).
|
|
12
|
+
*/
|
|
13
|
+
async getEffectiveLimit(organizationId, userId) {
|
|
14
|
+
// Check per-user override first
|
|
15
|
+
const userLimit = await this.db.chatUserCostLimit.findUnique({
|
|
16
|
+
where: {
|
|
17
|
+
organizationId_userId: { organizationId, userId },
|
|
18
|
+
},
|
|
19
|
+
});
|
|
20
|
+
if (userLimit) {
|
|
21
|
+
return userLimit.costLimitCents;
|
|
22
|
+
}
|
|
23
|
+
// Fall back to org default
|
|
24
|
+
const org = await this.db.organization.findUnique({
|
|
25
|
+
where: { id: organizationId },
|
|
26
|
+
select: { metadata: true },
|
|
27
|
+
});
|
|
28
|
+
const meta = (org?.metadata ?? {});
|
|
29
|
+
return meta.chatLimits?.defaultUserCostCentsPerMonth ?? null;
|
|
30
|
+
}
|
|
31
|
+
/**
|
|
32
|
+
* Check whether a user is allowed to send a chat message
|
|
33
|
+
* based on their current monthly spend vs. their effective limit.
|
|
34
|
+
*/
|
|
35
|
+
async checkChatCostQuota(organizationId, userId) {
|
|
36
|
+
const limitCents = await this.getEffectiveLimit(organizationId, userId);
|
|
37
|
+
// No limit configured — always allowed, no warning
|
|
38
|
+
if (limitCents === null) {
|
|
39
|
+
const usage = await this.chatUsageSvc.getUserUsage(organizationId, userId);
|
|
40
|
+
return {
|
|
41
|
+
allowed: true,
|
|
42
|
+
warning: false,
|
|
43
|
+
currentCostCents: usage.totalCostCents,
|
|
44
|
+
limitCents: null,
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
const usage = await this.chatUsageSvc.getUserUsage(organizationId, userId);
|
|
48
|
+
const currentCost = usage.totalCostCents;
|
|
49
|
+
return {
|
|
50
|
+
allowed: currentCost < limitCents,
|
|
51
|
+
warning: currentCost >= limitCents * WARNING_THRESHOLD,
|
|
52
|
+
currentCostCents: currentCost,
|
|
53
|
+
limitCents,
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
/**
|
|
57
|
+
* Set (upsert) a per-user cost limit override.
|
|
58
|
+
*/
|
|
59
|
+
async setUserLimit(organizationId, userId, costLimitCents, updatedBy) {
|
|
60
|
+
return this.db.chatUserCostLimit.upsert({
|
|
61
|
+
where: {
|
|
62
|
+
organizationId_userId: { organizationId, userId },
|
|
63
|
+
},
|
|
64
|
+
create: {
|
|
65
|
+
organizationId,
|
|
66
|
+
userId,
|
|
67
|
+
costLimitCents,
|
|
68
|
+
updatedBy,
|
|
69
|
+
},
|
|
70
|
+
update: {
|
|
71
|
+
costLimitCents,
|
|
72
|
+
updatedBy,
|
|
73
|
+
updatedAt: new Date(),
|
|
74
|
+
},
|
|
75
|
+
});
|
|
76
|
+
}
|
|
77
|
+
/**
|
|
78
|
+
* Remove a per-user cost limit override (reverts to org default).
|
|
79
|
+
*/
|
|
80
|
+
async removeUserLimit(organizationId, userId) {
|
|
81
|
+
await this.db.chatUserCostLimit.deleteMany({
|
|
82
|
+
where: { organizationId, userId },
|
|
83
|
+
});
|
|
84
|
+
}
|
|
85
|
+
/**
|
|
86
|
+
* List all per-user cost limit overrides for an org.
|
|
87
|
+
*/
|
|
88
|
+
async listUserLimits(organizationId) {
|
|
89
|
+
return this.db.chatUserCostLimit.findMany({
|
|
90
|
+
where: { organizationId },
|
|
91
|
+
orderBy: { updatedAt: "desc" },
|
|
92
|
+
});
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
//# sourceMappingURL=chat-cost-limit-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-cost-limit-service.js","sourceRoot":"","sources":["../../src/services/chat-cost-limit-service.ts"],"names":[],"mappings":"AAoBA,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAE9B,MAAM,OAAO,oBAAoB;IAErB;IACA;IAFV,YACU,EAAgB,EAChB,YAA8B;QAD9B,OAAE,GAAF,EAAE,CAAc;QAChB,iBAAY,GAAZ,YAAY,CAAkB;IACrC,CAAC;IAEJ;;;OAGG;IACH,KAAK,CAAC,iBAAiB,CACrB,cAAsB,EACtB,MAAc;QAEd,gCAAgC;QAChC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC;YAC3D,KAAK,EAAE;gBACL,qBAAqB,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE;aAClD;SACF,CAAC,CAAC;QAEH,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,SAAS,CAAC,cAAc,CAAC;QAClC,CAAC;QAED,2BAA2B;QAC3B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC;YAChD,KAAK,EAAE,EAAE,EAAE,EAAE,cAAc,EAAE;YAC7B,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC3B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,CAAC,GAAG,EAAE,QAAQ,IAAI,EAAE,CAAgB,CAAC;QAClD,OAAO,IAAI,CAAC,UAAU,EAAE,4BAA4B,IAAI,IAAI,CAAC;IAC/D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,kBAAkB,CACtB,cAAsB,EACtB,MAAc;QAEd,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAExE,mDAAmD;QACnD,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YAC3E,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,KAAK;gBACd,gBAAgB,EAAE,KAAK,CAAC,cAAc;gBACtC,UAAU,EAAE,IAAI;aACjB,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC3E,MAAM,WAAW,GAAG,KAAK,CAAC,cAAc,CAAC;QAEzC,OAAO;YACL,OAAO,EAAE,WAAW,GAAG,UAAU;YACjC,OAAO,EAAE,WAAW,IAAI,UAAU,GAAG,iBAAiB;YACtD,gBAAgB,EAAE,WAAW;YAC7B,UAAU;SACX,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,cAAsB,EACtB,MAAc,EACd,cAAsB,EACtB,SAAiB;QAEjB,OAAO,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC;YACtC,KAAK,EAAE;gBACL,qBAAqB,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE;aAClD;YACD,MAAM,EAAE;gBACN,cAAc;gBACd,MAAM;gBACN,cAAc;gBACd,SAAS;aACV;YACD,MAAM,EAAE;gBACN,cAAc;gBACd,SAAS;gBACT,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB;SACF,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,cAAsB,EACtB,MAAc;QAEd,MAAM,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC;YACzC,KAAK,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE;SAClC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAClB,cAAsB;QAEtB,OAAO,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;YACxC,KAAK,EAAE,EAAE,cAAc,EAAE;YACzB,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-cost-limit-service.test.d.ts","sourceRoot":"","sources":["../../src/services/chat-cost-limit-service.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,184 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { ChatCostLimitService } from "./chat-cost-limit-service.js";
|
|
3
|
+
const ORG_ID = "org-1";
|
|
4
|
+
const USER_ID = "user-1";
|
|
5
|
+
function makeMockChatUsageSvc(totalCostCents = 0) {
|
|
6
|
+
return {
|
|
7
|
+
getUserUsage: vi.fn(async () => ({
|
|
8
|
+
totalMessages: 10,
|
|
9
|
+
totalTokensIn: 1000,
|
|
10
|
+
totalTokensOut: 500,
|
|
11
|
+
totalCostCents,
|
|
12
|
+
})),
|
|
13
|
+
};
|
|
14
|
+
}
|
|
15
|
+
function makeMockDb(opts) {
|
|
16
|
+
return {
|
|
17
|
+
chatUserCostLimit: {
|
|
18
|
+
findUnique: vi.fn(async () => opts?.userLimit ?? null),
|
|
19
|
+
upsert: vi.fn(async ({ create }) => ({
|
|
20
|
+
id: "limit-1",
|
|
21
|
+
...create,
|
|
22
|
+
updatedAt: new Date(),
|
|
23
|
+
})),
|
|
24
|
+
deleteMany: vi.fn(async () => ({ count: 1 })),
|
|
25
|
+
findMany: vi.fn(async () => opts?.userLimit
|
|
26
|
+
? [{ id: "limit-1", organizationId: ORG_ID, userId: USER_ID, ...opts.userLimit, updatedAt: new Date(), updatedBy: "admin" }]
|
|
27
|
+
: []),
|
|
28
|
+
},
|
|
29
|
+
organization: {
|
|
30
|
+
findUnique: vi.fn(async () => ({
|
|
31
|
+
metadata: opts?.orgMetadata ?? {},
|
|
32
|
+
})),
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
describe("ChatCostLimitService", () => {
|
|
37
|
+
describe("getEffectiveLimit", () => {
|
|
38
|
+
it("returns null when no config exists (unlimited)", async () => {
|
|
39
|
+
const db = makeMockDb();
|
|
40
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
41
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
42
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
43
|
+
const limit = await svc.getEffectiveLimit(ORG_ID, USER_ID);
|
|
44
|
+
expect(limit).toBeNull();
|
|
45
|
+
});
|
|
46
|
+
it("returns org default when no user override exists", async () => {
|
|
47
|
+
const db = makeMockDb({
|
|
48
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
49
|
+
});
|
|
50
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
51
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
52
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
53
|
+
const limit = await svc.getEffectiveLimit(ORG_ID, USER_ID);
|
|
54
|
+
expect(limit).toBe(500);
|
|
55
|
+
});
|
|
56
|
+
it("returns user override when it exists (takes precedence over org default)", async () => {
|
|
57
|
+
const db = makeMockDb({
|
|
58
|
+
userLimit: { costLimitCents: 1000 },
|
|
59
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
60
|
+
});
|
|
61
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
62
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
63
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
64
|
+
const limit = await svc.getEffectiveLimit(ORG_ID, USER_ID);
|
|
65
|
+
expect(limit).toBe(1000);
|
|
66
|
+
});
|
|
67
|
+
});
|
|
68
|
+
describe("checkChatCostQuota", () => {
|
|
69
|
+
it("allows when no limit is set", async () => {
|
|
70
|
+
const db = makeMockDb();
|
|
71
|
+
const usageSvc = makeMockChatUsageSvc(200);
|
|
72
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
73
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
74
|
+
const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
|
|
75
|
+
expect(result.allowed).toBe(true);
|
|
76
|
+
expect(result.warning).toBe(false);
|
|
77
|
+
expect(result.limitCents).toBeNull();
|
|
78
|
+
expect(result.currentCostCents).toBe(200);
|
|
79
|
+
});
|
|
80
|
+
it("allows when under limit with no warning", async () => {
|
|
81
|
+
const db = makeMockDb({
|
|
82
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
83
|
+
});
|
|
84
|
+
const usageSvc = makeMockChatUsageSvc(200); // 200/500 = 40%
|
|
85
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
86
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
87
|
+
const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
|
|
88
|
+
expect(result.allowed).toBe(true);
|
|
89
|
+
expect(result.warning).toBe(false);
|
|
90
|
+
expect(result.limitCents).toBe(500);
|
|
91
|
+
});
|
|
92
|
+
it("warns at 80% threshold", async () => {
|
|
93
|
+
const db = makeMockDb({
|
|
94
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
95
|
+
});
|
|
96
|
+
const usageSvc = makeMockChatUsageSvc(400); // 400/500 = 80%
|
|
97
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
98
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
99
|
+
const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
|
|
100
|
+
expect(result.allowed).toBe(true);
|
|
101
|
+
expect(result.warning).toBe(true);
|
|
102
|
+
expect(result.currentCostCents).toBe(400);
|
|
103
|
+
});
|
|
104
|
+
it("blocks at 100%", async () => {
|
|
105
|
+
const db = makeMockDb({
|
|
106
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
107
|
+
});
|
|
108
|
+
const usageSvc = makeMockChatUsageSvc(500); // 500/500 = 100%
|
|
109
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
110
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
111
|
+
const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
|
|
112
|
+
expect(result.allowed).toBe(false);
|
|
113
|
+
expect(result.warning).toBe(true);
|
|
114
|
+
expect(result.limitCents).toBe(500);
|
|
115
|
+
});
|
|
116
|
+
it("blocks when over limit", async () => {
|
|
117
|
+
const db = makeMockDb({
|
|
118
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
119
|
+
});
|
|
120
|
+
const usageSvc = makeMockChatUsageSvc(600); // 600/500 = 120%
|
|
121
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
122
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
123
|
+
const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
|
|
124
|
+
expect(result.allowed).toBe(false);
|
|
125
|
+
expect(result.warning).toBe(true);
|
|
126
|
+
});
|
|
127
|
+
it("uses user override for limit check", async () => {
|
|
128
|
+
const db = makeMockDb({
|
|
129
|
+
userLimit: { costLimitCents: 1000 },
|
|
130
|
+
orgMetadata: { chatLimits: { defaultUserCostCentsPerMonth: 500 } },
|
|
131
|
+
});
|
|
132
|
+
const usageSvc = makeMockChatUsageSvc(600); // 600/1000 = 60%, under 80%
|
|
133
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
134
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
135
|
+
const result = await svc.checkChatCostQuota(ORG_ID, USER_ID);
|
|
136
|
+
expect(result.allowed).toBe(true);
|
|
137
|
+
expect(result.warning).toBe(false);
|
|
138
|
+
expect(result.limitCents).toBe(1000);
|
|
139
|
+
});
|
|
140
|
+
});
|
|
141
|
+
describe("setUserLimit", () => {
|
|
142
|
+
it("upserts a per-user limit", async () => {
|
|
143
|
+
const db = makeMockDb();
|
|
144
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
145
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
146
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
147
|
+
const result = await svc.setUserLimit(ORG_ID, USER_ID, 1000, "admin@test.com");
|
|
148
|
+
expect(result.costLimitCents).toBe(1000);
|
|
149
|
+
expect(db.chatUserCostLimit.upsert).toHaveBeenCalled();
|
|
150
|
+
});
|
|
151
|
+
});
|
|
152
|
+
describe("removeUserLimit", () => {
|
|
153
|
+
it("deletes the per-user limit", async () => {
|
|
154
|
+
const db = makeMockDb({ userLimit: { costLimitCents: 1000 } });
|
|
155
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
156
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
157
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
158
|
+
await svc.removeUserLimit(ORG_ID, USER_ID);
|
|
159
|
+
expect(db.chatUserCostLimit.deleteMany).toHaveBeenCalledWith({
|
|
160
|
+
where: { organizationId: ORG_ID, userId: USER_ID },
|
|
161
|
+
});
|
|
162
|
+
});
|
|
163
|
+
});
|
|
164
|
+
describe("listUserLimits", () => {
|
|
165
|
+
it("returns all overrides for org", async () => {
|
|
166
|
+
const db = makeMockDb({ userLimit: { costLimitCents: 1000 } });
|
|
167
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
168
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
169
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
170
|
+
const limits = await svc.listUserLimits(ORG_ID);
|
|
171
|
+
expect(limits).toHaveLength(1);
|
|
172
|
+
expect(limits[0].costLimitCents).toBe(1000);
|
|
173
|
+
});
|
|
174
|
+
it("returns empty array when no overrides", async () => {
|
|
175
|
+
const db = makeMockDb();
|
|
176
|
+
const usageSvc = makeMockChatUsageSvc();
|
|
177
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
178
|
+
const svc = new ChatCostLimitService(db, usageSvc);
|
|
179
|
+
const limits = await svc.listUserLimits(ORG_ID);
|
|
180
|
+
expect(limits).toHaveLength(0);
|
|
181
|
+
});
|
|
182
|
+
});
|
|
183
|
+
});
|
|
184
|
+
//# sourceMappingURL=chat-cost-limit-service.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -11,6 +11,13 @@ export interface ChatUsageSummary {
|
|
|
11
11
|
totalTokensOut: number;
|
|
12
12
|
totalCostCents: number;
|
|
13
13
|
}
|
|
14
|
+
export interface ChatModelUsage {
|
|
15
|
+
model: string;
|
|
16
|
+
totalMessages: number;
|
|
17
|
+
totalTokensIn: number;
|
|
18
|
+
totalTokensOut: number;
|
|
19
|
+
totalCostCents: number;
|
|
20
|
+
}
|
|
14
21
|
/**
|
|
15
22
|
* Extends UsageService with chat-specific metrics. Uses the existing
|
|
16
23
|
* incrementMetric() pattern from UsageService to track:
|
|
@@ -32,5 +39,13 @@ export declare class ChatUsageService {
|
|
|
32
39
|
start: Date;
|
|
33
40
|
end: Date;
|
|
34
41
|
}): Promise<ChatUsageSummary>;
|
|
42
|
+
getUserUsageByModel(organizationId: string, userId: string, period?: {
|
|
43
|
+
start: Date;
|
|
44
|
+
end: Date;
|
|
45
|
+
}): Promise<ChatModelUsage[]>;
|
|
46
|
+
getOrgUsageByModel(organizationId: string, period?: {
|
|
47
|
+
start: Date;
|
|
48
|
+
end: Date;
|
|
49
|
+
}): Promise<ChatModelUsage[]>;
|
|
35
50
|
}
|
|
36
51
|
//# sourceMappingURL=chat-usage-service.d.ts.map
|
|
@@ -1 +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;
|
|
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,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"}
|
|
@@ -85,5 +85,70 @@ export class ChatUsageService {
|
|
|
85
85
|
totalCostCents: result._sum.costCents ?? 0,
|
|
86
86
|
};
|
|
87
87
|
}
|
|
88
|
+
async getUserUsageByModel(organizationId, userId, period) {
|
|
89
|
+
const where = {
|
|
90
|
+
organizationId,
|
|
91
|
+
session: { userId },
|
|
92
|
+
model: { not: null },
|
|
93
|
+
};
|
|
94
|
+
if (period) {
|
|
95
|
+
where.createdAt = { gte: period.start, lte: period.end };
|
|
96
|
+
}
|
|
97
|
+
else {
|
|
98
|
+
const now = new Date();
|
|
99
|
+
where.createdAt = {
|
|
100
|
+
gte: new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1)),
|
|
101
|
+
};
|
|
102
|
+
}
|
|
103
|
+
const groups = await this.db.chatMessage.groupBy({
|
|
104
|
+
by: ["model"],
|
|
105
|
+
where,
|
|
106
|
+
_count: { id: true },
|
|
107
|
+
_sum: {
|
|
108
|
+
tokensIn: true,
|
|
109
|
+
tokensOut: true,
|
|
110
|
+
costCents: true,
|
|
111
|
+
},
|
|
112
|
+
});
|
|
113
|
+
return groups.map((g) => ({
|
|
114
|
+
model: g.model,
|
|
115
|
+
totalMessages: g._count.id,
|
|
116
|
+
totalTokensIn: g._sum.tokensIn ?? 0,
|
|
117
|
+
totalTokensOut: g._sum.tokensOut ?? 0,
|
|
118
|
+
totalCostCents: g._sum.costCents ?? 0,
|
|
119
|
+
}));
|
|
120
|
+
}
|
|
121
|
+
async getOrgUsageByModel(organizationId, period) {
|
|
122
|
+
const where = {
|
|
123
|
+
organizationId,
|
|
124
|
+
model: { not: null },
|
|
125
|
+
};
|
|
126
|
+
if (period) {
|
|
127
|
+
where.createdAt = { gte: period.start, lte: period.end };
|
|
128
|
+
}
|
|
129
|
+
else {
|
|
130
|
+
const now = new Date();
|
|
131
|
+
where.createdAt = {
|
|
132
|
+
gte: new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1)),
|
|
133
|
+
};
|
|
134
|
+
}
|
|
135
|
+
const groups = await this.db.chatMessage.groupBy({
|
|
136
|
+
by: ["model"],
|
|
137
|
+
where,
|
|
138
|
+
_count: { id: true },
|
|
139
|
+
_sum: {
|
|
140
|
+
tokensIn: true,
|
|
141
|
+
tokensOut: true,
|
|
142
|
+
costCents: true,
|
|
143
|
+
},
|
|
144
|
+
});
|
|
145
|
+
return groups.map((g) => ({
|
|
146
|
+
model: g.model,
|
|
147
|
+
totalMessages: g._count.id,
|
|
148
|
+
totalTokensIn: g._sum.tokensIn ?? 0,
|
|
149
|
+
totalTokensOut: g._sum.tokensOut ?? 0,
|
|
150
|
+
totalCostCents: g._sum.costCents ?? 0,
|
|
151
|
+
}));
|
|
152
|
+
}
|
|
88
153
|
}
|
|
89
154
|
//# sourceMappingURL=chat-usage-service.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chat-usage-service.js","sourceRoot":"","sources":["../../src/services/chat-usage-service.ts"],"names":[],"mappings":"
|
|
1
|
+
{"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"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import type { PrismaClient } from "@prisma/client";
|
|
2
|
+
export interface ModelPricing {
|
|
3
|
+
inputCentsPerMillion: number;
|
|
4
|
+
outputCentsPerMillion: number;
|
|
5
|
+
}
|
|
6
|
+
export interface ModelPricingDefaultRecord {
|
|
7
|
+
id: string;
|
|
8
|
+
model: string;
|
|
9
|
+
inputCentsPerMillion: number;
|
|
10
|
+
outputCentsPerMillion: number;
|
|
11
|
+
updatedAt: Date;
|
|
12
|
+
updatedBy: string;
|
|
13
|
+
}
|
|
14
|
+
/**
|
|
15
|
+
* Resolves model pricing through a three-tier hierarchy:
|
|
16
|
+
* 1. Org override (OrgMetadata.modelPricing)
|
|
17
|
+
* 2. Apart system default (model_pricing_defaults table)
|
|
18
|
+
* 3. Hardcoded fallback
|
|
19
|
+
*
|
|
20
|
+
* Also provides CRUD for system-level pricing defaults.
|
|
21
|
+
*/
|
|
22
|
+
export declare class ModelPricingService {
|
|
23
|
+
private db;
|
|
24
|
+
constructor(db: PrismaClient);
|
|
25
|
+
/**
|
|
26
|
+
* Resolve the effective pricing for a model within an org.
|
|
27
|
+
* Priority: org override → system default → hardcoded fallback.
|
|
28
|
+
*/
|
|
29
|
+
resolvePricing(organizationId: string, model: string): Promise<ModelPricing>;
|
|
30
|
+
/**
|
|
31
|
+
* Calculate cost in integer cents for a model invocation using resolved pricing.
|
|
32
|
+
*/
|
|
33
|
+
calculateCostCents(pricing: ModelPricing, tokensIn: number, tokensOut: number): number;
|
|
34
|
+
listSystemDefaults(): Promise<ModelPricingDefaultRecord[]>;
|
|
35
|
+
getSystemDefault(model: string): Promise<ModelPricingDefaultRecord | null>;
|
|
36
|
+
setSystemDefault(model: string, inputCentsPerMillion: number, outputCentsPerMillion: number, updatedBy: string): Promise<ModelPricingDefaultRecord>;
|
|
37
|
+
deleteSystemDefault(model: string): Promise<boolean>;
|
|
38
|
+
}
|
|
39
|
+
//# sourceMappingURL=model-pricing-service.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-pricing-service.d.ts","sourceRoot":"","sources":["../../src/services/model-pricing-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAGnD,MAAM,WAAW,YAAY;IAC3B,oBAAoB,EAAE,MAAM,CAAC;IAC7B,qBAAqB,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,yBAAyB;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,oBAAoB,EAAE,MAAM,CAAC;IAC7B,qBAAqB,EAAE,MAAM,CAAC;IAC9B,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAQD;;;;;;;GAOG;AACH,qBAAa,mBAAmB;IAClB,OAAO,CAAC,EAAE;gBAAF,EAAE,EAAE,YAAY;IAEpC;;;OAGG;IACG,cAAc,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgClF;;OAEG;IACH,kBAAkB,CAAC,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM;IAQhF,kBAAkB,IAAI,OAAO,CAAC,yBAAyB,EAAE,CAAC;IAM1D,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,GAAG,IAAI,CAAC;IAM1E,gBAAgB,CACpB,KAAK,EAAE,MAAM,EACb,oBAAoB,EAAE,MAAM,EAC5B,qBAAqB,EAAE,MAAM,EAC7B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,yBAAyB,CAAC;IAkB/B,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAQ3D"}
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
/** Hardcoded last-resort fallback for models with no configured pricing. */
|
|
2
|
+
const HARDCODED_FALLBACK = {
|
|
3
|
+
inputCentsPerMillion: 300,
|
|
4
|
+
outputCentsPerMillion: 1500,
|
|
5
|
+
};
|
|
6
|
+
/**
|
|
7
|
+
* Resolves model pricing through a three-tier hierarchy:
|
|
8
|
+
* 1. Org override (OrgMetadata.modelPricing)
|
|
9
|
+
* 2. Apart system default (model_pricing_defaults table)
|
|
10
|
+
* 3. Hardcoded fallback
|
|
11
|
+
*
|
|
12
|
+
* Also provides CRUD for system-level pricing defaults.
|
|
13
|
+
*/
|
|
14
|
+
export class ModelPricingService {
|
|
15
|
+
db;
|
|
16
|
+
constructor(db) {
|
|
17
|
+
this.db = db;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Resolve the effective pricing for a model within an org.
|
|
21
|
+
* Priority: org override → system default → hardcoded fallback.
|
|
22
|
+
*/
|
|
23
|
+
async resolvePricing(organizationId, model) {
|
|
24
|
+
// 1. Check org override
|
|
25
|
+
const org = await this.db.organization.findUnique({
|
|
26
|
+
where: { id: organizationId },
|
|
27
|
+
select: { metadata: true },
|
|
28
|
+
});
|
|
29
|
+
const meta = (org?.metadata ?? {});
|
|
30
|
+
const orgOverride = meta.modelPricing?.[model];
|
|
31
|
+
if (orgOverride) {
|
|
32
|
+
return {
|
|
33
|
+
inputCentsPerMillion: orgOverride.inputCentsPerMillion,
|
|
34
|
+
outputCentsPerMillion: orgOverride.outputCentsPerMillion,
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
// 2. Check system default
|
|
38
|
+
const systemDefault = await this.db.modelPricingDefault.findUnique({
|
|
39
|
+
where: { model },
|
|
40
|
+
});
|
|
41
|
+
if (systemDefault) {
|
|
42
|
+
return {
|
|
43
|
+
inputCentsPerMillion: systemDefault.inputCentsPerMillion,
|
|
44
|
+
outputCentsPerMillion: systemDefault.outputCentsPerMillion,
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
// 3. Hardcoded fallback
|
|
48
|
+
return HARDCODED_FALLBACK;
|
|
49
|
+
}
|
|
50
|
+
/**
|
|
51
|
+
* Calculate cost in integer cents for a model invocation using resolved pricing.
|
|
52
|
+
*/
|
|
53
|
+
calculateCostCents(pricing, tokensIn, tokensOut) {
|
|
54
|
+
const inputCost = Math.ceil((tokensIn * pricing.inputCentsPerMillion) / 1_000_000);
|
|
55
|
+
const outputCost = Math.ceil((tokensOut * pricing.outputCentsPerMillion) / 1_000_000);
|
|
56
|
+
return inputCost + outputCost;
|
|
57
|
+
}
|
|
58
|
+
// ── System default CRUD ─────────────────────────────────────────────
|
|
59
|
+
async listSystemDefaults() {
|
|
60
|
+
return this.db.modelPricingDefault.findMany({
|
|
61
|
+
orderBy: { model: "asc" },
|
|
62
|
+
});
|
|
63
|
+
}
|
|
64
|
+
async getSystemDefault(model) {
|
|
65
|
+
return this.db.modelPricingDefault.findUnique({
|
|
66
|
+
where: { model },
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
async setSystemDefault(model, inputCentsPerMillion, outputCentsPerMillion, updatedBy) {
|
|
70
|
+
return this.db.modelPricingDefault.upsert({
|
|
71
|
+
where: { model },
|
|
72
|
+
create: {
|
|
73
|
+
model,
|
|
74
|
+
inputCentsPerMillion,
|
|
75
|
+
outputCentsPerMillion,
|
|
76
|
+
updatedBy,
|
|
77
|
+
},
|
|
78
|
+
update: {
|
|
79
|
+
inputCentsPerMillion,
|
|
80
|
+
outputCentsPerMillion,
|
|
81
|
+
updatedBy,
|
|
82
|
+
updatedAt: new Date(),
|
|
83
|
+
},
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
async deleteSystemDefault(model) {
|
|
87
|
+
try {
|
|
88
|
+
await this.db.modelPricingDefault.delete({ where: { model } });
|
|
89
|
+
return true;
|
|
90
|
+
}
|
|
91
|
+
catch {
|
|
92
|
+
return false;
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
//# sourceMappingURL=model-pricing-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-pricing-service.js","sourceRoot":"","sources":["../../src/services/model-pricing-service.ts"],"names":[],"mappings":"AAiBA,4EAA4E;AAC5E,MAAM,kBAAkB,GAAiB;IACvC,oBAAoB,EAAE,GAAG;IACzB,qBAAqB,EAAE,IAAI;CAC5B,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,OAAO,mBAAmB;IACV;IAApB,YAAoB,EAAgB;QAAhB,OAAE,GAAF,EAAE,CAAc;IAAG,CAAC;IAExC;;;OAGG;IACH,KAAK,CAAC,cAAc,CAAC,cAAsB,EAAE,KAAa;QACxD,wBAAwB;QACxB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC;YAChD,KAAK,EAAE,EAAE,EAAE,EAAE,cAAc,EAAE;YAC7B,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC3B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,CAAC,GAAG,EAAE,QAAQ,IAAI,EAAE,CAAgB,CAAC;QAClD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO;gBACL,oBAAoB,EAAE,WAAW,CAAC,oBAAoB;gBACtD,qBAAqB,EAAE,WAAW,CAAC,qBAAqB;aACzD,CAAC;QACJ,CAAC;QAED,0BAA0B;QAC1B,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,mBAAmB,CAAC,UAAU,CAAC;YACjE,KAAK,EAAE,EAAE,KAAK,EAAE;SACjB,CAAC,CAAC;QAEH,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO;gBACL,oBAAoB,EAAE,aAAa,CAAC,oBAAoB;gBACxD,qBAAqB,EAAE,aAAa,CAAC,qBAAqB;aAC3D,CAAC;QACJ,CAAC;QAED,wBAAwB;QACxB,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,OAAqB,EAAE,QAAgB,EAAE,SAAiB;QAC3E,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,GAAG,OAAO,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAC,CAAC;QACnF,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,SAAS,GAAG,OAAO,CAAC,qBAAqB,CAAC,GAAG,SAAS,CAAC,CAAC;QACtF,OAAO,SAAS,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,uEAAuE;IAEvE,KAAK,CAAC,kBAAkB;QACtB,OAAO,IAAI,CAAC,EAAE,CAAC,mBAAmB,CAAC,QAAQ,CAAC;YAC1C,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;SAC1B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAa;QAClC,OAAO,IAAI,CAAC,EAAE,CAAC,mBAAmB,CAAC,UAAU,CAAC;YAC5C,KAAK,EAAE,EAAE,KAAK,EAAE;SACjB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,oBAA4B,EAC5B,qBAA6B,EAC7B,SAAiB;QAEjB,OAAO,IAAI,CAAC,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,EAAE,KAAK,EAAE;YAChB,MAAM,EAAE;gBACN,KAAK;gBACL,oBAAoB;gBACpB,qBAAqB;gBACrB,SAAS;aACV;YACD,MAAM,EAAE;gBACN,oBAAoB;gBACpB,qBAAqB;gBACrB,SAAS;gBACT,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,KAAa;QACrC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YAC/D,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-pricing-service.test.d.ts","sourceRoot":"","sources":["../../src/services/model-pricing-service.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,134 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
2
|
+
import { ModelPricingService } from "./model-pricing-service.js";
|
|
3
|
+
function createMockDb(overrides = {}) {
|
|
4
|
+
return {
|
|
5
|
+
organization: {
|
|
6
|
+
findUnique: vi.fn().mockResolvedValue(null),
|
|
7
|
+
},
|
|
8
|
+
modelPricingDefault: {
|
|
9
|
+
findUnique: vi.fn().mockResolvedValue(null),
|
|
10
|
+
findMany: vi.fn().mockResolvedValue([]),
|
|
11
|
+
upsert: vi.fn(),
|
|
12
|
+
delete: vi.fn(),
|
|
13
|
+
},
|
|
14
|
+
...overrides,
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
describe("ModelPricingService", () => {
|
|
18
|
+
let db;
|
|
19
|
+
let svc;
|
|
20
|
+
beforeEach(() => {
|
|
21
|
+
db = createMockDb();
|
|
22
|
+
svc = new ModelPricingService(db);
|
|
23
|
+
});
|
|
24
|
+
describe("resolvePricing", () => {
|
|
25
|
+
it("returns hardcoded fallback when nothing is configured", async () => {
|
|
26
|
+
const pricing = await svc.resolvePricing("org-1", "unknown-model");
|
|
27
|
+
expect(pricing).toEqual({
|
|
28
|
+
inputCentsPerMillion: 300,
|
|
29
|
+
outputCentsPerMillion: 1500,
|
|
30
|
+
});
|
|
31
|
+
});
|
|
32
|
+
it("returns system default when no org override exists", async () => {
|
|
33
|
+
db.modelPricingDefault.findUnique.mockResolvedValue({
|
|
34
|
+
model: "gpt-4o",
|
|
35
|
+
inputCentsPerMillion: 250,
|
|
36
|
+
outputCentsPerMillion: 1000,
|
|
37
|
+
});
|
|
38
|
+
const pricing = await svc.resolvePricing("org-1", "gpt-4o");
|
|
39
|
+
expect(pricing).toEqual({
|
|
40
|
+
inputCentsPerMillion: 250,
|
|
41
|
+
outputCentsPerMillion: 1000,
|
|
42
|
+
});
|
|
43
|
+
});
|
|
44
|
+
it("org override takes precedence over system default", async () => {
|
|
45
|
+
db.organization.findUnique.mockResolvedValue({
|
|
46
|
+
metadata: {
|
|
47
|
+
modelPricing: {
|
|
48
|
+
"gpt-4o": { inputCentsPerMillion: 200, outputCentsPerMillion: 800 },
|
|
49
|
+
},
|
|
50
|
+
},
|
|
51
|
+
});
|
|
52
|
+
db.modelPricingDefault.findUnique.mockResolvedValue({
|
|
53
|
+
model: "gpt-4o",
|
|
54
|
+
inputCentsPerMillion: 250,
|
|
55
|
+
outputCentsPerMillion: 1000,
|
|
56
|
+
});
|
|
57
|
+
const pricing = await svc.resolvePricing("org-1", "gpt-4o");
|
|
58
|
+
expect(pricing).toEqual({
|
|
59
|
+
inputCentsPerMillion: 200,
|
|
60
|
+
outputCentsPerMillion: 800,
|
|
61
|
+
});
|
|
62
|
+
// Should not even query the system default table
|
|
63
|
+
expect(db.modelPricingDefault.findUnique).not.toHaveBeenCalled();
|
|
64
|
+
});
|
|
65
|
+
it("falls through org metadata without modelPricing to system default", async () => {
|
|
66
|
+
db.organization.findUnique.mockResolvedValue({
|
|
67
|
+
metadata: { search: { defaultLimit: 10 } },
|
|
68
|
+
});
|
|
69
|
+
db.modelPricingDefault.findUnique.mockResolvedValue({
|
|
70
|
+
model: "gpt-4o",
|
|
71
|
+
inputCentsPerMillion: 250,
|
|
72
|
+
outputCentsPerMillion: 1000,
|
|
73
|
+
});
|
|
74
|
+
const pricing = await svc.resolvePricing("org-1", "gpt-4o");
|
|
75
|
+
expect(pricing).toEqual({
|
|
76
|
+
inputCentsPerMillion: 250,
|
|
77
|
+
outputCentsPerMillion: 1000,
|
|
78
|
+
});
|
|
79
|
+
});
|
|
80
|
+
});
|
|
81
|
+
describe("calculateCostCents", () => {
|
|
82
|
+
it("calculates cost correctly with ceiling", () => {
|
|
83
|
+
const pricing = { inputCentsPerMillion: 250, outputCentsPerMillion: 1000 };
|
|
84
|
+
// 1000 tokens in @ 250 cents/M = 0.25 cents → ceil to 1
|
|
85
|
+
// 500 tokens out @ 1000 cents/M = 0.50 cents → ceil to 1
|
|
86
|
+
expect(svc.calculateCostCents(pricing, 1000, 500)).toBe(2);
|
|
87
|
+
});
|
|
88
|
+
it("returns 0 for zero tokens", () => {
|
|
89
|
+
const pricing = { inputCentsPerMillion: 250, outputCentsPerMillion: 1000 };
|
|
90
|
+
expect(svc.calculateCostCents(pricing, 0, 0)).toBe(0);
|
|
91
|
+
});
|
|
92
|
+
it("handles large token counts", () => {
|
|
93
|
+
const pricing = { inputCentsPerMillion: 250, outputCentsPerMillion: 1000 };
|
|
94
|
+
// 1M tokens in @ 250 cents/M = 250 cents
|
|
95
|
+
// 1M tokens out @ 1000 cents/M = 1000 cents
|
|
96
|
+
expect(svc.calculateCostCents(pricing, 1_000_000, 1_000_000)).toBe(1250);
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
describe("system default CRUD", () => {
|
|
100
|
+
it("lists system defaults", async () => {
|
|
101
|
+
const rows = [
|
|
102
|
+
{ id: "1", model: "gpt-4o", inputCentsPerMillion: 250, outputCentsPerMillion: 1000, updatedAt: new Date(), updatedBy: "admin" },
|
|
103
|
+
];
|
|
104
|
+
db.modelPricingDefault.findMany.mockResolvedValue(rows);
|
|
105
|
+
const result = await svc.listSystemDefaults();
|
|
106
|
+
expect(result).toEqual(rows);
|
|
107
|
+
expect(db.modelPricingDefault.findMany).toHaveBeenCalledWith({ orderBy: { model: "asc" } });
|
|
108
|
+
});
|
|
109
|
+
it("sets (upserts) a system default", async () => {
|
|
110
|
+
const row = {
|
|
111
|
+
id: "1",
|
|
112
|
+
model: "gpt-4o",
|
|
113
|
+
inputCentsPerMillion: 250,
|
|
114
|
+
outputCentsPerMillion: 1000,
|
|
115
|
+
updatedAt: new Date(),
|
|
116
|
+
updatedBy: "admin@apart.tech",
|
|
117
|
+
};
|
|
118
|
+
db.modelPricingDefault.upsert.mockResolvedValue(row);
|
|
119
|
+
const result = await svc.setSystemDefault("gpt-4o", 250, 1000, "admin@apart.tech");
|
|
120
|
+
expect(result).toEqual(row);
|
|
121
|
+
});
|
|
122
|
+
it("deletes a system default", async () => {
|
|
123
|
+
db.modelPricingDefault.delete.mockResolvedValue({});
|
|
124
|
+
const result = await svc.deleteSystemDefault("gpt-4o");
|
|
125
|
+
expect(result).toBe(true);
|
|
126
|
+
});
|
|
127
|
+
it("returns false when deleting a non-existent model", async () => {
|
|
128
|
+
db.modelPricingDefault.delete.mockRejectedValue(new Error("Not found"));
|
|
129
|
+
const result = await svc.deleteSystemDefault("non-existent");
|
|
130
|
+
expect(result).toBe(false);
|
|
131
|
+
});
|
|
132
|
+
});
|
|
133
|
+
});
|
|
134
|
+
//# sourceMappingURL=model-pricing-service.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-pricing-service.test.js","sourceRoot":"","sources":["../../src/services/model-pricing-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAEjE,SAAS,YAAY,CAAC,YAAqC,EAAE;IAC3D,OAAO;QACL,YAAY,EAAE;YACZ,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SAC5C;QACD,mBAAmB,EAAE;YACnB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;YAC3C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACvC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;SAChB;QACD,GAAG,SAAS;KACN,CAAC;AACX,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,EAAmC,CAAC;IACxC,IAAI,GAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,YAAY,EAAE,CAAC;QACpB,GAAG,GAAG,IAAI,mBAAmB,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YAEnE,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,EAAE,CAAC,mBAAmB,CAAC,UAAU,CAAC,iBAAiB,CAAC;gBAClD,KAAK,EAAE,QAAQ;gBACf,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE5D,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC;gBAC3C,QAAQ,EAAE;oBACR,YAAY,EAAE;wBACZ,QAAQ,EAAE,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,GAAG,EAAE;qBACpE;iBACF;aACF,CAAC,CAAC;YAEH,EAAE,CAAC,mBAAmB,CAAC,UAAU,CAAC,iBAAiB,CAAC;gBAClD,KAAK,EAAE,QAAQ;gBACf,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE5D,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,GAAG;aAC3B,CAAC,CAAC;YACH,iDAAiD;YACjD,MAAM,CAAC,EAAE,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC;gBAC3C,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE;aAC3C,CAAC,CAAC;YAEH,EAAE,CAAC,mBAAmB,CAAC,UAAU,CAAC,iBAAiB,CAAC;gBAClD,KAAK,EAAE,QAAQ;gBACf,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE5D,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,OAAO,GAAG,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;YAE3E,wDAAwD;YACxD,yDAAyD;YACzD,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,OAAO,GAAG,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;YAC3E,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,OAAO,GAAG,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;YAE3E,yCAAyC;YACzC,4CAA4C;YAC5C,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,OAAO,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,IAAI,GAAG;gBACX,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE;aAChI,CAAC;YACF,EAAE,CAAC,mBAAmB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAExD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,EAAE,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,EAAE,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,GAAG,GAAG;gBACV,EAAE,EAAE,GAAG;gBACP,KAAK,EAAE,QAAQ;gBACf,oBAAoB,EAAE,GAAG;gBACzB,qBAAqB,EAAE,IAAI;gBAC3B,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,SAAS,EAAE,kBAAkB;aAC9B,CAAC;YACF,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAErD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,gBAAgB,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAE,kBAAkB,CAAC,CAAC;YACnF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YACxE,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { PrismaClient } from "@prisma/client";
|
|
2
|
-
export type LlmProviderName = "openai" | "anthropic" | "google" | "mistral" | "ollama";
|
|
2
|
+
export type LlmProviderName = "openai" | "anthropic" | "google" | "mistral" | "ollama" | "xai" | "deepseek" | "moonshotai";
|
|
3
3
|
export interface OrgLlmProviderRecord {
|
|
4
4
|
id: string;
|
|
5
5
|
provider: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"org-llm-provider-service.d.ts","sourceRoot":"","sources":["../../src/services/org-llm-provider-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAYnD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;
|
|
1
|
+
{"version":3,"file":"org-llm-provider-service.d.ts","sourceRoot":"","sources":["../../src/services/org-llm-provider-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAYnD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,QAAQ,GAAG,KAAK,GAAG,UAAU,GAAG,YAAY,CAAC;AAE3H,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,aAAa,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,qBAAsB,SAAQ,oBAAoB;IACjE,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,sBAAsB;IACrC,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,qBAAa,qBAAqB;IACpB,OAAO,CAAC,EAAE;gBAAF,EAAE,EAAE,YAAY;IAE9B,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAwCzF,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC;IAYnF,UAAU,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC;IAiB3F,IAAI,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAQ7D,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAgBlE,mBAAmB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAC;QAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,EAAE,CAAC;IASjI,OAAO,CAAC,QAAQ;CA6BjB"}
|
package/dist/types/index.d.ts
CHANGED
|
@@ -149,6 +149,13 @@ export interface OrgMetadata {
|
|
|
149
149
|
editingEnabled?: boolean;
|
|
150
150
|
piiOutputPolicy?: "warn" | "block" | "strip";
|
|
151
151
|
};
|
|
152
|
+
chatLimits?: {
|
|
153
|
+
defaultUserCostCentsPerMonth?: number;
|
|
154
|
+
};
|
|
155
|
+
modelPricing?: Record<string, {
|
|
156
|
+
inputCentsPerMillion: number;
|
|
157
|
+
outputCentsPerMillion: number;
|
|
158
|
+
}>;
|
|
152
159
|
}
|
|
153
160
|
/** @deprecated Use ApartConfig instead */
|
|
154
161
|
export type CollectiveConfig = ApartConfig;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,IAAI,IAAI,UAAU,EAClB,IAAI,IAAI,UAAU,EAClB,MAAM,IAAI,YAAY,EACtB,SAAS,IAAI,eAAe,EAC5B,SAAS,IAAI,eAAe,EAC5B,aAAa,IAAI,mBAAmB,EACpC,OAAO,IAAI,aAAa,EACxB,cAAc,IAAI,oBAAoB,EACtC,YAAY,IAAI,kBAAkB,EACnC,MAAM,gBAAgB,CAAC;AAIxB,MAAM,MAAM,IAAI,GAAG,UAAU,CAAC;AAE9B,MAAM,MAAM,MAAM,GAAG,YAAY,CAAC;AAElC,MAAM,MAAM,SAAS,GAAG,eAAe,CAAC;AAExC,MAAM,MAAM,SAAS,GAAG,eAAe,CAAC;AAIxC,MAAM,MAAM,aAAa,GAAG,mBAAmB,CAAC;AAChD,MAAM,MAAM,OAAO,GAAG,aAAa,CAAC;AACpC,MAAM,MAAM,cAAc,GAAG,oBAAoB,CAAC;AAClD,MAAM,MAAM,YAAY,GAAG,kBAAkB,CAAC;AAE9C,MAAM,MAAM,OAAO,GAAG,MAAM,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC;AAClE,MAAM,MAAM,cAAc,GAAG,WAAW,GAAG,OAAO,CAAC;AACnD,MAAM,MAAM,SAAS,GAAG,WAAW,GAAG,UAAU,GAAG,UAAU,CAAC;AAC9D,MAAM,MAAM,kBAAkB,GAAG,YAAY,GAAG,eAAe,GAAG,gBAAgB,GAAG,WAAW,CAAC;AACjG,MAAM,MAAM,mBAAmB,GAAG,QAAQ,GAAG,YAAY,GAAG,SAAS,CAAC;AAEtE,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,MAAM,WAAW,GACnB,IAAI,GACJ,KAAK,GACL,IAAI,GACJ,KAAK,GACL,IAAI,GACJ,KAAK,GACL,SAAS,GACT,IAAI,GACJ,UAAU,GACV,QAAQ,CAAC;AAEb,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,EAAE,EAAE,WAAW,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IACxD,OAAO,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,QAAQ;IACvB,OAAO,EAAE,SAAS,EAAE,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,YAAY,CAAC,EAAE,KAAK,CAAC;QACnB,EAAE,EAAE,OAAO,GAAG,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,KAAK,CAAC;QAC5C,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IAChC,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,GAAG,WAAW,GAAG,cAAc,CAAC,GAAG;IAClG,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG,OAAO,GAAG,UAAU,GAAG,UAAU,GAAG,UAAU,CAAC;AAIxE,MAAM,MAAM,IAAI,GAAG,UAAU,CAAC;AAE9B,MAAM,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC;AAIrD,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAC;IACxB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,UAAU,CAAC,EAAE,SAAS,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,UAAU,GAAG,SAAS,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAC;IACvE,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAID,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC;IAChE,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACnC;AAID,MAAM,WAAW,iBAAiB;IAChC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,iBAAiB,KAAK,IAAI,CAAC;IACnD,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC;CAClC;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACpE,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;CAC/E;AAID,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,EAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,OAAO,CAAC;KACzB,CAAC;IACF,OAAO,CAAC,EAAE;QACR,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,QAAQ,CAAC,EAAE;QACT,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,CAAC;IACF,aAAa,CAAC,EAAE;QACd,UAAU,CAAC,EAAE;YACX,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,QAAQ,CAAC,EAAE,MAAM,CAAC;YAClB,MAAM,CAAC,EAAE,MAAM,CAAC;SACjB,CAAC;KACH,CAAC;IACF,kBAAkB,CAAC,EAAE;QACnB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC;KAC9C,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,IAAI,IAAI,UAAU,EAClB,IAAI,IAAI,UAAU,EAClB,MAAM,IAAI,YAAY,EACtB,SAAS,IAAI,eAAe,EAC5B,SAAS,IAAI,eAAe,EAC5B,aAAa,IAAI,mBAAmB,EACpC,OAAO,IAAI,aAAa,EACxB,cAAc,IAAI,oBAAoB,EACtC,YAAY,IAAI,kBAAkB,EACnC,MAAM,gBAAgB,CAAC;AAIxB,MAAM,MAAM,IAAI,GAAG,UAAU,CAAC;AAE9B,MAAM,MAAM,MAAM,GAAG,YAAY,CAAC;AAElC,MAAM,MAAM,SAAS,GAAG,eAAe,CAAC;AAExC,MAAM,MAAM,SAAS,GAAG,eAAe,CAAC;AAIxC,MAAM,MAAM,aAAa,GAAG,mBAAmB,CAAC;AAChD,MAAM,MAAM,OAAO,GAAG,aAAa,CAAC;AACpC,MAAM,MAAM,cAAc,GAAG,oBAAoB,CAAC;AAClD,MAAM,MAAM,YAAY,GAAG,kBAAkB,CAAC;AAE9C,MAAM,MAAM,OAAO,GAAG,MAAM,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC;AAClE,MAAM,MAAM,cAAc,GAAG,WAAW,GAAG,OAAO,CAAC;AACnD,MAAM,MAAM,SAAS,GAAG,WAAW,GAAG,UAAU,GAAG,UAAU,CAAC;AAC9D,MAAM,MAAM,kBAAkB,GAAG,YAAY,GAAG,eAAe,GAAG,gBAAgB,GAAG,WAAW,CAAC;AACjG,MAAM,MAAM,mBAAmB,GAAG,QAAQ,GAAG,YAAY,GAAG,SAAS,CAAC;AAEtE,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,MAAM,WAAW,GACnB,IAAI,GACJ,KAAK,GACL,IAAI,GACJ,KAAK,GACL,IAAI,GACJ,KAAK,GACL,SAAS,GACT,IAAI,GACJ,UAAU,GACV,QAAQ,CAAC;AAEb,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,EAAE,EAAE,WAAW,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IACxD,OAAO,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,QAAQ;IACvB,OAAO,EAAE,SAAS,EAAE,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,YAAY,CAAC,EAAE,KAAK,CAAC;QACnB,EAAE,EAAE,OAAO,GAAG,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,KAAK,CAAC;QAC5C,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IAChC,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,GAAG,WAAW,GAAG,cAAc,CAAC,GAAG;IAClG,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG,OAAO,GAAG,UAAU,GAAG,UAAU,GAAG,UAAU,CAAC;AAIxE,MAAM,MAAM,IAAI,GAAG,UAAU,CAAC;AAE9B,MAAM,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC;AAIrD,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAC;IACxB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,UAAU,CAAC,EAAE,SAAS,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,UAAU,GAAG,SAAS,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAC;IACvE,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAID,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC;IAChE,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACnC;AAID,MAAM,WAAW,iBAAiB;IAChC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,iBAAiB,KAAK,IAAI,CAAC;IACnD,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC;CAClC;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACpE,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;CAC/E;AAID,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,EAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,OAAO,CAAC;KACzB,CAAC;IACF,OAAO,CAAC,EAAE;QACR,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,QAAQ,CAAC,EAAE;QACT,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,CAAC;IACF,aAAa,CAAC,EAAE;QACd,UAAU,CAAC,EAAE;YACX,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,QAAQ,CAAC,EAAE,MAAM,CAAC;YAClB,MAAM,CAAC,EAAE,MAAM,CAAC;SACjB,CAAC;KACH,CAAC;IACF,kBAAkB,CAAC,EAAE;QACnB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC;KAC9C,CAAC;IACF,UAAU,CAAC,EAAE;QACX,4BAA4B,CAAC,EAAE,MAAM,CAAC;KACvC,CAAC;IACF,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE;QAC5B,oBAAoB,EAAE,MAAM,CAAC;QAC7B,qBAAqB,EAAE,MAAM,CAAC;KAC/B,CAAC,CAAC;CACJ;AAID,0CAA0C;AAC1C,MAAM,MAAM,gBAAgB,GAAG,WAAW,CAAC;AAE3C;;;;;;;;;;GAUG;AACH,MAAM,MAAM,WAAW,GAAG,OAAO,GAAG,KAAK,GAAG,SAAS,GAAG,MAAM,CAAC;AAE/D,MAAM,WAAW,WAAW;IAC1B,WAAW,EAAE,WAAW,CAAC;IACzB,QAAQ,EAAE;QACR,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,UAAU,EAAE;QACV,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE;YACN,OAAO,EAAE,OAAO,CAAC;YACjB,OAAO,EAAE,UAAU,GAAG,MAAM,CAAC;YAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;SACpB,CAAC;KACH,CAAC;IACF,MAAM,EAAE;QACN,cAAc,EAAE,MAAM,CAAC;QACvB,aAAa,EAAE,MAAM,CAAC;QACtB,eAAe,EAAE,MAAM,CAAC;QACxB,aAAa,EAAE,MAAM,CAAC;QACtB,gBAAgB,EAAE,MAAM,CAAC;QACzB,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,OAAO,CAAC;KACxB,CAAC;IACF,OAAO,EAAE;QACP,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,GAAG,EAAE;QACH,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IACF,OAAO,EAAE;QACP,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC;QAC9B,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,CAAC;IACF,KAAK,CAAC,EAAE;QACN,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;CACH"}
|
package/package.json
CHANGED
package/prisma/schema.prisma
CHANGED
|
@@ -53,6 +53,7 @@ model Organization {
|
|
|
53
53
|
contentPolicies OrgContentPolicy[]
|
|
54
54
|
contentPolicyLogs ContentPolicyLog[]
|
|
55
55
|
mergeProposals MergeProposal[]
|
|
56
|
+
chatUserCostLimits ChatUserCostLimit[]
|
|
56
57
|
|
|
57
58
|
@@map("organizations")
|
|
58
59
|
}
|
|
@@ -70,10 +71,11 @@ model User {
|
|
|
70
71
|
agentSecretsEncrypted String? @map("agent_secrets_encrypted") @db.Text
|
|
71
72
|
isSuperAdmin Boolean @default(false) @map("is_super_admin")
|
|
72
73
|
|
|
73
|
-
memberships
|
|
74
|
-
createdAgentTypes
|
|
75
|
-
invokedAgentRuns
|
|
76
|
-
scheduledAgents
|
|
74
|
+
memberships Membership[]
|
|
75
|
+
createdAgentTypes OrgAgentType[] @relation("OrgAgentTypeCreatedBy")
|
|
76
|
+
invokedAgentRuns AgentRun[] @relation("AgentRunInvokedBy")
|
|
77
|
+
scheduledAgents AgentSchedule[] @relation("AgentScheduleScheduledBy")
|
|
78
|
+
chatCostLimits ChatUserCostLimit[]
|
|
77
79
|
|
|
78
80
|
@@index([email], map: "idx_users_email")
|
|
79
81
|
@@map("users")
|
|
@@ -957,3 +959,34 @@ model MergeProposal {
|
|
|
957
959
|
@@index([organizationId, status], map: "idx_merge_proposals_org_status")
|
|
958
960
|
@@map("merge_proposals")
|
|
959
961
|
}
|
|
962
|
+
|
|
963
|
+
// ── Model Pricing Defaults ──────────────────────────────────────────────
|
|
964
|
+
|
|
965
|
+
model ModelPricingDefault {
|
|
966
|
+
id String @id @default(uuid()) @db.Uuid
|
|
967
|
+
model String @unique @db.VarChar(100)
|
|
968
|
+
inputCentsPerMillion Int @map("input_cents_per_million")
|
|
969
|
+
outputCentsPerMillion Int @map("output_cents_per_million")
|
|
970
|
+
updatedAt DateTime @default(now()) @map("updated_at") @db.Timestamptz
|
|
971
|
+
updatedBy String @map("updated_by") @db.VarChar(255)
|
|
972
|
+
|
|
973
|
+
@@map("model_pricing_defaults")
|
|
974
|
+
}
|
|
975
|
+
|
|
976
|
+
// ── Chat User Cost Limits ──────────────────────────────────────────────
|
|
977
|
+
|
|
978
|
+
model ChatUserCostLimit {
|
|
979
|
+
id String @id @default(uuid()) @db.Uuid
|
|
980
|
+
organizationId String @map("organization_id") @db.Uuid
|
|
981
|
+
userId String @map("user_id") @db.Uuid
|
|
982
|
+
costLimitCents Int @map("cost_limit_cents")
|
|
983
|
+
updatedAt DateTime @default(now()) @map("updated_at") @db.Timestamptz
|
|
984
|
+
updatedBy String @map("updated_by") @db.VarChar(255)
|
|
985
|
+
|
|
986
|
+
organization Organization @relation(fields: [organizationId], references: [id], onDelete: Cascade)
|
|
987
|
+
user User @relation(fields: [userId], references: [id], onDelete: Cascade)
|
|
988
|
+
|
|
989
|
+
@@unique([organizationId, userId], map: "uq_chat_user_cost_limit_org_user")
|
|
990
|
+
@@index([organizationId], map: "idx_chat_user_cost_limits_org")
|
|
991
|
+
@@map("chat_user_cost_limits")
|
|
992
|
+
}
|