@roackb2/heddle 0.0.9 → 0.0.12
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 +91 -11
- package/dist/examples/heartbeat-scheduler.d.ts +2 -0
- package/dist/examples/heartbeat-scheduler.d.ts.map +1 -0
- package/dist/examples/heartbeat-scheduler.js +88 -0
- package/dist/examples/heartbeat-scheduler.js.map +1 -0
- package/dist/src/cli/chat/App.d.ts.map +1 -1
- package/dist/src/cli/chat/App.js +7 -15
- package/dist/src/cli/chat/App.js.map +1 -1
- package/dist/src/cli/chat/hooks/useAgentRun.d.ts +1 -0
- package/dist/src/cli/chat/hooks/useAgentRun.d.ts.map +1 -1
- package/dist/src/cli/chat/hooks/useAgentRun.js +38 -4
- package/dist/src/cli/chat/hooks/useAgentRun.js.map +1 -1
- package/dist/src/cli/chat/state/storage.js +2 -2
- package/dist/src/cli/chat/state/storage.js.map +1 -1
- package/dist/src/cli/chat/utils/drift-footer.d.ts +5 -0
- package/dist/src/cli/chat/utils/drift-footer.d.ts.map +1 -0
- package/dist/src/cli/chat/utils/drift-footer.js +19 -0
- package/dist/src/cli/chat/utils/drift-footer.js.map +1 -0
- package/dist/src/cli/chat/utils/format.d.ts.map +1 -1
- package/dist/src/cli/chat/utils/format.js +29 -2
- package/dist/src/cli/chat/utils/format.js.map +1 -1
- package/dist/src/cli/heartbeat.d.ts +20 -0
- package/dist/src/cli/heartbeat.d.ts.map +1 -0
- package/dist/src/cli/heartbeat.js +451 -0
- package/dist/src/cli/heartbeat.js.map +1 -0
- package/dist/src/cli/main.js +15 -0
- package/dist/src/cli/main.js.map +1 -1
- package/dist/src/index.d.ts +2 -0
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/index.js +1 -0
- package/dist/src/index.js.map +1 -1
- package/dist/src/integrations/cyberloop-kinematics.d.ts +26 -1
- package/dist/src/integrations/cyberloop-kinematics.d.ts.map +1 -1
- package/dist/src/integrations/cyberloop-kinematics.js +28 -4
- package/dist/src/integrations/cyberloop-kinematics.js.map +1 -1
- package/dist/src/integrations/cyberloop.d.ts +4 -0
- package/dist/src/integrations/cyberloop.d.ts.map +1 -1
- package/dist/src/integrations/cyberloop.js +36 -10
- package/dist/src/integrations/cyberloop.js.map +1 -1
- package/dist/src/runtime/heartbeat-scheduler.d.ts +111 -0
- package/dist/src/runtime/heartbeat-scheduler.d.ts.map +1 -0
- package/dist/src/runtime/heartbeat-scheduler.js +236 -0
- package/dist/src/runtime/heartbeat-scheduler.js.map +1 -0
- package/dist/src/runtime/heartbeat.js +3 -0
- package/dist/src/runtime/heartbeat.js.map +1 -1
- package/dist/src/trace/format.js +28 -1
- package/dist/src/trace/format.js.map +1 -1
- package/package.json +4 -2
|
@@ -1,16 +1,41 @@
|
|
|
1
1
|
import type { TraceEvent } from '../types.js';
|
|
2
|
-
import { type CyberLoopObserver, type CyberLoopObserverAnnotation } from './cyberloop.js';
|
|
2
|
+
import { type CyberLoopCompatibleMiddleware, type CyberLoopObserver, type CyberLoopObserverAnnotation, type HeddleRuntimeFrame } from './cyberloop.js';
|
|
3
3
|
export type CyberLoopKinematicsObserver = {
|
|
4
4
|
observer: CyberLoopObserver;
|
|
5
5
|
annotations: TraceEvent[];
|
|
6
6
|
};
|
|
7
7
|
export type CreateCyberLoopKinematicsObserverOptions = {
|
|
8
8
|
goal: string;
|
|
9
|
+
referenceText?: string;
|
|
9
10
|
apiKey?: string;
|
|
10
11
|
embeddingModel?: string;
|
|
12
|
+
stabilityThreshold?: number;
|
|
11
13
|
moduleSpecifier?: string;
|
|
12
14
|
onAnnotation?: (annotation: CyberLoopObserverAnnotation) => void;
|
|
13
15
|
onError?: (error: unknown) => void;
|
|
16
|
+
_testOverrides?: {
|
|
17
|
+
advancedModule?: CyberLoopAdvancedModule;
|
|
18
|
+
embedText?: (text: string) => Promise<number[]>;
|
|
19
|
+
};
|
|
20
|
+
};
|
|
21
|
+
type CyberLoopAdvancedModule = {
|
|
22
|
+
kinematicsMiddleware: (options: {
|
|
23
|
+
embedder: {
|
|
24
|
+
embed: (state: HeddleRuntimeFrame) => Promise<number[]>;
|
|
25
|
+
};
|
|
26
|
+
goalEmbedding: number[];
|
|
27
|
+
pid?: {
|
|
28
|
+
Kp?: number;
|
|
29
|
+
Ki?: number;
|
|
30
|
+
Kd?: number;
|
|
31
|
+
stabilityThreshold?: number;
|
|
32
|
+
};
|
|
33
|
+
physics?: {
|
|
34
|
+
processNoise?: number;
|
|
35
|
+
measureNoise?: number;
|
|
36
|
+
};
|
|
37
|
+
}) => CyberLoopCompatibleMiddleware<HeddleRuntimeFrame>;
|
|
14
38
|
};
|
|
15
39
|
export declare function createCyberLoopKinematicsObserver(options: CreateCyberLoopKinematicsObserverOptions): Promise<CyberLoopKinematicsObserver>;
|
|
40
|
+
export {};
|
|
16
41
|
//# sourceMappingURL=cyberloop-kinematics.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cyberloop-kinematics.d.ts","sourceRoot":"","sources":["../../../src/integrations/cyberloop-kinematics.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,
|
|
1
|
+
{"version":3,"file":"cyberloop-kinematics.d.ts","sourceRoot":"","sources":["../../../src/integrations/cyberloop-kinematics.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAGL,KAAK,6BAA6B,EAClC,KAAK,iBAAiB,EACtB,KAAK,2BAA2B,EAChC,KAAK,kBAAkB,EACxB,MAAM,gBAAgB,CAAC;AAExB,MAAM,MAAM,2BAA2B,GAAG;IACxC,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,wCAAwC,GAAG;IACrD,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,CAAC,UAAU,EAAE,2BAA2B,KAAK,IAAI,CAAC;IACjE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IACnC,cAAc,CAAC,EAAE;QACf,cAAc,CAAC,EAAE,uBAAuB,CAAC;QACzC,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;KACjD,CAAC;CACH,CAAC;AAEF,KAAK,uBAAuB,GAAG;IAC7B,oBAAoB,EAAE,CAAC,OAAO,EAAE;QAC9B,QAAQ,EAAE;YAAE,KAAK,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;SAAE,CAAC;QACtE,aAAa,EAAE,MAAM,EAAE,CAAC;QACxB,GAAG,CAAC,EAAE;YACJ,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,kBAAkB,CAAC,EAAE,MAAM,CAAC;SAC7B,CAAC;QACF,OAAO,CAAC,EAAE;YACR,YAAY,CAAC,EAAE,MAAM,CAAC;YACtB,YAAY,CAAC,EAAE,MAAM,CAAC;SACvB,CAAC;KACH,KAAK,6BAA6B,CAAC,kBAAkB,CAAC,CAAC;CACzD,CAAC;AAIF,wBAAsB,iCAAiC,CACrD,OAAO,EAAE,wCAAwC,GAChD,OAAO,CAAC,2BAA2B,CAAC,CAuDtC"}
|
|
@@ -8,10 +8,10 @@ export async function createCyberLoopKinematicsObserver(options) {
|
|
|
8
8
|
if (!apiKey) {
|
|
9
9
|
throw new Error('CyberLoop drift detection requires OPENAI_API_KEY or PERSONAL_OPENAI_API_KEY for embeddings.');
|
|
10
10
|
}
|
|
11
|
-
const cyberloop = await importCyberLoopAdvanced(options.moduleSpecifier ?? process.env.HEDDLE_CYBERLOOP_ADVANCED_MODULE);
|
|
11
|
+
const cyberloop = options._testOverrides?.advancedModule ?? await importCyberLoopAdvanced(options.moduleSpecifier ?? process.env.HEDDLE_CYBERLOOP_ADVANCED_MODULE);
|
|
12
12
|
const client = new OpenAI({ apiKey });
|
|
13
13
|
const embeddingModel = options.embeddingModel ?? process.env.HEDDLE_DRIFT_EMBEDDING_MODEL ?? DEFAULT_EMBEDDING_MODEL;
|
|
14
|
-
const embedText = async (text) => {
|
|
14
|
+
const embedText = options._testOverrides?.embedText ?? (async (text) => {
|
|
15
15
|
const response = await client.embeddings.create({
|
|
16
16
|
model: embeddingModel,
|
|
17
17
|
input: text,
|
|
@@ -21,15 +21,31 @@ export async function createCyberLoopKinematicsObserver(options) {
|
|
|
21
21
|
throw new Error('OpenAI embeddings response did not include an embedding vector.');
|
|
22
22
|
}
|
|
23
23
|
return embedding;
|
|
24
|
-
};
|
|
24
|
+
});
|
|
25
25
|
const frameEmbedder = createRuntimeFrameEmbedder({ embedText });
|
|
26
|
-
const
|
|
26
|
+
const referenceText = options.referenceText?.trim();
|
|
27
|
+
const goalEmbedding = await embedText(referenceText || options.goal);
|
|
27
28
|
const annotations = [];
|
|
28
29
|
const observer = createCyberLoopObserver({
|
|
30
|
+
baselineFrame: referenceText ?
|
|
31
|
+
(event) => ({
|
|
32
|
+
runId: event.runId,
|
|
33
|
+
step: 0,
|
|
34
|
+
kind: 'assistant',
|
|
35
|
+
goal: event.goal,
|
|
36
|
+
text: referenceText,
|
|
37
|
+
timestamp: event.timestamp,
|
|
38
|
+
rawEvent: event,
|
|
39
|
+
})
|
|
40
|
+
: undefined,
|
|
41
|
+
shouldObserveFrame: (frame) => frame.kind === 'assistant' || frame.kind === 'final',
|
|
29
42
|
middleware: [
|
|
30
43
|
cyberloop.kinematicsMiddleware({
|
|
31
44
|
embedder: frameEmbedder,
|
|
32
45
|
goalEmbedding,
|
|
46
|
+
pid: {
|
|
47
|
+
stabilityThreshold: options.stabilityThreshold ?? readStabilityThreshold(),
|
|
48
|
+
},
|
|
33
49
|
}),
|
|
34
50
|
],
|
|
35
51
|
onAnnotation(annotation) {
|
|
@@ -40,6 +56,14 @@ export async function createCyberLoopKinematicsObserver(options) {
|
|
|
40
56
|
});
|
|
41
57
|
return { observer, annotations };
|
|
42
58
|
}
|
|
59
|
+
function readStabilityThreshold() {
|
|
60
|
+
const raw = process.env.HEDDLE_DRIFT_STABILITY_THRESHOLD;
|
|
61
|
+
if (!raw) {
|
|
62
|
+
return 0.05;
|
|
63
|
+
}
|
|
64
|
+
const parsed = Number(raw);
|
|
65
|
+
return Number.isFinite(parsed) && parsed > 0 ? parsed : 0.05;
|
|
66
|
+
}
|
|
43
67
|
function toCyberLoopTraceEvent(annotation) {
|
|
44
68
|
return {
|
|
45
69
|
type: 'cyberloop.annotation',
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cyberloop-kinematics.js","sourceRoot":"","sources":["../../../src/integrations/cyberloop-kinematics.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EACL,uBAAuB,EACvB,0BAA0B,GAK3B,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"cyberloop-kinematics.js","sourceRoot":"","sources":["../../../src/integrations/cyberloop-kinematics.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EACL,uBAAuB,EACvB,0BAA0B,GAK3B,MAAM,gBAAgB,CAAC;AAuCxB,MAAM,uBAAuB,GAAG,wBAAwB,CAAC;AAEzD,MAAM,CAAC,KAAK,UAAU,iCAAiC,CACrD,OAAiD;IAEjD,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrH,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,8FAA8F,CAAC,CAAC;IAClH,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,cAAc,EAAE,cAAc,IAAI,MAAM,uBAAuB,CAAC,OAAO,CAAC,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IACnK,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACtC,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,uBAAuB,CAAC;IACrH,MAAM,SAAS,GAAG,OAAO,CAAC,cAAc,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE,IAAY,EAAE,EAAE;QAC7E,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YAC9C,KAAK,EAAE,cAAc;YACrB,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC;QAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QACrF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC,CAAC;IACH,MAAM,aAAa,GAAG,0BAA0B,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;IAChE,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC;IACpD,MAAM,aAAa,GAAG,MAAM,SAAS,CAAC,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACrE,MAAM,WAAW,GAAiB,EAAE,CAAC;IAErC,MAAM,QAAQ,GAAG,uBAAuB,CAAC;QACvC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC5B,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBACV,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,IAAI,EAAE,CAAC;gBACP,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,QAAQ,EAAE,KAAK;aAChB,CAAC;YACJ,CAAC,CAAC,SAAS;QACX,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;QACnF,UAAU,EAAE;YACV,SAAS,CAAC,oBAAoB,CAAC;gBAC7B,QAAQ,EAAE,aAAa;gBACvB,aAAa;gBACb,GAAG,EAAE;oBACH,kBAAkB,EAAE,OAAO,CAAC,kBAAkB,IAAI,sBAAsB,EAAE;iBAC3E;aACF,CAAC;SACH;QACD,YAAY,CAAC,UAAU;YACrB,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAC;YACpD,OAAO,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,EAAE,OAAO,CAAC,OAAO;KACzB,CAAC,CAAC;IAEH,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;AACnC,CAAC;AAED,SAAS,sBAAsB;IAC7B,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC;IACzD,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;AAC/D,CAAC;AAED,SAAS,qBAAqB,CAAC,UAAuC;IACpE,OAAO;QACL,IAAI,EAAE,sBAAsB;QAC5B,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,SAAS,EAAE,UAAU,CAAC,KAAK,CAAC,IAAI;QAChC,UAAU,EAAE,UAAU,CAAC,UAAU;QACjC,aAAa,EAAE,UAAU,CAAC,aAAa;QACvC,QAAQ,EAAE,UAAU,CAAC,QAAQ;QAC7B,SAAS,EAAE,UAAU,CAAC,SAAS;KAChC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,eAAmC;IACxE,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,CAAC,wBAAwB,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC;IACrG,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,SAAS,CAAqC,CAAC;QACxE,IAAI,OAAO,GAAG,CAAC,oBAAoB,KAAK,UAAU,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,UAAU,SAAS,wCAAwC,CAAC,CAAC;QAC/E,CAAC;QACD,OAAO,EAAE,oBAAoB,EAAE,GAAG,CAAC,oBAAoB,EAAE,CAAC;IAC5D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACtE,MAAM,IAAI,KAAK,CACb,uDAAuD,SAAS,sKAAsK,MAAM,EAAE,CAC/O,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,wBAAwB,CAAC,SAAiB;IACjD,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC3D,OAAO,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;IAChD,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAG,MAAiC;IAChE,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACtF,CAAC"}
|
|
@@ -64,6 +64,10 @@ export type CyberLoopObserverAnnotation = {
|
|
|
64
64
|
};
|
|
65
65
|
export type CreateCyberLoopObserverOptions = {
|
|
66
66
|
middleware: CyberLoopCompatibleMiddleware<HeddleRuntimeFrame>[];
|
|
67
|
+
baselineFrame?: (event: Extract<AgentLoopEvent, {
|
|
68
|
+
type: 'loop.started';
|
|
69
|
+
}>) => HeddleRuntimeFrame | undefined;
|
|
70
|
+
shouldObserveFrame?: (frame: HeddleRuntimeFrame) => boolean;
|
|
67
71
|
onAnnotation?: (annotation: CyberLoopObserverAnnotation) => void;
|
|
68
72
|
onError?: (error: unknown) => void;
|
|
69
73
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cyberloop.d.ts","sourceRoot":"","sources":["../../../src/integrations/cyberloop.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAE3D,MAAM,MAAM,yBAAyB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAEhE,MAAM,MAAM,oBAAoB,CAAC,CAAC,GAAG,OAAO,IAAI;IAC9C,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,CAAC,CAAC;IACT,SAAS,CAAC,EAAE,CAAC,CAAC;IACd,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAC5C,QAAQ,EAAE,yBAAyB,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,mBAAmB,CAAC,CAAC,GAAG,OAAO,IAAI;IAC7C,KAAK,EAAE,CAAC,CAAC;IACT,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,6BAA6B,CAAC,CAAC,GAAG,OAAO,IAAI;IACvD,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,CAAC,GAAG,EAAE;QAAE,KAAK,EAAE,OAAO,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,UAAU,CAAC,EAAE,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC;IACzF,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5F,QAAQ,CAAC,EAAE,CAAC,GAAG,EAAE;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gCAAgC,CAAC,CAAC,GAAG,OAAO,IAAI;IAC1D,KAAK,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACxC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAC9B,WAAW,GACX,MAAM,GACN,YAAY,GACZ,WAAW,GACX,OAAO,CAAC;AAEZ,MAAM,MAAM,kBAAkB,GAAG;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,sBAAsB,CAAC;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,EAAE,CAAC,EAAE,OAAO,CAAC;IACb,QAAQ,EAAE,cAAc,CAAC;CAC1B,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAExE,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,kBAAkB,CAAC;IAC1B,QAAQ,EAAE,yBAAyB,CAAC;IACpC,UAAU,EAAE,mBAAmB,CAAC;IAChC,aAAa,EAAE,OAAO,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,UAAU,EAAE,6BAA6B,CAAC,kBAAkB,CAAC,EAAE,CAAC;IAChE,YAAY,CAAC,EAAE,CAAC,UAAU,EAAE,2BAA2B,KAAK,IAAI,CAAC;IACjE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;AAEnG,MAAM,MAAM,iCAAiC,GAAG;IAC9C,SAAS,EAAE,qBAAqB,CAAC;IACjC,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,WAAW,EAAE,CAAC,KAAK,EAAE,cAAc,KAAK,IAAI,CAAC;IAC7C,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5B,CAAC;AAEF,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,iCAAiC,GACzC,gCAAgC,CAAC,kBAAkB,CAAC,CAMtD;AAED,wBAAgB,8BAA8B,CAC5C,KAAK,EAAE,kBAAkB,EACzB,OAAO,GAAE,IAAI,CAAC,iCAAiC,EAAE,aAAa,GAAG,eAAe,CAAM,GACrF,MAAM,CAYR;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,8BAA8B,GAAG,iBAAiB,
|
|
1
|
+
{"version":3,"file":"cyberloop.d.ts","sourceRoot":"","sources":["../../../src/integrations/cyberloop.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAE3D,MAAM,MAAM,yBAAyB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAEhE,MAAM,MAAM,oBAAoB,CAAC,CAAC,GAAG,OAAO,IAAI;IAC9C,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,CAAC,CAAC;IACT,SAAS,CAAC,EAAE,CAAC,CAAC;IACd,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAC5C,QAAQ,EAAE,yBAAyB,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,mBAAmB,CAAC,CAAC,GAAG,OAAO,IAAI;IAC7C,KAAK,EAAE,CAAC,CAAC;IACT,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,6BAA6B,CAAC,CAAC,GAAG,OAAO,IAAI;IACvD,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,CAAC,GAAG,EAAE;QAAE,KAAK,EAAE,OAAO,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,UAAU,CAAC,EAAE,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC;IACzF,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5F,QAAQ,CAAC,EAAE,CAAC,GAAG,EAAE;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gCAAgC,CAAC,CAAC,GAAG,OAAO,IAAI;IAC1D,KAAK,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACxC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAC9B,WAAW,GACX,MAAM,GACN,YAAY,GACZ,WAAW,GACX,OAAO,CAAC;AAEZ,MAAM,MAAM,kBAAkB,GAAG;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,sBAAsB,CAAC;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,EAAE,CAAC,EAAE,OAAO,CAAC;IACb,QAAQ,EAAE,cAAc,CAAC;CAC1B,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAExE,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,kBAAkB,CAAC;IAC1B,QAAQ,EAAE,yBAAyB,CAAC;IACpC,UAAU,EAAE,mBAAmB,CAAC;IAChC,aAAa,EAAE,OAAO,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,UAAU,EAAE,6BAA6B,CAAC,kBAAkB,CAAC,EAAE,CAAC;IAChE,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,cAAc,EAAE;QAAE,IAAI,EAAE,cAAc,CAAA;KAAE,CAAC,KAAK,kBAAkB,GAAG,SAAS,CAAC;IAC7G,kBAAkB,CAAC,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,OAAO,CAAC;IAC5D,YAAY,CAAC,EAAE,CAAC,UAAU,EAAE,2BAA2B,KAAK,IAAI,CAAC;IACjE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;AAEnG,MAAM,MAAM,iCAAiC,GAAG;IAC9C,SAAS,EAAE,qBAAqB,CAAC;IACjC,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,WAAW,EAAE,CAAC,KAAK,EAAE,cAAc,KAAK,IAAI,CAAC;IAC7C,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5B,CAAC;AAEF,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,iCAAiC,GACzC,gCAAgC,CAAC,kBAAkB,CAAC,CAMtD;AAED,wBAAgB,8BAA8B,CAC5C,KAAK,EAAE,kBAAkB,EACzB,OAAO,GAAE,IAAI,CAAC,iCAAiC,EAAE,aAAa,GAAG,eAAe,CAAM,GACrF,MAAM,CAYR;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,8BAA8B,GAAG,iBAAiB,CA+HlG;AAaD,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,cAAc,EACrB,OAAO,GAAE;IAAE,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAO,GAC9C,kBAAkB,GAAG,SAAS,CAoGhC;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,yBAAyB,GAAG,mBAAmB,CAmBxF"}
|
|
@@ -23,6 +23,7 @@ export function createCyberLoopObserver(options) {
|
|
|
23
23
|
let goal = '';
|
|
24
24
|
let initialized = false;
|
|
25
25
|
let previousFrame;
|
|
26
|
+
let observedFrameKeys = new Set();
|
|
26
27
|
const enqueue = (work) => {
|
|
27
28
|
queue = queue.then(work, work).catch((error) => {
|
|
28
29
|
options.onError?.(error);
|
|
@@ -32,10 +33,15 @@ export function createCyberLoopObserver(options) {
|
|
|
32
33
|
runId = event.runId;
|
|
33
34
|
goal = event.goal;
|
|
34
35
|
previousFrame = undefined;
|
|
36
|
+
observedFrameKeys = new Set();
|
|
35
37
|
initialized = true;
|
|
36
38
|
for (const middleware of options.middleware) {
|
|
37
39
|
await middleware.setup?.({ input: event });
|
|
38
40
|
}
|
|
41
|
+
const baselineFrame = options.baselineFrame?.(event);
|
|
42
|
+
if (baselineFrame) {
|
|
43
|
+
await observeFrame(baselineFrame, { annotate: false });
|
|
44
|
+
}
|
|
39
45
|
};
|
|
40
46
|
const teardown = async (reason) => {
|
|
41
47
|
if (!initialized) {
|
|
@@ -46,7 +52,7 @@ export function createCyberLoopObserver(options) {
|
|
|
46
52
|
}
|
|
47
53
|
initialized = false;
|
|
48
54
|
};
|
|
49
|
-
const observeFrame = async (frame) => {
|
|
55
|
+
const observeFrame = async (frame, observeOptions = {}) => {
|
|
50
56
|
if (!initialized) {
|
|
51
57
|
initialized = true;
|
|
52
58
|
runId = frame.runId;
|
|
@@ -55,6 +61,14 @@ export function createCyberLoopObserver(options) {
|
|
|
55
61
|
await middleware.setup?.({ input: frame.rawEvent });
|
|
56
62
|
}
|
|
57
63
|
}
|
|
64
|
+
if (options.shouldObserveFrame && !options.shouldObserveFrame(frame)) {
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
const frameKey = runtimeFrameKey(frame);
|
|
68
|
+
if (observedFrameKeys.has(frameKey)) {
|
|
69
|
+
return;
|
|
70
|
+
}
|
|
71
|
+
observedFrameKeys.add(frameKey);
|
|
58
72
|
let ctx = {
|
|
59
73
|
step: frame.step,
|
|
60
74
|
state: frame,
|
|
@@ -83,15 +97,17 @@ export function createCyberLoopObserver(options) {
|
|
|
83
97
|
await options.middleware[i]?.afterStep?.(ctx, result);
|
|
84
98
|
}
|
|
85
99
|
previousFrame = frame;
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
100
|
+
if (observeOptions.annotate !== false) {
|
|
101
|
+
options.onAnnotation?.({
|
|
102
|
+
runId: frame.runId,
|
|
103
|
+
step: frame.step,
|
|
104
|
+
frame,
|
|
105
|
+
metadata: ctx.metadata,
|
|
106
|
+
driftLevel: inferDriftLevel(ctx.metadata),
|
|
107
|
+
requestedHalt,
|
|
108
|
+
timestamp: new Date().toISOString(),
|
|
109
|
+
});
|
|
110
|
+
}
|
|
95
111
|
};
|
|
96
112
|
return {
|
|
97
113
|
handleEvent(event) {
|
|
@@ -114,6 +130,16 @@ export function createCyberLoopObserver(options) {
|
|
|
114
130
|
},
|
|
115
131
|
};
|
|
116
132
|
}
|
|
133
|
+
function runtimeFrameKey(frame) {
|
|
134
|
+
return [
|
|
135
|
+
frame.runId,
|
|
136
|
+
frame.step,
|
|
137
|
+
frame.kind,
|
|
138
|
+
frame.tool ?? '',
|
|
139
|
+
frame.ok === undefined ? '' : String(frame.ok),
|
|
140
|
+
frame.text,
|
|
141
|
+
].join('\u001f');
|
|
142
|
+
}
|
|
117
143
|
export function eventToRuntimeFrame(event, context = {}) {
|
|
118
144
|
const runId = 'runId' in event ? event.runId : context.runId;
|
|
119
145
|
if (!runId) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cyberloop.js","sourceRoot":"","sources":["../../../src/integrations/cyberloop.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"cyberloop.js","sourceRoot":"","sources":["../../../src/integrations/cyberloop.ts"],"names":[],"mappings":"AA6FA,MAAM,UAAU,0BAA0B,CACxC,OAA0C;IAE1C,OAAO;QACL,KAAK,CAAC,KAAK,CAAC,KAAK;YACf,OAAO,OAAO,CAAC,SAAS,CAAC,8BAA8B,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;QAClF,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,KAAyB,EACzB,UAAoF,EAAE;IAEtF,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,KAAK,CAAC;IACrD,MAAM,QAAQ,GAAG;QACf,SAAS,KAAK,CAAC,IAAI,EAAE;QACrB,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;QAC9C,OAAO,KAAK,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS;QAC7D,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;QACvD,EAAE;QACF,KAAK,CAAC,IAAI;KACX,CAAC,MAAM,CAAC,CAAC,OAAO,EAAqB,EAAE,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC;IAEhE,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,OAAuC;IAC7E,IAAI,KAAK,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAC9B,IAAI,KAAyB,CAAC;IAC9B,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,IAAI,aAA6C,CAAC;IAClD,IAAI,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAE1C,MAAM,OAAO,GAAG,CAAC,IAAyB,EAAE,EAAE;QAC5C,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACtD,OAAO,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,KAAK,GAAG,KAAK,EAAE,KAAwD,EAAE,EAAE;QAC/E,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACpB,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QAClB,aAAa,GAAG,SAAS,CAAC;QAC1B,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;QAC9B,WAAW,GAAG,IAAI,CAAC;QACnB,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC5C,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC,KAAK,CAAC,CAAC;QACrD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,YAAY,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAc,EAAE,EAAE;QACxC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC5C,MAAM,UAAU,CAAC,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC;QACD,WAAW,GAAG,KAAK,CAAC;IACtB,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,KAAK,EAAE,KAAyB,EAAE,iBAAyC,EAAE,EAAE,EAAE;QACpG,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,WAAW,GAAG,IAAI,CAAC;YACnB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YACpB,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YAClB,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBAC5C,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAED,IAAI,OAAO,CAAC,kBAAkB,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YACrE,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpC,OAAO;QACT,CAAC;QACD,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEhC,IAAI,GAAG,GAA6C;YAClD,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,aAAa;YACxB,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,MAAM,CAAC,iBAAiB,EAAE;YACjE,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,IAAI,aAAa,GAAG,KAAK,CAAC;QAC1B,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC5C,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC;YAChD,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBACpB,aAAa,GAAG,IAAI,CAAC;gBACrB,MAAM;YACR,CAAC;YACD,IAAI,IAAI,EAAE,CAAC;gBACT,GAAG,GAAG,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAA4C;YACtD,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,KAAK,CAAC,IAAI;YAClB,QAAQ,EAAE,KAAK,CAAC,EAAE;YAClB,IAAI,EAAE,CAAC;SACR,CAAC;QAEF,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACxD,CAAC;QAED,aAAa,GAAG,KAAK,CAAC;QACtB,IAAI,cAAc,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;YACtC,OAAO,CAAC,YAAY,EAAE,CAAC;gBACrB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK;gBACL,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,UAAU,EAAE,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC;gBACzC,aAAa;gBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;IAEF,OAAO;QACL,WAAW,CAAC,KAAqB;YAC/B,OAAO,CAAC,KAAK,IAAI,EAAE;gBACjB,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;oBAClC,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;oBACnB,OAAO;gBACT,CAAC;gBAED,MAAM,KAAK,GAAG,mBAAmB,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC1D,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,YAAY,CAAC,KAAK,CAAC,CAAC;gBAC5B,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;oBACnC,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QACD,KAAK;YACH,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAyB;IAChD,OAAO;QACL,KAAK,CAAC,KAAK;QACX,KAAK,CAAC,IAAI;QACV,KAAK,CAAC,IAAI;QACV,KAAK,CAAC,IAAI,IAAI,EAAE;QAChB,KAAK,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9C,KAAK,CAAC,IAAI;KACX,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,KAAqB,EACrB,UAA6C,EAAE;IAE/C,MAAM,KAAK,GAAG,OAAO,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;IAC7D,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAE/E,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QACpD,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,IAAI,EAAE,WAAW;YACjB,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;QACpC,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,IAAI,EAAE,MAAM;YACZ,IAAI;YACJ,IAAI,EAAE,uBAAuB,CAAC,KAAK,CAAC;YACpC,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE;YACnB,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;QACpE,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;YACtB,IAAI,EAAE,WAAW;YACjB,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO;YACzB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;QACjE,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;YACtB,IAAI,EAAE,MAAM;YACZ,IAAI;YACJ,IAAI,EAAE,yBAAyB,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;YACrE,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;YACtB,EAAE,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;YACzB,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;QACtC,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,IAAI,EAAE,YAAY;YAClB,IAAI;YACJ,IAAI,EAAE,wBAAwB,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE;YACtD,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;QACxC,OAAO;YACL,KAAK;YACL,IAAI,EAAE,CAAC;YACP,IAAI,EAAE,WAAW;YACjB,IAAI;YACJ,IAAI,EAAE,uBAAuB,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,OAAO,EAAE;YAC/D,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,EAAE,EAAE,KAAK,CAAC,QAAQ,KAAK,UAAU;YACjC,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;QACnC,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM;YAC9B,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;YACtB,IAAI,EAAE,KAAK,CAAC,OAAO;YACnB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,EAAE,EAAE,KAAK,CAAC,OAAO,KAAK,MAAM;YAC5B,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,QAAmC;IACjE,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAgF,CAAC;IAC3G,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAgF,CAAC;IAC/G,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAyG,CAAC;IACtI,MAAM,aAAa,GAAG,QAAQ,CAAC,oBAAoB,IAAI,IAAI,CAAC;IAE5D,IAAI,QAAQ,EAAE,UAAU,KAAK,IAAI,IAAI,YAAY,EAAE,UAAU,KAAK,IAAI,EAAE,CAAC;QACvE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,UAAU,EAAE,QAAQ,KAAK,KAAK,IAAI,aAAa,EAAE,CAAC;QACpD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,IAAI,QAAQ,IAAI,YAAY,IAAI,UAAU,EAAE,CAAC;QAC3C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,uBAAuB,CAAC,KAA0D;IACzF,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACrB,OAAO,QAAQ,KAAK,CAAC,IAAI,YAAY,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC;IAC/E,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;IACnC,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,QAAQ,KAAK,CAAC,IAAI,eAAe,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;IACnE,CAAC;IAED,OAAO,QAAQ,KAAK,CAAC,IAAI,eAAe,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;AACnF,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAY,EAAE,MAAyD;IACxG,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACf,OAAO,QAAQ,IAAI,YAAY,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC;IACnE,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACtC,OAAO,QAAQ,IAAI,eAAe,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;IACpE,CAAC;IAED,OAAO,QAAQ,IAAI,eAAe,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;AACpF,CAAC"}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
import type { AgentHeartbeatResult, HeartbeatDecision, RunAgentHeartbeatOptions } from './heartbeat.js';
|
|
2
|
+
import type { AgentLoopCheckpoint, AgentLoopState } from './events.js';
|
|
3
|
+
import type { LlmUsage } from '../llm/types.js';
|
|
4
|
+
export type HeartbeatTask = {
|
|
5
|
+
id: string;
|
|
6
|
+
task: string;
|
|
7
|
+
name?: string;
|
|
8
|
+
enabled: boolean;
|
|
9
|
+
intervalMs: number;
|
|
10
|
+
nextRunAt?: string;
|
|
11
|
+
checkpointPath?: string;
|
|
12
|
+
model?: string;
|
|
13
|
+
maxSteps?: number;
|
|
14
|
+
workspaceRoot?: string;
|
|
15
|
+
stateDir?: string;
|
|
16
|
+
memoryDir?: string;
|
|
17
|
+
searchIgnoreDirs?: string[];
|
|
18
|
+
systemContext?: string;
|
|
19
|
+
lastRunAt?: string;
|
|
20
|
+
lastDecision?: HeartbeatDecision;
|
|
21
|
+
lastOutcome?: string;
|
|
22
|
+
lastSummary?: string;
|
|
23
|
+
lastError?: string;
|
|
24
|
+
updatedAt?: string;
|
|
25
|
+
};
|
|
26
|
+
export type HeartbeatTaskRunRecord = {
|
|
27
|
+
task: HeartbeatTask;
|
|
28
|
+
result: AgentHeartbeatResult;
|
|
29
|
+
loadedCheckpoint: boolean;
|
|
30
|
+
};
|
|
31
|
+
export type HeartbeatTaskRunRecordEntry = {
|
|
32
|
+
id: string;
|
|
33
|
+
path: string;
|
|
34
|
+
taskId: string;
|
|
35
|
+
runId: string;
|
|
36
|
+
createdAt: string;
|
|
37
|
+
record: HeartbeatTaskRunRecord;
|
|
38
|
+
};
|
|
39
|
+
export type HeartbeatTaskStore = {
|
|
40
|
+
listTasks: () => Promise<HeartbeatTask[]>;
|
|
41
|
+
saveTask: (task: HeartbeatTask) => Promise<void>;
|
|
42
|
+
loadCheckpoint: (task: HeartbeatTask) => Promise<AgentLoopCheckpoint | undefined>;
|
|
43
|
+
saveCheckpoint: (task: HeartbeatTask, checkpoint: AgentLoopCheckpoint) => Promise<void>;
|
|
44
|
+
saveRunRecord?: (record: HeartbeatTaskRunRecord) => Promise<void>;
|
|
45
|
+
listRunRecords?: (options?: {
|
|
46
|
+
taskId?: string;
|
|
47
|
+
limit?: number;
|
|
48
|
+
}) => Promise<HeartbeatTaskRunRecordEntry[]>;
|
|
49
|
+
loadRunRecord?: (id: string) => Promise<HeartbeatTaskRunRecordEntry | undefined>;
|
|
50
|
+
};
|
|
51
|
+
export type FileHeartbeatTaskStoreOptions = {
|
|
52
|
+
dir: string;
|
|
53
|
+
};
|
|
54
|
+
export type HeartbeatSchedulerEvent = {
|
|
55
|
+
type: 'heartbeat.scheduler.started';
|
|
56
|
+
timestamp: string;
|
|
57
|
+
} | {
|
|
58
|
+
type: 'heartbeat.scheduler.stopped';
|
|
59
|
+
reason: 'aborted' | 'completed' | 'error';
|
|
60
|
+
timestamp: string;
|
|
61
|
+
} | {
|
|
62
|
+
type: 'heartbeat.task.due';
|
|
63
|
+
taskId: string;
|
|
64
|
+
timestamp: string;
|
|
65
|
+
} | {
|
|
66
|
+
type: 'heartbeat.task.started';
|
|
67
|
+
taskId: string;
|
|
68
|
+
loadedCheckpoint: boolean;
|
|
69
|
+
timestamp: string;
|
|
70
|
+
} | {
|
|
71
|
+
type: 'heartbeat.task.finished';
|
|
72
|
+
taskId: string;
|
|
73
|
+
decision: HeartbeatDecision;
|
|
74
|
+
outcome: string;
|
|
75
|
+
summary: string;
|
|
76
|
+
runId: string;
|
|
77
|
+
usage?: LlmUsage;
|
|
78
|
+
nextRunAt?: string;
|
|
79
|
+
enabled: boolean;
|
|
80
|
+
timestamp: string;
|
|
81
|
+
} | {
|
|
82
|
+
type: 'heartbeat.task.failed';
|
|
83
|
+
taskId: string;
|
|
84
|
+
error: string;
|
|
85
|
+
nextRunAt?: string;
|
|
86
|
+
timestamp: string;
|
|
87
|
+
};
|
|
88
|
+
export type HeartbeatTaskRunner = (task: HeartbeatTask, checkpoint: AgentLoopState | AgentLoopCheckpoint | undefined) => Promise<AgentHeartbeatResult>;
|
|
89
|
+
export type RunDueHeartbeatTasksOptions = {
|
|
90
|
+
store: HeartbeatTaskStore;
|
|
91
|
+
runner?: HeartbeatTaskRunner;
|
|
92
|
+
heartbeat?: Omit<RunAgentHeartbeatOptions, 'task' | 'checkpoint'>;
|
|
93
|
+
now?: () => Date;
|
|
94
|
+
onEvent?: (event: HeartbeatSchedulerEvent) => void;
|
|
95
|
+
failureRetryMs?: number;
|
|
96
|
+
};
|
|
97
|
+
export type RunDueHeartbeatTasksResult = {
|
|
98
|
+
checked: number;
|
|
99
|
+
ran: number;
|
|
100
|
+
failed: number;
|
|
101
|
+
records: HeartbeatTaskRunRecord[];
|
|
102
|
+
};
|
|
103
|
+
export type RunHeartbeatSchedulerOptions = RunDueHeartbeatTasksOptions & {
|
|
104
|
+
pollIntervalMs?: number;
|
|
105
|
+
signal?: AbortSignal;
|
|
106
|
+
sleep?: (ms: number, signal?: AbortSignal) => Promise<void>;
|
|
107
|
+
};
|
|
108
|
+
export declare function createFileHeartbeatTaskStore(options: FileHeartbeatTaskStoreOptions): HeartbeatTaskStore;
|
|
109
|
+
export declare function runDueHeartbeatTasks(options: RunDueHeartbeatTasksOptions): Promise<RunDueHeartbeatTasksResult>;
|
|
110
|
+
export declare function runHeartbeatScheduler(options: RunHeartbeatSchedulerOptions): Promise<void>;
|
|
111
|
+
//# sourceMappingURL=heartbeat-scheduler.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"heartbeat-scheduler.d.ts","sourceRoot":"","sources":["../../../src/runtime/heartbeat-scheduler.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC;AACxG,OAAO,KAAK,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACvE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAGhD,MAAM,MAAM,aAAa,GAAG;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,iBAAiB,CAAC;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE,aAAa,CAAC;IACpB,MAAM,EAAE,oBAAoB,CAAC;IAC7B,gBAAgB,EAAE,OAAO,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,sBAAsB,CAAC;CAChC,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,SAAS,EAAE,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IAC1C,QAAQ,EAAE,CAAC,IAAI,EAAE,aAAa,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACjD,cAAc,EAAE,CAAC,IAAI,EAAE,aAAa,KAAK,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;IAClF,cAAc,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxF,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,sBAAsB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAClE,cAAc,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,2BAA2B,EAAE,CAAC,CAAC;IAC3G,aAAa,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,2BAA2B,GAAG,SAAS,CAAC,CAAC;CAClF,CAAC;AAEF,MAAM,MAAM,6BAA6B,GAAG;IAC1C,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAC/B;IAAE,IAAI,EAAE,6BAA6B,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GAC1D;IAAE,IAAI,EAAE,6BAA6B,CAAC;IAAC,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,OAAO,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACrG;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACjE;IAAE,IAAI,EAAE,wBAAwB,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,gBAAgB,EAAE,OAAO,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GAChG;IACE,IAAI,EAAE,yBAAyB,CAAC;IAChC,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,QAAQ,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,uBAAuB,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC;AAE5G,MAAM,MAAM,mBAAmB,GAAG,CAChC,IAAI,EAAE,aAAa,EACnB,UAAU,EAAE,cAAc,GAAG,mBAAmB,GAAG,SAAS,KACzD,OAAO,CAAC,oBAAoB,CAAC,CAAC;AAEnC,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,CAAC,EAAE,mBAAmB,CAAC;IAC7B,SAAS,CAAC,EAAE,IAAI,CAAC,wBAAwB,EAAE,MAAM,GAAG,YAAY,CAAC,CAAC;IAClE,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,uBAAuB,KAAK,IAAI,CAAC;IACnD,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACvC,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,sBAAsB,EAAE,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,2BAA2B,GAAG;IACvE,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7D,CAAC;AAIF,wBAAgB,4BAA4B,CAAC,OAAO,EAAE,6BAA6B,GAAG,kBAAkB,CAqEvG;AAED,wBAAsB,oBAAoB,CAAC,OAAO,EAAE,2BAA2B,GAAG,OAAO,CAAC,0BAA0B,CAAC,CA0DpH;AAED,wBAAsB,qBAAqB,CAAC,OAAO,EAAE,4BAA4B,GAAG,OAAO,CAAC,IAAI,CAAC,CAiBhG"}
|
|
@@ -0,0 +1,236 @@
|
|
|
1
|
+
import { existsSync, mkdirSync, readdirSync, readFileSync, writeFileSync } from 'node:fs';
|
|
2
|
+
import { basename, dirname, join } from 'node:path';
|
|
3
|
+
import { runAgentHeartbeat } from './heartbeat.js';
|
|
4
|
+
import { suggestNextHeartbeatDelayMs } from './heartbeat-store.js';
|
|
5
|
+
const DEFAULT_FAILURE_RETRY_MS = 5 * 60_000;
|
|
6
|
+
export function createFileHeartbeatTaskStore(options) {
|
|
7
|
+
const tasksDir = join(options.dir, 'tasks');
|
|
8
|
+
const checkpointsDir = join(options.dir, 'checkpoints');
|
|
9
|
+
const runsDir = join(options.dir, 'runs');
|
|
10
|
+
return {
|
|
11
|
+
async listTasks() {
|
|
12
|
+
if (!existsSync(tasksDir)) {
|
|
13
|
+
return [];
|
|
14
|
+
}
|
|
15
|
+
return readdirSync(tasksDir)
|
|
16
|
+
.filter((entry) => entry.endsWith('.json'))
|
|
17
|
+
.map((entry) => JSON.parse(readFileSync(join(tasksDir, entry), 'utf8')))
|
|
18
|
+
.sort((left, right) => left.id.localeCompare(right.id));
|
|
19
|
+
},
|
|
20
|
+
async saveTask(task) {
|
|
21
|
+
const path = join(tasksDir, `${safeTaskFileName(task.id)}.json`);
|
|
22
|
+
mkdirSync(dirname(path), { recursive: true });
|
|
23
|
+
writeFileSync(path, JSON.stringify(normalizeTaskForSave(task), null, 2));
|
|
24
|
+
},
|
|
25
|
+
async loadCheckpoint(task) {
|
|
26
|
+
const path = checkpointPathForTask(task, checkpointsDir);
|
|
27
|
+
if (!existsSync(path)) {
|
|
28
|
+
return undefined;
|
|
29
|
+
}
|
|
30
|
+
return JSON.parse(readFileSync(path, 'utf8'));
|
|
31
|
+
},
|
|
32
|
+
async saveCheckpoint(task, checkpoint) {
|
|
33
|
+
const path = checkpointPathForTask(task, checkpointsDir);
|
|
34
|
+
mkdirSync(dirname(path), { recursive: true });
|
|
35
|
+
writeFileSync(path, JSON.stringify(checkpoint, null, 2));
|
|
36
|
+
},
|
|
37
|
+
async saveRunRecord(record) {
|
|
38
|
+
const timestamp = new Date().toISOString().replaceAll(':', '-');
|
|
39
|
+
const path = join(runsDir, `${timestamp}-${safeTaskFileName(record.task.id)}.json`);
|
|
40
|
+
mkdirSync(dirname(path), { recursive: true });
|
|
41
|
+
writeFileSync(path, JSON.stringify(record, null, 2));
|
|
42
|
+
},
|
|
43
|
+
async listRunRecords(options = {}) {
|
|
44
|
+
if (!existsSync(runsDir)) {
|
|
45
|
+
return [];
|
|
46
|
+
}
|
|
47
|
+
const entries = readdirSync(runsDir)
|
|
48
|
+
.filter((entry) => entry.endsWith('.json'))
|
|
49
|
+
.flatMap((entry) => {
|
|
50
|
+
const path = join(runsDir, entry);
|
|
51
|
+
try {
|
|
52
|
+
const record = JSON.parse(readFileSync(path, 'utf8'));
|
|
53
|
+
if (options.taskId && record.task.id !== options.taskId) {
|
|
54
|
+
return [];
|
|
55
|
+
}
|
|
56
|
+
return [runRecordEntryFromPath(path, record)];
|
|
57
|
+
}
|
|
58
|
+
catch {
|
|
59
|
+
return [];
|
|
60
|
+
}
|
|
61
|
+
})
|
|
62
|
+
.sort((left, right) => right.createdAt.localeCompare(left.createdAt));
|
|
63
|
+
return options.limit ? entries.slice(0, options.limit) : entries;
|
|
64
|
+
},
|
|
65
|
+
async loadRunRecord(id) {
|
|
66
|
+
const entries = await this.listRunRecords?.();
|
|
67
|
+
return entries?.find((entry) => entry.id === id || entry.runId === id);
|
|
68
|
+
},
|
|
69
|
+
};
|
|
70
|
+
}
|
|
71
|
+
export async function runDueHeartbeatTasks(options) {
|
|
72
|
+
const now = options.now?.() ?? new Date();
|
|
73
|
+
const tasks = await options.store.listTasks();
|
|
74
|
+
const dueTasks = tasks.filter((task) => isTaskDue(task, now));
|
|
75
|
+
const records = [];
|
|
76
|
+
let failed = 0;
|
|
77
|
+
for (const task of dueTasks) {
|
|
78
|
+
options.onEvent?.({ type: 'heartbeat.task.due', taskId: task.id, timestamp: now.toISOString() });
|
|
79
|
+
try {
|
|
80
|
+
const checkpoint = await options.store.loadCheckpoint(task);
|
|
81
|
+
options.onEvent?.({
|
|
82
|
+
type: 'heartbeat.task.started',
|
|
83
|
+
taskId: task.id,
|
|
84
|
+
loadedCheckpoint: Boolean(checkpoint),
|
|
85
|
+
timestamp: now.toISOString(),
|
|
86
|
+
});
|
|
87
|
+
const result = await runHeartbeatTask(task, checkpoint, options);
|
|
88
|
+
const record = { task, result, loadedCheckpoint: Boolean(checkpoint) };
|
|
89
|
+
await options.store.saveCheckpoint(task, result.checkpoint);
|
|
90
|
+
await options.store.saveRunRecord?.(record);
|
|
91
|
+
records.push(record);
|
|
92
|
+
const nextTask = updateTaskAfterResult(task, result, now);
|
|
93
|
+
await options.store.saveTask(nextTask);
|
|
94
|
+
options.onEvent?.({
|
|
95
|
+
type: 'heartbeat.task.finished',
|
|
96
|
+
taskId: task.id,
|
|
97
|
+
decision: result.decision,
|
|
98
|
+
outcome: result.state.outcome,
|
|
99
|
+
summary: result.summary,
|
|
100
|
+
runId: result.state.runId,
|
|
101
|
+
usage: result.state.usage,
|
|
102
|
+
nextRunAt: nextTask.nextRunAt,
|
|
103
|
+
enabled: nextTask.enabled,
|
|
104
|
+
timestamp: now.toISOString(),
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
catch (error) {
|
|
108
|
+
failed++;
|
|
109
|
+
const nextTask = updateTaskAfterFailure(task, error, now, options.failureRetryMs ?? DEFAULT_FAILURE_RETRY_MS);
|
|
110
|
+
await options.store.saveTask(nextTask);
|
|
111
|
+
options.onEvent?.({
|
|
112
|
+
type: 'heartbeat.task.failed',
|
|
113
|
+
taskId: task.id,
|
|
114
|
+
error: error instanceof Error ? error.message : String(error),
|
|
115
|
+
nextRunAt: nextTask.nextRunAt,
|
|
116
|
+
timestamp: now.toISOString(),
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
return {
|
|
121
|
+
checked: tasks.length,
|
|
122
|
+
ran: records.length,
|
|
123
|
+
failed,
|
|
124
|
+
records,
|
|
125
|
+
};
|
|
126
|
+
}
|
|
127
|
+
export async function runHeartbeatScheduler(options) {
|
|
128
|
+
options.onEvent?.({ type: 'heartbeat.scheduler.started', timestamp: (options.now?.() ?? new Date()).toISOString() });
|
|
129
|
+
try {
|
|
130
|
+
while (!options.signal?.aborted) {
|
|
131
|
+
await runDueHeartbeatTasks(options);
|
|
132
|
+
await (options.sleep ?? sleep)(options.pollIntervalMs ?? 60_000, options.signal);
|
|
133
|
+
}
|
|
134
|
+
options.onEvent?.({ type: 'heartbeat.scheduler.stopped', reason: 'aborted', timestamp: (options.now?.() ?? new Date()).toISOString() });
|
|
135
|
+
}
|
|
136
|
+
catch (error) {
|
|
137
|
+
if (options.signal?.aborted) {
|
|
138
|
+
options.onEvent?.({ type: 'heartbeat.scheduler.stopped', reason: 'aborted', timestamp: (options.now?.() ?? new Date()).toISOString() });
|
|
139
|
+
return;
|
|
140
|
+
}
|
|
141
|
+
options.onEvent?.({ type: 'heartbeat.scheduler.stopped', reason: 'error', timestamp: (options.now?.() ?? new Date()).toISOString() });
|
|
142
|
+
throw error;
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
function isTaskDue(task, now) {
|
|
146
|
+
if (!task.enabled) {
|
|
147
|
+
return false;
|
|
148
|
+
}
|
|
149
|
+
if (!task.nextRunAt) {
|
|
150
|
+
return true;
|
|
151
|
+
}
|
|
152
|
+
const nextRunAt = Date.parse(task.nextRunAt);
|
|
153
|
+
return Number.isFinite(nextRunAt) && nextRunAt <= now.getTime();
|
|
154
|
+
}
|
|
155
|
+
async function runHeartbeatTask(task, checkpoint, options) {
|
|
156
|
+
if (options.runner) {
|
|
157
|
+
return options.runner(task, checkpoint);
|
|
158
|
+
}
|
|
159
|
+
return runAgentHeartbeat({
|
|
160
|
+
...options.heartbeat,
|
|
161
|
+
task: task.task,
|
|
162
|
+
checkpoint,
|
|
163
|
+
model: task.model ?? options.heartbeat?.model,
|
|
164
|
+
maxSteps: task.maxSteps ?? options.heartbeat?.maxSteps,
|
|
165
|
+
workspaceRoot: task.workspaceRoot ?? options.heartbeat?.workspaceRoot,
|
|
166
|
+
stateDir: task.stateDir ?? options.heartbeat?.stateDir,
|
|
167
|
+
memoryDir: task.memoryDir ?? options.heartbeat?.memoryDir,
|
|
168
|
+
searchIgnoreDirs: task.searchIgnoreDirs ?? options.heartbeat?.searchIgnoreDirs,
|
|
169
|
+
systemContext: task.systemContext ?? options.heartbeat?.systemContext,
|
|
170
|
+
});
|
|
171
|
+
}
|
|
172
|
+
function updateTaskAfterResult(task, result, now) {
|
|
173
|
+
const terminal = result.decision === 'complete' || result.decision === 'escalate';
|
|
174
|
+
const delayMs = terminal ? undefined
|
|
175
|
+
: result.decision === 'continue' ? task.intervalMs
|
|
176
|
+
: suggestNextHeartbeatDelayMs(result.decision) ?? task.intervalMs;
|
|
177
|
+
return normalizeTaskForSave({
|
|
178
|
+
...task,
|
|
179
|
+
enabled: terminal ? false : task.enabled,
|
|
180
|
+
nextRunAt: delayMs === undefined ? undefined : new Date(now.getTime() + delayMs).toISOString(),
|
|
181
|
+
lastRunAt: now.toISOString(),
|
|
182
|
+
lastDecision: result.decision,
|
|
183
|
+
lastOutcome: result.state.outcome,
|
|
184
|
+
lastSummary: result.summary,
|
|
185
|
+
lastError: undefined,
|
|
186
|
+
updatedAt: now.toISOString(),
|
|
187
|
+
});
|
|
188
|
+
}
|
|
189
|
+
function updateTaskAfterFailure(task, error, now, retryMs) {
|
|
190
|
+
return normalizeTaskForSave({
|
|
191
|
+
...task,
|
|
192
|
+
nextRunAt: new Date(now.getTime() + retryMs).toISOString(),
|
|
193
|
+
lastRunAt: now.toISOString(),
|
|
194
|
+
lastError: error instanceof Error ? error.message : String(error),
|
|
195
|
+
updatedAt: now.toISOString(),
|
|
196
|
+
});
|
|
197
|
+
}
|
|
198
|
+
function normalizeTaskForSave(task) {
|
|
199
|
+
return {
|
|
200
|
+
...task,
|
|
201
|
+
intervalMs: Math.max(1, Math.trunc(task.intervalMs)),
|
|
202
|
+
};
|
|
203
|
+
}
|
|
204
|
+
function checkpointPathForTask(task, checkpointsDir) {
|
|
205
|
+
return task.checkpointPath ?? join(checkpointsDir, `${safeTaskFileName(task.id)}.json`);
|
|
206
|
+
}
|
|
207
|
+
function safeTaskFileName(id) {
|
|
208
|
+
if (!/^[a-zA-Z0-9._-]+$/.test(id)) {
|
|
209
|
+
throw new Error(`Invalid heartbeat task id "${id}". Use only letters, numbers, dots, underscores, and hyphens.`);
|
|
210
|
+
}
|
|
211
|
+
return id;
|
|
212
|
+
}
|
|
213
|
+
function runRecordEntryFromPath(path, record) {
|
|
214
|
+
const id = basename(path, '.json');
|
|
215
|
+
return {
|
|
216
|
+
id,
|
|
217
|
+
path,
|
|
218
|
+
taskId: record.task.id,
|
|
219
|
+
runId: record.result.state.runId,
|
|
220
|
+
createdAt: record.result.state.finishedAt,
|
|
221
|
+
record,
|
|
222
|
+
};
|
|
223
|
+
}
|
|
224
|
+
function sleep(ms, signal) {
|
|
225
|
+
if (signal?.aborted) {
|
|
226
|
+
return Promise.resolve();
|
|
227
|
+
}
|
|
228
|
+
return new Promise((resolve) => {
|
|
229
|
+
const timeout = setTimeout(resolve, ms);
|
|
230
|
+
signal?.addEventListener('abort', () => {
|
|
231
|
+
clearTimeout(timeout);
|
|
232
|
+
resolve();
|
|
233
|
+
}, { once: true });
|
|
234
|
+
});
|
|
235
|
+
}
|
|
236
|
+
//# sourceMappingURL=heartbeat-scheduler.js.map
|