@zhin.js/agent 0.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/lib/agent.d.ts +130 -0
- package/lib/agent.d.ts.map +1 -0
- package/lib/agent.js +702 -0
- package/lib/agent.js.map +1 -0
- package/lib/bootstrap.d.ts +91 -0
- package/lib/bootstrap.d.ts.map +1 -0
- package/lib/bootstrap.js +243 -0
- package/lib/bootstrap.js.map +1 -0
- package/lib/builtin-tools.d.ts +59 -0
- package/lib/builtin-tools.d.ts.map +1 -0
- package/lib/builtin-tools.js +777 -0
- package/lib/builtin-tools.js.map +1 -0
- package/lib/compaction.d.ts +132 -0
- package/lib/compaction.d.ts.map +1 -0
- package/lib/compaction.js +370 -0
- package/lib/compaction.js.map +1 -0
- package/lib/context-manager.d.ts +213 -0
- package/lib/context-manager.d.ts.map +1 -0
- package/lib/context-manager.js +313 -0
- package/lib/context-manager.js.map +1 -0
- package/lib/conversation-memory.d.ts +181 -0
- package/lib/conversation-memory.d.ts.map +1 -0
- package/lib/conversation-memory.js +581 -0
- package/lib/conversation-memory.js.map +1 -0
- package/lib/cron-engine.d.ts +91 -0
- package/lib/cron-engine.d.ts.map +1 -0
- package/lib/cron-engine.js +277 -0
- package/lib/cron-engine.js.map +1 -0
- package/lib/follow-up.d.ts +131 -0
- package/lib/follow-up.d.ts.map +1 -0
- package/lib/follow-up.js +265 -0
- package/lib/follow-up.js.map +1 -0
- package/lib/hooks.d.ts +143 -0
- package/lib/hooks.d.ts.map +1 -0
- package/lib/hooks.js +108 -0
- package/lib/hooks.js.map +1 -0
- package/lib/index.d.ts +43 -0
- package/lib/index.d.ts.map +1 -0
- package/lib/index.js +28 -0
- package/lib/index.js.map +1 -0
- package/lib/init.d.ts +30 -0
- package/lib/init.d.ts.map +1 -0
- package/lib/init.js +684 -0
- package/lib/init.js.map +1 -0
- package/lib/output.d.ts +93 -0
- package/lib/output.d.ts.map +1 -0
- package/lib/output.js +176 -0
- package/lib/output.js.map +1 -0
- package/lib/rate-limiter.d.ts +38 -0
- package/lib/rate-limiter.d.ts.map +1 -0
- package/lib/rate-limiter.js +86 -0
- package/lib/rate-limiter.js.map +1 -0
- package/lib/service.d.ts +86 -0
- package/lib/service.d.ts.map +1 -0
- package/lib/service.js +285 -0
- package/lib/service.js.map +1 -0
- package/lib/session.d.ts +186 -0
- package/lib/session.d.ts.map +1 -0
- package/lib/session.js +443 -0
- package/lib/session.js.map +1 -0
- package/lib/subagent.d.ts +50 -0
- package/lib/subagent.d.ts.map +1 -0
- package/lib/subagent.js +144 -0
- package/lib/subagent.js.map +1 -0
- package/lib/tone-detector.d.ts +19 -0
- package/lib/tone-detector.d.ts.map +1 -0
- package/lib/tone-detector.js +72 -0
- package/lib/tone-detector.js.map +1 -0
- package/lib/tools.d.ts +45 -0
- package/lib/tools.d.ts.map +1 -0
- package/lib/tools.js +206 -0
- package/lib/tools.js.map +1 -0
- package/lib/user-profile.d.ts +56 -0
- package/lib/user-profile.d.ts.map +1 -0
- package/lib/user-profile.js +130 -0
- package/lib/user-profile.js.map +1 -0
- package/lib/zhin-agent/builtin-tools.d.ts +17 -0
- package/lib/zhin-agent/builtin-tools.d.ts.map +1 -0
- package/lib/zhin-agent/builtin-tools.js +220 -0
- package/lib/zhin-agent/builtin-tools.js.map +1 -0
- package/lib/zhin-agent/config.d.ts +54 -0
- package/lib/zhin-agent/config.d.ts.map +1 -0
- package/lib/zhin-agent/config.js +76 -0
- package/lib/zhin-agent/config.js.map +1 -0
- package/lib/zhin-agent/exec-policy.d.ts +20 -0
- package/lib/zhin-agent/exec-policy.d.ts.map +1 -0
- package/lib/zhin-agent/exec-policy.js +71 -0
- package/lib/zhin-agent/exec-policy.js.map +1 -0
- package/lib/zhin-agent/index.d.ts +68 -0
- package/lib/zhin-agent/index.d.ts.map +1 -0
- package/lib/zhin-agent/index.js +404 -0
- package/lib/zhin-agent/index.js.map +1 -0
- package/lib/zhin-agent/prompt.d.ts +21 -0
- package/lib/zhin-agent/prompt.d.ts.map +1 -0
- package/lib/zhin-agent/prompt.js +111 -0
- package/lib/zhin-agent/prompt.js.map +1 -0
- package/lib/zhin-agent/tool-collector.d.ts +21 -0
- package/lib/zhin-agent/tool-collector.d.ts.map +1 -0
- package/lib/zhin-agent/tool-collector.js +218 -0
- package/lib/zhin-agent/tool-collector.js.map +1 -0
- package/package.json +37 -0
- package/src/agent.ts +831 -0
- package/src/bootstrap.ts +309 -0
- package/src/builtin-tools.ts +849 -0
- package/src/compaction.ts +529 -0
- package/src/context-manager.ts +440 -0
- package/src/conversation-memory.ts +774 -0
- package/src/cron-engine.ts +336 -0
- package/src/follow-up.ts +357 -0
- package/src/hooks.ts +223 -0
- package/src/index.ts +169 -0
- package/src/init.ts +750 -0
- package/src/output.ts +261 -0
- package/src/rate-limiter.ts +129 -0
- package/src/service.ts +329 -0
- package/src/session.ts +544 -0
- package/src/subagent.ts +209 -0
- package/src/tone-detector.ts +89 -0
- package/src/tools.ts +217 -0
- package/src/user-profile.ts +181 -0
- package/src/zhin-agent/builtin-tools.ts +247 -0
- package/src/zhin-agent/config.ts +113 -0
- package/src/zhin-agent/exec-policy.ts +78 -0
- package/src/zhin-agent/index.ts +505 -0
- package/src/zhin-agent/prompt.ts +131 -0
- package/src/zhin-agent/tool-collector.ts +242 -0
- package/tests/ai/agent.test.ts +565 -0
- package/tests/ai/context-manager.test.ts +413 -0
- package/tests/ai/conversation-memory.test.ts +128 -0
- package/tests/ai/follow-up.test.ts +175 -0
- package/tests/ai/integration.test.ts +582 -0
- package/tests/ai/output.test.ts +128 -0
- package/tests/ai/rate-limiter.test.ts +108 -0
- package/tests/ai/session.test.ts +334 -0
- package/tests/ai/setup.ts +250 -0
- package/tests/ai/subagent.test.ts +270 -0
- package/tests/ai/tone-detector.test.ts +80 -0
- package/tests/ai/tools-builtin.test.ts +310 -0
- package/tests/ai/user-profile.test.ts +73 -0
- package/tests/ai/zhin-agent.test.ts +177 -0
- package/tsconfig.json +22 -0
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 持久化定时任务
|
|
3
|
+
*
|
|
4
|
+
* 将定时任务持久化到 data/cron-jobs.json,进程重启后自动加载;
|
|
5
|
+
* 触发时以 prompt 调用 ZhinAgent,实现「到点执行 AI 任务」。
|
|
6
|
+
*
|
|
7
|
+
* - 存储:id, cronExpression, prompt, label?, enabled, createdAt
|
|
8
|
+
* - 启动时:读取文件 → 为每条启用的任务创建 Cron → 注册到 CronFeature
|
|
9
|
+
* - CLI / AI 工具:可对持久化任务做 list / add / remove / pause / resume(AI 侧立即生效)
|
|
10
|
+
*/
|
|
11
|
+
import { Cron, ZhinTool } from '@zhin.js/core';
|
|
12
|
+
export declare const CRON_JOBS_FILENAME = "cron-jobs.json";
|
|
13
|
+
export interface CronJobRecord {
|
|
14
|
+
id: string;
|
|
15
|
+
/** Cron 表达式,5 字段:分 时 日 月 周 */
|
|
16
|
+
cronExpression: string;
|
|
17
|
+
/** 触发时发给 AI 的 prompt */
|
|
18
|
+
prompt: string;
|
|
19
|
+
/** 可选标签,便于识别 */
|
|
20
|
+
label?: string;
|
|
21
|
+
/** 是否启用(暂停的任务不加载) */
|
|
22
|
+
enabled: boolean;
|
|
23
|
+
createdAt: number;
|
|
24
|
+
}
|
|
25
|
+
export declare function getCronJobsFilePath(dataDir: string): string;
|
|
26
|
+
export declare function readCronJobsFile(dataDir: string): Promise<CronJobRecord[]>;
|
|
27
|
+
export declare function writeCronJobsFile(dataDir: string, jobs: CronJobRecord[]): Promise<void>;
|
|
28
|
+
export type CronRunner = (prompt: string, jobId: string) => void | Promise<void>;
|
|
29
|
+
export type AddCronFn = (cron: Cron) => () => void;
|
|
30
|
+
export interface PersistentCronEngineOptions {
|
|
31
|
+
dataDir: string;
|
|
32
|
+
addCron: AddCronFn;
|
|
33
|
+
runner: CronRunner;
|
|
34
|
+
}
|
|
35
|
+
export declare class PersistentCronEngine {
|
|
36
|
+
private options;
|
|
37
|
+
/** jobId -> dispose */
|
|
38
|
+
private disposes;
|
|
39
|
+
constructor(options: PersistentCronEngineOptions);
|
|
40
|
+
getDataDir(): string;
|
|
41
|
+
/**
|
|
42
|
+
* 从文件加载任务并注册到 CronFeature;仅加载 enabled 的任务。
|
|
43
|
+
*/
|
|
44
|
+
load(): void;
|
|
45
|
+
private registerOne;
|
|
46
|
+
/**
|
|
47
|
+
* 列出所有持久化任务(从文件读取)
|
|
48
|
+
*/
|
|
49
|
+
listJobs(): Promise<CronJobRecord[]>;
|
|
50
|
+
/**
|
|
51
|
+
* 添加持久化任务并立即生效
|
|
52
|
+
*/
|
|
53
|
+
addJob(record: Omit<CronJobRecord, 'createdAt'> & {
|
|
54
|
+
createdAt?: number;
|
|
55
|
+
}): Promise<CronJobRecord>;
|
|
56
|
+
/**
|
|
57
|
+
* 删除持久化任务并立即生效
|
|
58
|
+
*/
|
|
59
|
+
removeJob(id: string): Promise<boolean>;
|
|
60
|
+
/**
|
|
61
|
+
* 暂停任务(不删除,停止调度)
|
|
62
|
+
*/
|
|
63
|
+
pauseJob(id: string): Promise<boolean>;
|
|
64
|
+
/**
|
|
65
|
+
* 恢复已暂停的任务
|
|
66
|
+
*/
|
|
67
|
+
resumeJob(id: string): Promise<boolean>;
|
|
68
|
+
/**
|
|
69
|
+
* 卸载所有由本引擎注册的定时任务(用于 dispose)
|
|
70
|
+
*/
|
|
71
|
+
unload(): void;
|
|
72
|
+
}
|
|
73
|
+
/**
|
|
74
|
+
* 生成唯一 ID(用于 CLI / AI 添加时)
|
|
75
|
+
*/
|
|
76
|
+
export declare function generateCronJobId(): string;
|
|
77
|
+
export interface CronManager {
|
|
78
|
+
cronFeature: {
|
|
79
|
+
getStatus(): Array<{
|
|
80
|
+
expression: string;
|
|
81
|
+
running: boolean;
|
|
82
|
+
nextExecution: Date | null;
|
|
83
|
+
plugin: string;
|
|
84
|
+
}>;
|
|
85
|
+
};
|
|
86
|
+
engine: PersistentCronEngine | null;
|
|
87
|
+
}
|
|
88
|
+
export declare function setCronManager(m: CronManager | null): void;
|
|
89
|
+
export declare function getCronManager(): CronManager | null;
|
|
90
|
+
export declare function createCronTools(): ZhinTool[];
|
|
91
|
+
//# sourceMappingURL=cron-engine.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cron-engine.d.ts","sourceRoot":"","sources":["../src/cron-engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAIH,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAS/C,eAAO,MAAM,kBAAkB,mBAAmB,CAAC;AAEnD,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,8BAA8B;IAC9B,cAAc,EAAE,MAAM,CAAC;IACvB,wBAAwB;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,gBAAgB;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qBAAqB;IACrB,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE3D;AAED,wBAAsB,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC,CAYhF;AAED,wBAAsB,iBAAiB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAI7F;AAMD,MAAM,MAAM,UAAU,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAEjF,MAAM,MAAM,SAAS,GAAG,CAAC,IAAI,EAAE,IAAI,KAAK,MAAM,IAAI,CAAC;AAEnD,MAAM,WAAW,2BAA2B;IAC1C,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,SAAS,CAAC;IACnB,MAAM,EAAE,UAAU,CAAC;CACpB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,OAAO,CAA8B;IAC7C,uBAAuB;IACvB,OAAO,CAAC,QAAQ,CAAiC;gBAErC,OAAO,EAAE,2BAA2B;IAIhD,UAAU,IAAI,MAAM;IAIpB;;OAEG;IACH,IAAI,IAAI,IAAI;IAeZ,OAAO,CAAC,WAAW;IAkBnB;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;IAI1C;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,EAAE,WAAW,CAAC,GAAG;QAAE,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAevG;;OAEG;IACG,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAa7C;;OAEG;IACG,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAc5C;;OAEG;IACG,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAU7C;;OAEG;IACH,MAAM,IAAI,IAAI;CAQf;AAED;;GAEG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C;AAMD,MAAM,WAAW,WAAW;IAC1B,WAAW,EAAE;QAAE,SAAS,IAAI,KAAK,CAAC;YAAE,UAAU,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,OAAO,CAAC;YAAC,aAAa,EAAE,IAAI,GAAG,IAAI,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC,CAAA;KAAE,CAAC;IAC1H,MAAM,EAAE,oBAAoB,GAAG,IAAI,CAAC;CACrC;AAID,wBAAgB,cAAc,CAAC,CAAC,EAAE,WAAW,GAAG,IAAI,GAAG,IAAI,CAE1D;AAED,wBAAgB,cAAc,IAAI,WAAW,GAAG,IAAI,CAEnD;AAMD,wBAAgB,eAAe,IAAI,QAAQ,EAAE,CAiG5C"}
|
|
@@ -0,0 +1,277 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 持久化定时任务
|
|
3
|
+
*
|
|
4
|
+
* 将定时任务持久化到 data/cron-jobs.json,进程重启后自动加载;
|
|
5
|
+
* 触发时以 prompt 调用 ZhinAgent,实现「到点执行 AI 任务」。
|
|
6
|
+
*
|
|
7
|
+
* - 存储:id, cronExpression, prompt, label?, enabled, createdAt
|
|
8
|
+
* - 启动时:读取文件 → 为每条启用的任务创建 Cron → 注册到 CronFeature
|
|
9
|
+
* - CLI / AI 工具:可对持久化任务做 list / add / remove / pause / resume(AI 侧立即生效)
|
|
10
|
+
*/
|
|
11
|
+
import * as fs from 'fs';
|
|
12
|
+
import * as path from 'path';
|
|
13
|
+
import { Cron, ZhinTool } from '@zhin.js/core';
|
|
14
|
+
import { Logger } from '@zhin.js/logger';
|
|
15
|
+
const logger = new Logger(null, 'cron-engine');
|
|
16
|
+
// ─────────────────────────────────────────────────────────────────────────────
|
|
17
|
+
// 类型与存储路径
|
|
18
|
+
// ─────────────────────────────────────────────────────────────────────────────
|
|
19
|
+
export const CRON_JOBS_FILENAME = 'cron-jobs.json';
|
|
20
|
+
export function getCronJobsFilePath(dataDir) {
|
|
21
|
+
return path.join(dataDir, CRON_JOBS_FILENAME);
|
|
22
|
+
}
|
|
23
|
+
export async function readCronJobsFile(dataDir) {
|
|
24
|
+
const filePath = getCronJobsFilePath(dataDir);
|
|
25
|
+
try {
|
|
26
|
+
const raw = await fs.promises.readFile(filePath, 'utf-8');
|
|
27
|
+
const data = JSON.parse(raw);
|
|
28
|
+
if (!Array.isArray(data))
|
|
29
|
+
return [];
|
|
30
|
+
return data;
|
|
31
|
+
}
|
|
32
|
+
catch (e) {
|
|
33
|
+
if (e?.code === 'ENOENT')
|
|
34
|
+
return [];
|
|
35
|
+
logger.warn('读取定时任务文件失败: ' + (e?.message || String(e)));
|
|
36
|
+
return [];
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
export async function writeCronJobsFile(dataDir, jobs) {
|
|
40
|
+
const filePath = getCronJobsFilePath(dataDir);
|
|
41
|
+
await fs.promises.mkdir(path.dirname(filePath), { recursive: true });
|
|
42
|
+
await fs.promises.writeFile(filePath, JSON.stringify(jobs, null, 2), 'utf-8');
|
|
43
|
+
}
|
|
44
|
+
export class PersistentCronEngine {
|
|
45
|
+
options;
|
|
46
|
+
/** jobId -> dispose */
|
|
47
|
+
disposes = new Map();
|
|
48
|
+
constructor(options) {
|
|
49
|
+
this.options = options;
|
|
50
|
+
}
|
|
51
|
+
getDataDir() {
|
|
52
|
+
return this.options.dataDir;
|
|
53
|
+
}
|
|
54
|
+
/**
|
|
55
|
+
* 从文件加载任务并注册到 CronFeature;仅加载 enabled 的任务。
|
|
56
|
+
*/
|
|
57
|
+
load() {
|
|
58
|
+
const { dataDir, addCron, runner } = this.options;
|
|
59
|
+
readCronJobsFile(dataDir).then((jobs) => {
|
|
60
|
+
for (const job of jobs) {
|
|
61
|
+
if (!job.enabled)
|
|
62
|
+
continue;
|
|
63
|
+
this.registerOne(job, addCron, runner);
|
|
64
|
+
}
|
|
65
|
+
if (jobs.filter((j) => j.enabled).length > 0) {
|
|
66
|
+
logger.info(`已加载 ${this.disposes.size} 个持久化定时任务`);
|
|
67
|
+
}
|
|
68
|
+
}).catch((e) => {
|
|
69
|
+
logger.warn('加载持久化定时任务失败: ' + (e?.message || String(e)));
|
|
70
|
+
});
|
|
71
|
+
}
|
|
72
|
+
registerOne(job, addCron, runner) {
|
|
73
|
+
const { prompt, id: jobId, cronExpression } = job;
|
|
74
|
+
try {
|
|
75
|
+
const cron = new Cron(cronExpression, async () => {
|
|
76
|
+
await runner(prompt, jobId);
|
|
77
|
+
});
|
|
78
|
+
cron.id = jobId;
|
|
79
|
+
const dispose = addCron(cron);
|
|
80
|
+
this.disposes.set(jobId, dispose);
|
|
81
|
+
}
|
|
82
|
+
catch (e) {
|
|
83
|
+
logger.warn(`定时任务加载失败 [${jobId}]: ${e?.message || String(e)}`);
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
/**
|
|
87
|
+
* 列出所有持久化任务(从文件读取)
|
|
88
|
+
*/
|
|
89
|
+
async listJobs() {
|
|
90
|
+
return readCronJobsFile(this.options.dataDir);
|
|
91
|
+
}
|
|
92
|
+
/**
|
|
93
|
+
* 添加持久化任务并立即生效
|
|
94
|
+
*/
|
|
95
|
+
async addJob(record) {
|
|
96
|
+
const jobs = await readCronJobsFile(this.options.dataDir);
|
|
97
|
+
const full = {
|
|
98
|
+
...record,
|
|
99
|
+
createdAt: record.createdAt ?? Date.now(),
|
|
100
|
+
enabled: record.enabled ?? true,
|
|
101
|
+
};
|
|
102
|
+
jobs.push(full);
|
|
103
|
+
await writeCronJobsFile(this.options.dataDir, jobs);
|
|
104
|
+
if (full.enabled) {
|
|
105
|
+
this.registerOne(full, this.options.addCron, this.options.runner);
|
|
106
|
+
}
|
|
107
|
+
return full;
|
|
108
|
+
}
|
|
109
|
+
/**
|
|
110
|
+
* 删除持久化任务并立即生效
|
|
111
|
+
*/
|
|
112
|
+
async removeJob(id) {
|
|
113
|
+
const jobs = await readCronJobsFile(this.options.dataDir);
|
|
114
|
+
const next = jobs.filter((j) => j.id !== id);
|
|
115
|
+
if (next.length === jobs.length)
|
|
116
|
+
return false;
|
|
117
|
+
await writeCronJobsFile(this.options.dataDir, next);
|
|
118
|
+
const dispose = this.disposes.get(id);
|
|
119
|
+
if (dispose) {
|
|
120
|
+
dispose();
|
|
121
|
+
this.disposes.delete(id);
|
|
122
|
+
}
|
|
123
|
+
return true;
|
|
124
|
+
}
|
|
125
|
+
/**
|
|
126
|
+
* 暂停任务(不删除,停止调度)
|
|
127
|
+
*/
|
|
128
|
+
async pauseJob(id) {
|
|
129
|
+
const jobs = await readCronJobsFile(this.options.dataDir);
|
|
130
|
+
const j = jobs.find((x) => x.id === id);
|
|
131
|
+
if (!j)
|
|
132
|
+
return false;
|
|
133
|
+
j.enabled = false;
|
|
134
|
+
await writeCronJobsFile(this.options.dataDir, jobs);
|
|
135
|
+
const dispose = this.disposes.get(id);
|
|
136
|
+
if (dispose) {
|
|
137
|
+
dispose();
|
|
138
|
+
this.disposes.delete(id);
|
|
139
|
+
}
|
|
140
|
+
return true;
|
|
141
|
+
}
|
|
142
|
+
/**
|
|
143
|
+
* 恢复已暂停的任务
|
|
144
|
+
*/
|
|
145
|
+
async resumeJob(id) {
|
|
146
|
+
const jobs = await readCronJobsFile(this.options.dataDir);
|
|
147
|
+
const j = jobs.find((x) => x.id === id);
|
|
148
|
+
if (!j)
|
|
149
|
+
return false;
|
|
150
|
+
j.enabled = true;
|
|
151
|
+
await writeCronJobsFile(this.options.dataDir, jobs);
|
|
152
|
+
this.registerOne(j, this.options.addCron, this.options.runner);
|
|
153
|
+
return true;
|
|
154
|
+
}
|
|
155
|
+
/**
|
|
156
|
+
* 卸载所有由本引擎注册的定时任务(用于 dispose)
|
|
157
|
+
*/
|
|
158
|
+
unload() {
|
|
159
|
+
for (const dispose of this.disposes.values()) {
|
|
160
|
+
try {
|
|
161
|
+
dispose();
|
|
162
|
+
}
|
|
163
|
+
catch (_) { }
|
|
164
|
+
}
|
|
165
|
+
this.disposes.clear();
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
/**
|
|
169
|
+
* 生成唯一 ID(用于 CLI / AI 添加时)
|
|
170
|
+
*/
|
|
171
|
+
export function generateCronJobId() {
|
|
172
|
+
return `cron_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`;
|
|
173
|
+
}
|
|
174
|
+
let cronManager = null;
|
|
175
|
+
export function setCronManager(m) {
|
|
176
|
+
cronManager = m;
|
|
177
|
+
}
|
|
178
|
+
export function getCronManager() {
|
|
179
|
+
return cronManager;
|
|
180
|
+
}
|
|
181
|
+
// ─────────────────────────────────────────────────────────────────────────────
|
|
182
|
+
// AI 可调用的定时任务管理工具
|
|
183
|
+
// ─────────────────────────────────────────────────────────────────────────────
|
|
184
|
+
export function createCronTools() {
|
|
185
|
+
const listTool = new ZhinTool('cron_list')
|
|
186
|
+
.desc('列出所有定时任务:包括插件注册的内存任务与持久化任务(持久化任务有 id,可用于 cron_remove/cron_pause/cron_resume)')
|
|
187
|
+
.keyword('定时任务', 'cron', '计划任务', '任务列表')
|
|
188
|
+
.tag('cron', 'schedule')
|
|
189
|
+
.execute(async () => {
|
|
190
|
+
const m = getCronManager();
|
|
191
|
+
if (!m) {
|
|
192
|
+
return { error: '定时任务服务不可用' };
|
|
193
|
+
}
|
|
194
|
+
const memory = m.cronFeature.getStatus().map((s) => ({
|
|
195
|
+
type: 'memory',
|
|
196
|
+
expression: s.expression,
|
|
197
|
+
running: s.running,
|
|
198
|
+
nextExecution: s.nextExecution?.toISOString() ?? null,
|
|
199
|
+
plugin: s.plugin,
|
|
200
|
+
}));
|
|
201
|
+
const persistent = m.engine
|
|
202
|
+
? (await m.engine.listJobs()).map((j) => ({
|
|
203
|
+
type: 'persistent',
|
|
204
|
+
id: j.id,
|
|
205
|
+
cronExpression: j.cronExpression,
|
|
206
|
+
prompt: j.prompt,
|
|
207
|
+
label: j.label,
|
|
208
|
+
enabled: j.enabled,
|
|
209
|
+
createdAt: j.createdAt,
|
|
210
|
+
}))
|
|
211
|
+
: [];
|
|
212
|
+
return { memory, persistent };
|
|
213
|
+
});
|
|
214
|
+
const addTool = new ZhinTool('cron_add')
|
|
215
|
+
.desc('添加一条持久化定时任务:到点由 AI 执行指定 prompt,重启后仍保留')
|
|
216
|
+
.keyword('添加定时', '新建定时任务', 'cron add')
|
|
217
|
+
.tag('cron', 'schedule')
|
|
218
|
+
.param('cron_expression', { type: 'string', description: 'Cron 表达式,如 "0 9 * * *" 表示每天 9:00' }, true)
|
|
219
|
+
.param('prompt', { type: 'string', description: '到点触发时发给 AI 的提示词' }, true)
|
|
220
|
+
.param('label', { type: 'string', description: '可选标签,便于识别' })
|
|
221
|
+
.execute(async (args) => {
|
|
222
|
+
const m = getCronManager();
|
|
223
|
+
if (!m?.engine) {
|
|
224
|
+
return { error: '持久化定时任务引擎不可用' };
|
|
225
|
+
}
|
|
226
|
+
const id = generateCronJobId();
|
|
227
|
+
const job = await m.engine.addJob({
|
|
228
|
+
id,
|
|
229
|
+
cronExpression: args.cron_expression,
|
|
230
|
+
prompt: args.prompt,
|
|
231
|
+
label: args.label,
|
|
232
|
+
enabled: true,
|
|
233
|
+
});
|
|
234
|
+
return { success: true, id: job.id, message: '已添加并立即生效' };
|
|
235
|
+
});
|
|
236
|
+
const removeTool = new ZhinTool('cron_remove')
|
|
237
|
+
.desc('按 id 删除一条持久化定时任务')
|
|
238
|
+
.keyword('删除定时', '取消定时', 'cron remove')
|
|
239
|
+
.tag('cron', 'schedule')
|
|
240
|
+
.param('id', { type: 'string', description: '任务 ID(cron_list 中 persistent 的 id)' }, true)
|
|
241
|
+
.execute(async (args) => {
|
|
242
|
+
const m = getCronManager();
|
|
243
|
+
if (!m?.engine) {
|
|
244
|
+
return { error: '持久化定时任务引擎不可用' };
|
|
245
|
+
}
|
|
246
|
+
const ok = await m.engine.removeJob(args.id);
|
|
247
|
+
return ok ? { success: true, message: '已删除' } : { error: '未找到该任务' };
|
|
248
|
+
});
|
|
249
|
+
const pauseTool = new ZhinTool('cron_pause')
|
|
250
|
+
.desc('暂停一条持久化定时任务(不删除,可 cron_resume 恢复)')
|
|
251
|
+
.keyword('暂停定时', 'cron pause')
|
|
252
|
+
.tag('cron', 'schedule')
|
|
253
|
+
.param('id', { type: 'string', description: '任务 ID' }, true)
|
|
254
|
+
.execute(async (args) => {
|
|
255
|
+
const m = getCronManager();
|
|
256
|
+
if (!m?.engine) {
|
|
257
|
+
return { error: '持久化定时任务引擎不可用' };
|
|
258
|
+
}
|
|
259
|
+
const ok = await m.engine.pauseJob(args.id);
|
|
260
|
+
return ok ? { success: true, message: '已暂停' } : { error: '未找到该任务' };
|
|
261
|
+
});
|
|
262
|
+
const resumeTool = new ZhinTool('cron_resume')
|
|
263
|
+
.desc('恢复已暂停的持久化定时任务')
|
|
264
|
+
.keyword('恢复定时', 'cron resume')
|
|
265
|
+
.tag('cron', 'schedule')
|
|
266
|
+
.param('id', { type: 'string', description: '任务 ID' }, true)
|
|
267
|
+
.execute(async (args) => {
|
|
268
|
+
const m = getCronManager();
|
|
269
|
+
if (!m?.engine) {
|
|
270
|
+
return { error: '持久化定时任务引擎不可用' };
|
|
271
|
+
}
|
|
272
|
+
const ok = await m.engine.resumeJob(args.id);
|
|
273
|
+
return ok ? { success: true, message: '已恢复' } : { error: '未找到该任务' };
|
|
274
|
+
});
|
|
275
|
+
return [listTool, addTool, removeTool, pauseTool, resumeTool];
|
|
276
|
+
}
|
|
277
|
+
//# sourceMappingURL=cron-engine.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cron-engine.js","sourceRoot":"","sources":["../src/cron-engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEzC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;AAE/C,gFAAgF;AAChF,UAAU;AACV,gFAAgF;AAEhF,MAAM,CAAC,MAAM,kBAAkB,GAAG,gBAAgB,CAAC;AAenD,MAAM,UAAU,mBAAmB,CAAC,OAAe;IACjD,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,OAAe;IACpD,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;YAAE,OAAO,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,CAAC,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,EAAE,OAAO,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,OAAe,EAAE,IAAqB;IAC5E,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IAC9C,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrE,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;AAChF,CAAC;AAgBD,MAAM,OAAO,oBAAoB;IACvB,OAAO,CAA8B;IAC7C,uBAAuB;IACf,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;IAEjD,YAAY,OAAoC;QAC9C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,IAAI;QACF,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACtC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,IAAI,CAAC,GAAG,CAAC,OAAO;oBAAE,SAAS;gBAC3B,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACzC,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7C,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,WAAW,CAAC,CAAC;YACpD,CAAC;QACH,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACb,MAAM,CAAC,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC,EAAE,OAAO,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CACjB,GAAkB,EAClB,OAAkB,EAClB,MAAkB;QAElB,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,GAAG,CAAC;QAClD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;gBAC/C,MAAM,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC;YAChB,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,aAAa,KAAK,MAAM,CAAC,EAAE,OAAO,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,OAAO,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,MAAiE;QAC5E,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAkB;YAC1B,GAAG,MAAM;YACT,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;YACzC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI;SAChC,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7C,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC9C,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;YACV,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;QACrB,CAAC,CAAC,OAAO,GAAG,KAAK,CAAC;QAClB,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;YACV,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;QACrB,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;QACjB,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,OAAO,EAAE,CAAC;YACZ,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC,CAAA,CAAC;QAChB,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB;IAC/B,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AACrF,CAAC;AAWD,IAAI,WAAW,GAAuB,IAAI,CAAC;AAE3C,MAAM,UAAU,cAAc,CAAC,CAAqB;IAClD,WAAW,GAAG,CAAC,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,gFAAgF;AAChF,kBAAkB;AAClB,gFAAgF;AAEhF,MAAM,UAAU,eAAe;IAC7B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,WAAW,CAAC;SACvC,IAAI,CAAC,8EAA8E,CAAC;SACpF,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SACvC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;SACvB,OAAO,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,CAAC,GAAG,cAAc,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,CAAC;YACP,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;QAChC,CAAC;QACD,MAAM,MAAM,GAAG,CAAC,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnD,IAAI,EAAE,QAAiB;YACvB,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,aAAa,EAAE,CAAC,CAAC,aAAa,EAAE,WAAW,EAAE,IAAI,IAAI;YACrD,MAAM,EAAE,CAAC,CAAC,MAAM;SACjB,CAAC,CAAC,CAAC;QACJ,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM;YACzB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACtC,IAAI,EAAE,YAAqB;gBAC3B,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,cAAc,EAAE,CAAC,CAAC,cAAc;gBAChC,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,SAAS,EAAE,CAAC,CAAC,SAAS;aACvB,CAAC,CAAC;YACL,CAAC,CAAC,EAAE,CAAC;QACP,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEL,MAAM,OAAO,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC;SACrC,IAAI,CAAC,uCAAuC,CAAC;SAC7C,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC;SACrC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;SACvB,KAAK,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kCAAkC,EAAE,EAAE,IAAI,CAAC;SACnG,KAAK,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,iBAAiB,EAAE,EAAE,IAAI,CAAC;SACzE,KAAK,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;SAC5D,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACtB,MAAM,CAAC,GAAG,cAAc,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACnC,CAAC;QACD,MAAM,EAAE,GAAG,iBAAiB,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;YAChC,EAAE;YACF,cAAc,EAAE,IAAI,CAAC,eAAyB;YAC9C,MAAM,EAAE,IAAI,CAAC,MAAgB;YAC7B,KAAK,EAAE,IAAI,CAAC,KAA2B;YACvC,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEL,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC,aAAa,CAAC;SAC3C,IAAI,CAAC,kBAAkB,CAAC;SACxB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,CAAC;SACtC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;SACvB,KAAK,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,oCAAoC,EAAE,EAAE,IAAI,CAAC;SACxF,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACtB,MAAM,CAAC,GAAG,cAAc,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACnC,CAAC;QACD,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAY,CAAC,CAAC;QACvD,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;IACtE,CAAC,CAAC,CAAC;IAEL,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,YAAY,CAAC;SACzC,IAAI,CAAC,mCAAmC,CAAC;SACzC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC;SAC7B,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;SACvB,KAAK,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,CAAC;SAC3D,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACtB,MAAM,CAAC,GAAG,cAAc,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACnC,CAAC;QACD,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAY,CAAC,CAAC;QACtD,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;IACtE,CAAC,CAAC,CAAC;IAEL,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC,aAAa,CAAC;SAC3C,IAAI,CAAC,eAAe,CAAC;SACrB,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC;SAC9B,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;SACvB,KAAK,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,CAAC;SAC3D,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACtB,MAAM,CAAC,GAAG,cAAc,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACnC,CAAC;QACD,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAY,CAAC,CAAC;QACvD,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;IACtE,CAAC,CAAC,CAAC;IAEL,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;AAChE,CAAC"}
|
|
@@ -0,0 +1,131 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* FollowUpStore — 定时跟进任务的持久化存储
|
|
3
|
+
*
|
|
4
|
+
* 解决纯 setTimeout 重启即丢失的问题:
|
|
5
|
+
* 1. 创建任务时同步写入数据库
|
|
6
|
+
* 2. 执行完成 / 过期后从数据库中删除
|
|
7
|
+
* 3. 机器人启动时从数据库加载所有未完成任务,重新挂定时器
|
|
8
|
+
* 4. 同一会话创建新任务时,自动取消旧的 pending 任务(防止重复提醒)
|
|
9
|
+
*
|
|
10
|
+
* ai_followups 表:
|
|
11
|
+
* ┌──────────────────────────────────────────────────────────────────┐
|
|
12
|
+
* │ id (PK) | session_id | platform | sender_id | scene_id │
|
|
13
|
+
* │ | message | fire_at | created_at | status │
|
|
14
|
+
* └──────────────────────────────────────────────────────────────────┘
|
|
15
|
+
*/
|
|
16
|
+
export declare const AI_FOLLOWUP_MODEL: {
|
|
17
|
+
session_id: {
|
|
18
|
+
type: "text";
|
|
19
|
+
nullable: boolean;
|
|
20
|
+
};
|
|
21
|
+
platform: {
|
|
22
|
+
type: "text";
|
|
23
|
+
nullable: boolean;
|
|
24
|
+
};
|
|
25
|
+
bot_id: {
|
|
26
|
+
type: "text";
|
|
27
|
+
nullable: boolean;
|
|
28
|
+
};
|
|
29
|
+
sender_id: {
|
|
30
|
+
type: "text";
|
|
31
|
+
nullable: boolean;
|
|
32
|
+
};
|
|
33
|
+
scene_id: {
|
|
34
|
+
type: "text";
|
|
35
|
+
nullable: boolean;
|
|
36
|
+
};
|
|
37
|
+
scene_type: {
|
|
38
|
+
type: "text";
|
|
39
|
+
nullable: boolean;
|
|
40
|
+
};
|
|
41
|
+
message: {
|
|
42
|
+
type: "text";
|
|
43
|
+
nullable: boolean;
|
|
44
|
+
};
|
|
45
|
+
/** 触发时间戳 (ms) */
|
|
46
|
+
fire_at: {
|
|
47
|
+
type: "integer";
|
|
48
|
+
nullable: boolean;
|
|
49
|
+
};
|
|
50
|
+
created_at: {
|
|
51
|
+
type: "integer";
|
|
52
|
+
default: number;
|
|
53
|
+
};
|
|
54
|
+
/** pending | fired | cancelled */
|
|
55
|
+
status: {
|
|
56
|
+
type: "text";
|
|
57
|
+
default: string;
|
|
58
|
+
};
|
|
59
|
+
};
|
|
60
|
+
export interface FollowUpRecord {
|
|
61
|
+
id?: number;
|
|
62
|
+
session_id: string;
|
|
63
|
+
platform: string;
|
|
64
|
+
bot_id: string;
|
|
65
|
+
sender_id: string;
|
|
66
|
+
scene_id: string;
|
|
67
|
+
scene_type: string;
|
|
68
|
+
message: string;
|
|
69
|
+
fire_at: number;
|
|
70
|
+
created_at: number;
|
|
71
|
+
status: string;
|
|
72
|
+
}
|
|
73
|
+
/**
|
|
74
|
+
* 数据库模型接口(与 RelatedModel 的链式查询 API 对齐)
|
|
75
|
+
*/
|
|
76
|
+
interface DbModel {
|
|
77
|
+
select(...fields: string[]): any;
|
|
78
|
+
create(data: Record<string, any>): Promise<any>;
|
|
79
|
+
update(data: Partial<any>): any;
|
|
80
|
+
delete(condition: Record<string, any>): any;
|
|
81
|
+
}
|
|
82
|
+
/**
|
|
83
|
+
* 发送提醒的回调函数
|
|
84
|
+
*
|
|
85
|
+
* 由 ZhinAgent 在 init 阶段注入,负责把提醒消息发到正确的会话。
|
|
86
|
+
*/
|
|
87
|
+
export type FollowUpSender = (record: FollowUpRecord) => Promise<void>;
|
|
88
|
+
export declare class FollowUpManager {
|
|
89
|
+
private store;
|
|
90
|
+
/** 内存中活跃的定时器: recordId → timer */
|
|
91
|
+
private timers;
|
|
92
|
+
private sender;
|
|
93
|
+
constructor();
|
|
94
|
+
/** 注入消息发送回调 */
|
|
95
|
+
setSender(sender: FollowUpSender): void;
|
|
96
|
+
/** 升级到数据库存储 */
|
|
97
|
+
upgradeToDatabase(model: DbModel): void;
|
|
98
|
+
/**
|
|
99
|
+
* 创建一个跟进任务
|
|
100
|
+
*
|
|
101
|
+
* 重要:同一会话的旧 pending 任务会被自动取消,防止重复提醒。
|
|
102
|
+
*
|
|
103
|
+
* @returns 人类可读的确认文本
|
|
104
|
+
*/
|
|
105
|
+
schedule(params: {
|
|
106
|
+
sessionId: string;
|
|
107
|
+
platform: string;
|
|
108
|
+
botId: string;
|
|
109
|
+
senderId: string;
|
|
110
|
+
sceneId: string;
|
|
111
|
+
sceneType: string;
|
|
112
|
+
message: string;
|
|
113
|
+
delayMinutes: number;
|
|
114
|
+
}): Promise<string>;
|
|
115
|
+
/**
|
|
116
|
+
* 取消指定会话的所有 pending 任务
|
|
117
|
+
*/
|
|
118
|
+
cancelBySession(sessionId: string): Promise<number>;
|
|
119
|
+
/**
|
|
120
|
+
* 启动时从数据库恢复所有未完成的跟进任务
|
|
121
|
+
*/
|
|
122
|
+
restore(): Promise<number>;
|
|
123
|
+
/**
|
|
124
|
+
* 为一条记录挂定时器
|
|
125
|
+
*/
|
|
126
|
+
private scheduleTimer;
|
|
127
|
+
private scheduleTimerWithDelay;
|
|
128
|
+
dispose(): void;
|
|
129
|
+
}
|
|
130
|
+
export {};
|
|
131
|
+
//# sourceMappingURL=follow-up.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"follow-up.d.ts","sourceRoot":"","sources":["../src/follow-up.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAUH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAQ5B,iBAAiB;;;;;;;;;IAGjB,kCAAkC;;;;;CAEnC,CAAC;AAMF,MAAM,WAAW,cAAc;IAC7B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,UAAU,OAAO;IACf,MAAM,CAAC,GAAG,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC;IACjC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IAChD,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;IAChC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,CAAC;CAC7C;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,MAAM,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAuFvE,qBAAa,eAAe;IAC1B,OAAO,CAAC,KAAK,CAAiB;IAC9B,kCAAkC;IAClC,OAAO,CAAC,MAAM,CAAyD;IACvE,OAAO,CAAC,MAAM,CAA+B;;IAM7C,eAAe;IACf,SAAS,CAAC,MAAM,EAAE,cAAc,GAAG,IAAI;IAIvC,eAAe;IACf,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAOvC;;;;;;OAMG;IACG,QAAQ,CAAC,MAAM,EAAE;QACrB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,YAAY,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,MAAM,CAAC;IA4DnB;;OAEG;IACG,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBzD;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC;IA4BhC;;OAEG;IACH,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,sBAAsB;IA6B9B,OAAO,IAAI,IAAI;CAOhB"}
|