@peopl-health/nexus 5.45.0-dev.7136 → 5.45.0-dev.7142
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/README.md +32 -0
- package/lib/core/NexusMessaging.js +5 -0
- package/lib/core/WorkflowRunner.js +265 -0
- package/lib/index.d.ts +6 -4
- package/lib/index.js +0 -2
- package/lib/models/deferredWorkModel.js +24 -0
- package/lib/services/airtableService.js +7 -0
- package/package.json +1 -1
- package/lib/core/workflowRunner.js +0 -35
package/README.md
CHANGED
|
@@ -116,6 +116,38 @@ const bus = nexus.getMessaging().getEventBus();
|
|
|
116
116
|
bus.on('message:received', (m) => console.log('rx', m.id));
|
|
117
117
|
```
|
|
118
118
|
|
|
119
|
+
## Workflows (Optional)
|
|
120
|
+
|
|
121
|
+
Register a named unit of work, deduplicated by a key you derive from the trigger.
|
|
122
|
+
|
|
123
|
+
```js
|
|
124
|
+
const runner = nexus.getMessaging().getWorkflowRunner();
|
|
125
|
+
|
|
126
|
+
await runner.register({
|
|
127
|
+
kind: 'triage-projection',
|
|
128
|
+
dedupeKey: (trigger) => String(trigger.submissionId),
|
|
129
|
+
prepare: async (trigger, checkpoints, save) => { /* ... */ },
|
|
130
|
+
retrySchedule: [5, 10, 20, 40], // minutes; omit for fire-and-forget
|
|
131
|
+
});
|
|
132
|
+
|
|
133
|
+
await runner.enqueue('triage-projection', { submissionId });
|
|
134
|
+
```
|
|
135
|
+
|
|
136
|
+
With `retrySchedule` the trigger and a checkpoint journal are stored in Mongo and the schedule runs
|
|
137
|
+
on an in-process timer — deferred workflows never touch the queue adapter, so they add no Redis load.
|
|
138
|
+
`prepare` resumes from `checkpoints`; call `save(patch)` before each external call so a retry does
|
|
139
|
+
not repeat completed work. Throw an error carrying `permanent: true` to give up immediately instead
|
|
140
|
+
of walking the schedule; the Airtable helpers tag their own permanent 4xx errors that way, in
|
|
141
|
+
`withAirtableRetry`, which all five of them route through. They rethrow on failure too, so an
|
|
142
|
+
Airtable outage reaches `prepare` as an exception and walks the retry schedule on its own rather
|
|
143
|
+
than reading as success.
|
|
144
|
+
|
|
145
|
+
Recovery is automatic: registering sweeps for work left behind by a previous process, and the runner
|
|
146
|
+
keeps sweeping so work whose worker died mid-`prepare` is reclaimed once its claim goes stale.
|
|
147
|
+
`enqueue` on an abandoned key revives it, keeping its checkpoints. Work is retained for 24 hours
|
|
148
|
+
after its last update and then expires — the dedupe key expires with it, so idempotency has the
|
|
149
|
+
same horizon as retention.
|
|
150
|
+
|
|
119
151
|
## Assistants (Optional)
|
|
120
152
|
|
|
121
153
|
Register assistant classes and (optionally) a custom resolver. OpenAI is supported via `llm: 'openai'`.
|
|
@@ -43,6 +43,7 @@ const { isLlmMonitorEnabled } = require('../clinical/flags/llmMonitorConfig');
|
|
|
43
43
|
const { createQueueAdapter } = require('../queue');
|
|
44
44
|
const { ScheduledMessageJob } = require('../jobs/ScheduledMessageJob');
|
|
45
45
|
const { TemplateApprovalJob } = require('../jobs/TemplateApprovalJob');
|
|
46
|
+
const { WorkflowRunner } = require('./WorkflowRunner');
|
|
46
47
|
|
|
47
48
|
const { PhiProcessor } = require('./PhiProcessor');
|
|
48
49
|
|
|
@@ -127,6 +128,8 @@ class NexusMessaging {
|
|
|
127
128
|
queueAdapter: this.queueAdapter,
|
|
128
129
|
});
|
|
129
130
|
|
|
131
|
+
this.workflowRunner = new WorkflowRunner({ queueAdapter: this.queueAdapter });
|
|
132
|
+
|
|
130
133
|
this.phiProcessor = new PhiProcessor({
|
|
131
134
|
encode: config.phi?.encode || false,
|
|
132
135
|
ner: config.phi?.ner || null,
|
|
@@ -310,6 +313,7 @@ class NexusMessaging {
|
|
|
310
313
|
getAssistantProcessor() { return this.assistantProcessor; }
|
|
311
314
|
getLlmMonitor() { return this.llmMonitor; }
|
|
312
315
|
getQueueAdapter() { return this.queueAdapter; }
|
|
316
|
+
getWorkflowRunner() { return this.workflowRunner; }
|
|
313
317
|
getPhiProcessor() { return this.phiProcessor; }
|
|
314
318
|
isConnected() { return this.provider?.getConnectionStatus() ?? false; }
|
|
315
319
|
isProcessing(chatId) { return this.batchingManager.isProcessing(chatId); }
|
|
@@ -1012,6 +1016,7 @@ class NexusMessaging {
|
|
|
1012
1016
|
async disconnect() {
|
|
1013
1017
|
this.queueReconciliation?.stop();
|
|
1014
1018
|
if (this.provider) await this.provider.disconnect();
|
|
1019
|
+
if (this.workflowRunner) this.workflowRunner.stop();
|
|
1015
1020
|
if (this.queueAdapter) await this.queueAdapter.shutdown();
|
|
1016
1021
|
this.events.removeAllListeners();
|
|
1017
1022
|
}
|
|
@@ -0,0 +1,265 @@
|
|
|
1
|
+
const { describeThrown } = require('../utils/errorUtils');
|
|
2
|
+
const { logger } = require('../utils/logger');
|
|
3
|
+
|
|
4
|
+
const { DeferredWork, RETENTION_MS } = require('../models/deferredWorkModel');
|
|
5
|
+
|
|
6
|
+
const STALE_CLAIM_MS = 5 * 60 * 1000;
|
|
7
|
+
const MINUTE_MS = 60 * 1000;
|
|
8
|
+
const INDEX_TIMEOUT_MS = 1000;
|
|
9
|
+
const SWEEP_INTERVAL_MS = 60 * 1000;
|
|
10
|
+
|
|
11
|
+
let indexesReady = null;
|
|
12
|
+
|
|
13
|
+
function ensureIndexes() {
|
|
14
|
+
if (!indexesReady) {
|
|
15
|
+
indexesReady = DeferredWork.init().catch((error) => {
|
|
16
|
+
indexesReady = null;
|
|
17
|
+
logger.error('[WorkflowRunner] Index build failed, dedupe and retention are not guaranteed', { error: error.message });
|
|
18
|
+
});
|
|
19
|
+
}
|
|
20
|
+
return indexesReady;
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
function withIndexTimeout(promise) {
|
|
24
|
+
let timer;
|
|
25
|
+
const timeout = new Promise((resolve) => {
|
|
26
|
+
timer = setTimeout(resolve, INDEX_TIMEOUT_MS);
|
|
27
|
+
timer.unref();
|
|
28
|
+
});
|
|
29
|
+
return Promise.race([promise, timeout]).finally(() => clearTimeout(timer));
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
class WorkflowRunner {
|
|
33
|
+
constructor({ queueAdapter, sweepIntervalMs = SWEEP_INTERVAL_MS } = {}) {
|
|
34
|
+
if (!queueAdapter) throw new Error('WorkflowRunner requires a queueAdapter');
|
|
35
|
+
this.queueAdapter = queueAdapter;
|
|
36
|
+
this.sweepIntervalMs = sweepIntervalMs;
|
|
37
|
+
this.sweepTimer = null;
|
|
38
|
+
this.timers = new Map();
|
|
39
|
+
this.workflows = new Map();
|
|
40
|
+
this.stopped = false;
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
async register(workflow = {}) {
|
|
44
|
+
if (!workflow.kind || typeof workflow.kind !== 'string') throw new Error('workflow requires a kind');
|
|
45
|
+
if (workflow.kind.includes('__')) throw new Error(`workflow '${workflow.kind}' must not contain '__' (reserved as the jobId delimiter)`);
|
|
46
|
+
if (typeof workflow.dedupeKey !== 'function') throw new Error(`workflow '${workflow.kind}' requires a dedupeKey function`);
|
|
47
|
+
if (typeof workflow.prepare !== 'function') throw new Error(`workflow '${workflow.kind}' requires a prepare function`);
|
|
48
|
+
if (this.workflows.has(workflow.kind)) throw new Error(`workflow '${workflow.kind}' is already registered`);
|
|
49
|
+
if (workflow.retrySchedule !== undefined) {
|
|
50
|
+
if (!Array.isArray(workflow.retrySchedule) || !workflow.retrySchedule.length) {
|
|
51
|
+
throw new Error(`workflow '${workflow.kind}' requires a non-empty retrySchedule array`);
|
|
52
|
+
}
|
|
53
|
+
if (workflow.retrySchedule.some((minutes) => !Number.isFinite(minutes) || minutes <= 0)) {
|
|
54
|
+
throw new Error(`workflow '${workflow.kind}' requires retrySchedule entries to be positive minutes`);
|
|
55
|
+
}
|
|
56
|
+
if (workflow.retrySchedule.some((minutes) => minutes * MINUTE_MS >= RETENTION_MS)) {
|
|
57
|
+
throw new Error(`workflow '${workflow.kind}' requires retrySchedule entries under the ${RETENTION_MS / MINUTE_MS}m retention, or the record expires before the attempt fires`);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
this.workflows.set(workflow.kind, workflow);
|
|
62
|
+
if (workflow.retrySchedule) {
|
|
63
|
+
await this.sweep([workflow.kind]).catch((error) => {
|
|
64
|
+
logger.error('[WorkflowRunner] Sweep on register failed', { kind: workflow.kind, error: error.message });
|
|
65
|
+
});
|
|
66
|
+
this._startSweeping();
|
|
67
|
+
return workflow;
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
try {
|
|
71
|
+
await this.queueAdapter.process(workflow.kind, (trigger) => workflow.prepare(trigger));
|
|
72
|
+
} catch (error) {
|
|
73
|
+
this.workflows.delete(workflow.kind);
|
|
74
|
+
throw error;
|
|
75
|
+
}
|
|
76
|
+
return workflow;
|
|
77
|
+
}
|
|
78
|
+
|
|
79
|
+
enqueue(kind, trigger, options = {}) {
|
|
80
|
+
const workflow = this.workflows.get(kind);
|
|
81
|
+
if (!workflow) throw new Error(`no workflow registered for '${kind}'`);
|
|
82
|
+
const dedupeKey = workflow.dedupeKey(trigger);
|
|
83
|
+
if (dedupeKey == null || (typeof dedupeKey === 'string' && !dedupeKey.trim())) {
|
|
84
|
+
throw new Error(`workflow '${kind}' produced an empty dedupe key`);
|
|
85
|
+
}
|
|
86
|
+
if (typeof dedupeKey !== 'string') throw new Error(`workflow '${kind}' produced a non-string dedupe key`);
|
|
87
|
+
if (!workflow.retrySchedule) {
|
|
88
|
+
return this.queueAdapter.enqueue(kind, trigger, { ...options, jobId: `${kind}__${dedupeKey}` });
|
|
89
|
+
}
|
|
90
|
+
return this._armDeferred(workflow, trigger, dedupeKey);
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
async sweep(only = null) {
|
|
94
|
+
const deferredKinds = [...this.workflows.values()].filter((w) => w.retrySchedule).map((w) => w.kind);
|
|
95
|
+
const kinds = only ? deferredKinds.filter((kind) => only.includes(kind)) : deferredKinds;
|
|
96
|
+
if (!kinds.length) return [];
|
|
97
|
+
|
|
98
|
+
const orphaned = await DeferredWork.find({
|
|
99
|
+
kind: { $in: kinds },
|
|
100
|
+
$or: [
|
|
101
|
+
{ status: 'pending' },
|
|
102
|
+
{ status: 'processing', claimedAt: { $lt: new Date(Date.now() - STALE_CLAIM_MS) } }
|
|
103
|
+
]
|
|
104
|
+
});
|
|
105
|
+
|
|
106
|
+
const rearmed = orphaned.map((work) => this._arm(
|
|
107
|
+
this.workflows.get(work.kind),
|
|
108
|
+
work._id,
|
|
109
|
+
this._delayUntilDue(work),
|
|
110
|
+
));
|
|
111
|
+
if (rearmed.length) logger.info('[WorkflowRunner] Swept', { kinds, rearmed: rearmed.length });
|
|
112
|
+
return rearmed;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
_startSweeping() {
|
|
116
|
+
if (this.stopped || this.sweepTimer || !this.sweepIntervalMs) return;
|
|
117
|
+
this.sweepTimer = setInterval(() => {
|
|
118
|
+
this.sweep().catch((error) => logger.error('[WorkflowRunner] Periodic sweep failed', { error: error.message }));
|
|
119
|
+
}, this.sweepIntervalMs);
|
|
120
|
+
this.sweepTimer.unref();
|
|
121
|
+
}
|
|
122
|
+
|
|
123
|
+
stop() {
|
|
124
|
+
this.stopped = true;
|
|
125
|
+
if (this.sweepTimer) clearInterval(this.sweepTimer);
|
|
126
|
+
this.sweepTimer = null;
|
|
127
|
+
for (const timer of this.timers.values()) clearTimeout(timer);
|
|
128
|
+
this.timers.clear();
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
async _armDeferred(workflow, trigger, dedupeKey) {
|
|
132
|
+
const { kind } = workflow;
|
|
133
|
+
await withIndexTimeout(ensureIndexes());
|
|
134
|
+
const work = await DeferredWork.findOne({ kind, dedupeKey })
|
|
135
|
+
|| await DeferredWork.create({ kind, dedupeKey, trigger }).catch(async (error) => {
|
|
136
|
+
if (error.code !== 11000) throw error;
|
|
137
|
+
return await DeferredWork.findOne({ kind, dedupeKey });
|
|
138
|
+
});
|
|
139
|
+
if (work.status === 'abandoned') {
|
|
140
|
+
const { modifiedCount } = await DeferredWork.updateOne(
|
|
141
|
+
{ _id: work._id, status: 'abandoned' },
|
|
142
|
+
{ $set: { status: 'pending', attempt: 0, nextAttemptAt: new Date(), lastError: null, claimedAt: null } }
|
|
143
|
+
);
|
|
144
|
+
if (!modifiedCount) return null;
|
|
145
|
+
logger.info('[WorkflowRunner] Re-arming abandoned work', { kind, dedupeKey });
|
|
146
|
+
return this._arm(workflow, work._id, 0);
|
|
147
|
+
}
|
|
148
|
+
if (work.status !== 'pending') {
|
|
149
|
+
logger.info('[WorkflowRunner] Skipping, work is in flight or complete', { kind, dedupeKey, status: work.status });
|
|
150
|
+
return null;
|
|
151
|
+
}
|
|
152
|
+
return this._arm(workflow, work._id, this._delayUntilDue(work));
|
|
153
|
+
}
|
|
154
|
+
|
|
155
|
+
_delayUntilDue(work) {
|
|
156
|
+
return Math.max(0, new Date(work.nextAttemptAt).getTime() - Date.now());
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
_arm(workflow, workId, delayMs) {
|
|
160
|
+
const id = String(workId);
|
|
161
|
+
// An attempt still in flight when stop() lands must not re-arm after shutdown. The
|
|
162
|
+
// record stays pending in Mongo, so the next process recovers it on its sweep.
|
|
163
|
+
if (this.stopped) return id;
|
|
164
|
+
clearTimeout(this.timers.get(id));
|
|
165
|
+
const timer = setTimeout(() => {
|
|
166
|
+
this.timers.delete(id);
|
|
167
|
+
this._runDeferred(workflow, id).catch((error) => {
|
|
168
|
+
logger.error('[WorkflowRunner] Deferred run failed', { kind: workflow.kind, deferredWorkId: id, error: error.message });
|
|
169
|
+
});
|
|
170
|
+
}, delayMs);
|
|
171
|
+
timer.unref();
|
|
172
|
+
this.timers.set(id, timer);
|
|
173
|
+
return id;
|
|
174
|
+
}
|
|
175
|
+
|
|
176
|
+
async _runDeferred(workflow, deferredWorkId) {
|
|
177
|
+
const claimedAt = new Date();
|
|
178
|
+
const prior = await DeferredWork.findOneAndUpdate(
|
|
179
|
+
{
|
|
180
|
+
_id: deferredWorkId,
|
|
181
|
+
$or: [
|
|
182
|
+
{ status: 'pending', nextAttemptAt: { $lte: claimedAt } },
|
|
183
|
+
{ status: 'processing', claimedAt: { $lt: new Date(claimedAt.getTime() - STALE_CLAIM_MS) } }
|
|
184
|
+
]
|
|
185
|
+
},
|
|
186
|
+
{ $set: { status: 'processing', claimedAt } },
|
|
187
|
+
{ new: false }
|
|
188
|
+
);
|
|
189
|
+
if (!prior) {
|
|
190
|
+
logger.info('[WorkflowRunner] Claimed elsewhere or not due', { kind: workflow.kind, deferredWorkId });
|
|
191
|
+
return { claimed: false };
|
|
192
|
+
}
|
|
193
|
+
|
|
194
|
+
const reclaimed = prior.status === 'processing';
|
|
195
|
+
const work = { ...prior.toObject(), attempt: prior.attempt + (reclaimed ? 1 : 0) };
|
|
196
|
+
if (reclaimed) await this._write(work, claimedAt, { attempt: work.attempt });
|
|
197
|
+
|
|
198
|
+
const checkpoints = { ...(work.checkpoints || {}) };
|
|
199
|
+
const save = async (patch) => {
|
|
200
|
+
Object.assign(checkpoints, patch);
|
|
201
|
+
const $set = {};
|
|
202
|
+
for (const [key, value] of Object.entries(patch || {})) $set[`checkpoints.${key}`] = value;
|
|
203
|
+
if (Object.keys($set).length) await this._write(work, claimedAt, $set);
|
|
204
|
+
};
|
|
205
|
+
|
|
206
|
+
try {
|
|
207
|
+
await workflow.prepare(work.trigger, checkpoints, save);
|
|
208
|
+
} catch (error) {
|
|
209
|
+
return await this._rescheduleOrAbandon(workflow, work, claimedAt, error);
|
|
210
|
+
}
|
|
211
|
+
await this._write(work, claimedAt, { status: 'done', claimedAt: null });
|
|
212
|
+
return { status: 'done' };
|
|
213
|
+
}
|
|
214
|
+
|
|
215
|
+
// matchedCount, not modifiedCount: the fence is "do we still hold the claim", and
|
|
216
|
+
// modifiedCount is also 0 for a write that matched but set an identical value - a
|
|
217
|
+
// save() rewriting an unchanged checkpoint is not a lost claim.
|
|
218
|
+
async _write(work, claimedAt, $set) {
|
|
219
|
+
const { matchedCount } = await DeferredWork.updateOne({ _id: work._id, claimedAt }, { $set });
|
|
220
|
+
if (!matchedCount) logger.warn('[WorkflowRunner] Claim lost, write discarded', { deferredWorkId: String(work._id) });
|
|
221
|
+
return matchedCount > 0;
|
|
222
|
+
}
|
|
223
|
+
|
|
224
|
+
async _rescheduleOrAbandon(workflow, work, claimedAt, error) {
|
|
225
|
+
// prepare() is consumer code and can throw anything, including null. Dereferencing
|
|
226
|
+
// it here used to raise a TypeError out of this handler, so neither branch ran and
|
|
227
|
+
// the record stayed `processing` until stale recovery hit the same failure again.
|
|
228
|
+
const lastError = describeThrown(error);
|
|
229
|
+
const delayMinutes = workflow.retrySchedule[work.attempt];
|
|
230
|
+
if (error?.permanent === true || delayMinutes === undefined) {
|
|
231
|
+
await this._write(work, claimedAt, { status: 'abandoned', lastError, claimedAt: null });
|
|
232
|
+
logger.error('[WorkflowRunner] Abandoned', {
|
|
233
|
+
kind: workflow.kind,
|
|
234
|
+
dedupeKey: work.dedupeKey,
|
|
235
|
+
attempt: work.attempt,
|
|
236
|
+
permanent: error?.permanent === true,
|
|
237
|
+
error: lastError
|
|
238
|
+
});
|
|
239
|
+
return { status: 'abandoned' };
|
|
240
|
+
}
|
|
241
|
+
|
|
242
|
+
const attempt = work.attempt + 1;
|
|
243
|
+
const delayMs = delayMinutes * MINUTE_MS;
|
|
244
|
+
const kept = await this._write(work, claimedAt, {
|
|
245
|
+
status: 'pending',
|
|
246
|
+
attempt,
|
|
247
|
+
nextAttemptAt: new Date(Date.now() + delayMs),
|
|
248
|
+
lastError,
|
|
249
|
+
claimedAt: null
|
|
250
|
+
});
|
|
251
|
+
if (!kept) return { claimed: false };
|
|
252
|
+
|
|
253
|
+
logger.warn('[WorkflowRunner] Rescheduled', {
|
|
254
|
+
kind: workflow.kind,
|
|
255
|
+
dedupeKey: work.dedupeKey,
|
|
256
|
+
attempt,
|
|
257
|
+
delayMinutes,
|
|
258
|
+
error: lastError
|
|
259
|
+
});
|
|
260
|
+
this._arm(workflow, work._id, delayMs);
|
|
261
|
+
return { status: 'pending', attempt };
|
|
262
|
+
}
|
|
263
|
+
}
|
|
264
|
+
|
|
265
|
+
module.exports = { WorkflowRunner };
|
package/lib/index.d.ts
CHANGED
|
@@ -243,6 +243,7 @@ declare module '@peopl-health/nexus' {
|
|
|
243
243
|
getAssistantProcessor(): AssistantProcessor;
|
|
244
244
|
getLlmMonitor(): { start(options?: { scheduleDaily?: boolean }): Promise<{ enabled: boolean }> } | null;
|
|
245
245
|
initializeLlmMonitor(): Promise<{ enabled: boolean; error?: string } | undefined>;
|
|
246
|
+
getWorkflowRunner(): WorkflowRunner;
|
|
246
247
|
processInstruction(code: string, instruction: string, role?: string, options?: { triggeredBy?: string }): Promise<void>;
|
|
247
248
|
processSystemMessage(code: string, messages: string | string[], role?: string, options?: { triggeredBy?: string; reply?: boolean }): Promise<void>;
|
|
248
249
|
processOutreach(code: string, options?: { brief?: OutreachBrief | null; triggeredBy?: string | null; reason?: string | null; firstName?: string | null; dryRun?: boolean }): Promise<OutreachDecision>;
|
|
@@ -520,16 +521,17 @@ declare module '@peopl-health/nexus' {
|
|
|
520
521
|
export interface Workflow {
|
|
521
522
|
kind: string;
|
|
522
523
|
dedupeKey: (trigger: any) => string;
|
|
523
|
-
prepare: (trigger: any) => Promise<any>;
|
|
524
|
+
prepare: (trigger: any, checkpoints?: any, save?: (patch: any) => Promise<void>) => Promise<any>;
|
|
525
|
+
retrySchedule?: number[];
|
|
524
526
|
}
|
|
525
527
|
|
|
526
528
|
export interface WorkflowRunner {
|
|
527
529
|
register(workflow: Workflow): Promise<Workflow>;
|
|
528
|
-
enqueue(kind: string, trigger: any, options?: QueueJobOptions): Promise<string>;
|
|
530
|
+
enqueue(kind: string, trigger: any, options?: QueueJobOptions): Promise<string | null>;
|
|
531
|
+
sweep(only?: string[]): Promise<string[]>;
|
|
532
|
+
stop(): void;
|
|
529
533
|
}
|
|
530
534
|
|
|
531
|
-
export function createWorkflowRunner(options: { queueAdapter: QueueAdapter }): WorkflowRunner;
|
|
532
|
-
|
|
533
535
|
// Memory System
|
|
534
536
|
export interface PatientMemoryDocument {
|
|
535
537
|
_id: any;
|
package/lib/index.js
CHANGED
|
@@ -27,7 +27,6 @@ const { BaileysProvider } = require('./adapters/BaileysProvider');
|
|
|
27
27
|
const { setPreprocessingHandler, hasPreprocessingHandler, invokePreprocessingHandler } = require('./services/preprocessingService');
|
|
28
28
|
const { requestIdMiddleware, getRequestId } = require('./middleware/requestId');
|
|
29
29
|
const { QueueAdapter, LocalQueueAdapter, RedisQueueAdapter, createQueueAdapter, registerQueueAdapter } = require('./queue');
|
|
30
|
-
const { createWorkflowRunner } = require('./core/workflowRunner');
|
|
31
30
|
const routes = require('./routes');
|
|
32
31
|
const { resetAll } = require('./config/lifecycle');
|
|
33
32
|
const { EvalProvider } = require('./eval/EvalProvider');
|
|
@@ -229,7 +228,6 @@ class Nexus {
|
|
|
229
228
|
}
|
|
230
229
|
|
|
231
230
|
module.exports = {
|
|
232
|
-
createWorkflowRunner,
|
|
233
231
|
Nexus,
|
|
234
232
|
TwilioProvider,
|
|
235
233
|
BaileysProvider,
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
const mongoose = require('mongoose');
|
|
2
|
+
|
|
3
|
+
const STATUS_VALUES = ['pending', 'processing', 'done', 'abandoned'];
|
|
4
|
+
const RETENTION_MS = 24 * 60 * 60 * 1000;
|
|
5
|
+
|
|
6
|
+
const deferredWorkSchema = new mongoose.Schema({
|
|
7
|
+
kind: { type: String, required: true },
|
|
8
|
+
dedupeKey: { type: String, required: true },
|
|
9
|
+
trigger: { type: mongoose.Schema.Types.Mixed, default: null },
|
|
10
|
+
checkpoints: { type: mongoose.Schema.Types.Mixed, default: () => ({}) },
|
|
11
|
+
status: { type: String, enum: STATUS_VALUES, default: 'pending' },
|
|
12
|
+
attempt: { type: Number, default: 0 },
|
|
13
|
+
nextAttemptAt: { type: Date, default: Date.now },
|
|
14
|
+
claimedAt: { type: Date, default: null },
|
|
15
|
+
lastError: { type: String, default: null }
|
|
16
|
+
}, { timestamps: true });
|
|
17
|
+
|
|
18
|
+
deferredWorkSchema.index({ kind: 1, dedupeKey: 1 }, { unique: true, name: 'dedupe_idx' });
|
|
19
|
+
deferredWorkSchema.index({ status: 1, nextAttemptAt: 1 }, { name: 'due_idx' });
|
|
20
|
+
deferredWorkSchema.index({ updatedAt: 1 }, { expireAfterSeconds: RETENTION_MS / 1000, name: 'ttl_idx' });
|
|
21
|
+
|
|
22
|
+
const DeferredWork = mongoose.model('DeferredWork', deferredWorkSchema);
|
|
23
|
+
|
|
24
|
+
module.exports = { DeferredWork, RETENTION_MS };
|
|
@@ -15,6 +15,11 @@ const AIRTABLE_API_URL = 'https://api.airtable.com/v0';
|
|
|
15
15
|
const AXIOS_TIMEOUT_CODE = 'ECONNABORTED';
|
|
16
16
|
const RETRY_BASE_DELAY_MS = 1000;
|
|
17
17
|
|
|
18
|
+
function isPermanentAirtableError(error) {
|
|
19
|
+
const status = error?.statusCode;
|
|
20
|
+
return typeof status === 'number' && status >= 400 && status < 500 && status !== 429;
|
|
21
|
+
}
|
|
22
|
+
|
|
18
23
|
let isEvalMode = false;
|
|
19
24
|
|
|
20
25
|
function setEvalMode(enabled) {
|
|
@@ -42,6 +47,7 @@ async function withAirtableRetry(operation, label, { retryNetworkErrors = true }
|
|
|
42
47
|
const isTransient = TRANSIENT_STATUS_CODES.includes(error.statusCode)
|
|
43
48
|
|| (retryNetworkErrors && isTransientNetworkError(error));
|
|
44
49
|
if (!isTransient || attempt === MAX_ATTEMPTS) {
|
|
50
|
+
if (isPermanentAirtableError(error)) error.permanent = true;
|
|
45
51
|
throw error;
|
|
46
52
|
}
|
|
47
53
|
const baseDelay = error.statusCode === RATE_LIMITED_STATUS ? RATE_LIMIT_DELAY_MS : RETRY_BASE_DELAY_MS;
|
|
@@ -229,6 +235,7 @@ async function upsertRecord(baseID, tableName, fields, { mergeOn } = {}, context
|
|
|
229
235
|
|
|
230
236
|
module.exports = {
|
|
231
237
|
setEvalMode,
|
|
238
|
+
isPermanentAirtableError,
|
|
232
239
|
addRecord,
|
|
233
240
|
getRecords,
|
|
234
241
|
getRecordByFilter,
|
package/package.json
CHANGED
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
function createWorkflowRunner({ queueAdapter } = {}) {
|
|
2
|
-
if (!queueAdapter) throw new Error('createWorkflowRunner requires a queueAdapter');
|
|
3
|
-
const workflows = new Map();
|
|
4
|
-
|
|
5
|
-
async function register(workflow = {}) {
|
|
6
|
-
if (!workflow.kind || typeof workflow.kind !== 'string') throw new Error('workflow requires a kind');
|
|
7
|
-
if (workflow.kind.includes('__')) throw new Error(`workflow '${workflow.kind}' must not contain '__' (reserved as the jobId delimiter)`);
|
|
8
|
-
if (typeof workflow.dedupeKey !== 'function') throw new Error(`workflow '${workflow.kind}' requires a dedupeKey function`);
|
|
9
|
-
if (typeof workflow.prepare !== 'function') throw new Error(`workflow '${workflow.kind}' requires a prepare function`);
|
|
10
|
-
if (workflows.has(workflow.kind)) throw new Error(`workflow '${workflow.kind}' is already registered`);
|
|
11
|
-
workflows.set(workflow.kind, workflow);
|
|
12
|
-
try {
|
|
13
|
-
await queueAdapter.process(workflow.kind, (trigger) => workflow.prepare(trigger));
|
|
14
|
-
} catch (error) {
|
|
15
|
-
workflows.delete(workflow.kind);
|
|
16
|
-
throw error;
|
|
17
|
-
}
|
|
18
|
-
return workflow;
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
function enqueue(kind, trigger, options = {}) {
|
|
22
|
-
const workflow = workflows.get(kind);
|
|
23
|
-
if (!workflow) throw new Error(`no workflow registered for '${kind}'`);
|
|
24
|
-
const dedupeKey = workflow.dedupeKey(trigger);
|
|
25
|
-
if (dedupeKey == null || (typeof dedupeKey === 'string' && !dedupeKey.trim())) {
|
|
26
|
-
throw new Error(`workflow '${kind}' produced an empty dedupe key`);
|
|
27
|
-
}
|
|
28
|
-
if (typeof dedupeKey !== 'string') throw new Error(`workflow '${kind}' produced a non-string dedupe key`);
|
|
29
|
-
return queueAdapter.enqueue(kind, trigger, { ...options, jobId: `${kind}__${dedupeKey}` });
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
return { register, enqueue };
|
|
33
|
-
}
|
|
34
|
-
|
|
35
|
-
module.exports = { createWorkflowRunner };
|