@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
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import type { TimerEntry, TimerStore } from '@cogitator-ai/types';
|
|
2
|
+
export interface TimerStoreOptions {
|
|
3
|
+
persistPath?: string;
|
|
4
|
+
}
|
|
5
|
+
export declare class SimpleTimerStore implements TimerStore {
|
|
6
|
+
private timers;
|
|
7
|
+
private callbacks;
|
|
8
|
+
private readonly persistPath;
|
|
9
|
+
constructor(opts?: TimerStoreOptions);
|
|
10
|
+
schedule(entry: Omit<TimerEntry, 'id' | 'cancelled' | 'fired' | 'createdAt'>): Promise<string>;
|
|
11
|
+
cancel(id: string): Promise<void>;
|
|
12
|
+
get(id: string): Promise<TimerEntry | null>;
|
|
13
|
+
getByWorkflow(workflowId: string): Promise<TimerEntry[]>;
|
|
14
|
+
getByRun(runId: string): Promise<TimerEntry[]>;
|
|
15
|
+
getPending(): Promise<TimerEntry[]>;
|
|
16
|
+
getOverdue(): Promise<TimerEntry[]>;
|
|
17
|
+
markFired(id: string): Promise<void>;
|
|
18
|
+
cleanup(olderThan: number): Promise<number>;
|
|
19
|
+
update(id: string, patch: Partial<TimerEntry>): Promise<void>;
|
|
20
|
+
list(filter?: {
|
|
21
|
+
enabled?: boolean;
|
|
22
|
+
type?: string;
|
|
23
|
+
}): Promise<TimerEntry[]>;
|
|
24
|
+
onFire(callback: (entry: TimerEntry) => void): () => void;
|
|
25
|
+
private saveToDisk;
|
|
26
|
+
private loadFromDisk;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=simple-timer-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"simple-timer-store.d.ts","sourceRoot":"","sources":["../src/simple-timer-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAOlE,MAAM,WAAW,iBAAiB;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,gBAAiB,YAAW,UAAU;IACjD,OAAO,CAAC,MAAM,CAAiC;IAC/C,OAAO,CAAC,SAAS,CAAuC;IACxD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAgB;gBAEhC,IAAI,CAAC,EAAE,iBAAiB;IAO9B,QAAQ,CACZ,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,GAAG,WAAW,GAAG,OAAO,GAAG,WAAW,CAAC,GAClE,OAAO,CAAC,MAAM,CAAC;IAcZ,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQjC,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAI3C,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;IAIxD,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;IAI9C,UAAU,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;IAInC,UAAU,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;IAKnC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASpC,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAY3C,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAO7D,IAAI,CAAC,MAAM,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,OAAO,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;IAWhF,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,GAAG,MAAM,IAAI;IAQzD,OAAO,CAAC,UAAU;IAYlB,OAAO,CAAC,YAAY;CAYrB"}
|
|
@@ -0,0 +1,129 @@
|
|
|
1
|
+
import { readFileSync, writeFileSync, existsSync, mkdirSync, renameSync } from 'node:fs';
|
|
2
|
+
import { dirname, join } from 'node:path';
|
|
3
|
+
let counter = 0;
|
|
4
|
+
function nextId() {
|
|
5
|
+
return `timer_${++counter}_${Date.now()}`;
|
|
6
|
+
}
|
|
7
|
+
export class SimpleTimerStore {
|
|
8
|
+
timers = new Map();
|
|
9
|
+
callbacks = [];
|
|
10
|
+
persistPath;
|
|
11
|
+
constructor(opts) {
|
|
12
|
+
this.persistPath = opts?.persistPath ?? null;
|
|
13
|
+
if (this.persistPath) {
|
|
14
|
+
this.loadFromDisk();
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
async schedule(entry) {
|
|
18
|
+
const id = nextId();
|
|
19
|
+
const full = {
|
|
20
|
+
...entry,
|
|
21
|
+
id,
|
|
22
|
+
cancelled: false,
|
|
23
|
+
fired: false,
|
|
24
|
+
createdAt: Date.now(),
|
|
25
|
+
};
|
|
26
|
+
this.timers.set(id, full);
|
|
27
|
+
this.saveToDisk();
|
|
28
|
+
return id;
|
|
29
|
+
}
|
|
30
|
+
async cancel(id) {
|
|
31
|
+
const entry = this.timers.get(id);
|
|
32
|
+
if (entry) {
|
|
33
|
+
entry.cancelled = true;
|
|
34
|
+
this.saveToDisk();
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
async get(id) {
|
|
38
|
+
return this.timers.get(id) ?? null;
|
|
39
|
+
}
|
|
40
|
+
async getByWorkflow(workflowId) {
|
|
41
|
+
return [...this.timers.values()].filter((t) => t.workflowId === workflowId);
|
|
42
|
+
}
|
|
43
|
+
async getByRun(runId) {
|
|
44
|
+
return [...this.timers.values()].filter((t) => t.runId === runId);
|
|
45
|
+
}
|
|
46
|
+
async getPending() {
|
|
47
|
+
return [...this.timers.values()].filter((t) => !t.cancelled && !t.fired);
|
|
48
|
+
}
|
|
49
|
+
async getOverdue() {
|
|
50
|
+
const now = Date.now();
|
|
51
|
+
return [...this.timers.values()].filter((t) => !t.cancelled && !t.fired && t.firesAt <= now);
|
|
52
|
+
}
|
|
53
|
+
async markFired(id) {
|
|
54
|
+
const entry = this.timers.get(id);
|
|
55
|
+
if (entry) {
|
|
56
|
+
entry.fired = true;
|
|
57
|
+
this.saveToDisk();
|
|
58
|
+
for (const cb of this.callbacks)
|
|
59
|
+
cb(entry);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
async cleanup(olderThan) {
|
|
63
|
+
let count = 0;
|
|
64
|
+
for (const [id, entry] of this.timers) {
|
|
65
|
+
if ((entry.fired || entry.cancelled) && entry.createdAt < olderThan) {
|
|
66
|
+
this.timers.delete(id);
|
|
67
|
+
count++;
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
if (count > 0)
|
|
71
|
+
this.saveToDisk();
|
|
72
|
+
return count;
|
|
73
|
+
}
|
|
74
|
+
async update(id, patch) {
|
|
75
|
+
const entry = this.timers.get(id);
|
|
76
|
+
if (!entry)
|
|
77
|
+
return;
|
|
78
|
+
Object.assign(entry, patch);
|
|
79
|
+
this.saveToDisk();
|
|
80
|
+
}
|
|
81
|
+
async list(filter) {
|
|
82
|
+
let entries = [...this.timers.values()];
|
|
83
|
+
if (filter?.enabled !== undefined) {
|
|
84
|
+
entries = entries.filter((e) => (e.enabled ?? true) === filter.enabled);
|
|
85
|
+
}
|
|
86
|
+
if (filter?.type) {
|
|
87
|
+
entries = entries.filter((e) => e.type === filter.type);
|
|
88
|
+
}
|
|
89
|
+
return entries;
|
|
90
|
+
}
|
|
91
|
+
onFire(callback) {
|
|
92
|
+
this.callbacks.push(callback);
|
|
93
|
+
return () => {
|
|
94
|
+
const idx = this.callbacks.indexOf(callback);
|
|
95
|
+
if (idx >= 0)
|
|
96
|
+
this.callbacks.splice(idx, 1);
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
saveToDisk() {
|
|
100
|
+
if (!this.persistPath)
|
|
101
|
+
return;
|
|
102
|
+
try {
|
|
103
|
+
const dir = dirname(this.persistPath);
|
|
104
|
+
if (!existsSync(dir))
|
|
105
|
+
mkdirSync(dir, { recursive: true });
|
|
106
|
+
const data = [...this.timers.values()];
|
|
107
|
+
const tmpPath = join(dir, `.timer-store-${Date.now()}.tmp`);
|
|
108
|
+
writeFileSync(tmpPath, JSON.stringify(data));
|
|
109
|
+
renameSync(tmpPath, this.persistPath);
|
|
110
|
+
}
|
|
111
|
+
catch { }
|
|
112
|
+
}
|
|
113
|
+
loadFromDisk() {
|
|
114
|
+
if (!this.persistPath || !existsSync(this.persistPath))
|
|
115
|
+
return;
|
|
116
|
+
try {
|
|
117
|
+
const raw = readFileSync(this.persistPath, 'utf-8');
|
|
118
|
+
const entries = JSON.parse(raw);
|
|
119
|
+
for (const entry of entries) {
|
|
120
|
+
this.timers.set(entry.id, entry);
|
|
121
|
+
const numPart = /^timer_(\d+)/.exec(entry.id);
|
|
122
|
+
if (numPart)
|
|
123
|
+
counter = Math.max(counter, Number(numPart[1]));
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
catch { }
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
//# sourceMappingURL=simple-timer-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"simple-timer-store.js","sourceRoot":"","sources":["../src/simple-timer-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACzF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAG1C,IAAI,OAAO,GAAG,CAAC,CAAC;AAChB,SAAS,MAAM;IACb,OAAO,SAAS,EAAE,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AAC5C,CAAC;AAMD,MAAM,OAAO,gBAAgB;IACnB,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;IACvC,SAAS,GAAoC,EAAE,CAAC;IACvC,WAAW,CAAgB;IAE5C,YAAY,IAAwB;QAClC,IAAI,CAAC,WAAW,GAAG,IAAI,EAAE,WAAW,IAAI,IAAI,CAAC;QAC7C,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,KAAmE;QAEnE,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QACpB,MAAM,IAAI,GAAe;YACvB,GAAG,KAAK;YACR,EAAE;YACF,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC1B,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,UAAkB;QACpC,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,UAAU,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAa;QAC1B,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,IAAI,GAAG,CAAC,CAAC;IAC/F,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;YACnB,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,SAAS;gBAAE,EAAE,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB;QAC7B,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACtC,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,SAAS,GAAG,SAAS,EAAE,CAAC;gBACpE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACvB,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,IAAI,KAAK,GAAG,CAAC;YAAE,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,KAA0B;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC5B,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAA6C;QACtD,IAAI,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACxC,IAAI,MAAM,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;YAClC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,MAAM,EAAE,IAAI,EAAE,CAAC;YACjB,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1D,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,CAAC,QAAqC;QAC1C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9B,OAAO,GAAG,EAAE;YACV,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC7C,IAAI,GAAG,IAAI,CAAC;gBAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC;IACJ,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO;QAC9B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACtC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YACvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,gBAAgB,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC5D,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;YAC7C,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACxC,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC;YAAE,OAAO;QAC/D,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAiB,CAAC;YAChD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;gBACjC,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC9C,IAAI,OAAO;oBAAE,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { Channel, StatusPhase, StatusReactionConfig } from '@cogitator-ai/types';
|
|
2
|
+
export declare class StatusReactionTracker {
|
|
3
|
+
private readonly channel;
|
|
4
|
+
private readonly channelId;
|
|
5
|
+
private readonly messageId;
|
|
6
|
+
private currentPhase;
|
|
7
|
+
private debounceTimer;
|
|
8
|
+
private stallSoftTimer;
|
|
9
|
+
private stallHardTimer;
|
|
10
|
+
private readonly debounceMs;
|
|
11
|
+
private readonly stallSoftMs;
|
|
12
|
+
private readonly stallHardMs;
|
|
13
|
+
private readonly emojis;
|
|
14
|
+
private disposed;
|
|
15
|
+
constructor(channel: Channel, channelId: string, messageId: string, config?: StatusReactionConfig);
|
|
16
|
+
setPhase(phase: StatusPhase): void;
|
|
17
|
+
dispose(): void;
|
|
18
|
+
private startStallTimers;
|
|
19
|
+
private clearStallTimers;
|
|
20
|
+
private react;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=status-reactions.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"status-reactions.d.ts","sourceRoot":"","sources":["../src/status-reactions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAatF,qBAAa,qBAAqB;IAY9B,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,SAAS;IAb5B,OAAO,CAAC,YAAY,CAA4B;IAChD,OAAO,CAAC,aAAa,CAA8C;IACnE,OAAO,CAAC,cAAc,CAA8C;IACpE,OAAO,CAAC,cAAc,CAA8C;IACpE,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,QAAQ,CAAS;gBAGN,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EAClC,MAAM,CAAC,EAAE,oBAAoB;IAQ/B,QAAQ,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI;IAqBlC,OAAO,IAAI,IAAI;IAMf,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,gBAAgB;YAWV,KAAK;CAKpB"}
|
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
const DEFAULT_EMOJIS = {
|
|
2
|
+
queued: '\u{1F440}',
|
|
3
|
+
thinking: '\u{1F914}',
|
|
4
|
+
tool: '\u{1F525}',
|
|
5
|
+
done: '\u{1F44D}',
|
|
6
|
+
error: '\u{1F631}',
|
|
7
|
+
};
|
|
8
|
+
const STALL_SOFT_EMOJI = '\u{1F971}';
|
|
9
|
+
const STALL_HARD_EMOJI = '\u{1F628}';
|
|
10
|
+
export class StatusReactionTracker {
|
|
11
|
+
channel;
|
|
12
|
+
channelId;
|
|
13
|
+
messageId;
|
|
14
|
+
currentPhase = null;
|
|
15
|
+
debounceTimer = null;
|
|
16
|
+
stallSoftTimer = null;
|
|
17
|
+
stallHardTimer = null;
|
|
18
|
+
debounceMs;
|
|
19
|
+
stallSoftMs;
|
|
20
|
+
stallHardMs;
|
|
21
|
+
emojis;
|
|
22
|
+
disposed = false;
|
|
23
|
+
constructor(channel, channelId, messageId, config) {
|
|
24
|
+
this.channel = channel;
|
|
25
|
+
this.channelId = channelId;
|
|
26
|
+
this.messageId = messageId;
|
|
27
|
+
this.debounceMs = config?.debounceMs ?? 700;
|
|
28
|
+
this.stallSoftMs = config?.stallSoftMs ?? 10_000;
|
|
29
|
+
this.stallHardMs = config?.stallHardMs ?? 30_000;
|
|
30
|
+
this.emojis = { ...DEFAULT_EMOJIS, ...config?.emojis };
|
|
31
|
+
}
|
|
32
|
+
setPhase(phase) {
|
|
33
|
+
if (this.disposed)
|
|
34
|
+
return;
|
|
35
|
+
if (phase === this.currentPhase)
|
|
36
|
+
return;
|
|
37
|
+
if (this.debounceTimer)
|
|
38
|
+
clearTimeout(this.debounceTimer);
|
|
39
|
+
this.clearStallTimers();
|
|
40
|
+
if (phase === 'done' || phase === 'error') {
|
|
41
|
+
this.currentPhase = phase;
|
|
42
|
+
void this.react(this.emojis[phase]);
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
this.debounceTimer = setTimeout(() => {
|
|
46
|
+
if (this.disposed)
|
|
47
|
+
return;
|
|
48
|
+
this.currentPhase = phase;
|
|
49
|
+
void this.react(this.emojis[phase]);
|
|
50
|
+
this.startStallTimers();
|
|
51
|
+
}, this.debounceMs);
|
|
52
|
+
}
|
|
53
|
+
dispose() {
|
|
54
|
+
this.disposed = true;
|
|
55
|
+
if (this.debounceTimer)
|
|
56
|
+
clearTimeout(this.debounceTimer);
|
|
57
|
+
this.clearStallTimers();
|
|
58
|
+
}
|
|
59
|
+
startStallTimers() {
|
|
60
|
+
this.stallSoftTimer = setTimeout(() => {
|
|
61
|
+
if (!this.disposed)
|
|
62
|
+
void this.react(STALL_SOFT_EMOJI);
|
|
63
|
+
}, this.stallSoftMs);
|
|
64
|
+
this.stallHardTimer = setTimeout(() => {
|
|
65
|
+
if (!this.disposed)
|
|
66
|
+
void this.react(STALL_HARD_EMOJI);
|
|
67
|
+
}, this.stallHardMs);
|
|
68
|
+
}
|
|
69
|
+
clearStallTimers() {
|
|
70
|
+
if (this.stallSoftTimer) {
|
|
71
|
+
clearTimeout(this.stallSoftTimer);
|
|
72
|
+
this.stallSoftTimer = null;
|
|
73
|
+
}
|
|
74
|
+
if (this.stallHardTimer) {
|
|
75
|
+
clearTimeout(this.stallHardTimer);
|
|
76
|
+
this.stallHardTimer = null;
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
async react(emoji) {
|
|
80
|
+
try {
|
|
81
|
+
await this.channel.setReaction?.(this.channelId, this.messageId, emoji);
|
|
82
|
+
}
|
|
83
|
+
catch { }
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
//# sourceMappingURL=status-reactions.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"status-reactions.js","sourceRoot":"","sources":["../src/status-reactions.ts"],"names":[],"mappings":"AAEA,MAAM,cAAc,GAAgC;IAClD,MAAM,EAAE,WAAW;IACnB,QAAQ,EAAE,WAAW;IACrB,IAAI,EAAE,WAAW;IACjB,IAAI,EAAE,WAAW;IACjB,KAAK,EAAE,WAAW;CACnB,CAAC;AAEF,MAAM,gBAAgB,GAAG,WAAW,CAAC;AACrC,MAAM,gBAAgB,GAAG,WAAW,CAAC;AAErC,MAAM,OAAO,qBAAqB;IAYb;IACA;IACA;IAbX,YAAY,GAAuB,IAAI,CAAC;IACxC,aAAa,GAAyC,IAAI,CAAC;IAC3D,cAAc,GAAyC,IAAI,CAAC;IAC5D,cAAc,GAAyC,IAAI,CAAC;IACnD,UAAU,CAAS;IACnB,WAAW,CAAS;IACpB,WAAW,CAAS;IACpB,MAAM,CAA8B;IAC7C,QAAQ,GAAG,KAAK,CAAC;IAEzB,YACmB,OAAgB,EAChB,SAAiB,EACjB,SAAiB,EAClC,MAA6B;QAHZ,YAAO,GAAP,OAAO,CAAS;QAChB,cAAS,GAAT,SAAS,CAAQ;QACjB,cAAS,GAAT,SAAS,CAAQ;QAGlC,IAAI,CAAC,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,GAAG,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,MAAM,CAAC;QACjD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,MAAM,CAAC;QACjD,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,CAAC;IACzD,CAAC;IAED,QAAQ,CAAC,KAAkB;QACzB,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,IAAI,KAAK,KAAK,IAAI,CAAC,YAAY;YAAE,OAAO;QAExC,IAAI,IAAI,CAAC,aAAa;YAAE,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACzD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAExB,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAC1B,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACpC,OAAO;QACT,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;YACnC,IAAI,IAAI,CAAC,QAAQ;gBAAE,OAAO;YAC1B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAC1B,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACpC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IACtB,CAAC;IAED,OAAO;QACL,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,IAAI,CAAC,aAAa;YAAE,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACzD,IAAI,CAAC,gBAAgB,EAAE,CAAC;IAC1B,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAAE,KAAK,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACxD,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAErB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAAE,KAAK,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACxD,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACvB,CAAC;IAEO,gBAAgB;QACtB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,KAAa;QAC/B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC1E,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;CACF"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import type { Channel, StreamConfig } from '@cogitator-ai/types';
|
|
2
|
+
export declare class StreamBuffer {
|
|
3
|
+
private readonly channel;
|
|
4
|
+
private readonly channelId;
|
|
5
|
+
private readonly config;
|
|
6
|
+
private readonly replyTo?;
|
|
7
|
+
private buffer;
|
|
8
|
+
private messageId;
|
|
9
|
+
private readonly messageIds;
|
|
10
|
+
private timer;
|
|
11
|
+
private flushPromise;
|
|
12
|
+
private readonly draftId;
|
|
13
|
+
private draftFailed;
|
|
14
|
+
private initialSent;
|
|
15
|
+
private lastSentText;
|
|
16
|
+
private generation;
|
|
17
|
+
private lastFlushEnd;
|
|
18
|
+
private stopped;
|
|
19
|
+
constructor(channel: Channel, channelId: string, config?: StreamConfig, replyTo?: string | undefined, useDraft?: boolean);
|
|
20
|
+
start(): void;
|
|
21
|
+
append(token: string): void;
|
|
22
|
+
forceNewMessage(): void;
|
|
23
|
+
getMessageIds(): readonly string[];
|
|
24
|
+
finish(): Promise<string>;
|
|
25
|
+
abort(): Promise<void>;
|
|
26
|
+
private trackMessageId;
|
|
27
|
+
private scheduleNext;
|
|
28
|
+
private flush;
|
|
29
|
+
private doFlush;
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=stream-buffer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-buffer.d.ts","sourceRoot":"","sources":["../src/stream-buffer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAKjE,qBAAa,YAAY;IAerB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAIvB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;IApB3B,OAAO,CAAC,MAAM,CAAM;IACpB,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAgB;IAC3C,OAAO,CAAC,KAAK,CAA8C;IAC3D,OAAO,CAAC,YAAY,CAA8B;IAClD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAuB;IAC/C,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,YAAY,CAAM;IAC1B,OAAO,CAAC,UAAU,CAAK;IACvB,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,OAAO,CAAS;gBAGL,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,EACjB,MAAM,GAAE,YAGxB,EACgB,OAAO,CAAC,EAAE,MAAM,YAAA,EACjC,QAAQ,UAAQ;IAOlB,KAAK,IAAI,IAAI;IAKb,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAY3B,eAAe,IAAI,IAAI;IAQvB,aAAa,IAAI,SAAS,MAAM,EAAE;IAI5B,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC;IA0BzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB5B,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,YAAY;YAUN,KAAK;YA2BL,OAAO;CA0BtB"}
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
const DEFAULT_FLUSH_INTERVAL = 500;
|
|
2
|
+
const DEFAULT_MIN_CHUNK_SIZE = 20;
|
|
3
|
+
export class StreamBuffer {
|
|
4
|
+
channel;
|
|
5
|
+
channelId;
|
|
6
|
+
config;
|
|
7
|
+
replyTo;
|
|
8
|
+
buffer = '';
|
|
9
|
+
messageId = null;
|
|
10
|
+
messageIds = [];
|
|
11
|
+
timer = null;
|
|
12
|
+
flushPromise = null;
|
|
13
|
+
draftId = null;
|
|
14
|
+
draftFailed = false;
|
|
15
|
+
initialSent = false;
|
|
16
|
+
lastSentText = '';
|
|
17
|
+
generation = 0;
|
|
18
|
+
lastFlushEnd = 0;
|
|
19
|
+
stopped = false;
|
|
20
|
+
constructor(channel, channelId, config = {
|
|
21
|
+
flushInterval: DEFAULT_FLUSH_INTERVAL,
|
|
22
|
+
minChunkSize: DEFAULT_MIN_CHUNK_SIZE,
|
|
23
|
+
}, replyTo, useDraft = false) {
|
|
24
|
+
this.channel = channel;
|
|
25
|
+
this.channelId = channelId;
|
|
26
|
+
this.config = config;
|
|
27
|
+
this.replyTo = replyTo;
|
|
28
|
+
if (useDraft && channel.sendDraft) {
|
|
29
|
+
this.draftId = Math.floor(Math.random() * 2_147_483_646) + 1;
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
start() {
|
|
33
|
+
this.stopped = false;
|
|
34
|
+
this.scheduleNext();
|
|
35
|
+
}
|
|
36
|
+
append(token) {
|
|
37
|
+
this.buffer += token;
|
|
38
|
+
const limit = this.config.maxMessageChars;
|
|
39
|
+
if (limit && this.buffer.length > limit) {
|
|
40
|
+
const overflow = this.buffer.slice(limit);
|
|
41
|
+
this.buffer = this.buffer.slice(0, limit);
|
|
42
|
+
this.forceNewMessage();
|
|
43
|
+
this.buffer = overflow;
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
forceNewMessage() {
|
|
47
|
+
this.generation++;
|
|
48
|
+
this.buffer = '';
|
|
49
|
+
this.messageId = null;
|
|
50
|
+
this.initialSent = false;
|
|
51
|
+
this.lastSentText = '';
|
|
52
|
+
}
|
|
53
|
+
getMessageIds() {
|
|
54
|
+
return this.messageIds;
|
|
55
|
+
}
|
|
56
|
+
async finish() {
|
|
57
|
+
this.stopped = true;
|
|
58
|
+
if (this.timer) {
|
|
59
|
+
clearTimeout(this.timer);
|
|
60
|
+
this.timer = null;
|
|
61
|
+
}
|
|
62
|
+
if (this.flushPromise) {
|
|
63
|
+
await this.flushPromise;
|
|
64
|
+
}
|
|
65
|
+
if (this.draftId && !this.draftFailed) {
|
|
66
|
+
const text = this.buffer;
|
|
67
|
+
if (!text)
|
|
68
|
+
return '';
|
|
69
|
+
const msgId = await this.channel.sendText(this.channelId, text, {
|
|
70
|
+
replyTo: this.replyTo,
|
|
71
|
+
format: 'markdown',
|
|
72
|
+
});
|
|
73
|
+
this.trackMessageId(msgId);
|
|
74
|
+
return msgId;
|
|
75
|
+
}
|
|
76
|
+
await this.flush(true);
|
|
77
|
+
return this.messageId ?? '';
|
|
78
|
+
}
|
|
79
|
+
async abort() {
|
|
80
|
+
this.stopped = true;
|
|
81
|
+
if (this.timer) {
|
|
82
|
+
clearTimeout(this.timer);
|
|
83
|
+
this.timer = null;
|
|
84
|
+
}
|
|
85
|
+
if (this.flushPromise) {
|
|
86
|
+
await this.flushPromise;
|
|
87
|
+
}
|
|
88
|
+
if (this.config.deleteOnAbort && this.channel.deleteMessage) {
|
|
89
|
+
for (const id of this.messageIds) {
|
|
90
|
+
try {
|
|
91
|
+
await this.channel.deleteMessage(this.channelId, id);
|
|
92
|
+
}
|
|
93
|
+
catch { }
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
trackMessageId(id) {
|
|
98
|
+
if (id && !this.messageIds.includes(id)) {
|
|
99
|
+
this.messageIds.push(id);
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
scheduleNext() {
|
|
103
|
+
if (this.stopped)
|
|
104
|
+
return;
|
|
105
|
+
const elapsed = this.lastFlushEnd ? Date.now() - this.lastFlushEnd : 0;
|
|
106
|
+
const delay = Math.max(0, this.config.flushInterval - elapsed);
|
|
107
|
+
this.timer = setTimeout(() => {
|
|
108
|
+
const p = this.flush();
|
|
109
|
+
void p.then(() => this.scheduleNext());
|
|
110
|
+
}, delay);
|
|
111
|
+
}
|
|
112
|
+
async flush(force = false) {
|
|
113
|
+
if (this.flushPromise)
|
|
114
|
+
return;
|
|
115
|
+
if (this.buffer.length === 0)
|
|
116
|
+
return;
|
|
117
|
+
if (!force && !this.initialSent && this.config.minInitialChars) {
|
|
118
|
+
if (this.buffer.length < this.config.minInitialChars)
|
|
119
|
+
return;
|
|
120
|
+
}
|
|
121
|
+
if (!force && this.buffer.length < this.config.minChunkSize)
|
|
122
|
+
return;
|
|
123
|
+
const text = this.buffer;
|
|
124
|
+
if (!force && this.messageId && text === this.lastSentText)
|
|
125
|
+
return;
|
|
126
|
+
const gen = this.generation;
|
|
127
|
+
const promise = this.doFlush(text, gen);
|
|
128
|
+
this.flushPromise = promise;
|
|
129
|
+
try {
|
|
130
|
+
await promise;
|
|
131
|
+
}
|
|
132
|
+
finally {
|
|
133
|
+
this.flushPromise = null;
|
|
134
|
+
this.lastFlushEnd = Date.now();
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
async doFlush(text, gen) {
|
|
138
|
+
try {
|
|
139
|
+
if (this.draftId && !this.draftFailed) {
|
|
140
|
+
await this.channel.sendDraft(this.channelId, this.draftId, text);
|
|
141
|
+
}
|
|
142
|
+
else if (!this.messageId) {
|
|
143
|
+
const msgId = await this.channel.sendText(this.channelId, text, {
|
|
144
|
+
replyTo: this.replyTo,
|
|
145
|
+
format: 'markdown',
|
|
146
|
+
});
|
|
147
|
+
if (gen !== this.generation)
|
|
148
|
+
return;
|
|
149
|
+
this.messageId = msgId;
|
|
150
|
+
this.trackMessageId(msgId);
|
|
151
|
+
this.initialSent = true;
|
|
152
|
+
this.lastSentText = text;
|
|
153
|
+
}
|
|
154
|
+
else {
|
|
155
|
+
if (gen !== this.generation)
|
|
156
|
+
return;
|
|
157
|
+
await this.channel.editText(this.channelId, this.messageId, text);
|
|
158
|
+
if (gen !== this.generation)
|
|
159
|
+
return;
|
|
160
|
+
this.lastSentText = text;
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
catch {
|
|
164
|
+
if (this.draftId && !this.draftFailed) {
|
|
165
|
+
this.draftFailed = true;
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
//# sourceMappingURL=stream-buffer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-buffer.js","sourceRoot":"","sources":["../src/stream-buffer.ts"],"names":[],"mappings":"AAEA,MAAM,sBAAsB,GAAG,GAAG,CAAC;AACnC,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAElC,MAAM,OAAO,YAAY;IAeJ;IACA;IACA;IAIA;IApBX,MAAM,GAAG,EAAE,CAAC;IACZ,SAAS,GAAkB,IAAI,CAAC;IACvB,UAAU,GAAa,EAAE,CAAC;IACnC,KAAK,GAAyC,IAAI,CAAC;IACnD,YAAY,GAAyB,IAAI,CAAC;IACjC,OAAO,GAAkB,IAAI,CAAC;IACvC,WAAW,GAAG,KAAK,CAAC;IACpB,WAAW,GAAG,KAAK,CAAC;IACpB,YAAY,GAAG,EAAE,CAAC;IAClB,UAAU,GAAG,CAAC,CAAC;IACf,YAAY,GAAG,CAAC,CAAC;IACjB,OAAO,GAAG,KAAK,CAAC;IAExB,YACmB,OAAgB,EAChB,SAAiB,EACjB,SAAuB;QACtC,aAAa,EAAE,sBAAsB;QACrC,YAAY,EAAE,sBAAsB;KACrC,EACgB,OAAgB,EACjC,QAAQ,GAAG,KAAK;QAPC,YAAO,GAAP,OAAO,CAAS;QAChB,cAAS,GAAT,SAAS,CAAQ;QACjB,WAAM,GAAN,MAAM,CAGtB;QACgB,YAAO,GAAP,OAAO,CAAS;QAGjC,IAAI,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED,MAAM,CAAC,KAAa;QAClB,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC;QAErB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;QAC1C,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC1C,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;QACzB,CAAC;IACH,CAAC;IAED,eAAe;QACb,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;IACzB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,MAAM;QACV,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,YAAY,CAAC;QAC1B,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;YACzB,IAAI,CAAC,IAAI;gBAAE,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE;gBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,MAAM,EAAE,UAAU;aACnB,CAAC,CAAC;YACH,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC3B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,YAAY,CAAC;QAC1B,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC5D,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACjC,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;gBACvD,CAAC;gBAAC,MAAM,CAAC,CAAA,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,EAAU;QAC/B,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAEO,YAAY;QAClB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,OAAO,CAAC,CAAC;QAC/D,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC3B,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YACvB,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACzC,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK;QAC/B,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAErC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YAC/D,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe;gBAAE,OAAO;QAC/D,CAAC;QAED,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY;YAAE,OAAO;QAEpE,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;QAEzB,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,KAAK,IAAI,CAAC,YAAY;YAAE,OAAO;QAEnE,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC;QAE5B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACxC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;QAE5B,IAAI,CAAC;YACH,MAAM,OAAO,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YACzB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,GAAW;QAC7C,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtC,MAAM,IAAI,CAAC,OAAO,CAAC,SAAU,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACpE,CAAC;iBAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC3B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE;oBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;gBACH,IAAI,GAAG,KAAK,IAAI,CAAC,UAAU;oBAAE,OAAO;gBACpC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;gBAC3B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBACxB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,IAAI,GAAG,KAAK,IAAI,CAAC,UAAU;oBAAE,OAAO;gBACpC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAClE,IAAI,GAAG,KAAK,IAAI,CAAC,UAAU;oBAAE,OAAO;gBACpC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC3B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { Tool } from '@cogitator-ai/types';
|
|
2
|
+
export declare function createSelfConfigTools(opts: {
|
|
3
|
+
configPath: string;
|
|
4
|
+
parseYaml: (s: string) => unknown;
|
|
5
|
+
stringifyYaml: (o: unknown) => string;
|
|
6
|
+
validateConfig: (o: unknown) => unknown;
|
|
7
|
+
onConfigUpdated?: () => void;
|
|
8
|
+
}): Tool[];
|
|
9
|
+
//# sourceMappingURL=self-config.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"self-config.d.ts","sourceRoot":"","sources":["../../src/tools/self-config.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AA0BhD,wBAAgB,qBAAqB,CAAC,IAAI,EAAE;IAC1C,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC;IAClC,aAAa,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,MAAM,CAAC;IACtC,cAAc,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC;IACxC,eAAe,CAAC,EAAE,MAAM,IAAI,CAAC;CAC9B,GAAG,IAAI,EAAE,CA8GT"}
|
|
@@ -0,0 +1,122 @@
|
|
|
1
|
+
import { readFileSync, writeFileSync, existsSync } from 'node:fs';
|
|
2
|
+
import { dirname, join } from 'node:path';
|
|
3
|
+
import { z } from 'zod';
|
|
4
|
+
import { tool } from '@cogitator-ai/core';
|
|
5
|
+
function deepMerge(target, source) {
|
|
6
|
+
const result = { ...target };
|
|
7
|
+
for (const key of Object.keys(source)) {
|
|
8
|
+
const srcVal = source[key];
|
|
9
|
+
const tgtVal = target[key];
|
|
10
|
+
if (srcVal !== null &&
|
|
11
|
+
typeof srcVal === 'object' &&
|
|
12
|
+
!Array.isArray(srcVal) &&
|
|
13
|
+
tgtVal !== null &&
|
|
14
|
+
typeof tgtVal === 'object' &&
|
|
15
|
+
!Array.isArray(tgtVal)) {
|
|
16
|
+
result[key] = deepMerge(tgtVal, srcVal);
|
|
17
|
+
}
|
|
18
|
+
else {
|
|
19
|
+
result[key] = srcVal;
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
return result;
|
|
23
|
+
}
|
|
24
|
+
export function createSelfConfigTools(opts) {
|
|
25
|
+
const { configPath, parseYaml, stringifyYaml, validateConfig, onConfigUpdated } = opts;
|
|
26
|
+
const configRead = tool({
|
|
27
|
+
name: 'config_read',
|
|
28
|
+
description: 'Read the current assistant configuration (cogitator.yml)',
|
|
29
|
+
parameters: z.object({}),
|
|
30
|
+
execute: async () => {
|
|
31
|
+
const raw = readFileSync(configPath, 'utf-8');
|
|
32
|
+
return { config: parseYaml(raw) };
|
|
33
|
+
},
|
|
34
|
+
});
|
|
35
|
+
const configUpdate = tool({
|
|
36
|
+
name: 'config_update',
|
|
37
|
+
description: 'Update assistant configuration. Provide a partial config object that will be deep-merged with the current config. Changes are validated before saving. The assistant will restart automatically after a successful update.',
|
|
38
|
+
parameters: z.object({
|
|
39
|
+
updates: z
|
|
40
|
+
.record(z.string(), z.unknown())
|
|
41
|
+
.describe('Partial config to deep-merge with current config'),
|
|
42
|
+
}),
|
|
43
|
+
execute: async ({ updates }) => {
|
|
44
|
+
const raw = readFileSync(configPath, 'utf-8');
|
|
45
|
+
const current = parseYaml(raw);
|
|
46
|
+
const merged = deepMerge(current, updates);
|
|
47
|
+
try {
|
|
48
|
+
validateConfig(merged);
|
|
49
|
+
}
|
|
50
|
+
catch (err) {
|
|
51
|
+
return {
|
|
52
|
+
success: false,
|
|
53
|
+
error: `Validation failed: ${err instanceof Error ? err.message : String(err)}`,
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
writeFileSync(configPath, stringifyYaml(merged));
|
|
57
|
+
onConfigUpdated?.();
|
|
58
|
+
return { success: true, message: 'Config updated. Restarting...' };
|
|
59
|
+
},
|
|
60
|
+
});
|
|
61
|
+
const envPath = join(dirname(configPath), '.env');
|
|
62
|
+
const KNOWN_VARS = [
|
|
63
|
+
'GOOGLE_API_KEY',
|
|
64
|
+
'GEMINI_API_KEY',
|
|
65
|
+
'OPENAI_API_KEY',
|
|
66
|
+
'ANTHROPIC_API_KEY',
|
|
67
|
+
'OLLAMA_URL',
|
|
68
|
+
'OLLAMA_API_KEY',
|
|
69
|
+
'GITHUB_TOKEN',
|
|
70
|
+
'TG_TOKEN',
|
|
71
|
+
'DISCORD_TOKEN',
|
|
72
|
+
'SLACK_BOT_TOKEN',
|
|
73
|
+
'SLACK_SIGNING_SECRET',
|
|
74
|
+
'DEEPGRAM_API_KEY',
|
|
75
|
+
'GROQ_API_KEY',
|
|
76
|
+
];
|
|
77
|
+
const envCheck = tool({
|
|
78
|
+
name: 'env_check',
|
|
79
|
+
description: 'Check which environment variables are set. Returns true/false for each known variable (never exposes values). Use this before switching LLM providers to verify required keys are configured.',
|
|
80
|
+
parameters: z.object({
|
|
81
|
+
vars: z
|
|
82
|
+
.array(z.string())
|
|
83
|
+
.optional()
|
|
84
|
+
.describe('Specific vars to check. If omitted, checks all known vars.'),
|
|
85
|
+
}),
|
|
86
|
+
execute: async ({ vars }) => {
|
|
87
|
+
const toCheck = vars?.length ? vars : KNOWN_VARS;
|
|
88
|
+
const result = {};
|
|
89
|
+
for (const v of toCheck) {
|
|
90
|
+
result[v] = !!process.env[v];
|
|
91
|
+
}
|
|
92
|
+
return result;
|
|
93
|
+
},
|
|
94
|
+
});
|
|
95
|
+
const envSet = tool({
|
|
96
|
+
name: 'env_set',
|
|
97
|
+
description: 'Write environment variables to the .env file. Merges with existing vars. The assistant will restart automatically to pick up new values.',
|
|
98
|
+
parameters: z.object({
|
|
99
|
+
vars: z.record(z.string(), z.string()).describe('Key-value pairs to write to .env'),
|
|
100
|
+
}),
|
|
101
|
+
execute: async ({ vars }) => {
|
|
102
|
+
const existing = existsSync(envPath) ? readFileSync(envPath, 'utf-8') : '';
|
|
103
|
+
const map = new Map();
|
|
104
|
+
for (const line of existing.split('\n')) {
|
|
105
|
+
const eq = line.indexOf('=');
|
|
106
|
+
if (eq > 0)
|
|
107
|
+
map.set(line.slice(0, eq), line);
|
|
108
|
+
}
|
|
109
|
+
for (const [key, value] of Object.entries(vars)) {
|
|
110
|
+
map.set(key, `${key}=${value}`);
|
|
111
|
+
}
|
|
112
|
+
writeFileSync(envPath, [...map.values()].join('\n') + '\n');
|
|
113
|
+
onConfigUpdated?.();
|
|
114
|
+
return {
|
|
115
|
+
success: true,
|
|
116
|
+
message: `Set ${Object.keys(vars).join(', ')} in .env. Restarting...`,
|
|
117
|
+
};
|
|
118
|
+
},
|
|
119
|
+
});
|
|
120
|
+
return [configRead, configUpdate, envCheck, envSet];
|
|
121
|
+
}
|
|
122
|
+
//# sourceMappingURL=self-config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"self-config.js","sourceRoot":"","sources":["../../src/tools/self-config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAG1C,SAAS,SAAS,CAChB,MAA+B,EAC/B,MAA+B;IAE/B,MAAM,MAAM,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;IAC7B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,IACE,MAAM,KAAK,IAAI;YACf,OAAO,MAAM,KAAK,QAAQ;YAC1B,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACtB,MAAM,KAAK,IAAI;YACf,OAAO,MAAM,KAAK,QAAQ;YAC1B,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EACtB,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,MAAiC,EAAE,MAAiC,CAAC,CAAC;QAChG,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;QACvB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,IAMrC;IACC,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,aAAa,EAAE,cAAc,EAAE,eAAe,EAAE,GAAG,IAAI,CAAC;IAEvF,MAAM,UAAU,GAAG,IAAI,CAAC;QACtB,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,0DAA0D;QACvE,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,GAAG,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;QACpC,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,IAAI,CAAC;QACxB,IAAI,EAAE,eAAe;QACrB,WAAW,EACT,4NAA4N;QAC9N,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,OAAO,EAAE,CAAC;iBACP,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;iBAC/B,QAAQ,CAAC,kDAAkD,CAAC;SAChE,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;YAC7B,MAAM,GAAG,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC9C,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAA4B,CAAC;YAC1D,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE3C,IAAI,CAAC;gBACH,cAAc,CAAC,MAAM,CAAC,CAAC;YACzB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,sBAAsB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE;iBAChF,CAAC;YACJ,CAAC;YAED,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;YACjD,eAAe,EAAE,EAAE,CAAC;YACpB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,+BAA+B,EAAE,CAAC;QACrE,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,CAAC;IAElD,MAAM,UAAU,GAAG;QACjB,gBAAgB;QAChB,gBAAgB;QAChB,gBAAgB;QAChB,mBAAmB;QACnB,YAAY;QACZ,gBAAgB;QAChB,cAAc;QACd,UAAU;QACV,eAAe;QACf,iBAAiB;QACjB,sBAAsB;QACtB,kBAAkB;QAClB,cAAc;KACf,CAAC;IAEF,MAAM,QAAQ,GAAG,IAAI,CAAC;QACpB,IAAI,EAAE,WAAW;QACjB,WAAW,EACT,+LAA+L;QACjM,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,IAAI,EAAE,CAAC;iBACJ,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBACjB,QAAQ,EAAE;iBACV,QAAQ,CAAC,4DAA4D,CAAC;SAC1E,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;YAC1B,MAAM,OAAO,GAAG,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC;YACjD,MAAM,MAAM,GAA4B,EAAE,CAAC;YAC3C,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,IAAI,CAAC;QAClB,IAAI,EAAE,SAAS;QACf,WAAW,EACT,0IAA0I;QAC5I,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC;SACpF,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;YAC1B,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3E,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YAEtC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;gBAC7B,IAAI,EAAE,GAAG,CAAC;oBAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YAC/C,CAAC;YAED,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAChD,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,KAAK,EAAE,CAAC,CAAC;YAClC,CAAC;YAED,aAAa,CAAC,OAAO,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;YAC5D,eAAe,EAAE,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,yBAAyB;aACtE,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;AACtD,CAAC"}
|