@apart-tech/intelligence-core 1.21.0 → 1.21.1
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 +9 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -1
- package/dist/services/__tests__/node-service.test.js +3 -3
- package/dist/services/__tests__/node-service.test.js.map +1 -1
- package/dist/services/agent-cli-reference.d.ts +1 -1
- package/dist/services/agent-cli-reference.d.ts.map +1 -1
- package/dist/services/agent-cli-reference.js +2 -2
- package/dist/services/chat-capture-service.d.ts.map +1 -1
- package/dist/services/chat-capture-service.js +1 -0
- package/dist/services/chat-capture-service.js.map +1 -1
- package/dist/services/chat-tool-definitions.d.ts +178 -0
- package/dist/services/chat-tool-definitions.d.ts.map +1 -0
- package/dist/services/chat-tool-definitions.js +120 -0
- package/dist/services/chat-tool-definitions.js.map +1 -0
- package/dist/services/chat-tool-definitions.test.d.ts +2 -0
- package/dist/services/chat-tool-definitions.test.d.ts.map +1 -0
- package/dist/services/chat-tool-definitions.test.js +150 -0
- package/dist/services/chat-tool-definitions.test.js.map +1 -0
- package/dist/services/document-post-check-service.d.ts +47 -0
- package/dist/services/document-post-check-service.d.ts.map +1 -0
- package/dist/services/document-post-check-service.js +130 -0
- package/dist/services/document-post-check-service.js.map +1 -0
- package/dist/services/document-post-check-service.test.d.ts +2 -0
- package/dist/services/document-post-check-service.test.d.ts.map +1 -0
- package/dist/services/document-post-check-service.test.js +150 -0
- package/dist/services/document-post-check-service.test.js.map +1 -0
- package/dist/services/document-render-service.d.ts +27 -0
- package/dist/services/document-render-service.d.ts.map +1 -0
- package/dist/services/document-render-service.js +186 -0
- package/dist/services/document-render-service.js.map +1 -0
- package/dist/services/document-render-service.test.d.ts +2 -0
- package/dist/services/document-render-service.test.d.ts.map +1 -0
- package/dist/services/document-render-service.test.js +144 -0
- package/dist/services/document-render-service.test.js.map +1 -0
- package/dist/services/document-service.d.ts +4 -1
- package/dist/services/document-service.d.ts.map +1 -1
- package/dist/services/document-service.js +24 -1
- package/dist/services/document-service.js.map +1 -1
- package/dist/services/document-tool-handler.d.ts +44 -0
- package/dist/services/document-tool-handler.d.ts.map +1 -0
- package/dist/services/document-tool-handler.js +179 -0
- package/dist/services/document-tool-handler.js.map +1 -0
- package/dist/services/document-tool-handler.test.d.ts +2 -0
- package/dist/services/document-tool-handler.test.d.ts.map +1 -0
- package/dist/services/document-tool-handler.test.js +149 -0
- package/dist/services/document-tool-handler.test.js.map +1 -0
- package/dist/services/node-service.js +2 -2
- package/dist/services/node-service.js.map +1 -1
- package/dist/services/quota-service.d.ts +2 -0
- package/dist/services/quota-service.d.ts.map +1 -1
- package/dist/services/quota-service.js +17 -1
- package/dist/services/quota-service.js.map +1 -1
- package/dist/services/usage-service.d.ts +21 -1
- package/dist/services/usage-service.d.ts.map +1 -1
- package/dist/services/usage-service.js +37 -1
- package/dist/services/usage-service.js.map +1 -1
- package/dist/types/index.d.ts +5 -0
- package/dist/types/index.d.ts.map +1 -1
- package/package.json +11 -1
- package/prisma/schema.prisma +24 -18
|
@@ -0,0 +1,179 @@
|
|
|
1
|
+
import { CreateSpreadsheetSchema, CreatePresentationSchema, CreateDocumentSchema, CreateCsvSchema, extractTextFromToolArgs, } from "./chat-tool-definitions.js";
|
|
2
|
+
const SIGNED_URL_EXPIRY_MS = 15 * 60 * 1000;
|
|
3
|
+
export class DocumentToolHandler {
|
|
4
|
+
renderService;
|
|
5
|
+
contentPolicyService;
|
|
6
|
+
piiDetector;
|
|
7
|
+
storage;
|
|
8
|
+
bucketName;
|
|
9
|
+
tenantCtx;
|
|
10
|
+
db;
|
|
11
|
+
piiOutputPolicy;
|
|
12
|
+
constructor(renderService, contentPolicyService, piiDetector, storage, bucketName, tenantCtx, db, piiOutputPolicy = "warn") {
|
|
13
|
+
this.renderService = renderService;
|
|
14
|
+
this.contentPolicyService = contentPolicyService;
|
|
15
|
+
this.piiDetector = piiDetector;
|
|
16
|
+
this.storage = storage;
|
|
17
|
+
this.bucketName = bucketName;
|
|
18
|
+
this.tenantCtx = tenantCtx;
|
|
19
|
+
this.db = db;
|
|
20
|
+
this.piiOutputPolicy = piiOutputPolicy;
|
|
21
|
+
}
|
|
22
|
+
async handleToolCall(input) {
|
|
23
|
+
const { toolName, args, sessionId, userId, model } = input;
|
|
24
|
+
// 1. Validate args against the matching schema
|
|
25
|
+
const validated = this.validateArgs(toolName, args);
|
|
26
|
+
if (!validated.success) {
|
|
27
|
+
return { success: false, error: `Invalid arguments: ${validated.error}` };
|
|
28
|
+
}
|
|
29
|
+
// 2. Content policy check on text fields
|
|
30
|
+
const extractedText = extractTextFromToolArgs(toolName, validated.data);
|
|
31
|
+
if (this.contentPolicyService) {
|
|
32
|
+
try {
|
|
33
|
+
const evaluation = await this.contentPolicyService.evaluate(extractedText);
|
|
34
|
+
if (!evaluation.allowed) {
|
|
35
|
+
const block = evaluation.violations.find((v) => v.action === "block");
|
|
36
|
+
return {
|
|
37
|
+
success: false,
|
|
38
|
+
policyViolation: true,
|
|
39
|
+
policyMessage: `Content blocked by policy "${block?.policyName ?? "unknown"}" (${block?.category ?? "unknown"}).`,
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
catch (err) {
|
|
44
|
+
console.error("[DocumentToolHandler] Policy evaluation failed, proceeding:", err);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
// 3. PII check on text fields
|
|
48
|
+
let piiWarning = false;
|
|
49
|
+
if (this.piiDetector) {
|
|
50
|
+
try {
|
|
51
|
+
const piiResult = this.piiDetector.detect(extractedText);
|
|
52
|
+
if (piiResult.matches.length > 0) {
|
|
53
|
+
if (this.piiOutputPolicy === "block") {
|
|
54
|
+
return {
|
|
55
|
+
success: false,
|
|
56
|
+
policyViolation: true,
|
|
57
|
+
policyMessage: "Generated document content contains PII that is not allowed by organization policy.",
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
piiWarning = true;
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
catch (err) {
|
|
64
|
+
console.error("[DocumentToolHandler] PII detection failed, proceeding:", err);
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
// 4. Render the document
|
|
68
|
+
let renderResult;
|
|
69
|
+
try {
|
|
70
|
+
renderResult = await this.render(toolName, validated.data);
|
|
71
|
+
}
|
|
72
|
+
catch (err) {
|
|
73
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
74
|
+
return { success: false, error: `Rendering failed: ${msg}` };
|
|
75
|
+
}
|
|
76
|
+
// 5. Upload to GCS
|
|
77
|
+
const docId = crypto.randomUUID();
|
|
78
|
+
const storagePath = `${this.tenantCtx.organizationId}/${docId}/generated.${renderResult.extension}`;
|
|
79
|
+
try {
|
|
80
|
+
const file = this.storage.bucket(this.bucketName).file(storagePath);
|
|
81
|
+
await file.save(renderResult.buffer, { contentType: renderResult.mimeType });
|
|
82
|
+
}
|
|
83
|
+
catch (err) {
|
|
84
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
85
|
+
return { success: false, error: `Upload failed: ${msg}` };
|
|
86
|
+
}
|
|
87
|
+
// 6. Create Document record
|
|
88
|
+
try {
|
|
89
|
+
await this.db.document.create({
|
|
90
|
+
data: {
|
|
91
|
+
id: docId,
|
|
92
|
+
organizationId: this.tenantCtx.organizationId,
|
|
93
|
+
fileName: renderResult.fileName,
|
|
94
|
+
mimeType: renderResult.mimeType,
|
|
95
|
+
fileSizeBytes: renderResult.buffer.length,
|
|
96
|
+
storagePath,
|
|
97
|
+
status: "ready",
|
|
98
|
+
source: "generated",
|
|
99
|
+
sourceMessageId: input.messageId ?? null,
|
|
100
|
+
generationModel: model ?? null,
|
|
101
|
+
createdBy: userId,
|
|
102
|
+
},
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
catch (err) {
|
|
106
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
107
|
+
return { success: false, error: `Failed to create document record: ${msg}` };
|
|
108
|
+
}
|
|
109
|
+
// 7. Generate signed download URL
|
|
110
|
+
let downloadUrl;
|
|
111
|
+
try {
|
|
112
|
+
const file = this.storage.bucket(this.bucketName).file(storagePath);
|
|
113
|
+
const [url] = await file.getSignedUrl({
|
|
114
|
+
version: "v4",
|
|
115
|
+
action: "read",
|
|
116
|
+
expires: Date.now() + SIGNED_URL_EXPIRY_MS,
|
|
117
|
+
responseDisposition: `attachment; filename="${renderResult.fileName}"`,
|
|
118
|
+
});
|
|
119
|
+
downloadUrl = url;
|
|
120
|
+
}
|
|
121
|
+
catch (err) {
|
|
122
|
+
// Document is created — URL can be fetched later via the download endpoint
|
|
123
|
+
downloadUrl = "";
|
|
124
|
+
}
|
|
125
|
+
return {
|
|
126
|
+
success: true,
|
|
127
|
+
documentId: docId,
|
|
128
|
+
downloadUrl,
|
|
129
|
+
fileName: renderResult.fileName,
|
|
130
|
+
mimeType: renderResult.mimeType,
|
|
131
|
+
piiWarning,
|
|
132
|
+
};
|
|
133
|
+
}
|
|
134
|
+
validateArgs(toolName, args) {
|
|
135
|
+
switch (toolName) {
|
|
136
|
+
case "create_spreadsheet": {
|
|
137
|
+
const result = CreateSpreadsheetSchema.safeParse(args);
|
|
138
|
+
if (!result.success)
|
|
139
|
+
return { success: false, error: result.error.message };
|
|
140
|
+
return { success: true, data: result.data };
|
|
141
|
+
}
|
|
142
|
+
case "create_presentation": {
|
|
143
|
+
const result = CreatePresentationSchema.safeParse(args);
|
|
144
|
+
if (!result.success)
|
|
145
|
+
return { success: false, error: result.error.message };
|
|
146
|
+
return { success: true, data: result.data };
|
|
147
|
+
}
|
|
148
|
+
case "create_document": {
|
|
149
|
+
const result = CreateDocumentSchema.safeParse(args);
|
|
150
|
+
if (!result.success)
|
|
151
|
+
return { success: false, error: result.error.message };
|
|
152
|
+
return { success: true, data: result.data };
|
|
153
|
+
}
|
|
154
|
+
case "create_csv": {
|
|
155
|
+
const result = CreateCsvSchema.safeParse(args);
|
|
156
|
+
if (!result.success)
|
|
157
|
+
return { success: false, error: result.error.message };
|
|
158
|
+
return { success: true, data: result.data };
|
|
159
|
+
}
|
|
160
|
+
default:
|
|
161
|
+
return { success: false, error: `Unknown tool: ${toolName}` };
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
async render(toolName, args) {
|
|
165
|
+
switch (toolName) {
|
|
166
|
+
case "create_spreadsheet":
|
|
167
|
+
return this.renderService.renderSpreadsheet(args);
|
|
168
|
+
case "create_presentation":
|
|
169
|
+
return this.renderService.renderPresentation(args);
|
|
170
|
+
case "create_document":
|
|
171
|
+
return this.renderService.renderDocument(args);
|
|
172
|
+
case "create_csv":
|
|
173
|
+
return this.renderService.renderCsv(args);
|
|
174
|
+
default:
|
|
175
|
+
throw new Error(`Unknown tool: ${toolName}`);
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
//# sourceMappingURL=document-tool-handler.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-tool-handler.js","sourceRoot":"","sources":["../../src/services/document-tool-handler.ts"],"names":[],"mappings":"AAWA,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,EACpB,eAAe,EACf,uBAAuB,GACxB,MAAM,4BAA4B,CAAC;AAuBpC,MAAM,oBAAoB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAE5C,MAAM,OAAO,mBAAmB;IAEpB;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IARV,YACU,aAAoC,EACpC,oBAAsD,EACtD,WAA2C,EAC3C,OAAmB,EACnB,UAAkB,EAClB,SAAwB,EACxB,EAAgB,EAChB,kBAA8C,MAAM;QAPpD,kBAAa,GAAb,aAAa,CAAuB;QACpC,yBAAoB,GAApB,oBAAoB,CAAkC;QACtD,gBAAW,GAAX,WAAW,CAAgC;QAC3C,YAAO,GAAP,OAAO,CAAY;QACnB,eAAU,GAAV,UAAU,CAAQ;QAClB,cAAS,GAAT,SAAS,CAAe;QACxB,OAAE,GAAF,EAAE,CAAc;QAChB,oBAAe,GAAf,eAAe,CAAqC;IAC3D,CAAC;IAEJ,KAAK,CAAC,cAAc,CAAC,KAAwB;QAC3C,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAE3D,+CAA+C;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;YACvB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,sBAAsB,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC;QAC5E,CAAC;QAED,yCAAyC;QACzC,MAAM,aAAa,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;QACxE,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;gBAC3E,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;oBACxB,MAAM,KAAK,GAAG,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC;oBACtE,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,eAAe,EAAE,IAAI;wBACrB,aAAa,EAAE,8BAA8B,KAAK,EAAE,UAAU,IAAI,SAAS,MAAM,KAAK,EAAE,QAAQ,IAAI,SAAS,IAAI;qBAClH,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,6DAA6D,EAAE,GAAG,CAAC,CAAC;YACpF,CAAC;QACH,CAAC;QAED,8BAA8B;QAC9B,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACzD,IAAI,SAAS,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACjC,IAAI,IAAI,CAAC,eAAe,KAAK,OAAO,EAAE,CAAC;wBACrC,OAAO;4BACL,OAAO,EAAE,KAAK;4BACd,eAAe,EAAE,IAAI;4BACrB,aAAa,EAAE,qFAAqF;yBACrG,CAAC;oBACJ,CAAC;oBACD,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,yDAAyD,EAAE,GAAG,CAAC,CAAC;YAChF,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,IAAI,YAA0B,CAAC;QAC/B,IAAI,CAAC;YACH,YAAY,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,qBAAqB,GAAG,EAAE,EAAE,CAAC;QAC/D,CAAC;QAED,mBAAmB;QACnB,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,IAAI,KAAK,cAAc,YAAY,CAAC,SAAS,EAAE,CAAC;QACpG,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACpE,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC/E,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAkB,GAAG,EAAE,EAAE,CAAC;QAC5D,CAAC;QAED,4BAA4B;QAC5B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,IAAI,EAAE;oBACJ,EAAE,EAAE,KAAK;oBACT,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc;oBAC7C,QAAQ,EAAE,YAAY,CAAC,QAAQ;oBAC/B,QAAQ,EAAE,YAAY,CAAC,QAAQ;oBAC/B,aAAa,EAAE,YAAY,CAAC,MAAM,CAAC,MAAM;oBACzC,WAAW;oBACX,MAAM,EAAE,OAAO;oBACf,MAAM,EAAE,WAAW;oBACnB,eAAe,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI;oBACxC,eAAe,EAAE,KAAK,IAAI,IAAI;oBAC9B,SAAS,EAAE,MAAM;iBAClB;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,qCAAqC,GAAG,EAAE,EAAE,CAAC;QAC/E,CAAC;QAED,kCAAkC;QAClC,IAAI,WAAmB,CAAC;QACxB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACpE,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC;gBACpC,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;gBAC1C,mBAAmB,EAAE,yBAAyB,YAAY,CAAC,QAAQ,GAAG;aACvE,CAAC,CAAC;YACH,WAAW,GAAG,GAAG,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,2EAA2E;YAC3E,WAAW,GAAG,EAAE,CAAC;QACnB,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,KAAK;YACjB,WAAW;YACX,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,UAAU;SACX,CAAC;IACJ,CAAC;IAEO,YAAY,CAClB,QAAgB,EAChB,IAAa;QAEb,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,oBAAoB,CAAC,CAAC,CAAC;gBAC1B,MAAM,MAAM,GAAG,uBAAuB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACvD,IAAI,CAAC,MAAM,CAAC,OAAO;oBAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;YAC9C,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,MAAM,MAAM,GAAG,wBAAwB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACxD,IAAI,CAAC,MAAM,CAAC,OAAO;oBAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;YAC9C,CAAC;YACD,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,MAAM,GAAG,oBAAoB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,CAAC,MAAM,CAAC,OAAO;oBAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;YAC9C,CAAC;YACD,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,MAAM,GAAG,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBAC/C,IAAI,CAAC,MAAM,CAAC,OAAO;oBAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;YAC9C,CAAC;YACD;gBACE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,QAAQ,EAAE,EAAE,CAAC;QAClE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,QAAgB,EAAE,IAAa;QAClD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,oBAAoB;gBACvB,OAAO,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,IAAW,CAAC,CAAC;YAC3D,KAAK,qBAAqB;gBACxB,OAAO,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,IAAW,CAAC,CAAC;YAC5D,KAAK,iBAAiB;gBACpB,OAAO,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,IAAW,CAAC,CAAC;YACxD,KAAK,YAAY;gBACf,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,IAAW,CAAC,CAAC;YACnD;gBACE,MAAM,IAAI,KAAK,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-tool-handler.test.d.ts","sourceRoot":"","sources":["../../src/services/document-tool-handler.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
2
|
+
import { DocumentToolHandler } from "./document-tool-handler.js";
|
|
3
|
+
function makeMockRenderService() {
|
|
4
|
+
const result = {
|
|
5
|
+
buffer: Buffer.from("fake-file"),
|
|
6
|
+
mimeType: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
7
|
+
extension: "xlsx",
|
|
8
|
+
fileName: "test.xlsx",
|
|
9
|
+
};
|
|
10
|
+
return {
|
|
11
|
+
renderSpreadsheet: vi.fn().mockResolvedValue(result),
|
|
12
|
+
renderPresentation: vi.fn().mockResolvedValue({ ...result, mimeType: "application/pptx", extension: "pptx", fileName: "test.pptx" }),
|
|
13
|
+
renderDocument: vi.fn().mockResolvedValue({ ...result, mimeType: "application/docx", extension: "docx", fileName: "test.docx" }),
|
|
14
|
+
renderCsv: vi.fn().mockResolvedValue({ ...result, mimeType: "text/csv", extension: "csv", fileName: "test.csv" }),
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
function makeMockGcs() {
|
|
18
|
+
const mockFile = {
|
|
19
|
+
save: vi.fn().mockResolvedValue(undefined),
|
|
20
|
+
getSignedUrl: vi.fn().mockResolvedValue(["https://signed-url.example.com"]),
|
|
21
|
+
exists: vi.fn().mockResolvedValue([true]),
|
|
22
|
+
download: vi.fn().mockResolvedValue([Buffer.from("")]),
|
|
23
|
+
};
|
|
24
|
+
return {
|
|
25
|
+
bucket: vi.fn().mockReturnValue({
|
|
26
|
+
file: vi.fn().mockReturnValue(mockFile),
|
|
27
|
+
}),
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
function makeMockDb() {
|
|
31
|
+
return {
|
|
32
|
+
document: {
|
|
33
|
+
create: vi.fn().mockResolvedValue({ id: "doc-123" }),
|
|
34
|
+
},
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
const tenantCtx = { organizationId: "org-111" };
|
|
38
|
+
const validSpreadsheetArgs = {
|
|
39
|
+
fileName: "report",
|
|
40
|
+
sheets: [
|
|
41
|
+
{
|
|
42
|
+
name: "Data",
|
|
43
|
+
columns: [{ header: "Name" }],
|
|
44
|
+
rows: [["Alice"]],
|
|
45
|
+
},
|
|
46
|
+
],
|
|
47
|
+
};
|
|
48
|
+
const baseInput = {
|
|
49
|
+
toolName: "create_spreadsheet",
|
|
50
|
+
args: validSpreadsheetArgs,
|
|
51
|
+
sessionId: "sess-1",
|
|
52
|
+
userId: "user-1",
|
|
53
|
+
model: "gpt-4o",
|
|
54
|
+
};
|
|
55
|
+
describe("DocumentToolHandler", () => {
|
|
56
|
+
let renderSvc;
|
|
57
|
+
let gcs;
|
|
58
|
+
let db;
|
|
59
|
+
beforeEach(() => {
|
|
60
|
+
renderSvc = makeMockRenderService();
|
|
61
|
+
gcs = makeMockGcs();
|
|
62
|
+
db = makeMockDb();
|
|
63
|
+
});
|
|
64
|
+
it("succeeds for valid spreadsheet args without policy/PII services", async () => {
|
|
65
|
+
const handler = new DocumentToolHandler(renderSvc, undefined, undefined, gcs, "bucket", tenantCtx, db);
|
|
66
|
+
const result = await handler.handleToolCall(baseInput);
|
|
67
|
+
expect(result.success).toBe(true);
|
|
68
|
+
expect(result.documentId).toBeDefined();
|
|
69
|
+
expect(result.downloadUrl).toBe("https://signed-url.example.com");
|
|
70
|
+
expect(result.fileName).toBe("test.xlsx");
|
|
71
|
+
expect(db.document.create).toHaveBeenCalledOnce();
|
|
72
|
+
});
|
|
73
|
+
it("blocks when content policy returns not allowed", async () => {
|
|
74
|
+
const policySvc = {
|
|
75
|
+
evaluate: vi.fn().mockResolvedValue({
|
|
76
|
+
allowed: false,
|
|
77
|
+
violations: [{ action: "block", policyName: "No PII", category: "pii" }],
|
|
78
|
+
}),
|
|
79
|
+
};
|
|
80
|
+
const handler = new DocumentToolHandler(renderSvc, policySvc, undefined, gcs, "bucket", tenantCtx, db);
|
|
81
|
+
const result = await handler.handleToolCall(baseInput);
|
|
82
|
+
expect(result.success).toBe(false);
|
|
83
|
+
expect(result.policyViolation).toBe(true);
|
|
84
|
+
expect(result.policyMessage).toContain("No PII");
|
|
85
|
+
expect(renderSvc.renderSpreadsheet).not.toHaveBeenCalled();
|
|
86
|
+
});
|
|
87
|
+
it("blocks when PII policy is 'block' and PII is detected", async () => {
|
|
88
|
+
const piiDetector = {
|
|
89
|
+
detect: vi.fn().mockReturnValue({
|
|
90
|
+
matches: [{ type: "email", text: "a@b.com", start: 0, end: 7 }],
|
|
91
|
+
matchCount: 1,
|
|
92
|
+
}),
|
|
93
|
+
};
|
|
94
|
+
const handler = new DocumentToolHandler(renderSvc, undefined, piiDetector, gcs, "bucket", tenantCtx, db, "block");
|
|
95
|
+
const result = await handler.handleToolCall(baseInput);
|
|
96
|
+
expect(result.success).toBe(false);
|
|
97
|
+
expect(result.policyViolation).toBe(true);
|
|
98
|
+
expect(renderSvc.renderSpreadsheet).not.toHaveBeenCalled();
|
|
99
|
+
});
|
|
100
|
+
it("warns but proceeds when PII policy is 'warn' and PII is detected", async () => {
|
|
101
|
+
const piiDetector = {
|
|
102
|
+
detect: vi.fn().mockReturnValue({
|
|
103
|
+
matches: [{ type: "email", text: "a@b.com", start: 0, end: 7 }],
|
|
104
|
+
matchCount: 1,
|
|
105
|
+
}),
|
|
106
|
+
};
|
|
107
|
+
const handler = new DocumentToolHandler(renderSvc, undefined, piiDetector, gcs, "bucket", tenantCtx, db, "warn");
|
|
108
|
+
const result = await handler.handleToolCall(baseInput);
|
|
109
|
+
expect(result.success).toBe(true);
|
|
110
|
+
expect(result.piiWarning).toBe(true);
|
|
111
|
+
expect(renderSvc.renderSpreadsheet).toHaveBeenCalledOnce();
|
|
112
|
+
});
|
|
113
|
+
it("returns error for invalid args", async () => {
|
|
114
|
+
const handler = new DocumentToolHandler(renderSvc, undefined, undefined, gcs, "bucket", tenantCtx, db);
|
|
115
|
+
const result = await handler.handleToolCall({
|
|
116
|
+
...baseInput,
|
|
117
|
+
args: { bad: "data" },
|
|
118
|
+
});
|
|
119
|
+
expect(result.success).toBe(false);
|
|
120
|
+
expect(result.error).toContain("Invalid arguments");
|
|
121
|
+
});
|
|
122
|
+
it("returns error for unknown tool", async () => {
|
|
123
|
+
const handler = new DocumentToolHandler(renderSvc, undefined, undefined, gcs, "bucket", tenantCtx, db);
|
|
124
|
+
const result = await handler.handleToolCall({
|
|
125
|
+
...baseInput,
|
|
126
|
+
toolName: "unknown_tool",
|
|
127
|
+
});
|
|
128
|
+
expect(result.success).toBe(false);
|
|
129
|
+
expect(result.error).toContain("Unknown tool");
|
|
130
|
+
});
|
|
131
|
+
it("creates Document record with correct source fields", async () => {
|
|
132
|
+
const handler = new DocumentToolHandler(renderSvc, undefined, undefined, gcs, "bucket", tenantCtx, db);
|
|
133
|
+
await handler.handleToolCall({
|
|
134
|
+
...baseInput,
|
|
135
|
+
messageId: "msg-1",
|
|
136
|
+
});
|
|
137
|
+
expect(db.document.create).toHaveBeenCalledWith(expect.objectContaining({
|
|
138
|
+
data: expect.objectContaining({
|
|
139
|
+
organizationId: "org-111",
|
|
140
|
+
source: "generated",
|
|
141
|
+
sourceMessageId: "msg-1",
|
|
142
|
+
generationModel: "gpt-4o",
|
|
143
|
+
status: "ready",
|
|
144
|
+
createdBy: "user-1",
|
|
145
|
+
}),
|
|
146
|
+
}));
|
|
147
|
+
});
|
|
148
|
+
});
|
|
149
|
+
//# sourceMappingURL=document-tool-handler.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-tool-handler.test.js","sourceRoot":"","sources":["../../src/services/document-tool-handler.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;AAOjE,SAAS,qBAAqB;IAC5B,MAAM,MAAM,GAAiB;QAC3B,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC;QAChC,QAAQ,EAAE,mEAAmE;QAC7E,SAAS,EAAE,MAAM;QACjB,QAAQ,EAAE,WAAW;KACtB,CAAC;IACF,OAAO;QACL,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC;QACpD,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;QACpI,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;QAChI,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;KAC9E,CAAC;AACxC,CAAC;AAED,SAAS,WAAW;IAClB,MAAM,QAAQ,GAAG;QACf,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC1C,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,gCAAgC,CAAC,CAAC;QAC3E,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC;QACzC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;KACvD,CAAC;IACF,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC9B,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;SACxC,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,UAAU;IACjB,OAAO;QACL,QAAQ,EAAE;YACR,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;SACrD;KACF,CAAC;AACJ,CAAC;AAED,MAAM,SAAS,GAAkB,EAAE,cAAc,EAAE,SAAS,EAAE,CAAC;AAE/D,MAAM,oBAAoB,GAAG;IAC3B,QAAQ,EAAE,QAAQ;IAClB,MAAM,EAAE;QACN;YACE,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;YAC7B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC;SAClB;KACF;CACF,CAAC;AAEF,MAAM,SAAS,GAAG;IAChB,QAAQ,EAAE,oBAAoB;IAC9B,IAAI,EAAE,oBAAoB;IAC1B,SAAS,EAAE,QAAQ;IACnB,MAAM,EAAE,QAAQ;IAChB,KAAK,EAAE,QAAQ;CAChB,CAAC;AAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,SAAgC,CAAC;IACrC,IAAI,GAAe,CAAC;IACpB,IAAI,EAAO,CAAC;IAEZ,UAAU,CAAC,GAAG,EAAE;QACd,SAAS,GAAG,qBAAqB,EAAE,CAAC;QACpC,GAAG,GAAG,WAAW,EAAE,CAAC;QACpB,EAAE,GAAG,UAAU,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvG,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,SAAS,GAAG;YAChB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAClC,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;aACzE,CAAC;SACgC,CAAC;QAErC,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvG,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,WAAW,GAAG;YAClB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC9B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;gBAC/D,UAAU,EAAE,CAAC;aACd,CAAC;SAC8B,CAAC;QAEnC,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;QAClH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,WAAW,GAAG;YAClB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC9B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;gBAC/D,UAAU,EAAE,CAAC;aACd,CAAC;SAC8B,CAAC;QAEnC,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QACjH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvG,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC;YAC1C,GAAG,SAAS;YACZ,IAAI,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvG,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC;YAC1C,GAAG,SAAS;YACZ,QAAQ,EAAE,cAAc;SACzB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,OAAO,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvG,MAAM,OAAO,CAAC,cAAc,CAAC;YAC3B,GAAG,SAAS;YACZ,SAAS,EAAE,OAAO;SACnB,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAC7C,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC5B,cAAc,EAAE,SAAS;gBACzB,MAAM,EAAE,WAAW;gBACnB,eAAe,EAAE,OAAO;gBACxB,eAAe,EAAE,QAAQ;gBACzB,MAAM,EAAE,OAAO;gBACf,SAAS,EAAE,QAAQ;aACpB,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -28,7 +28,7 @@ export class NodeService {
|
|
|
28
28
|
title,
|
|
29
29
|
content,
|
|
30
30
|
metadata: (metadata ?? {}),
|
|
31
|
-
status: input.status ?? "
|
|
31
|
+
status: input.status ?? "approved",
|
|
32
32
|
createdBy: input.createdBy,
|
|
33
33
|
domainId: input.domainId ?? null,
|
|
34
34
|
hasPii,
|
|
@@ -61,7 +61,7 @@ export class NodeService {
|
|
|
61
61
|
title,
|
|
62
62
|
content,
|
|
63
63
|
metadata: (metadata ?? {}),
|
|
64
|
-
status: input.status ?? "
|
|
64
|
+
status: input.status ?? "approved",
|
|
65
65
|
createdBy: input.createdBy,
|
|
66
66
|
domainId: input.domainId ?? null,
|
|
67
67
|
hasPii,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"node-service.js","sourceRoot":"","sources":["../../src/services/node-service.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAsB,MAAM,iBAAiB,CAAC;AAKxF,MAAM,OAAO,WAAW;IAIZ;IACA;IAEA;IACA;IACA;IARF,SAAS,CAAgB;IAEjC,YACU,EAAgB,EAChB,UAA6B,EACrC,SAAyB,EACjB,aAAoC,EACpC,UAAiC,EACjC,YAA2B;QAL3B,OAAE,GAAF,EAAE,CAAc;QAChB,eAAU,GAAV,UAAU,CAAmB;QAE7B,kBAAa,GAAb,aAAa,CAAuB;QACpC,eAAU,GAAV,UAAU,CAAuB;QACjC,iBAAY,GAAZ,YAAY,CAAe;QAEnC,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,EAAE,cAAc,EAAE,oBAAoB,EAAE,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAQZ;QACC,kEAAkE;QAClE,MAAM,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAC7C,GAAG,KAAK,CAAC,KAAK,OAAO,KAAK,CAAC,OAAO,EAAE,CACrC,CAAC;QACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAChE,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAEhD,kDAAkD;QAClD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CACxD,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,QAAQ,CACf,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACrC,IAAI,EAAE;gBACJ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK;gBACL,OAAO;gBACP,QAAQ,EAAE,CAAC,QAAQ,IAAI,EAAE,CAA0B;gBACnD,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,
|
|
1
|
+
{"version":3,"file":"node-service.js","sourceRoot":"","sources":["../../src/services/node-service.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAsB,MAAM,iBAAiB,CAAC;AAKxF,MAAM,OAAO,WAAW;IAIZ;IACA;IAEA;IACA;IACA;IARF,SAAS,CAAgB;IAEjC,YACU,EAAgB,EAChB,UAA6B,EACrC,SAAyB,EACjB,aAAoC,EACpC,UAAiC,EACjC,YAA2B;QAL3B,OAAE,GAAF,EAAE,CAAc;QAChB,eAAU,GAAV,UAAU,CAAmB;QAE7B,kBAAa,GAAb,aAAa,CAAuB;QACpC,eAAU,GAAV,UAAU,CAAuB;QACjC,iBAAY,GAAZ,YAAY,CAAe;QAEnC,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,EAAE,cAAc,EAAE,oBAAoB,EAAE,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAQZ;QACC,kEAAkE;QAClE,MAAM,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAC7C,GAAG,KAAK,CAAC,KAAK,OAAO,KAAK,CAAC,OAAO,EAAE,CACrC,CAAC;QACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAChE,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAEhD,kDAAkD;QAClD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CACxD,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,QAAQ,CACf,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACrC,IAAI,EAAE;gBACJ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK;gBACL,OAAO;gBACP,QAAQ,EAAE,CAAC,QAAQ,IAAI,EAAE,CAA0B;gBACnD,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,UAAU;gBAClC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;gBAChC,MAAM;gBACN,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc;aAC9C;SACF,CAAC,CAAC;QAEH,uDAAuD;QACvD,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAA;;wBAEL,YAAY;mDACe,KAAK,CAAC,KAAK,cAAc,KAAK,CAAC,OAAO;iCACxD,OAAO;mBACrB,IAAI,CAAC,EAAE;cACZ,SAAS;KAClB,CAAC;QAEF,gDAAgD;QAChD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7E,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,KAQC,EACD,SAAmB;QAEnB,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAEhD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CACxD,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,QAAQ,CACf,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACrC,IAAI,EAAE;gBACJ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK;gBACL,OAAO;gBACP,QAAQ,EAAE,CAAC,QAAQ,IAAI,EAAE,CAA0B;gBACnD,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,UAAU;gBAClC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;gBAChC,MAAM;gBACN,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc;aAC9C;SACF,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAA;;wBAEL,YAAY;mDACe,KAAK,CAAC,KAAK,cAAc,KAAK,CAAC,OAAO;iCACxD,OAAO;mBACrB,IAAI,CAAC,EAAE;cACZ,SAAS;KAClB,CAAC;QAEF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7E,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,EAAU,EACV,KAMC,EACD,SAAmB;QAEnB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAE3B,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC;QACpD,MAAM,eAAe,GAAG,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC;QAE1D,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CACxD,KAAK,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,EAC7B,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,EACjC,KAAK,CAAC,QAAQ,CACf,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,IAAI,EAAE;gBACJ,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC3C,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC/C,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,QAAiC,EAAE,CAAC;gBACpF,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBACrD,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACjE,MAAM;gBACN,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO,GAAG,CAAC;aAC9B;SACF,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAChD,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAA;;wBAEL,YAAY;mDACe,aAAa,cAAc,eAAe;iCAC5D,OAAO;mBACrB,EAAE;cACP,SAAS;KAClB,CAAC;QAEF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,EAAE,aAAa,EAAE,eAAe,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,UAAkB;QAChC,uBAAuB;QACvB,IAAI,iEAAiE,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACvF,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,oCAAoC;QACpC,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,SAAS,CAAkB;;4BAEjC,UAAU,GAAG,GAAG;cAC9B,SAAS;;KAElB,CAAC;QAEF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACtC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,wBAAwB,UAAU,kDAAkD,CAAC,CAAC;QACxG,CAAC;QACD,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,EAAU;QACtB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC;QAC7B,OAAO,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,MAAM,CACV,EAAU,EACV,KAMC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAE3B,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC;QACpD,MAAM,eAAe,GAAG,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC;QAE1D,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CACxD,aAAa,EACb,eAAe,EACf,KAAK,CAAC,QAAQ,CACf,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,IAAI,EAAE;gBACJ,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC3C,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC/C,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,QAAiC,EAAE,CAAC;gBACpF,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBACrD,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACjE,MAAM;gBACN,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO,GAAG,CAAC;aAC9B;SACF,CAAC,CAAC;QAEH,kDAAkD;QAClD,IAAI,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACjC,MAAM,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAC7C,GAAG,aAAa,OAAO,eAAe,EAAE,CACzC,CAAC;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAChE,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YAEtC,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAA;;0BAEL,YAAY;qDACe,aAAa,cAAc,eAAe;mCAC5D,OAAO;qBACrB,EAAE;gBACP,SAAS;OAClB,CAAC;YAEF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,EAAE,aAAa,EAAE,eAAe,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU,EAAE,MAAkB;QAC5C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU;gBAAE,OAAO,IAAI,CAAC;YAC7B,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC/B,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;gBACzB,IAAI,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAK,GAAG,EAAE;QACzB,OAAO,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC3B,KAAK,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;YAC1B,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;YAC9B,IAAI,EAAE,KAAK;SACZ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU;gBAAE,OAAO,KAAK,CAAC;YAC9B,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,EAAE;aAC5E,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;YACzD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;YACxC,EAAE,EAAE,CAAC,MAAM,CAAC;YACZ,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACtB,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;SACtC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACrE,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,IAAY;QACpC,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACnF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC;IACjF,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC;QACzE,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACK,QAAQ,CACd,KAAa,EACb,OAAe,EACf,QAAkC;QAOlC,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACnF,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QACrD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CACjD,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAC5B,IAAI,CAAC,UAAU,CAChB,CAAC;QAEF,mCAAmC;QACnC,MAAM,aAAa,GAAG;YACpB,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,QAAQ,IAAI,EAAE,CAAC;YACtC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvE,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,QAAQ,EAAE,aAAa;YACvB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC;SACrC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -20,6 +20,8 @@ export declare class QuotaService {
|
|
|
20
20
|
checkOcrQuota(orgId: string, pagesToProcess: number): Promise<void>;
|
|
21
21
|
checkStorageQuota(orgId: string, bytesToAdd: number): Promise<void>;
|
|
22
22
|
checkAgentRunQuota(orgId: string): Promise<void>;
|
|
23
|
+
checkDocGenerationQuota(orgId: string): Promise<void>;
|
|
24
|
+
checkDocEditsQuota(orgId: string): Promise<void>;
|
|
23
25
|
checkNodeQuota(orgId: string): Promise<void>;
|
|
24
26
|
checkMemberQuota(orgId: string): Promise<void>;
|
|
25
27
|
checkFileSizeLimit(orgId: string, sizeBytes: number): Promise<void>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"quota-service.d.ts","sourceRoot":"","sources":["../../src/services/quota-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,KAAK,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEnE,qBAAa,UAAW,SAAQ,KAAK;aAEjB,QAAQ,EAAE,MAAM;aAChB,OAAO,EAAE,MAAM;aACf,KAAK,EAAE,MAAM;aACb,IAAI,EAAE,MAAM;gBAHZ,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM;CAK/B;AAED;;;;;GAKG;AACH,qBAAa,YAAY;IAErB,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,YAAY;gBADZ,EAAE,EAAE,YAAY,EAChB,YAAY,EAAE,YAAY;IAG9B,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOnE,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQnE,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOhD,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ5C,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ9C,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQnE,yBAAyB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQhF;;OAEG;IACG,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"quota-service.d.ts","sourceRoot":"","sources":["../../src/services/quota-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,KAAK,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEnE,qBAAa,UAAW,SAAQ,KAAK;aAEjB,QAAQ,EAAE,MAAM;aAChB,OAAO,EAAE,MAAM;aACf,KAAK,EAAE,MAAM;aACb,IAAI,EAAE,MAAM;gBAHZ,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM;CAK/B;AAED;;;;;GAKG;AACH,qBAAa,YAAY;IAErB,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,YAAY;gBADZ,EAAE,EAAE,YAAY,EAChB,YAAY,EAAE,YAAY;IAG9B,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOnE,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQnE,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOhD,uBAAuB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOrD,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOhD,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ5C,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ9C,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQnE,yBAAyB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQhF;;OAEG;IACG,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IAuB/E;;;OAGG;YACW,YAAY;IAmC1B;;OAEG;YACW,iBAAiB;CAwBhC"}
|
|
@@ -45,6 +45,18 @@ export class QuotaService {
|
|
|
45
45
|
throw new QuotaError("agent_runs", current, limit, plan);
|
|
46
46
|
}
|
|
47
47
|
}
|
|
48
|
+
async checkDocGenerationQuota(orgId) {
|
|
49
|
+
const { plan, current, limit } = await this.getEffectiveUsage(orgId, "maxDocGenerationsPerMonth", "doc_generations");
|
|
50
|
+
if (current >= limit) {
|
|
51
|
+
throw new QuotaError("doc_generations", current, limit, plan);
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
async checkDocEditsQuota(orgId) {
|
|
55
|
+
const { plan, current, limit } = await this.getEffectiveUsage(orgId, "maxDocEditsPerMonth", "doc_edits");
|
|
56
|
+
if (current >= limit) {
|
|
57
|
+
throw new QuotaError("doc_edits", current, limit, plan);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
48
60
|
async checkNodeQuota(orgId) {
|
|
49
61
|
const snapshot = await this.usageService.getUsage(orgId);
|
|
50
62
|
const limit = await this.resolveLimit(orgId, snapshot.plan, "maxNodes", "nodes");
|
|
@@ -91,6 +103,8 @@ export class QuotaService {
|
|
|
91
103
|
maxAgentRunsPerMonth: await this.resolveLimit(orgId, snapshot.plan, "maxAgentRunsPerMonth", "agent_runs"),
|
|
92
104
|
maxFileSizeBytes: await this.resolveLimit(orgId, snapshot.plan, "maxFileSizeBytes", "file_size_bytes"),
|
|
93
105
|
maxPagesPerDocument: await this.resolveLimit(orgId, snapshot.plan, "maxPagesPerDocument", "pages_per_document"),
|
|
106
|
+
maxDocGenerationsPerMonth: await this.resolveLimit(orgId, snapshot.plan, "maxDocGenerationsPerMonth", "doc_generations"),
|
|
107
|
+
maxDocEditsPerMonth: await this.resolveLimit(orgId, snapshot.plan, "maxDocEditsPerMonth", "doc_edits"),
|
|
94
108
|
};
|
|
95
109
|
return { ...effectiveLimits, plan: snapshot.plan };
|
|
96
110
|
}
|
|
@@ -135,6 +149,8 @@ export class QuotaService {
|
|
|
135
149
|
agent_runs: snapshot.usage.agentRuns,
|
|
136
150
|
storage_bytes: snapshot.usage.storageBytes,
|
|
137
151
|
api_calls: snapshot.usage.apiCalls,
|
|
152
|
+
doc_generations: snapshot.usage.docGenerations,
|
|
153
|
+
doc_edits: snapshot.usage.docEdits,
|
|
138
154
|
};
|
|
139
155
|
return {
|
|
140
156
|
plan: snapshot.plan,
|
|
@@ -144,5 +160,5 @@ export class QuotaService {
|
|
|
144
160
|
}
|
|
145
161
|
}
|
|
146
162
|
/** Metrics that scale linearly with seat count on pro plans. */
|
|
147
|
-
const PER_SEAT_METRICS = ["ocr_pages", "storage_bytes", "agent_runs"];
|
|
163
|
+
const PER_SEAT_METRICS = ["ocr_pages", "storage_bytes", "agent_runs", "doc_generations", "doc_edits"];
|
|
148
164
|
//# sourceMappingURL=quota-service.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"quota-service.js","sourceRoot":"","sources":["../../src/services/quota-service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAmB,MAAM,oBAAoB,CAAC;AAEnE,MAAM,OAAO,UAAW,SAAQ,KAAK;IAEjB;IACA;IACA;IACA;IAJlB,YACkB,QAAgB,EAChB,OAAe,EACf,KAAa,EACb,IAAY;QAE5B,KAAK,CAAC,sBAAsB,QAAQ,KAAK,OAAO,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,CAAC;QAL7D,aAAQ,GAAR,QAAQ,CAAQ;QAChB,YAAO,GAAP,OAAO,CAAQ;QACf,UAAK,GAAL,KAAK,CAAQ;QACb,SAAI,GAAJ,IAAI,CAAQ;QAG5B,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IAC3B,CAAC;CACF;AAED;;;;;GAKG;AACH,MAAM,OAAO,YAAY;IAEb;IACA;IAFV,YACU,EAAgB,EAChB,YAA0B;QAD1B,OAAE,GAAF,EAAE,CAAc;QAChB,iBAAY,GAAZ,YAAY,CAAc;IACjC,CAAC;IAEJ,KAAK,CAAC,aAAa,CAAC,KAAa,EAAE,cAAsB;QACvD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,qBAAqB,EAAE,WAAW,CAAC,CAAC;QACzG,IAAI,OAAO,GAAG,cAAc,GAAG,KAAK,EAAE,CAAC;YACrC,MAAM,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,KAAa,EAAE,UAAkB;QACvD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,iBAAiB,EAAE,eAAe,CAAC,CAAC;QAChG,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC9D,IAAI,OAAO,GAAG,UAAU,GAAG,KAAK,EAAE,CAAC;YACjC,MAAM,IAAI,UAAU,CAAC,eAAe,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,KAAa;QACpC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,sBAAsB,EAAE,YAAY,CAAC,CAAC;QAC3G,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;YACrB,MAAM,IAAI,UAAU,CAAC,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACjF,IAAI,QAAQ,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,EAAE,CAAC;YAClC,MAAM,IAAI,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAa;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QACrF,IAAI,QAAQ,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,EAAE,CAAC;YACpC,MAAM,IAAI,UAAU,CAAC,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,KAAa,EAAE,SAAiB;QACvD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;QACnG,IAAI,SAAS,GAAG,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,KAAa,EAAE,SAAiB;QAC9D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,oBAAoB,CAAC,CAAC;QACzG,IAAI,SAAS,GAAG,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,UAAU,CAAC,oBAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,KAAa;QACpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC;QACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC;QAErC,mDAAmD;QACnD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,KAAK,KAAK,CAAC;QACtC,MAAM,eAAe,GAAe;YAClC,QAAQ,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC;YAC5E,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,WAAW,CAAC;YACtG,UAAU,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,SAAS,CAAC;YAClF,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,WAAW,CAAC;YACtG,eAAe,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,iBAAiB,EAAE,eAAe,CAAC;YAClG,oBAAoB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,sBAAsB,EAAE,YAAY,CAAC;YACzG,gBAAgB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,CAAC;YACtG,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,oBAAoB,CAAC;
|
|
1
|
+
{"version":3,"file":"quota-service.js","sourceRoot":"","sources":["../../src/services/quota-service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAmB,MAAM,oBAAoB,CAAC;AAEnE,MAAM,OAAO,UAAW,SAAQ,KAAK;IAEjB;IACA;IACA;IACA;IAJlB,YACkB,QAAgB,EAChB,OAAe,EACf,KAAa,EACb,IAAY;QAE5B,KAAK,CAAC,sBAAsB,QAAQ,KAAK,OAAO,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,CAAC;QAL7D,aAAQ,GAAR,QAAQ,CAAQ;QAChB,YAAO,GAAP,OAAO,CAAQ;QACf,UAAK,GAAL,KAAK,CAAQ;QACb,SAAI,GAAJ,IAAI,CAAQ;QAG5B,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IAC3B,CAAC;CACF;AAED;;;;;GAKG;AACH,MAAM,OAAO,YAAY;IAEb;IACA;IAFV,YACU,EAAgB,EAChB,YAA0B;QAD1B,OAAE,GAAF,EAAE,CAAc;QAChB,iBAAY,GAAZ,YAAY,CAAc;IACjC,CAAC;IAEJ,KAAK,CAAC,aAAa,CAAC,KAAa,EAAE,cAAsB;QACvD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,qBAAqB,EAAE,WAAW,CAAC,CAAC;QACzG,IAAI,OAAO,GAAG,cAAc,GAAG,KAAK,EAAE,CAAC;YACrC,MAAM,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,KAAa,EAAE,UAAkB;QACvD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,iBAAiB,EAAE,eAAe,CAAC,CAAC;QAChG,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC9D,IAAI,OAAO,GAAG,UAAU,GAAG,KAAK,EAAE,CAAC;YACjC,MAAM,IAAI,UAAU,CAAC,eAAe,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,KAAa;QACpC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,sBAAsB,EAAE,YAAY,CAAC,CAAC;QAC3G,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;YACrB,MAAM,IAAI,UAAU,CAAC,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,KAAa;QACzC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,2BAA2B,EAAE,iBAAiB,CAAC,CAAC;QACrH,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;YACrB,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,KAAa;QACpC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,qBAAqB,EAAE,WAAW,CAAC,CAAC;QACzG,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;YACrB,MAAM,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACjF,IAAI,QAAQ,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,EAAE,CAAC;YAClC,MAAM,IAAI,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAa;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QACrF,IAAI,QAAQ,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,EAAE,CAAC;YACpC,MAAM,IAAI,UAAU,CAAC,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,KAAa,EAAE,SAAiB;QACvD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;QACnG,IAAI,SAAS,GAAG,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,KAAa,EAAE,SAAiB;QAC9D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,oBAAoB,CAAC,CAAC;QACzG,IAAI,SAAS,GAAG,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,UAAU,CAAC,oBAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,KAAa;QACpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC;QACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC;QAErC,mDAAmD;QACnD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,KAAK,KAAK,CAAC;QACtC,MAAM,eAAe,GAAe;YAClC,QAAQ,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC;YAC5E,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,WAAW,CAAC;YACtG,UAAU,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,SAAS,CAAC;YAClF,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,WAAW,CAAC;YACtG,eAAe,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,iBAAiB,EAAE,eAAe,CAAC;YAClG,oBAAoB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,sBAAsB,EAAE,YAAY,CAAC;YACzG,gBAAgB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,CAAC;YACtG,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,oBAAoB,CAAC;YAC/G,yBAAyB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,2BAA2B,EAAE,iBAAiB,CAAC;YACxH,mBAAmB,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,qBAAqB,EAAE,WAAW,CAAC;SACvG,CAAC;QAEF,OAAO,EAAE,GAAG,eAAe,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC;IACrD,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,YAAY,CACxB,KAAa,EACb,IAAY,EACZ,QAA0B,EAC1B,QAAgB;QAEhB,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC;YACrD,KAAK,EAAE;gBACL,cAAc,EAAE,KAAK;gBACrB,QAAQ;gBACR,EAAE,EAAE;oBACF,EAAE,SAAS,EAAE,IAAI,EAAE;oBACnB,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,IAAI,IAAI,EAAE,EAAE,EAAE;iBAClC;aACF;YACD,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC/B,CAAC,CAAC;QAEH,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC,aAAa,CAAC;QAChC,CAAC;QAED,MAAM,UAAU,GAAG,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACpD,MAAM,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEvC,uDAAuD;QACvD,IAAI,IAAI,KAAK,KAAK,IAAI,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACnF,OAAO,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAC7B,KAAa,EACb,QAA0B,EAC1B,QAAgB;QAEhB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAEhF,8BAA8B;QAC9B,MAAM,QAAQ,GAA2B;YACvC,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,QAAQ;YAClC,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,SAAS;YACpC,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;YAC1C,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,QAAQ;YAClC,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,cAAc;YAC9C,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,QAAQ;SACnC,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;YAChC,KAAK;SACN,CAAC;IACJ,CAAC;CACF;AAED,gEAAgE;AAChE,MAAM,gBAAgB,GAAG,CAAC,WAAW,EAAE,eAAe,EAAE,YAAY,EAAE,iBAAiB,EAAE,WAAW,CAAC,CAAC"}
|
|
@@ -8,8 +8,10 @@ export interface PlanLimits {
|
|
|
8
8
|
maxAgentRunsPerMonth: number;
|
|
9
9
|
maxFileSizeBytes: number;
|
|
10
10
|
maxPagesPerDocument: number;
|
|
11
|
+
maxDocGenerationsPerMonth: number;
|
|
12
|
+
maxDocEditsPerMonth: number;
|
|
11
13
|
}
|
|
12
|
-
export type UsageMetric = "nodes" | "apiCalls" | "members" | "ocrPages" | "storageBytes" | "agentRuns";
|
|
14
|
+
export type UsageMetric = "nodes" | "apiCalls" | "members" | "ocrPages" | "storageBytes" | "agentRuns" | "docGenerations" | "docEdits";
|
|
13
15
|
export interface UsageSnapshot {
|
|
14
16
|
plan: string;
|
|
15
17
|
limits: PlanLimits;
|
|
@@ -20,6 +22,8 @@ export interface UsageSnapshot {
|
|
|
20
22
|
ocrPages: number;
|
|
21
23
|
storageBytes: number;
|
|
22
24
|
agentRuns: number;
|
|
25
|
+
docGenerations: number;
|
|
26
|
+
docEdits: number;
|
|
23
27
|
};
|
|
24
28
|
stripeCustomerId: string | null;
|
|
25
29
|
stripeSubscriptionId: string | null;
|
|
@@ -64,6 +68,22 @@ export declare class UsageService {
|
|
|
64
68
|
* Get agent runs used in the current period.
|
|
65
69
|
*/
|
|
66
70
|
getAgentRunsUsed(organizationId: string): Promise<number>;
|
|
71
|
+
/**
|
|
72
|
+
* Increment the document generation counter for the current month.
|
|
73
|
+
*/
|
|
74
|
+
incrementDocGenerations(organizationId: string, count?: number): Promise<void>;
|
|
75
|
+
/**
|
|
76
|
+
* Get document generations used in the current period.
|
|
77
|
+
*/
|
|
78
|
+
getDocGenerationsUsed(organizationId: string): Promise<number>;
|
|
79
|
+
/**
|
|
80
|
+
* Increment the document edits counter for the current month.
|
|
81
|
+
*/
|
|
82
|
+
incrementDocEdits(organizationId: string, count?: number): Promise<void>;
|
|
83
|
+
/**
|
|
84
|
+
* Get document edits used in the current period.
|
|
85
|
+
*/
|
|
86
|
+
getDocEditsUsed(organizationId: string): Promise<number>;
|
|
67
87
|
/**
|
|
68
88
|
* Get total storage used by an org, computed as SUM(fileSizeBytes) from documents.
|
|
69
89
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"usage-service.d.ts","sourceRoot":"","sources":["../../src/services/usage-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAEnD,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,eAAe,EAAE,MAAM,CAAC;IACxB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,mBAAmB,EAAE,MAAM,CAAC;CAC7B;
|
|
1
|
+
{"version":3,"file":"usage-service.d.ts","sourceRoot":"","sources":["../../src/services/usage-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAEnD,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,eAAe,EAAE,MAAM,CAAC;IACxB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,yBAAyB,EAAE,MAAM,CAAC;IAClC,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AA6BD,MAAM,MAAM,WAAW,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,GAAG,UAAU,GAAG,cAAc,GAAG,WAAW,GAAG,gBAAgB,GAAG,UAAU,CAAC;AAEvI,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,UAAU,CAAC;IACnB,KAAK,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,YAAY,EAAE,MAAM,CAAC;QACrB,SAAS,EAAE,MAAM,CAAC;QAClB,cAAc,EAAE,MAAM,CAAC;QACvB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,gBAAgB,EAAE,IAAI,GAAG,IAAI,CAAC;CAC/B;AAED;;;;;;GAMG;AACH,qBAAa,YAAY;IACX,OAAO,CAAC,EAAE;gBAAF,EAAE,EAAE,YAAY;IAEpC;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU;IAI9C;;OAEG;IACG,QAAQ,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAsC9D;;;OAGG;IACG,iBAAiB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjF;;OAEG;IACG,iBAAiB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjF;;OAEG;IACG,kBAAkB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlF;;OAEG;IACG,eAAe,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAI9D;;OAEG;IACG,gBAAgB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAI/D;;OAEG;IACG,uBAAuB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvF;;OAEG;IACG,qBAAqB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAIpE;;OAEG;IACG,iBAAiB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjF;;OAEG;IACG,eAAe,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAI9D;;OAEG;IACG,cAAc,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ7D;;OAEG;IACG,eAAe,CACnB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,EACd,MAAM,GAAE,MAAU,GACjB,OAAO,CAAC;QAAE,WAAW,EAAE,IAAI,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAmBlD;;OAEG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IAwBhF;;OAEG;YACW,eAAe;IAwB7B;;OAEG;YACW,cAAc;IAiB5B;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAI3B"}
|