@cogitator-ai/channels 0.2.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/README.md +349 -0
- package/dist/assistant-config.d.ts +92 -0
- package/dist/assistant-config.d.ts.map +1 -0
- package/dist/assistant-config.js +89 -0
- package/dist/assistant-config.js.map +1 -0
- package/dist/capabilities.d.ts +14 -0
- package/dist/capabilities.d.ts.map +1 -0
- package/dist/capabilities.js +28 -0
- package/dist/capabilities.js.map +1 -0
- package/dist/channels/discord.d.ts +26 -0
- package/dist/channels/discord.d.ts.map +1 -0
- package/dist/channels/discord.js +153 -0
- package/dist/channels/discord.js.map +1 -0
- package/dist/channels/slack.d.ts +23 -0
- package/dist/channels/slack.d.ts.map +1 -0
- package/dist/channels/slack.js +89 -0
- package/dist/channels/slack.js.map +1 -0
- package/dist/channels/telegram.d.ts +30 -0
- package/dist/channels/telegram.d.ts.map +1 -0
- package/dist/channels/telegram.js +236 -0
- package/dist/channels/telegram.js.map +1 -0
- package/dist/channels/terminal.d.ts +30 -0
- package/dist/channels/terminal.d.ts.map +1 -0
- package/dist/channels/terminal.js +110 -0
- package/dist/channels/terminal.js.map +1 -0
- package/dist/channels/webchat.d.ts +24 -0
- package/dist/channels/webchat.d.ts.map +1 -0
- package/dist/channels/webchat.js +106 -0
- package/dist/channels/webchat.js.map +1 -0
- package/dist/channels/whatsapp.d.ts +28 -0
- package/dist/channels/whatsapp.d.ts.map +1 -0
- package/dist/channels/whatsapp.js +117 -0
- package/dist/channels/whatsapp.js.map +1 -0
- package/dist/envelope.d.ts +4 -0
- package/dist/envelope.d.ts.map +1 -0
- package/dist/envelope.js +64 -0
- package/dist/envelope.js.map +1 -0
- package/dist/formatters/discord-chunker.d.ts +6 -0
- package/dist/formatters/discord-chunker.d.ts.map +1 -0
- package/dist/formatters/discord-chunker.js +169 -0
- package/dist/formatters/discord-chunker.js.map +1 -0
- package/dist/formatters/markdown.d.ts +5 -0
- package/dist/formatters/markdown.d.ts.map +1 -0
- package/dist/formatters/markdown.js +59 -0
- package/dist/formatters/markdown.js.map +1 -0
- package/dist/formatters/whatsapp-markdown.d.ts +2 -0
- package/dist/formatters/whatsapp-markdown.d.ts.map +1 -0
- package/dist/formatters/whatsapp-markdown.js +26 -0
- package/dist/formatters/whatsapp-markdown.js.map +1 -0
- package/dist/gateway.d.ts +38 -0
- package/dist/gateway.d.ts.map +1 -0
- package/dist/gateway.js +347 -0
- package/dist/gateway.js.map +1 -0
- package/dist/heartbeat.d.ts +28 -0
- package/dist/heartbeat.d.ts.map +1 -0
- package/dist/heartbeat.js +143 -0
- package/dist/heartbeat.js.map +1 -0
- package/dist/hooks.d.ts +4 -0
- package/dist/hooks.d.ts.map +1 -0
- package/dist/hooks.js +30 -0
- package/dist/hooks.js.map +1 -0
- package/dist/inbound-debounce.d.ts +13 -0
- package/dist/inbound-debounce.d.ts.map +1 -0
- package/dist/inbound-debounce.js +61 -0
- package/dist/inbound-debounce.js.map +1 -0
- package/dist/index.d.ts +54 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +33 -0
- package/dist/index.js.map +1 -0
- package/dist/media/deepgram-stt.d.ts +14 -0
- package/dist/media/deepgram-stt.d.ts.map +1 -0
- package/dist/media/deepgram-stt.js +30 -0
- package/dist/media/deepgram-stt.js.map +1 -0
- package/dist/media/index.d.ts +7 -0
- package/dist/media/index.d.ts.map +1 -0
- package/dist/media/index.js +5 -0
- package/dist/media/index.js.map +1 -0
- package/dist/media/media-processor.d.ts +20 -0
- package/dist/media/media-processor.d.ts.map +1 -0
- package/dist/media/media-processor.js +80 -0
- package/dist/media/media-processor.js.map +1 -0
- package/dist/media/whisper-api.d.ts +22 -0
- package/dist/media/whisper-api.d.ts.map +1 -0
- package/dist/media/whisper-api.js +53 -0
- package/dist/media/whisper-api.js.map +1 -0
- package/dist/media/whisper-local.d.ts +16 -0
- package/dist/media/whisper-local.d.ts.map +1 -0
- package/dist/media/whisper-local.js +155 -0
- package/dist/media/whisper-local.js.map +1 -0
- package/dist/media/whisper-tool.d.ts +4 -0
- package/dist/media/whisper-tool.d.ts.map +1 -0
- package/dist/media/whisper-tool.js +19 -0
- package/dist/media/whisper-tool.js.map +1 -0
- package/dist/message-queue.d.ts +17 -0
- package/dist/message-queue.d.ts.map +1 -0
- package/dist/message-queue.js +115 -0
- package/dist/message-queue.js.map +1 -0
- package/dist/middleware/auto-extract.d.ts +36 -0
- package/dist/middleware/auto-extract.d.ts.map +1 -0
- package/dist/middleware/auto-extract.js +73 -0
- package/dist/middleware/auto-extract.js.map +1 -0
- package/dist/middleware/dm-policy.d.ts +39 -0
- package/dist/middleware/dm-policy.d.ts.map +1 -0
- package/dist/middleware/dm-policy.js +177 -0
- package/dist/middleware/dm-policy.js.map +1 -0
- package/dist/middleware/owner-commands.d.ts +25 -0
- package/dist/middleware/owner-commands.d.ts.map +1 -0
- package/dist/middleware/owner-commands.js +131 -0
- package/dist/middleware/owner-commands.js.map +1 -0
- package/dist/middleware/pairing.d.ts +20 -0
- package/dist/middleware/pairing.d.ts.map +1 -0
- package/dist/middleware/pairing.js +70 -0
- package/dist/middleware/pairing.js.map +1 -0
- package/dist/middleware/rate-limit.d.ts +14 -0
- package/dist/middleware/rate-limit.d.ts.map +1 -0
- package/dist/middleware/rate-limit.js +31 -0
- package/dist/middleware/rate-limit.js.map +1 -0
- package/dist/runtime-builder.d.ts +42 -0
- package/dist/runtime-builder.d.ts.map +1 -0
- package/dist/runtime-builder.js +607 -0
- package/dist/runtime-builder.js.map +1 -0
- package/dist/simple-timer-store.d.ts +28 -0
- package/dist/simple-timer-store.d.ts.map +1 -0
- package/dist/simple-timer-store.js +129 -0
- package/dist/simple-timer-store.js.map +1 -0
- package/dist/status-reactions.d.ts +22 -0
- package/dist/status-reactions.d.ts.map +1 -0
- package/dist/status-reactions.js +86 -0
- package/dist/status-reactions.js.map +1 -0
- package/dist/stream-buffer.d.ts +31 -0
- package/dist/stream-buffer.d.ts.map +1 -0
- package/dist/stream-buffer.js +170 -0
- package/dist/stream-buffer.js.map +1 -0
- package/dist/tools/self-config.d.ts +9 -0
- package/dist/tools/self-config.d.ts.map +1 -0
- package/dist/tools/self-config.js +122 -0
- package/dist/tools/self-config.js.map +1 -0
- package/package.json +102 -0
package/dist/gateway.js
ADDED
|
@@ -0,0 +1,347 @@
|
|
|
1
|
+
import { SessionManager, CompactionService } from '@cogitator-ai/memory';
|
|
2
|
+
import { StreamBuffer } from './stream-buffer';
|
|
3
|
+
import { adaptMarkdown, chunkMessage, getPlatformLimit } from './formatters/markdown';
|
|
4
|
+
import { StatusReactionTracker } from './status-reactions';
|
|
5
|
+
import { InboundDebouncer } from './inbound-debounce';
|
|
6
|
+
import { formatEnvelope } from './envelope';
|
|
7
|
+
import { MessageQueue } from './message-queue';
|
|
8
|
+
export class Gateway {
|
|
9
|
+
config;
|
|
10
|
+
channels;
|
|
11
|
+
sessionManager;
|
|
12
|
+
compactionService;
|
|
13
|
+
middlewares;
|
|
14
|
+
streamConfig;
|
|
15
|
+
debouncer;
|
|
16
|
+
messageQueue;
|
|
17
|
+
hooks;
|
|
18
|
+
lastMessageTime = new Map();
|
|
19
|
+
running = false;
|
|
20
|
+
startedAt = null;
|
|
21
|
+
messageCount = 0;
|
|
22
|
+
constructor(config) {
|
|
23
|
+
this.config = config;
|
|
24
|
+
this.channels = config.channels;
|
|
25
|
+
this.middlewares = config.middleware ?? [];
|
|
26
|
+
this.streamConfig = config.stream ?? { flushInterval: 500, minChunkSize: 20 };
|
|
27
|
+
if (config.memory) {
|
|
28
|
+
this.sessionManager = config.sessionManager ? null : new SessionManager(config.memory);
|
|
29
|
+
this.compactionService = config.session?.compaction
|
|
30
|
+
? new CompactionService({
|
|
31
|
+
adapter: config.memory,
|
|
32
|
+
summarize: this.createSummarizeFn(),
|
|
33
|
+
})
|
|
34
|
+
: null;
|
|
35
|
+
}
|
|
36
|
+
else {
|
|
37
|
+
this.sessionManager = null;
|
|
38
|
+
this.compactionService = null;
|
|
39
|
+
}
|
|
40
|
+
this.hooks = config.hooks ?? null;
|
|
41
|
+
this.debouncer = config.debounce?.enabled
|
|
42
|
+
? new InboundDebouncer(config.debounce, (merged) => this.processMessage(merged))
|
|
43
|
+
: null;
|
|
44
|
+
this.messageQueue =
|
|
45
|
+
config.queueMode && config.queueMode !== 'parallel'
|
|
46
|
+
? new MessageQueue(config.queueMode, (msg) => this.processMessage(msg))
|
|
47
|
+
: null;
|
|
48
|
+
for (const channel of this.channels) {
|
|
49
|
+
channel.onMessage((msg) => this.handleIncoming(msg));
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
async start() {
|
|
53
|
+
if (this.running)
|
|
54
|
+
return;
|
|
55
|
+
this.running = true;
|
|
56
|
+
this.startedAt = Date.now();
|
|
57
|
+
await Promise.all(this.channels.map((ch) => ch.start()));
|
|
58
|
+
}
|
|
59
|
+
async stop() {
|
|
60
|
+
if (!this.running)
|
|
61
|
+
return;
|
|
62
|
+
this.running = false;
|
|
63
|
+
if (this.debouncer)
|
|
64
|
+
await this.debouncer.flushAll();
|
|
65
|
+
this.messageQueue?.dispose();
|
|
66
|
+
await Promise.all(this.channels.map((ch) => ch.stop()));
|
|
67
|
+
}
|
|
68
|
+
get stats() {
|
|
69
|
+
return {
|
|
70
|
+
uptime: this.startedAt ? Date.now() - this.startedAt : 0,
|
|
71
|
+
activeSessions: 0,
|
|
72
|
+
totalSessions: 0,
|
|
73
|
+
messagesToday: this.messageCount,
|
|
74
|
+
connectedChannels: this.channels.map((ch) => ch.type),
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
async injectMessage(msg) {
|
|
78
|
+
return this.handleIncoming(msg);
|
|
79
|
+
}
|
|
80
|
+
async handleIncoming(msg) {
|
|
81
|
+
if (this.debouncer) {
|
|
82
|
+
this.debouncer.enqueue(msg);
|
|
83
|
+
}
|
|
84
|
+
else if (this.messageQueue) {
|
|
85
|
+
const threadId = this.getThreadId(msg);
|
|
86
|
+
this.messageQueue.push(msg, threadId);
|
|
87
|
+
}
|
|
88
|
+
else {
|
|
89
|
+
await this.processMessage(msg);
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
async processMessage(msg) {
|
|
93
|
+
try {
|
|
94
|
+
const channel = this.channels.find((ch) => ch.type === msg.channelType);
|
|
95
|
+
if (!channel)
|
|
96
|
+
return;
|
|
97
|
+
const user = {
|
|
98
|
+
id: msg.userId,
|
|
99
|
+
channelType: msg.channelType,
|
|
100
|
+
name: msg.userName,
|
|
101
|
+
};
|
|
102
|
+
const threadId = this.getThreadId(msg);
|
|
103
|
+
const ctx = this.createMiddlewareContext(threadId, user, channel);
|
|
104
|
+
let middlewarePassed = true;
|
|
105
|
+
if (this.middlewares.length > 0) {
|
|
106
|
+
let nextCalled = false;
|
|
107
|
+
const runAll = async (index) => {
|
|
108
|
+
if (index >= this.middlewares.length) {
|
|
109
|
+
nextCalled = true;
|
|
110
|
+
return;
|
|
111
|
+
}
|
|
112
|
+
await this.middlewares[index].handle(msg, ctx, () => runAll(index + 1));
|
|
113
|
+
};
|
|
114
|
+
await runAll(0);
|
|
115
|
+
middlewarePassed = nextCalled;
|
|
116
|
+
}
|
|
117
|
+
if (!middlewarePassed)
|
|
118
|
+
return;
|
|
119
|
+
await this.hooks?.emit('message:received', { msg, threadId, user });
|
|
120
|
+
this.messageCount++;
|
|
121
|
+
if (this.config.envelope?.enabled) {
|
|
122
|
+
const prevTime = this.lastMessageTime.get(threadId);
|
|
123
|
+
msg = { ...msg, text: formatEnvelope(msg, this.config.envelope, prevTime) };
|
|
124
|
+
this.lastMessageTime.set(threadId, Date.now());
|
|
125
|
+
}
|
|
126
|
+
if (this.sessionManager) {
|
|
127
|
+
const session = await this.sessionManager.getOrCreate({
|
|
128
|
+
userId: msg.userId,
|
|
129
|
+
channelType: msg.channelType,
|
|
130
|
+
channelId: msg.channelId,
|
|
131
|
+
agentId: this.getAgentName(),
|
|
132
|
+
});
|
|
133
|
+
await this.sessionManager.incrementMessageCount(session.id);
|
|
134
|
+
if (this.compactionService && this.config.session?.compaction) {
|
|
135
|
+
const compaction = this.config.session.compaction;
|
|
136
|
+
if (session.messageCount >= compaction.threshold) {
|
|
137
|
+
await this.compactionService.compact(session.id, compaction);
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
const agent = await this.resolveAgent(user);
|
|
142
|
+
const reactionsEnabled = this.config.reactions?.enabled && channel.setReaction;
|
|
143
|
+
let tracker;
|
|
144
|
+
if (reactionsEnabled) {
|
|
145
|
+
tracker = new StatusReactionTracker(channel, msg.channelId, msg.id, this.config.reactions);
|
|
146
|
+
tracker.setPhase('queued');
|
|
147
|
+
}
|
|
148
|
+
await channel.sendTyping(msg.channelId);
|
|
149
|
+
const typingInterval = setInterval(() => {
|
|
150
|
+
channel.sendTyping(msg.channelId).catch(() => { });
|
|
151
|
+
}, 4000);
|
|
152
|
+
try {
|
|
153
|
+
tracker?.setPhase('thinking');
|
|
154
|
+
if (this.config.stream) {
|
|
155
|
+
await this.runStreaming(agent, msg, channel, threadId);
|
|
156
|
+
}
|
|
157
|
+
else {
|
|
158
|
+
await this.runDirect(agent, msg, channel, threadId);
|
|
159
|
+
}
|
|
160
|
+
tracker?.setPhase('done');
|
|
161
|
+
}
|
|
162
|
+
catch (error) {
|
|
163
|
+
tracker?.setPhase('error');
|
|
164
|
+
throw error;
|
|
165
|
+
}
|
|
166
|
+
finally {
|
|
167
|
+
clearInterval(typingInterval);
|
|
168
|
+
tracker?.dispose();
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
catch (error) {
|
|
172
|
+
if (this.config.onError) {
|
|
173
|
+
this.config.onError(error instanceof Error ? error : new Error(String(error)), msg);
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
}
|
|
177
|
+
async extractMedia(msg, agent) {
|
|
178
|
+
let input = msg.text;
|
|
179
|
+
let images;
|
|
180
|
+
if (this.config.mediaProcessor && msg.attachments?.length) {
|
|
181
|
+
console.log(`[gateway] Processing ${msg.attachments.length} attachment(s), types: ${msg.attachments.map((a) => a.type).join(', ')}`);
|
|
182
|
+
const result = await this.config.mediaProcessor.process(msg.attachments, agent.model);
|
|
183
|
+
console.log(`[gateway] Media result: images=${result.images.length}, text=${!!result.transcribedText}, notes=${result.systemNotes.length}`);
|
|
184
|
+
if (result.images.length > 0)
|
|
185
|
+
images = result.images;
|
|
186
|
+
if (result.transcribedText) {
|
|
187
|
+
input = result.transcribedText + (input ? `\n${input}` : '');
|
|
188
|
+
}
|
|
189
|
+
if (result.systemNotes.length > 0) {
|
|
190
|
+
input = (input || '') + '\n' + result.systemNotes.join('\n');
|
|
191
|
+
}
|
|
192
|
+
}
|
|
193
|
+
if (!input && !images?.length) {
|
|
194
|
+
input = '[empty message]';
|
|
195
|
+
}
|
|
196
|
+
return { input, images };
|
|
197
|
+
}
|
|
198
|
+
async runDirect(agent, msg, channel, threadId) {
|
|
199
|
+
const { input, images } = await this.extractMedia(msg, agent);
|
|
200
|
+
const replyTo = msg.raw?.scheduled ? undefined : msg.id;
|
|
201
|
+
await this.hooks?.emit('agent:before_run', { msg, threadId, agent: agent.name });
|
|
202
|
+
let result;
|
|
203
|
+
try {
|
|
204
|
+
result = await this.config.cogitator.run(agent, {
|
|
205
|
+
input,
|
|
206
|
+
threadId,
|
|
207
|
+
useMemory: !!this.config.memory,
|
|
208
|
+
userId: msg.userId,
|
|
209
|
+
channelType: msg.channelType,
|
|
210
|
+
channelId: msg.channelId,
|
|
211
|
+
...(images ? { images } : {}),
|
|
212
|
+
});
|
|
213
|
+
}
|
|
214
|
+
catch (error) {
|
|
215
|
+
await this.hooks?.emit('agent:error', { msg, threadId, error });
|
|
216
|
+
throw error;
|
|
217
|
+
}
|
|
218
|
+
await this.hooks?.emit('agent:after_run', { msg, threadId, output: result.output });
|
|
219
|
+
const output = adaptMarkdown(result.output, msg.channelType);
|
|
220
|
+
await this.hooks?.emit('message:sending', {
|
|
221
|
+
msg,
|
|
222
|
+
threadId,
|
|
223
|
+
text: output,
|
|
224
|
+
channelId: msg.channelId,
|
|
225
|
+
});
|
|
226
|
+
const limit = getPlatformLimit(msg.channelType);
|
|
227
|
+
const chunks = chunkMessage(output, limit);
|
|
228
|
+
let sentId = '';
|
|
229
|
+
for (let i = 0; i < chunks.length; i++) {
|
|
230
|
+
sentId = await channel.sendText(msg.channelId, chunks[i], {
|
|
231
|
+
...(i === 0 && replyTo ? { replyTo } : {}),
|
|
232
|
+
format: 'markdown',
|
|
233
|
+
});
|
|
234
|
+
}
|
|
235
|
+
await this.hooks?.emit('message:sent', {
|
|
236
|
+
msg,
|
|
237
|
+
threadId,
|
|
238
|
+
text: output,
|
|
239
|
+
messageId: sentId,
|
|
240
|
+
});
|
|
241
|
+
}
|
|
242
|
+
async runStreaming(agent, msg, channel, threadId) {
|
|
243
|
+
const { input, images } = await this.extractMedia(msg, agent);
|
|
244
|
+
console.log(`[gateway] Running LLM, input: "${input.slice(0, 200)}", images: ${images?.length ?? 0}`);
|
|
245
|
+
const replyTo = msg.raw?.scheduled ? undefined : msg.id;
|
|
246
|
+
const useDraft = !!channel.sendDraft;
|
|
247
|
+
const streamCfg = {
|
|
248
|
+
...this.streamConfig,
|
|
249
|
+
maxMessageChars: this.streamConfig.maxMessageChars ?? getPlatformLimit(msg.channelType),
|
|
250
|
+
};
|
|
251
|
+
const stream = new StreamBuffer(channel, msg.channelId, streamCfg, replyTo, useDraft);
|
|
252
|
+
stream.start();
|
|
253
|
+
await this.hooks?.emit('agent:before_run', { msg, threadId, agent: agent.name });
|
|
254
|
+
await this.hooks?.emit('stream:started', { msg, threadId });
|
|
255
|
+
try {
|
|
256
|
+
let tokenCount = 0;
|
|
257
|
+
const runPromise = this.config.cogitator.run(agent, {
|
|
258
|
+
input,
|
|
259
|
+
threadId,
|
|
260
|
+
useMemory: !!this.config.memory,
|
|
261
|
+
userId: msg.userId,
|
|
262
|
+
channelType: msg.channelType,
|
|
263
|
+
channelId: msg.channelId,
|
|
264
|
+
stream: true,
|
|
265
|
+
onToken: (token) => {
|
|
266
|
+
tokenCount++;
|
|
267
|
+
stream.append(token);
|
|
268
|
+
},
|
|
269
|
+
...(images ? { images } : {}),
|
|
270
|
+
});
|
|
271
|
+
const timeout = new Promise((_, reject) => setTimeout(() => reject(new Error('LLM response timeout (90s)')), 90_000));
|
|
272
|
+
const result = await Promise.race([runPromise, timeout]);
|
|
273
|
+
if (tokenCount > 0) {
|
|
274
|
+
await stream.finish();
|
|
275
|
+
}
|
|
276
|
+
else {
|
|
277
|
+
await stream.abort();
|
|
278
|
+
if (result.output) {
|
|
279
|
+
const output = adaptMarkdown(result.output, msg.channelType);
|
|
280
|
+
const limit = getPlatformLimit(msg.channelType);
|
|
281
|
+
const chunks = chunkMessage(output, limit);
|
|
282
|
+
for (let i = 0; i < chunks.length; i++) {
|
|
283
|
+
await channel.sendText(msg.channelId, chunks[i], {
|
|
284
|
+
...(i === 0 && replyTo ? { replyTo } : {}),
|
|
285
|
+
format: 'markdown',
|
|
286
|
+
});
|
|
287
|
+
}
|
|
288
|
+
}
|
|
289
|
+
}
|
|
290
|
+
await this.hooks?.emit('agent:after_run', { msg, threadId, output: result.output });
|
|
291
|
+
await this.hooks?.emit('stream:finished', {
|
|
292
|
+
msg,
|
|
293
|
+
threadId,
|
|
294
|
+
messageIds: stream.getMessageIds(),
|
|
295
|
+
});
|
|
296
|
+
}
|
|
297
|
+
catch (error) {
|
|
298
|
+
console.error('[gateway] LLM run error:', error.message);
|
|
299
|
+
await stream.abort();
|
|
300
|
+
await this.hooks?.emit('agent:error', { msg, threadId, error });
|
|
301
|
+
throw error;
|
|
302
|
+
}
|
|
303
|
+
}
|
|
304
|
+
getThreadId(msg) {
|
|
305
|
+
if (this.config.session?.threadKey) {
|
|
306
|
+
return this.config.session.threadKey(msg);
|
|
307
|
+
}
|
|
308
|
+
return `${msg.channelType}:${msg.userId}`;
|
|
309
|
+
}
|
|
310
|
+
async resolveAgent(user) {
|
|
311
|
+
const agentOrFactory = this.config.agent;
|
|
312
|
+
if (typeof agentOrFactory === 'function') {
|
|
313
|
+
return agentOrFactory(user);
|
|
314
|
+
}
|
|
315
|
+
return agentOrFactory;
|
|
316
|
+
}
|
|
317
|
+
getAgentName() {
|
|
318
|
+
const agentOrFactory = this.config.agent;
|
|
319
|
+
if (typeof agentOrFactory === 'function') {
|
|
320
|
+
return 'dynamic';
|
|
321
|
+
}
|
|
322
|
+
return agentOrFactory.name;
|
|
323
|
+
}
|
|
324
|
+
createMiddlewareContext(threadId, user, channel) {
|
|
325
|
+
const store = new Map();
|
|
326
|
+
return {
|
|
327
|
+
threadId,
|
|
328
|
+
user,
|
|
329
|
+
channel,
|
|
330
|
+
set(key, value) {
|
|
331
|
+
store.set(key, value);
|
|
332
|
+
},
|
|
333
|
+
get(key) {
|
|
334
|
+
return store.get(key);
|
|
335
|
+
},
|
|
336
|
+
};
|
|
337
|
+
}
|
|
338
|
+
createSummarizeFn() {
|
|
339
|
+
return async (messages) => {
|
|
340
|
+
const text = messages
|
|
341
|
+
.map((m) => `${m.role}: ${typeof m.content === 'string' ? m.content : JSON.stringify(m.content)}`)
|
|
342
|
+
.join('\n');
|
|
343
|
+
return `[Summary of ${messages.length} messages]: ${text.slice(0, 500)}`;
|
|
344
|
+
};
|
|
345
|
+
}
|
|
346
|
+
}
|
|
347
|
+
//# sourceMappingURL=gateway.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gateway.js","sourceRoot":"","sources":["../src/gateway.ts"],"names":[],"mappings":"AAcA,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzE,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEtF,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAO/C,MAAM,OAAO,OAAO;IACD,MAAM,CAAoB;IAC1B,QAAQ,CAAY;IACpB,cAAc,CAAwB;IACtC,iBAAiB,CAA2B;IAC5C,WAAW,CAAsB;IACjC,YAAY,CAAe;IAC3B,SAAS,CAA0B;IACnC,YAAY,CAAsB;IAClC,KAAK,CAAsB;IAC3B,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrD,OAAO,GAAG,KAAK,CAAC;IAChB,SAAS,GAAkB,IAAI,CAAC;IAChC,YAAY,GAAG,CAAC,CAAC;IAEzB,YAAY,MAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;QAE9E,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACvF,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,OAAO,EAAE,UAAU;gBACjD,CAAC,CAAC,IAAI,iBAAiB,CAAC;oBACpB,OAAO,EAAE,MAAM,CAAC,MAAM;oBACtB,SAAS,EAAE,IAAI,CAAC,iBAAiB,EAAE;iBACpC,CAAC;gBACJ,CAAC,CAAC,IAAI,CAAC;QACX,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;QAChC,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC;QAElC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,QAAQ,EAAE,OAAO;YACvC,CAAC,CAAC,IAAI,gBAAgB,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAChF,CAAC,CAAC,IAAI,CAAC;QAET,IAAI,CAAC,YAAY;YACf,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,KAAK,UAAU;gBACjD,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;gBACvE,CAAC,CAAC,IAAI,CAAC;QAEX,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE5B,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO;QAC1B,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QAErB,IAAI,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;QACpD,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;QAE7B,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,KAAK;QACP,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACxD,cAAc,EAAE,CAAC;YACjB,aAAa,EAAE,CAAC;YAChB,aAAa,EAAE,IAAI,CAAC,YAAY;YAChC,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;SACtD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAmB;QACrC,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,GAAmB;QAC9C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC;aAAM,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACvC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,GAAmB;QAC9C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,GAAG,CAAC,WAAW,CAAC,CAAC;YACxE,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,MAAM,IAAI,GAAgB;gBACxB,EAAE,EAAE,GAAG,CAAC,MAAM;gBACd,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,IAAI,EAAE,GAAG,CAAC,QAAQ;aACnB,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACvC,MAAM,GAAG,GAAG,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAElE,IAAI,gBAAgB,GAAG,IAAI,CAAC;YAC5B,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,IAAI,UAAU,GAAG,KAAK,CAAC;gBACvB,MAAM,MAAM,GAAG,KAAK,EAAE,KAAa,EAAiB,EAAE;oBACpD,IAAI,KAAK,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;wBACrC,UAAU,GAAG,IAAI,CAAC;wBAClB,OAAO;oBACT,CAAC;oBACD,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC1E,CAAC,CAAC;gBACF,MAAM,MAAM,CAAC,CAAC,CAAC,CAAC;gBAChB,gBAAgB,GAAG,UAAU,CAAC;YAChC,CAAC;YAED,IAAI,CAAC,gBAAgB;gBAAE,OAAO;YAE9B,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,kBAAkB,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;YAEpE,IAAI,CAAC,YAAY,EAAE,CAAC;YAEpB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACpD,GAAG,GAAG,EAAE,GAAG,GAAG,EAAE,IAAI,EAAE,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;gBAC5E,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YACjD,CAAC;YAED,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;oBACpD,MAAM,EAAE,GAAG,CAAC,MAAM;oBAClB,WAAW,EAAE,GAAG,CAAC,WAAW;oBAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;iBAC7B,CAAC,CAAC;gBAEH,MAAM,IAAI,CAAC,cAAc,CAAC,qBAAqB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAE5D,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC;oBAC9D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;oBAClD,IAAI,OAAO,CAAC,YAAY,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;wBACjD,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;oBAC/D,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAE5C,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,IAAI,OAAO,CAAC,WAAW,CAAC;YAC/E,IAAI,OAA0C,CAAC;YAE/C,IAAI,gBAAgB,EAAE,CAAC;gBACrB,OAAO,GAAG,IAAI,qBAAqB,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC3F,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC7B,CAAC;YAED,MAAM,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACxC,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;gBACtC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACpD,CAAC,EAAE,IAAI,CAAC,CAAC;YAET,IAAI,CAAC;gBACH,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;gBAE9B,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;oBACvB,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;gBACzD,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;gBACtD,CAAC;gBAED,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC3B,MAAM,KAAK,CAAC;YACd,CAAC;oBAAS,CAAC;gBACT,aAAa,CAAC,cAAc,CAAC,CAAC;gBAC9B,OAAO,EAAE,OAAO,EAAE,CAAC;YACrB,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACtF,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,GAAmB,EACnB,KAAY;QAEZ,IAAI,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;QACrB,IAAI,MAAgC,CAAC;QAErC,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,GAAG,CAAC,WAAW,EAAE,MAAM,EAAE,CAAC;YAC1D,OAAO,CAAC,GAAG,CACT,wBAAwB,GAAG,CAAC,WAAW,CAAC,MAAM,0BAA0B,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxH,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACtF,OAAO,CAAC,GAAG,CACT,kCAAkC,MAAM,CAAC,MAAM,CAAC,MAAM,UAAU,CAAC,CAAC,MAAM,CAAC,eAAe,WAAW,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAC/H,CAAC;YAEF,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YACrD,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;gBAC3B,KAAK,GAAG,MAAM,CAAC,eAAe,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC/D,CAAC;YACD,IAAI,MAAM,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,IAAI,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAC9B,KAAK,GAAG,iBAAiB,CAAC;QAC5B,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,SAAS,CACrB,KAAY,EACZ,GAAmB,EACnB,OAAgB,EAChB,QAAgB;QAEhB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAI,GAAG,CAAC,GAA+B,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;QAErF,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,kBAAkB,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAEjF,IAAI,MAAM,CAAC;QACX,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;gBAC9C,KAAK;gBACL,QAAQ;gBACR,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM;gBAC/B,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC9B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAEpF,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;QAE7D,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE;YACxC,GAAG;YACH,QAAQ;YACR,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,GAAG,CAAC,SAAS;SACzB,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;gBACxD,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC1C,MAAM,EAAE,UAAU;aACnB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;YACrC,GAAG;YACH,QAAQ;YACR,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,KAAY,EACZ,GAAmB,EACnB,OAAgB,EAChB,QAAgB;QAEhB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC9D,OAAO,CAAC,GAAG,CACT,kCAAkC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,cAAc,MAAM,EAAE,MAAM,IAAI,CAAC,EAAE,CACzF,CAAC;QACF,MAAM,OAAO,GAAI,GAAG,CAAC,GAA+B,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;QACrF,MAAM,QAAQ,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;QACrC,MAAM,SAAS,GAAG;YAChB,GAAG,IAAI,CAAC,YAAY;YACpB,eAAe,EAAE,IAAI,CAAC,YAAY,CAAC,eAAe,IAAI,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC;SACxF,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QACtF,MAAM,CAAC,KAAK,EAAE,CAAC;QAEf,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,kBAAkB,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACjF,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,IAAI,UAAU,GAAG,CAAC,CAAC;YAEnB,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;gBAClD,KAAK;gBACL,QAAQ;gBACR,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM;gBAC/B,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,UAAU,EAAE,CAAC;oBACb,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;gBACD,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC9B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAC/C,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,EAAE,MAAM,CAAC,CAC1E,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;YAEzD,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClB,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;oBAC7D,MAAM,KAAK,GAAG,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;oBAChD,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;oBAC3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;wBACvC,MAAM,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;4BAC/C,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;4BAC1C,MAAM,EAAE,UAAU;yBACnB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YACpF,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE;gBACxC,GAAG;gBACH,QAAQ;gBACR,UAAU,EAAE,MAAM,CAAC,aAAa,EAAE;aACnC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;YACpE,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,GAAmB;QACrC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;IAC5C,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,IAAiB;QAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QACzC,IAAI,OAAO,cAAc,KAAK,UAAU,EAAE,CAAC;YACzC,OAAO,cAAc,CAAC,IAAI,CAAmB,CAAC;QAChD,CAAC;QACD,OAAO,cAAuB,CAAC;IACjC,CAAC;IAEO,YAAY;QAClB,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QACzC,IAAI,OAAO,cAAc,KAAK,UAAU,EAAE,CAAC;YACzC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAQ,cAAwB,CAAC,IAAI,CAAC;IACxC,CAAC;IAEO,uBAAuB,CAC7B,QAAgB,EAChB,IAAiB,EACjB,OAAgB;QAEhB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAmB,CAAC;QACzC,OAAO;YACL,QAAQ;YACR,IAAI;YACJ,OAAO;YACP,GAAG,CAAC,GAAW,EAAE,KAAc;gBAC7B,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACxB,CAAC;YACD,GAAG,CAAI,GAAW;gBAChB,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,CAAkB,CAAC;YACzC,CAAC;SACF,CAAC;IACJ,CAAC;IAEO,iBAAiB;QACvB,OAAO,KAAK,EAAE,QAAQ,EAAE,EAAE;YACxB,MAAM,IAAI,GAAG,QAAQ;iBAClB,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACJ,GAAG,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CACxF;iBACA,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,OAAO,eAAe,QAAQ,CAAC,MAAM,eAAe,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;QAC3E,CAAC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import type { ChannelMessage, TimerStore, TimerEntry } from '@cogitator-ai/types';
|
|
2
|
+
export interface HeartbeatConfig {
|
|
3
|
+
onFire: (msg: ChannelMessage) => Promise<void> | void;
|
|
4
|
+
pollInterval?: number;
|
|
5
|
+
getNextCronMs?: (cron: string) => number;
|
|
6
|
+
maxRetries?: number;
|
|
7
|
+
staggerMs?: number;
|
|
8
|
+
onRunComplete?: (entry: TimerEntry, status: 'ok' | 'error', error?: string, durationMs?: number) => void;
|
|
9
|
+
}
|
|
10
|
+
export declare class HeartbeatScheduler {
|
|
11
|
+
private store;
|
|
12
|
+
private config;
|
|
13
|
+
private timer;
|
|
14
|
+
private processing;
|
|
15
|
+
private readonly maxRetries;
|
|
16
|
+
constructor(store: TimerStore, config: HeartbeatConfig);
|
|
17
|
+
start(): void;
|
|
18
|
+
stop(): void;
|
|
19
|
+
listJobs(): Promise<TimerEntry[]>;
|
|
20
|
+
getJob(id: string): Promise<TimerEntry | null>;
|
|
21
|
+
cancelJob(id: string): Promise<void>;
|
|
22
|
+
enableJob(id: string): Promise<void>;
|
|
23
|
+
disableJob(id: string): Promise<void>;
|
|
24
|
+
private processOverdue;
|
|
25
|
+
private reschedule;
|
|
26
|
+
private rescheduleInterval;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=heartbeat.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"heartbeat.d.ts","sourceRoot":"","sources":["../src/heartbeat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAElF,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,CAAC,GAAG,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACtD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,CAAC;IACzC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,CACd,KAAK,EAAE,UAAU,EACjB,MAAM,EAAE,IAAI,GAAG,OAAO,EACtB,KAAK,CAAC,EAAE,MAAM,EACd,UAAU,CAAC,EAAE,MAAM,KAChB,IAAI,CAAC;CACX;AAED,qBAAa,kBAAkB;IAM3B,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,MAAM;IANhB,OAAO,CAAC,KAAK,CAA+C;IAC5D,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;gBAG1B,KAAK,EAAE,UAAU,EACjB,MAAM,EAAE,eAAe;IAKjC,KAAK,IAAI,IAAI;IAkBb,IAAI,IAAI,IAAI;IAON,QAAQ,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;IAIjC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAI9C,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpC,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YAI7B,cAAc;YA2Ed,UAAU;YAiBV,kBAAkB;CAejC"}
|
|
@@ -0,0 +1,143 @@
|
|
|
1
|
+
export class HeartbeatScheduler {
|
|
2
|
+
store;
|
|
3
|
+
config;
|
|
4
|
+
timer = null;
|
|
5
|
+
processing = false;
|
|
6
|
+
maxRetries;
|
|
7
|
+
constructor(store, config) {
|
|
8
|
+
this.store = store;
|
|
9
|
+
this.config = config;
|
|
10
|
+
this.maxRetries = config.maxRetries ?? 5;
|
|
11
|
+
}
|
|
12
|
+
start() {
|
|
13
|
+
const stagger = this.config.staggerMs ? Math.floor(Math.random() * this.config.staggerMs) : 0;
|
|
14
|
+
const begin = () => {
|
|
15
|
+
void this.processOverdue();
|
|
16
|
+
this.timer = setInterval(() => void this.processOverdue(), this.config.pollInterval ?? 30_000);
|
|
17
|
+
};
|
|
18
|
+
if (stagger > 0) {
|
|
19
|
+
setTimeout(begin, stagger);
|
|
20
|
+
}
|
|
21
|
+
else {
|
|
22
|
+
begin();
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
stop() {
|
|
26
|
+
if (this.timer) {
|
|
27
|
+
clearInterval(this.timer);
|
|
28
|
+
this.timer = null;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
async listJobs() {
|
|
32
|
+
return this.store.list();
|
|
33
|
+
}
|
|
34
|
+
async getJob(id) {
|
|
35
|
+
return this.store.get(id);
|
|
36
|
+
}
|
|
37
|
+
async cancelJob(id) {
|
|
38
|
+
return this.store.cancel(id);
|
|
39
|
+
}
|
|
40
|
+
async enableJob(id) {
|
|
41
|
+
await this.store.update(id, { enabled: true, consecutiveErrors: 0 });
|
|
42
|
+
}
|
|
43
|
+
async disableJob(id) {
|
|
44
|
+
await this.store.update(id, { enabled: false });
|
|
45
|
+
}
|
|
46
|
+
async processOverdue() {
|
|
47
|
+
if (this.processing)
|
|
48
|
+
return;
|
|
49
|
+
this.processing = true;
|
|
50
|
+
let overdue;
|
|
51
|
+
try {
|
|
52
|
+
overdue = await this.store.getOverdue();
|
|
53
|
+
}
|
|
54
|
+
catch {
|
|
55
|
+
this.processing = false;
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
for (const entry of overdue) {
|
|
59
|
+
if (entry.enabled === false)
|
|
60
|
+
continue;
|
|
61
|
+
if ((entry.consecutiveErrors ?? 0) >= this.maxRetries) {
|
|
62
|
+
await this.store.update(entry.id, { lastRunStatus: 'skipped' });
|
|
63
|
+
continue;
|
|
64
|
+
}
|
|
65
|
+
const meta = (entry.metadata ?? {});
|
|
66
|
+
const description = meta.description ?? '';
|
|
67
|
+
const msg = {
|
|
68
|
+
id: `heartbeat_${entry.id}`,
|
|
69
|
+
channelType: meta.channel ?? 'system',
|
|
70
|
+
channelId: meta.channelId ?? meta.userId ?? 'system',
|
|
71
|
+
userId: meta.userId ?? 'system',
|
|
72
|
+
text: `[SCHEDULED TASK] Execute this task that was scheduled earlier: "${description}". If it's a reminder — deliver it in a friendly way. If it's an action (e.g. check something, fetch data, run a tool) — do it and report the result.`,
|
|
73
|
+
raw: { scheduled: true, taskId: entry.id },
|
|
74
|
+
};
|
|
75
|
+
const startedAt = Date.now();
|
|
76
|
+
let status = 'ok';
|
|
77
|
+
let errorMsg;
|
|
78
|
+
try {
|
|
79
|
+
await this.config.onFire(msg);
|
|
80
|
+
}
|
|
81
|
+
catch (err) {
|
|
82
|
+
status = 'error';
|
|
83
|
+
errorMsg = err instanceof Error ? err.message : String(err);
|
|
84
|
+
}
|
|
85
|
+
const durationMs = Date.now() - startedAt;
|
|
86
|
+
const patch = {
|
|
87
|
+
lastRunAt: startedAt,
|
|
88
|
+
lastRunStatus: status,
|
|
89
|
+
};
|
|
90
|
+
if (status === 'error') {
|
|
91
|
+
patch.lastError = errorMsg;
|
|
92
|
+
if (meta.bestEffort) {
|
|
93
|
+
patch.consecutiveErrors = 0;
|
|
94
|
+
}
|
|
95
|
+
else {
|
|
96
|
+
patch.consecutiveErrors = (entry.consecutiveErrors ?? 0) + 1;
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
else {
|
|
100
|
+
patch.consecutiveErrors = 0;
|
|
101
|
+
patch.lastError = undefined;
|
|
102
|
+
}
|
|
103
|
+
await this.store.update(entry.id, patch);
|
|
104
|
+
await this.store.markFired(entry.id);
|
|
105
|
+
this.config.onRunComplete?.(entry, status, errorMsg, durationMs);
|
|
106
|
+
if (entry.cron) {
|
|
107
|
+
await this.reschedule(entry, meta);
|
|
108
|
+
}
|
|
109
|
+
else if (entry.type === 'recurring' && entry.interval) {
|
|
110
|
+
await this.rescheduleInterval(entry, meta);
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
this.processing = false;
|
|
114
|
+
}
|
|
115
|
+
async reschedule(entry, meta) {
|
|
116
|
+
const nextFire = this.config.getNextCronMs
|
|
117
|
+
? this.config.getNextCronMs(entry.cron)
|
|
118
|
+
: Date.now() + 60_000;
|
|
119
|
+
await this.store.schedule({
|
|
120
|
+
workflowId: 'heartbeat',
|
|
121
|
+
runId: 'scheduler',
|
|
122
|
+
nodeId: 'task',
|
|
123
|
+
firesAt: nextFire,
|
|
124
|
+
type: 'cron',
|
|
125
|
+
cron: entry.cron,
|
|
126
|
+
metadata: meta,
|
|
127
|
+
name: entry.name,
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
async rescheduleInterval(entry, meta) {
|
|
131
|
+
await this.store.schedule({
|
|
132
|
+
workflowId: 'heartbeat',
|
|
133
|
+
runId: 'scheduler',
|
|
134
|
+
nodeId: 'task',
|
|
135
|
+
firesAt: Date.now() + entry.interval,
|
|
136
|
+
type: 'recurring',
|
|
137
|
+
interval: entry.interval,
|
|
138
|
+
metadata: meta,
|
|
139
|
+
name: entry.name,
|
|
140
|
+
});
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
//# sourceMappingURL=heartbeat.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"heartbeat.js","sourceRoot":"","sources":["../src/heartbeat.ts"],"names":[],"mappings":"AAgBA,MAAM,OAAO,kBAAkB;IAMnB;IACA;IANF,KAAK,GAA0C,IAAI,CAAC;IACpD,UAAU,GAAG,KAAK,CAAC;IACV,UAAU,CAAS;IAEpC,YACU,KAAiB,EACjB,MAAuB;QADvB,UAAK,GAAL,KAAK,CAAY;QACjB,WAAM,GAAN,MAAM,CAAiB;QAE/B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9F,MAAM,KAAK,GAAG,GAAG,EAAE;YACjB,KAAK,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3B,IAAI,CAAC,KAAK,GAAG,WAAW,CACtB,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,cAAc,EAAE,EAChC,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,MAAM,CACnC,CAAC;QACJ,CAAC,CAAC;QAEF,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,KAAK,EAAE,CAAC;QACV,CAAC;IACH,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU;QACzB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAC5B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,IAAI,OAAqB,CAAC;QAC1B,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;YACxB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,OAAO,KAAK,KAAK;gBAAE,SAAS;YACtC,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACtD,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,CAAC;gBAChE,SAAS;YACX,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAA4B,CAAC;YAC/D,MAAM,WAAW,GAAI,IAAI,CAAC,WAAsB,IAAI,EAAE,CAAC;YACvD,MAAM,GAAG,GAAmB;gBAC1B,EAAE,EAAE,aAAa,KAAK,CAAC,EAAE,EAAE;gBAC3B,WAAW,EAAG,IAAI,CAAC,OAAkB,IAAI,QAAQ;gBACjD,SAAS,EAAG,IAAI,CAAC,SAAoB,IAAK,IAAI,CAAC,MAAiB,IAAI,QAAQ;gBAC5E,MAAM,EAAG,IAAI,CAAC,MAAiB,IAAI,QAAQ;gBAC3C,IAAI,EAAE,mEAAmE,WAAW,uJAAuJ;gBAC3O,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE;aAC3C,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,MAAM,GAAmB,IAAI,CAAC;YAClC,IAAI,QAA4B,CAAC;YAEjC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,OAAO,CAAC;gBACjB,QAAQ,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC9D,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE1C,MAAM,KAAK,GAAwB;gBACjC,SAAS,EAAE,SAAS;gBACpB,aAAa,EAAE,MAAM;aACtB,CAAC;YAEF,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;gBACvB,KAAK,CAAC,SAAS,GAAG,QAAQ,CAAC;gBAC3B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACpB,KAAK,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBAC9B,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,iBAAiB,GAAG,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBAC5B,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;YAC9B,CAAC;YAED,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YACzC,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAErC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;YAEjE,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACrC,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACxD,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAiB,EAAE,IAA6B;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAK,CAAC;YACxC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC;QAExB,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACxB,UAAU,EAAE,WAAW;YACvB,KAAK,EAAE,WAAW;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,QAAQ;YACjB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,KAAK,CAAC,IAAI;SACjB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,KAAiB,EACjB,IAA6B;QAE7B,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACxB,UAAU,EAAE,WAAW;YACvB,KAAK,EAAE,WAAW;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,QAAS;YACrC,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,KAAK,CAAC,IAAI;SACjB,CAAC,CAAC;IACL,CAAC;CACF"}
|
package/dist/hooks.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../src/hooks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAE/E,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;AAEpD,wBAAgB,kBAAkB,IAAI,YAAY,CA6BjD"}
|
package/dist/hooks.js
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
export function createHookRegistry() {
|
|
2
|
+
const hooks = new Map();
|
|
3
|
+
return {
|
|
4
|
+
on(hook, handler) {
|
|
5
|
+
let set = hooks.get(hook);
|
|
6
|
+
if (!set) {
|
|
7
|
+
set = new Set();
|
|
8
|
+
hooks.set(hook, set);
|
|
9
|
+
}
|
|
10
|
+
set.add(handler);
|
|
11
|
+
},
|
|
12
|
+
off(hook, handler) {
|
|
13
|
+
hooks.get(hook)?.delete(handler);
|
|
14
|
+
},
|
|
15
|
+
async emit(hook, event) {
|
|
16
|
+
const set = hooks.get(hook);
|
|
17
|
+
if (!set)
|
|
18
|
+
return;
|
|
19
|
+
for (const handler of set) {
|
|
20
|
+
try {
|
|
21
|
+
await handler(event);
|
|
22
|
+
}
|
|
23
|
+
catch (err) {
|
|
24
|
+
console.error(`[hooks] Error in "${hook}" handler:`, err);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
},
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=hooks.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.js","sourceRoot":"","sources":["../src/hooks.ts"],"names":[],"mappings":"AAIA,MAAM,UAAU,kBAAkB;IAChC,MAAM,KAAK,GAAG,IAAI,GAAG,EAA8B,CAAC;IAEpD,OAAO;QACL,EAAE,CAAC,IAAc,EAAE,OAAoB;YACrC,IAAI,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC1B,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;gBAChB,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACvB,CAAC;YACD,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACnB,CAAC;QAED,GAAG,CAAC,IAAc,EAAE,OAAoB;YACtC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,IAAc,EAAE,KAAc;YACvC,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,CAAC,GAAG;gBAAE,OAAO;YACjB,KAAK,MAAM,OAAO,IAAI,GAAG,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,qBAAqB,IAAI,YAAY,EAAE,GAAG,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { ChannelMessage, DebounceConfig } from '@cogitator-ai/types';
|
|
2
|
+
export declare class InboundDebouncer {
|
|
3
|
+
private readonly config;
|
|
4
|
+
private readonly onFlush;
|
|
5
|
+
private buffers;
|
|
6
|
+
private readonly defaultDelay;
|
|
7
|
+
constructor(config: DebounceConfig, onFlush: (merged: ChannelMessage) => Promise<void>);
|
|
8
|
+
enqueue(msg: ChannelMessage): void;
|
|
9
|
+
flushAll(): Promise<void>;
|
|
10
|
+
dispose(): void;
|
|
11
|
+
private flush;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=inbound-debounce.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"inbound-debounce.d.ts","sourceRoot":"","sources":["../src/inbound-debounce.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAO1E,qBAAa,gBAAgB;IAKzB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAL1B,OAAO,CAAC,OAAO,CAAkC;IACjD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;gBAGnB,MAAM,EAAE,cAAc,EACtB,OAAO,EAAE,CAAC,MAAM,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC;IAKrE,OAAO,CAAC,GAAG,EAAE,cAAc,GAAG,IAAI;IAe5B,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAK/B,OAAO,IAAI,IAAI;YAOD,KAAK;CA2BpB"}
|