jfl 0.5.0 → 0.6.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/dist/commands/context-hub.d.ts +1 -0
- package/dist/commands/context-hub.d.ts.map +1 -1
- package/dist/commands/context-hub.js +246 -2
- package/dist/commands/context-hub.js.map +1 -1
- package/dist/commands/peter.d.ts +2 -0
- package/dist/commands/peter.d.ts.map +1 -1
- package/dist/commands/peter.js +242 -52
- package/dist/commands/peter.js.map +1 -1
- package/dist/commands/setup.d.ts +12 -0
- package/dist/commands/setup.d.ts.map +1 -0
- package/dist/commands/setup.js +322 -0
- package/dist/commands/setup.js.map +1 -0
- package/dist/commands/train.d.ts +33 -0
- package/dist/commands/train.d.ts.map +1 -0
- package/dist/commands/train.js +510 -0
- package/dist/commands/train.js.map +1 -0
- package/dist/commands/verify.d.ts +14 -0
- package/dist/commands/verify.d.ts.map +1 -0
- package/dist/commands/verify.js +276 -0
- package/dist/commands/verify.js.map +1 -0
- package/dist/dashboard-static/assets/index-CW9ZxqX8.css +1 -0
- package/dist/dashboard-static/assets/index-DNN__p4K.js +121 -0
- package/dist/dashboard-static/index.html +2 -2
- package/dist/index.js +99 -3
- package/dist/index.js.map +1 -1
- package/dist/lib/agent-session.d.ts.map +1 -1
- package/dist/lib/agent-session.js +12 -4
- package/dist/lib/agent-session.js.map +1 -1
- package/dist/lib/eval-snapshot.js +1 -1
- package/dist/lib/eval-snapshot.js.map +1 -1
- package/dist/lib/pi-sky/bridge.d.ts +55 -0
- package/dist/lib/pi-sky/bridge.d.ts.map +1 -0
- package/dist/lib/pi-sky/bridge.js +264 -0
- package/dist/lib/pi-sky/bridge.js.map +1 -0
- package/dist/lib/pi-sky/cost-monitor.d.ts +21 -0
- package/dist/lib/pi-sky/cost-monitor.d.ts.map +1 -0
- package/dist/lib/pi-sky/cost-monitor.js +126 -0
- package/dist/lib/pi-sky/cost-monitor.js.map +1 -0
- package/dist/lib/pi-sky/eval-sweep.d.ts +27 -0
- package/dist/lib/pi-sky/eval-sweep.d.ts.map +1 -0
- package/dist/lib/pi-sky/eval-sweep.js +141 -0
- package/dist/lib/pi-sky/eval-sweep.js.map +1 -0
- package/dist/lib/pi-sky/event-router.d.ts +32 -0
- package/dist/lib/pi-sky/event-router.d.ts.map +1 -0
- package/dist/lib/pi-sky/event-router.js +176 -0
- package/dist/lib/pi-sky/event-router.js.map +1 -0
- package/dist/lib/pi-sky/experiment.d.ts +9 -0
- package/dist/lib/pi-sky/experiment.d.ts.map +1 -0
- package/dist/lib/pi-sky/experiment.js +83 -0
- package/dist/lib/pi-sky/experiment.js.map +1 -0
- package/dist/lib/pi-sky/index.d.ts +16 -0
- package/dist/lib/pi-sky/index.d.ts.map +1 -0
- package/dist/lib/pi-sky/index.js +16 -0
- package/dist/lib/pi-sky/index.js.map +1 -0
- package/dist/lib/pi-sky/stratus-gate.d.ts +28 -0
- package/dist/lib/pi-sky/stratus-gate.d.ts.map +1 -0
- package/dist/lib/pi-sky/stratus-gate.js +61 -0
- package/dist/lib/pi-sky/stratus-gate.js.map +1 -0
- package/dist/lib/pi-sky/swarm.d.ts +28 -0
- package/dist/lib/pi-sky/swarm.d.ts.map +1 -0
- package/dist/lib/pi-sky/swarm.js +208 -0
- package/dist/lib/pi-sky/swarm.js.map +1 -0
- package/dist/lib/pi-sky/types.d.ts +139 -0
- package/dist/lib/pi-sky/types.d.ts.map +1 -0
- package/dist/lib/pi-sky/types.js +2 -0
- package/dist/lib/pi-sky/types.js.map +1 -0
- package/dist/lib/pi-sky/voice-bridge.d.ts +20 -0
- package/dist/lib/pi-sky/voice-bridge.d.ts.map +1 -0
- package/dist/lib/pi-sky/voice-bridge.js +91 -0
- package/dist/lib/pi-sky/voice-bridge.js.map +1 -0
- package/dist/lib/policy-head.d.ts +16 -1
- package/dist/lib/policy-head.d.ts.map +1 -1
- package/dist/lib/policy-head.js +117 -19
- package/dist/lib/policy-head.js.map +1 -1
- package/dist/lib/predictor.d.ts +10 -0
- package/dist/lib/predictor.d.ts.map +1 -1
- package/dist/lib/predictor.js +46 -7
- package/dist/lib/predictor.js.map +1 -1
- package/dist/lib/setup/agent-generator.d.ts +18 -0
- package/dist/lib/setup/agent-generator.d.ts.map +1 -0
- package/dist/lib/setup/agent-generator.js +114 -0
- package/dist/lib/setup/agent-generator.js.map +1 -0
- package/dist/lib/setup/context-analyzer.d.ts +16 -0
- package/dist/lib/setup/context-analyzer.d.ts.map +1 -0
- package/dist/lib/setup/context-analyzer.js +112 -0
- package/dist/lib/setup/context-analyzer.js.map +1 -0
- package/dist/lib/setup/doc-auditor.d.ts +54 -0
- package/dist/lib/setup/doc-auditor.d.ts.map +1 -0
- package/dist/lib/setup/doc-auditor.js +629 -0
- package/dist/lib/setup/doc-auditor.js.map +1 -0
- package/dist/lib/setup/domain-generator.d.ts +7 -0
- package/dist/lib/setup/domain-generator.d.ts.map +1 -0
- package/dist/lib/setup/domain-generator.js +58 -0
- package/dist/lib/setup/domain-generator.js.map +1 -0
- package/dist/lib/setup/smart-eval-generator.d.ts +38 -0
- package/dist/lib/setup/smart-eval-generator.d.ts.map +1 -0
- package/dist/lib/setup/smart-eval-generator.js +378 -0
- package/dist/lib/setup/smart-eval-generator.js.map +1 -0
- package/dist/lib/setup/smart-recommender.d.ts +63 -0
- package/dist/lib/setup/smart-recommender.d.ts.map +1 -0
- package/dist/lib/setup/smart-recommender.js +329 -0
- package/dist/lib/setup/smart-recommender.js.map +1 -0
- package/dist/lib/setup/spec-generator.d.ts +63 -0
- package/dist/lib/setup/spec-generator.d.ts.map +1 -0
- package/dist/lib/setup/spec-generator.js +310 -0
- package/dist/lib/setup/spec-generator.js.map +1 -0
- package/dist/lib/setup/violation-agent-generator.d.ts +32 -0
- package/dist/lib/setup/violation-agent-generator.d.ts.map +1 -0
- package/dist/lib/setup/violation-agent-generator.js +255 -0
- package/dist/lib/setup/violation-agent-generator.js.map +1 -0
- package/package.json +1 -1
- package/packages/pi/extensions/context.ts +88 -55
- package/packages/pi/extensions/hub-resolver.ts +63 -0
- package/packages/pi/extensions/index.ts +16 -3
- package/packages/pi/extensions/memory-tool.ts +9 -4
- package/packages/pi/extensions/session.ts +68 -16
- package/packages/pi/extensions/tool-renderers.ts +23 -8
- package/scripts/train/requirements.txt +5 -0
- package/scripts/train/train-policy-head.py +477 -0
- package/scripts/train/v2/dataset.py +81 -0
- package/scripts/train/v2/domain.json +18 -0
- package/scripts/train/v2/eval.py +196 -0
- package/scripts/train/v2/generate_data.py +219 -0
- package/scripts/train/v2/infer.py +188 -0
- package/scripts/train/v2/model.py +112 -0
- package/scripts/train/v2/precompute.py +132 -0
- package/scripts/train/v2/train.py +302 -0
- package/scripts/train/v2/transform_buffer.py +227 -0
- package/scripts/train/v2/validate_data.py +115 -0
- package/template/.claude/settings.json +2 -15
- package/template/scripts/session/session-cleanup.sh +2 -11
- package/template/scripts/session/session-end-hub.sh +72 -0
- package/template/scripts/session/session-start-hub.sh +105 -0
- package/dist/dashboard-static/assets/index-B6b867Pv.js +0 -121
- package/dist/dashboard-static/assets/index-Y4BrqxV-.css +0 -1
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
import { EventEmitter } from "events";
|
|
2
|
+
import type { PiRpcCommand, PiRpcResponse, BridgeOptions } from "./types.js";
|
|
3
|
+
export declare class PiRpcBridge extends EventEmitter {
|
|
4
|
+
private proc;
|
|
5
|
+
private pending;
|
|
6
|
+
private options;
|
|
7
|
+
private buffer;
|
|
8
|
+
private decoder;
|
|
9
|
+
private _started;
|
|
10
|
+
private _exited;
|
|
11
|
+
constructor(options?: BridgeOptions);
|
|
12
|
+
get started(): boolean;
|
|
13
|
+
get exited(): boolean;
|
|
14
|
+
get pid(): number | undefined;
|
|
15
|
+
start(): Promise<void>;
|
|
16
|
+
shutdown(): Promise<void>;
|
|
17
|
+
prompt(message: string, options?: {
|
|
18
|
+
images?: unknown[];
|
|
19
|
+
streamingBehavior?: "steer" | "followUp";
|
|
20
|
+
}): Promise<PiRpcResponse>;
|
|
21
|
+
steer(message: string, images?: unknown[]): Promise<PiRpcResponse>;
|
|
22
|
+
followUp(message: string, images?: unknown[]): Promise<PiRpcResponse>;
|
|
23
|
+
abort(): Promise<PiRpcResponse>;
|
|
24
|
+
newSession(parentSession?: string): Promise<PiRpcResponse>;
|
|
25
|
+
getState(): Promise<PiRpcResponse>;
|
|
26
|
+
getMessages(): Promise<PiRpcResponse>;
|
|
27
|
+
setModel(provider: string, modelId: string): Promise<PiRpcResponse>;
|
|
28
|
+
cycleModel(): Promise<PiRpcResponse>;
|
|
29
|
+
getAvailableModels(): Promise<PiRpcResponse>;
|
|
30
|
+
setThinkingLevel(level: "off" | "minimal" | "low" | "medium" | "high" | "xhigh"): Promise<PiRpcResponse>;
|
|
31
|
+
cycleThinkingLevel(): Promise<PiRpcResponse>;
|
|
32
|
+
setSteeringMode(mode: "all" | "one-at-a-time"): Promise<PiRpcResponse>;
|
|
33
|
+
setFollowUpMode(mode: "all" | "one-at-a-time"): Promise<PiRpcResponse>;
|
|
34
|
+
compact(customInstructions?: string): Promise<PiRpcResponse>;
|
|
35
|
+
setAutoCompaction(enabled: boolean): Promise<PiRpcResponse>;
|
|
36
|
+
setAutoRetry(enabled: boolean): Promise<PiRpcResponse>;
|
|
37
|
+
abortRetry(): Promise<PiRpcResponse>;
|
|
38
|
+
bash(command: string): Promise<PiRpcResponse>;
|
|
39
|
+
abortBash(): Promise<PiRpcResponse>;
|
|
40
|
+
getSessionStats(): Promise<PiRpcResponse>;
|
|
41
|
+
exportHtml(outputPath?: string): Promise<PiRpcResponse>;
|
|
42
|
+
switchSession(sessionPath: string): Promise<PiRpcResponse>;
|
|
43
|
+
fork(entryId: string): Promise<PiRpcResponse>;
|
|
44
|
+
getForkMessages(): Promise<PiRpcResponse>;
|
|
45
|
+
getLastAssistantText(): Promise<string | null>;
|
|
46
|
+
setSessionName(name: string): Promise<PiRpcResponse>;
|
|
47
|
+
getCommands(): Promise<PiRpcResponse>;
|
|
48
|
+
respondToExtensionUi(id: string, payload: Record<string, unknown>): Promise<void>;
|
|
49
|
+
send(cmd: PiRpcCommand, timeoutMs?: number): Promise<PiRpcResponse>;
|
|
50
|
+
private write;
|
|
51
|
+
private handleLine;
|
|
52
|
+
private attachReader;
|
|
53
|
+
private rejectAllPending;
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=bridge.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bridge.d.ts","sourceRoot":"","sources":["../../../src/lib/pi-sky/bridge.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AAErC,OAAO,KAAK,EACV,YAAY,EACZ,aAAa,EAEb,aAAa,EAId,MAAM,YAAY,CAAA;AAQnB,qBAAa,WAAY,SAAQ,YAAY;IAC3C,OAAO,CAAC,IAAI,CAA4B;IACxC,OAAO,CAAC,OAAO,CAAoC;IACnD,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,MAAM,CAAK;IACnB,OAAO,CAAC,OAAO,CAA4B;IAC3C,OAAO,CAAC,QAAQ,CAAQ;IACxB,OAAO,CAAC,OAAO,CAAQ;gBAEX,OAAO,GAAE,aAAkB;IAKvC,IAAI,OAAO,IAAI,OAAO,CAAyB;IAC/C,IAAI,MAAM,IAAI,OAAO,CAAwB;IAC7C,IAAI,GAAG,IAAI,MAAM,GAAG,SAAS,CAA0B;IAEjD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA4CtB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAoBzB,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC;QAAC,iBAAiB,CAAC,EAAE,OAAO,GAAG,UAAU,CAAA;KAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAO3H,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAMlE,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAMrE,KAAK,IAAI,OAAO,CAAC,aAAa,CAAC;IAI/B,UAAU,CAAC,aAAa,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAM1D,QAAQ,IAAI,OAAO,CAAC,aAAa,CAAC;IAIlC,WAAW,IAAI,OAAO,CAAC,aAAa,CAAC;IAIrC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAInE,UAAU,IAAI,OAAO,CAAC,aAAa,CAAC;IAIpC,kBAAkB,IAAI,OAAO,CAAC,aAAa,CAAC;IAI5C,gBAAgB,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC;IAIxG,kBAAkB,IAAI,OAAO,CAAC,aAAa,CAAC;IAI5C,eAAe,CAAC,IAAI,EAAE,KAAK,GAAG,eAAe,GAAG,OAAO,CAAC,aAAa,CAAC;IAItE,eAAe,CAAC,IAAI,EAAE,KAAK,GAAG,eAAe,GAAG,OAAO,CAAC,aAAa,CAAC;IAItE,OAAO,CAAC,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAM5D,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC;IAI3D,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC;IAItD,UAAU,IAAI,OAAO,CAAC,aAAa,CAAC;IAIpC,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAI7C,SAAS,IAAI,OAAO,CAAC,aAAa,CAAC;IAInC,eAAe,IAAI,OAAO,CAAC,aAAa,CAAC;IAIzC,UAAU,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAMvD,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAI1D,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAI7C,eAAe,IAAI,OAAO,CAAC,aAAa,CAAC;IAIzC,oBAAoB,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAM9C,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAIpD,WAAW,IAAI,OAAO,CAAC,aAAa,CAAC;IAIrC,oBAAoB,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjF,IAAI,CAAC,GAAG,EAAE,YAAY,EAAE,SAAS,SAAU,GAAG,OAAO,CAAC,aAAa,CAAC;IAiB1E,OAAO,CAAC,KAAK;IAKb,OAAO,CAAC,UAAU;IAyBlB,OAAO,CAAC,YAAY;IAuBpB,OAAO,CAAC,gBAAgB;CAOzB"}
|
|
@@ -0,0 +1,264 @@
|
|
|
1
|
+
import { spawn } from "child_process";
|
|
2
|
+
import { StringDecoder } from "string_decoder";
|
|
3
|
+
import { EventEmitter } from "events";
|
|
4
|
+
import { randomUUID } from "crypto";
|
|
5
|
+
export class PiRpcBridge extends EventEmitter {
|
|
6
|
+
proc = null;
|
|
7
|
+
pending = new Map();
|
|
8
|
+
options;
|
|
9
|
+
buffer = "";
|
|
10
|
+
decoder = new StringDecoder("utf8");
|
|
11
|
+
_started = false;
|
|
12
|
+
_exited = false;
|
|
13
|
+
constructor(options = {}) {
|
|
14
|
+
super();
|
|
15
|
+
this.options = options;
|
|
16
|
+
}
|
|
17
|
+
get started() { return this._started; }
|
|
18
|
+
get exited() { return this._exited; }
|
|
19
|
+
get pid() { return this.proc?.pid; }
|
|
20
|
+
async start() {
|
|
21
|
+
if (this._started)
|
|
22
|
+
throw new Error("Bridge already started");
|
|
23
|
+
const args = ["--mode", "rpc"];
|
|
24
|
+
if (this.options.extensionPath)
|
|
25
|
+
args.push("--extension", this.options.extensionPath);
|
|
26
|
+
if (this.options.skillsPath)
|
|
27
|
+
args.push("--skill", this.options.skillsPath);
|
|
28
|
+
if (this.options.themePath)
|
|
29
|
+
args.push("--theme", this.options.themePath);
|
|
30
|
+
if (this.options.yolo)
|
|
31
|
+
args.push("--yolo");
|
|
32
|
+
if (this.options.noSession)
|
|
33
|
+
args.push("--no-session");
|
|
34
|
+
if (this.options.sessionDir)
|
|
35
|
+
args.push("--session-dir", this.options.sessionDir);
|
|
36
|
+
if (this.options.provider)
|
|
37
|
+
args.push("--provider", this.options.provider);
|
|
38
|
+
if (this.options.model)
|
|
39
|
+
args.push("--model", this.options.model);
|
|
40
|
+
const env = { ...process.env, ...this.options.env };
|
|
41
|
+
this.proc = spawn("pi", args, {
|
|
42
|
+
cwd: this.options.cwd ?? process.cwd(),
|
|
43
|
+
stdio: ["pipe", "pipe", "pipe"],
|
|
44
|
+
env,
|
|
45
|
+
});
|
|
46
|
+
this.proc.on("exit", (code, signal) => {
|
|
47
|
+
this._exited = true;
|
|
48
|
+
this.rejectAllPending(new Error(`Pi process exited: code=${code} signal=${signal}`));
|
|
49
|
+
this.emit("exit", { code, signal });
|
|
50
|
+
});
|
|
51
|
+
this.proc.on("error", (err) => {
|
|
52
|
+
this._exited = true;
|
|
53
|
+
this.rejectAllPending(err);
|
|
54
|
+
this.emit("error", err);
|
|
55
|
+
});
|
|
56
|
+
this.attachReader(this.proc.stdout, (line) => this.handleLine(line));
|
|
57
|
+
this.proc.stderr?.on("data", (chunk) => {
|
|
58
|
+
const text = chunk.toString();
|
|
59
|
+
if (text.trim())
|
|
60
|
+
this.emit("stderr", text);
|
|
61
|
+
});
|
|
62
|
+
this._started = true;
|
|
63
|
+
}
|
|
64
|
+
async shutdown() {
|
|
65
|
+
if (!this.proc || this._exited)
|
|
66
|
+
return;
|
|
67
|
+
await this.send({ type: "abort" }).catch(() => { });
|
|
68
|
+
return new Promise((resolve) => {
|
|
69
|
+
const timer = setTimeout(() => {
|
|
70
|
+
this.proc?.kill("SIGKILL");
|
|
71
|
+
resolve();
|
|
72
|
+
}, 5000);
|
|
73
|
+
this.proc.on("exit", () => {
|
|
74
|
+
clearTimeout(timer);
|
|
75
|
+
resolve();
|
|
76
|
+
});
|
|
77
|
+
this.proc.kill("SIGTERM");
|
|
78
|
+
});
|
|
79
|
+
}
|
|
80
|
+
async prompt(message, options) {
|
|
81
|
+
const cmd = { type: "prompt", message };
|
|
82
|
+
if (options?.images)
|
|
83
|
+
cmd.images = options.images;
|
|
84
|
+
if (options?.streamingBehavior)
|
|
85
|
+
cmd.streamingBehavior = options.streamingBehavior;
|
|
86
|
+
return this.send(cmd);
|
|
87
|
+
}
|
|
88
|
+
async steer(message, images) {
|
|
89
|
+
const cmd = { type: "steer", message };
|
|
90
|
+
if (images)
|
|
91
|
+
cmd.images = images;
|
|
92
|
+
return this.send(cmd);
|
|
93
|
+
}
|
|
94
|
+
async followUp(message, images) {
|
|
95
|
+
const cmd = { type: "follow_up", message };
|
|
96
|
+
if (images)
|
|
97
|
+
cmd.images = images;
|
|
98
|
+
return this.send(cmd);
|
|
99
|
+
}
|
|
100
|
+
async abort() {
|
|
101
|
+
return this.send({ type: "abort" });
|
|
102
|
+
}
|
|
103
|
+
async newSession(parentSession) {
|
|
104
|
+
const cmd = { type: "new_session" };
|
|
105
|
+
if (parentSession)
|
|
106
|
+
cmd.parentSession = parentSession;
|
|
107
|
+
return this.send(cmd);
|
|
108
|
+
}
|
|
109
|
+
async getState() {
|
|
110
|
+
return this.send({ type: "get_state" });
|
|
111
|
+
}
|
|
112
|
+
async getMessages() {
|
|
113
|
+
return this.send({ type: "get_messages" });
|
|
114
|
+
}
|
|
115
|
+
async setModel(provider, modelId) {
|
|
116
|
+
return this.send({ type: "set_model", provider, modelId });
|
|
117
|
+
}
|
|
118
|
+
async cycleModel() {
|
|
119
|
+
return this.send({ type: "cycle_model" });
|
|
120
|
+
}
|
|
121
|
+
async getAvailableModels() {
|
|
122
|
+
return this.send({ type: "get_available_models" });
|
|
123
|
+
}
|
|
124
|
+
async setThinkingLevel(level) {
|
|
125
|
+
return this.send({ type: "set_thinking_level", level });
|
|
126
|
+
}
|
|
127
|
+
async cycleThinkingLevel() {
|
|
128
|
+
return this.send({ type: "cycle_thinking_level" });
|
|
129
|
+
}
|
|
130
|
+
async setSteeringMode(mode) {
|
|
131
|
+
return this.send({ type: "set_steering_mode", mode });
|
|
132
|
+
}
|
|
133
|
+
async setFollowUpMode(mode) {
|
|
134
|
+
return this.send({ type: "set_follow_up_mode", mode });
|
|
135
|
+
}
|
|
136
|
+
async compact(customInstructions) {
|
|
137
|
+
const cmd = { type: "compact" };
|
|
138
|
+
if (customInstructions)
|
|
139
|
+
cmd.customInstructions = customInstructions;
|
|
140
|
+
return this.send(cmd);
|
|
141
|
+
}
|
|
142
|
+
async setAutoCompaction(enabled) {
|
|
143
|
+
return this.send({ type: "set_auto_compaction", enabled });
|
|
144
|
+
}
|
|
145
|
+
async setAutoRetry(enabled) {
|
|
146
|
+
return this.send({ type: "set_auto_retry", enabled });
|
|
147
|
+
}
|
|
148
|
+
async abortRetry() {
|
|
149
|
+
return this.send({ type: "abort_retry" });
|
|
150
|
+
}
|
|
151
|
+
async bash(command) {
|
|
152
|
+
return this.send({ type: "bash", command });
|
|
153
|
+
}
|
|
154
|
+
async abortBash() {
|
|
155
|
+
return this.send({ type: "abort_bash" });
|
|
156
|
+
}
|
|
157
|
+
async getSessionStats() {
|
|
158
|
+
return this.send({ type: "get_session_stats" });
|
|
159
|
+
}
|
|
160
|
+
async exportHtml(outputPath) {
|
|
161
|
+
const cmd = { type: "export_html" };
|
|
162
|
+
if (outputPath)
|
|
163
|
+
cmd.outputPath = outputPath;
|
|
164
|
+
return this.send(cmd);
|
|
165
|
+
}
|
|
166
|
+
async switchSession(sessionPath) {
|
|
167
|
+
return this.send({ type: "switch_session", sessionPath });
|
|
168
|
+
}
|
|
169
|
+
async fork(entryId) {
|
|
170
|
+
return this.send({ type: "fork", entryId });
|
|
171
|
+
}
|
|
172
|
+
async getForkMessages() {
|
|
173
|
+
return this.send({ type: "get_fork_messages" });
|
|
174
|
+
}
|
|
175
|
+
async getLastAssistantText() {
|
|
176
|
+
const resp = await this.send({ type: "get_last_assistant_text" });
|
|
177
|
+
if (!resp.success)
|
|
178
|
+
return null;
|
|
179
|
+
return resp.data?.text ?? null;
|
|
180
|
+
}
|
|
181
|
+
async setSessionName(name) {
|
|
182
|
+
return this.send({ type: "set_session_name", name });
|
|
183
|
+
}
|
|
184
|
+
async getCommands() {
|
|
185
|
+
return this.send({ type: "get_commands" });
|
|
186
|
+
}
|
|
187
|
+
async respondToExtensionUi(id, payload) {
|
|
188
|
+
this.write({ type: "extension_ui_response", id, ...payload });
|
|
189
|
+
}
|
|
190
|
+
async send(cmd, timeoutMs = 120_000) {
|
|
191
|
+
if (!this.proc || this._exited)
|
|
192
|
+
throw new Error("Bridge not running");
|
|
193
|
+
const id = cmd.id ?? randomUUID();
|
|
194
|
+
cmd.id = id;
|
|
195
|
+
return new Promise((resolve, reject) => {
|
|
196
|
+
const timer = timeoutMs > 0 ? setTimeout(() => {
|
|
197
|
+
this.pending.delete(id);
|
|
198
|
+
reject(new Error(`RPC timeout after ${timeoutMs}ms: ${cmd.type}`));
|
|
199
|
+
}, timeoutMs) : undefined;
|
|
200
|
+
this.pending.set(id, { resolve, reject, timer });
|
|
201
|
+
this.write(cmd);
|
|
202
|
+
});
|
|
203
|
+
}
|
|
204
|
+
write(obj) {
|
|
205
|
+
if (!this.proc?.stdin?.writable)
|
|
206
|
+
throw new Error("stdin not writable");
|
|
207
|
+
this.proc.stdin.write(JSON.stringify(obj) + "\n");
|
|
208
|
+
}
|
|
209
|
+
handleLine(line) {
|
|
210
|
+
if (!line.trim())
|
|
211
|
+
return;
|
|
212
|
+
let parsed;
|
|
213
|
+
try {
|
|
214
|
+
parsed = JSON.parse(line);
|
|
215
|
+
}
|
|
216
|
+
catch {
|
|
217
|
+
this.emit("parse_error", line);
|
|
218
|
+
return;
|
|
219
|
+
}
|
|
220
|
+
if (parsed.type === "response" && parsed.id) {
|
|
221
|
+
const pending = this.pending.get(parsed.id);
|
|
222
|
+
if (pending) {
|
|
223
|
+
this.pending.delete(parsed.id);
|
|
224
|
+
if (pending.timer)
|
|
225
|
+
clearTimeout(pending.timer);
|
|
226
|
+
pending.resolve(parsed);
|
|
227
|
+
return;
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
this.emit(parsed.type, parsed);
|
|
231
|
+
this.emit("raw", parsed);
|
|
232
|
+
}
|
|
233
|
+
attachReader(stream, onLine) {
|
|
234
|
+
stream.on("data", (chunk) => {
|
|
235
|
+
this.buffer += typeof chunk === "string" ? chunk : this.decoder.write(chunk);
|
|
236
|
+
while (true) {
|
|
237
|
+
const idx = this.buffer.indexOf("\n");
|
|
238
|
+
if (idx === -1)
|
|
239
|
+
break;
|
|
240
|
+
let line = this.buffer.slice(0, idx);
|
|
241
|
+
this.buffer = this.buffer.slice(idx + 1);
|
|
242
|
+
if (line.endsWith("\r"))
|
|
243
|
+
line = line.slice(0, -1);
|
|
244
|
+
onLine(line);
|
|
245
|
+
}
|
|
246
|
+
});
|
|
247
|
+
stream.on("end", () => {
|
|
248
|
+
this.buffer += this.decoder.end();
|
|
249
|
+
if (this.buffer.length > 0) {
|
|
250
|
+
onLine(this.buffer.endsWith("\r") ? this.buffer.slice(0, -1) : this.buffer);
|
|
251
|
+
this.buffer = "";
|
|
252
|
+
}
|
|
253
|
+
});
|
|
254
|
+
}
|
|
255
|
+
rejectAllPending(err) {
|
|
256
|
+
for (const [id, pending] of this.pending) {
|
|
257
|
+
if (pending.timer)
|
|
258
|
+
clearTimeout(pending.timer);
|
|
259
|
+
pending.reject(err);
|
|
260
|
+
}
|
|
261
|
+
this.pending.clear();
|
|
262
|
+
}
|
|
263
|
+
}
|
|
264
|
+
//# sourceMappingURL=bridge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bridge.js","sourceRoot":"","sources":["../../../src/lib/pi-sky/bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,eAAe,CAAA;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAA;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AAiBnC,MAAM,OAAO,WAAY,SAAQ,YAAY;IACnC,IAAI,GAAwB,IAAI,CAAA;IAChC,OAAO,GAAG,IAAI,GAAG,EAA0B,CAAA;IAC3C,OAAO,CAAe;IACtB,MAAM,GAAG,EAAE,CAAA;IACX,OAAO,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAA;IACnC,QAAQ,GAAG,KAAK,CAAA;IAChB,OAAO,GAAG,KAAK,CAAA;IAEvB,YAAY,UAAyB,EAAE;QACrC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;IACxB,CAAC;IAED,IAAI,OAAO,KAAc,OAAO,IAAI,CAAC,QAAQ,CAAA,CAAC,CAAC;IAC/C,IAAI,MAAM,KAAc,OAAO,IAAI,CAAC,OAAO,CAAA,CAAC,CAAC;IAC7C,IAAI,GAAG,KAAyB,OAAO,IAAI,CAAC,IAAI,EAAE,GAAG,CAAA,CAAC,CAAC;IAEvD,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAA;QAE5D,MAAM,IAAI,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;QAE9B,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa;YAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;QACpF,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU;YAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;QAC1E,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QACxE,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI;YAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC1C,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QACrD,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU;YAAE,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;QAChF,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ;YAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QACzE,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAEhE,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;QAEnD,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE;YAC5B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE;YACtC,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;YAC/B,GAAG;SACJ,CAAC,CAAA;QAEF,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YACpC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;YACnB,IAAI,CAAC,gBAAgB,CAAC,IAAI,KAAK,CAAC,2BAA2B,IAAI,WAAW,MAAM,EAAE,CAAC,CAAC,CAAA;YACpF,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;QACrC,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YAC5B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;YACnB,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAA;YAC1B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,MAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAA;QAErE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YAC7C,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAA;YAC7B,IAAI,IAAI,CAAC,IAAI,EAAE;gBAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO;YAAE,OAAM;QAEtC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAElD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;gBAC1B,OAAO,EAAE,CAAA;YACX,CAAC,EAAE,IAAI,CAAC,CAAA;YAER,IAAI,CAAC,IAAK,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACzB,YAAY,CAAC,KAAK,CAAC,CAAA;gBACnB,OAAO,EAAE,CAAA;YACX,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,IAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC5B,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,OAA0E;QACtG,MAAM,GAAG,GAAiB,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAA;QACrD,IAAI,OAAO,EAAE,MAAM;YAAE,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAA;QAChD,IAAI,OAAO,EAAE,iBAAiB;YAAE,GAAG,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAA;QACjF,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAAe,EAAE,MAAkB;QAC7C,MAAM,GAAG,GAAiB,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAA;QACpD,IAAI,MAAM;YAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAA;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAe,EAAE,MAAkB;QAChD,MAAM,GAAG,GAAiB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAA;QACxD,IAAI,MAAM;YAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAA;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;IACrC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,aAAsB;QACrC,MAAM,GAAG,GAAiB,EAAE,IAAI,EAAE,aAAa,EAAE,CAAA;QACjD,IAAI,aAAa;YAAE,GAAG,CAAC,aAAa,GAAG,aAAa,CAAA;QACpD,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAA;IACzC,CAAC;IAED,KAAK,CAAC,WAAW;QACf,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAA;IAC5C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,OAAe;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAA;IAC5D,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,kBAAkB;QACtB,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAA;IACpD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAA8D;QACnF,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAAC,CAAA;IACzD,CAAC;IAED,KAAK,CAAC,kBAAkB;QACtB,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAA;IACpD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAA6B;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAA;IACvD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAA6B;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,CAAC,CAAA;IACxD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,kBAA2B;QACvC,MAAM,GAAG,GAAiB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAA;QAC7C,IAAI,kBAAkB;YAAE,GAAG,CAAC,kBAAkB,GAAG,kBAAkB,CAAA;QACnE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,OAAgB;QACtC,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAA;IAC5D,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAAgB;QACjC,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,CAAC,CAAA;IACvD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAe;QACxB,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAA;IAC7C,CAAC;IAED,KAAK,CAAC,SAAS;QACb,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAA;IAC1C,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAA;IACjD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,UAAmB;QAClC,MAAM,GAAG,GAAiB,EAAE,IAAI,EAAE,aAAa,EAAE,CAAA;QACjD,IAAI,UAAU;YAAE,GAAG,CAAC,UAAU,GAAG,UAAU,CAAA;QAC3C,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,WAAmB;QACrC,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,CAAC,CAAA;IAC3D,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAe;QACxB,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAA;IAC7C,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAA;IACjD,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC,CAAA;QACjE,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAA;QAC9B,OAAQ,IAAI,CAAC,IAAY,EAAE,IAAI,IAAI,IAAI,CAAA;IACzC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAAY;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,WAAW;QACf,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAA;IAC5C,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,EAAU,EAAE,OAAgC;QACrE,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,CAAC,CAAA;IAC/D,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,GAAiB,EAAE,SAAS,GAAG,OAAO;QAC/C,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAA;QAErE,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,IAAI,UAAU,EAAE,CAAA;QACjC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAA;QAEX,OAAO,IAAI,OAAO,CAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACpD,MAAM,KAAK,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE;gBAC5C,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;gBACvB,MAAM,CAAC,IAAI,KAAK,CAAC,qBAAqB,SAAS,OAAO,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;YACpE,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAEzB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;YAChD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;QACjB,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,KAAK,CAAC,GAA4B;QACxC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAA;QACtE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAA;IACnD,CAAC;IAEO,UAAU,CAAC,IAAY;QAC7B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,OAAM;QAExB,IAAI,MAA+B,CAAA;QACnC,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC3B,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,CAAA;YAC9B,OAAM;QACR,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAY,CAAC,CAAA;YACrD,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAY,CAAC,CAAA;gBACxC,IAAI,OAAO,CAAC,KAAK;oBAAE,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;gBAC9C,OAAO,CAAC,OAAO,CAAC,MAAkC,CAAC,CAAA;gBACnD,OAAM;YACR,CAAC;QACH,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAc,EAAE,MAAM,CAAC,CAAA;QACxC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;IAC1B,CAAC;IAEO,YAAY,CAAC,MAA6B,EAAE,MAA8B;QAChF,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YAClC,IAAI,CAAC,MAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;YAE5E,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;gBACrC,IAAI,GAAG,KAAK,CAAC,CAAC;oBAAE,MAAK;gBACrB,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA;gBACpC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAA;gBACxC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAAE,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;gBACjD,MAAM,CAAC,IAAI,CAAC,CAAA;YACd,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;YACpB,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;YACjC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;gBAC3E,IAAI,CAAC,MAAM,GAAG,EAAE,CAAA;YAClB,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,gBAAgB,CAAC,GAAU;QACjC,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,OAAO,CAAC,KAAK;gBAAE,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;YAC9C,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;QACrB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;IACtB,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { EventEmitter } from "events";
|
|
2
|
+
import type { PiRpcBridge } from "./bridge.js";
|
|
3
|
+
import type { CostBudgetConfig } from "./types.js";
|
|
4
|
+
export declare class CostMonitor extends EventEmitter {
|
|
5
|
+
private config;
|
|
6
|
+
private bridges;
|
|
7
|
+
private costs;
|
|
8
|
+
private downgraded;
|
|
9
|
+
constructor(config?: Partial<CostBudgetConfig>);
|
|
10
|
+
get totalCost(): number;
|
|
11
|
+
get budget(): number;
|
|
12
|
+
get remaining(): number;
|
|
13
|
+
registerBridge(name: string, bridge: PiRpcBridge): void;
|
|
14
|
+
unregisterBridge(name: string): void;
|
|
15
|
+
checkCost(agentName: string): Promise<void>;
|
|
16
|
+
checkCriticalPath(agentName: string, promptText: string): Promise<boolean>;
|
|
17
|
+
checkAll(): Promise<void>;
|
|
18
|
+
private downgrade;
|
|
19
|
+
private upgrade;
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=cost-monitor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-monitor.d.ts","sourceRoot":"","sources":["../../../src/lib/pi-sky/cost-monitor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AAC9C,OAAO,KAAK,EAAE,gBAAgB,EAAgB,MAAM,YAAY,CAAA;AAWhE,qBAAa,WAAY,SAAQ,YAAY;IAC3C,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,OAAO,CAAiC;IAChD,OAAO,CAAC,KAAK,CAA4B;IACzC,OAAO,CAAC,UAAU,CAAoB;gBAE1B,MAAM,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC;IAK9C,IAAI,SAAS,IAAI,MAAM,CAItB;IAED,IAAI,MAAM,IAAI,MAAM,CAA+B;IACnD,IAAI,SAAS,IAAI,MAAM,CAA6D;IAEpF,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,IAAI;IASvD,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAM9B,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAyB3C,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAiB1E,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;YAMjB,SAAS;YAuBT,OAAO;CAqBtB"}
|
|
@@ -0,0 +1,126 @@
|
|
|
1
|
+
import { EventEmitter } from "events";
|
|
2
|
+
const DEFAULT_CONFIG = {
|
|
3
|
+
maxCost: 5.0,
|
|
4
|
+
downgradeModel: { provider: "anthropic", modelId: "claude-haiku-4-5-20251001" },
|
|
5
|
+
downgradeThinkingLevel: "low",
|
|
6
|
+
upgradeModel: { provider: "anthropic", modelId: "claude-sonnet-4-20250514" },
|
|
7
|
+
upgradeThinkingLevel: "medium",
|
|
8
|
+
criticalKeywords: ["security", "auth", "payment", "migration", "deploy", "production"],
|
|
9
|
+
};
|
|
10
|
+
export class CostMonitor extends EventEmitter {
|
|
11
|
+
config;
|
|
12
|
+
bridges = new Map();
|
|
13
|
+
costs = new Map();
|
|
14
|
+
downgraded = new Set();
|
|
15
|
+
constructor(config) {
|
|
16
|
+
super();
|
|
17
|
+
this.config = { ...DEFAULT_CONFIG, ...config };
|
|
18
|
+
}
|
|
19
|
+
get totalCost() {
|
|
20
|
+
let sum = 0;
|
|
21
|
+
for (const c of this.costs.values())
|
|
22
|
+
sum += c;
|
|
23
|
+
return sum;
|
|
24
|
+
}
|
|
25
|
+
get budget() { return this.config.maxCost; }
|
|
26
|
+
get remaining() { return Math.max(0, this.config.maxCost - this.totalCost); }
|
|
27
|
+
registerBridge(name, bridge) {
|
|
28
|
+
this.bridges.set(name, bridge);
|
|
29
|
+
this.costs.set(name, 0);
|
|
30
|
+
bridge.on("agent_end", async () => {
|
|
31
|
+
await this.checkCost(name);
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
unregisterBridge(name) {
|
|
35
|
+
this.bridges.delete(name);
|
|
36
|
+
this.costs.delete(name);
|
|
37
|
+
this.downgraded.delete(name);
|
|
38
|
+
}
|
|
39
|
+
async checkCost(agentName) {
|
|
40
|
+
const bridge = this.bridges.get(agentName);
|
|
41
|
+
if (!bridge || bridge.exited)
|
|
42
|
+
return;
|
|
43
|
+
try {
|
|
44
|
+
const resp = await bridge.getSessionStats();
|
|
45
|
+
if (!resp.success)
|
|
46
|
+
return;
|
|
47
|
+
const stats = resp.data;
|
|
48
|
+
this.costs.set(agentName, stats.cost);
|
|
49
|
+
this.emit("cost_update", {
|
|
50
|
+
agent: agentName,
|
|
51
|
+
cost: stats.cost,
|
|
52
|
+
totalCost: this.totalCost,
|
|
53
|
+
remaining: this.remaining,
|
|
54
|
+
});
|
|
55
|
+
if (this.totalCost >= this.config.maxCost && !this.downgraded.has(agentName)) {
|
|
56
|
+
await this.downgrade(agentName, bridge);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
catch {
|
|
60
|
+
// agent may have exited
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
async checkCriticalPath(agentName, promptText) {
|
|
64
|
+
if (!this.config.criticalKeywords?.length)
|
|
65
|
+
return false;
|
|
66
|
+
const lower = promptText.toLowerCase();
|
|
67
|
+
const isCritical = this.config.criticalKeywords.some(kw => lower.includes(kw));
|
|
68
|
+
if (isCritical && this.downgraded.has(agentName)) {
|
|
69
|
+
const bridge = this.bridges.get(agentName);
|
|
70
|
+
if (bridge && !bridge.exited) {
|
|
71
|
+
await this.upgrade(agentName, bridge);
|
|
72
|
+
return true;
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
return isCritical;
|
|
76
|
+
}
|
|
77
|
+
async checkAll() {
|
|
78
|
+
for (const name of this.bridges.keys()) {
|
|
79
|
+
await this.checkCost(name);
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
async downgrade(name, bridge) {
|
|
83
|
+
const { downgradeModel, downgradeThinkingLevel } = this.config;
|
|
84
|
+
if (!downgradeModel)
|
|
85
|
+
return;
|
|
86
|
+
try {
|
|
87
|
+
await bridge.setModel(downgradeModel.provider, downgradeModel.modelId);
|
|
88
|
+
if (downgradeThinkingLevel) {
|
|
89
|
+
await bridge.setThinkingLevel(downgradeThinkingLevel);
|
|
90
|
+
}
|
|
91
|
+
this.downgraded.add(name);
|
|
92
|
+
this.emit("downgrade", {
|
|
93
|
+
agent: name,
|
|
94
|
+
model: downgradeModel.modelId,
|
|
95
|
+
thinking: downgradeThinkingLevel,
|
|
96
|
+
totalCost: this.totalCost,
|
|
97
|
+
budget: this.config.maxCost,
|
|
98
|
+
});
|
|
99
|
+
}
|
|
100
|
+
catch {
|
|
101
|
+
// model may not be available
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
async upgrade(name, bridge) {
|
|
105
|
+
const { upgradeModel, upgradeThinkingLevel } = this.config;
|
|
106
|
+
if (!upgradeModel)
|
|
107
|
+
return;
|
|
108
|
+
try {
|
|
109
|
+
await bridge.setModel(upgradeModel.provider, upgradeModel.modelId);
|
|
110
|
+
if (upgradeThinkingLevel) {
|
|
111
|
+
await bridge.setThinkingLevel(upgradeThinkingLevel);
|
|
112
|
+
}
|
|
113
|
+
this.downgraded.delete(name);
|
|
114
|
+
this.emit("upgrade", {
|
|
115
|
+
agent: name,
|
|
116
|
+
model: upgradeModel.modelId,
|
|
117
|
+
thinking: upgradeThinkingLevel,
|
|
118
|
+
reason: "critical_path",
|
|
119
|
+
});
|
|
120
|
+
}
|
|
121
|
+
catch {
|
|
122
|
+
// model may not be available
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
//# sourceMappingURL=cost-monitor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-monitor.js","sourceRoot":"","sources":["../../../src/lib/pi-sky/cost-monitor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AAIrC,MAAM,cAAc,GAAqB;IACvC,OAAO,EAAE,GAAG;IACZ,cAAc,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,2BAA2B,EAAE;IAC/E,sBAAsB,EAAE,KAAK;IAC7B,YAAY,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,0BAA0B,EAAE;IAC5E,oBAAoB,EAAE,QAAQ;IAC9B,gBAAgB,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,CAAC;CACvF,CAAA;AAED,MAAM,OAAO,WAAY,SAAQ,YAAY;IACnC,MAAM,CAAkB;IACxB,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAA;IACxC,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAA;IACjC,UAAU,GAAG,IAAI,GAAG,EAAU,CAAA;IAEtC,YAAY,MAAkC;QAC5C,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAA;IAChD,CAAC;IAED,IAAI,SAAS;QACX,IAAI,GAAG,GAAG,CAAC,CAAA;QACX,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE;YAAE,GAAG,IAAI,CAAC,CAAA;QAC7C,OAAO,GAAG,CAAA;IACZ,CAAC;IAED,IAAI,MAAM,KAAa,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAA,CAAC,CAAC;IACnD,IAAI,SAAS,KAAa,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAA,CAAC,CAAC;IAEpF,cAAc,CAAC,IAAY,EAAE,MAAmB;QAC9C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;QAC9B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;QAEvB,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE;YAChC,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;QAC5B,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,gBAAgB,CAAC,IAAY;QAC3B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QACzB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QACvB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAiB;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QAC1C,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM;YAAE,OAAM;QAEpC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,eAAe,EAAE,CAAA;YAC3C,IAAI,CAAC,IAAI,CAAC,OAAO;gBAAE,OAAM;YACzB,MAAM,KAAK,GAAG,IAAI,CAAC,IAAoB,CAAA;YACvC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,CAAA;YAErC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;gBACvB,KAAK,EAAE,SAAS;gBAChB,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAA;YAEF,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7E,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;YACzC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,SAAiB,EAAE,UAAkB;QAC3D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,MAAM;YAAE,OAAO,KAAK,CAAA;QAEvD,MAAM,KAAK,GAAG,UAAU,CAAC,WAAW,EAAE,CAAA;QACtC,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAE9E,IAAI,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACjD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;YAC1C,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;gBACrC,OAAO,IAAI,CAAA;YACb,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAA;IACnB,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACvC,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;QAC5B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,MAAmB;QACvD,MAAM,EAAE,cAAc,EAAE,sBAAsB,EAAE,GAAG,IAAI,CAAC,MAAM,CAAA;QAC9D,IAAI,CAAC,cAAc;YAAE,OAAM;QAE3B,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,cAAc,CAAC,OAAO,CAAC,CAAA;YACtE,IAAI,sBAAsB,EAAE,CAAC;gBAC3B,MAAM,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,CAAA;YACvD,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;YAEzB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBACrB,KAAK,EAAE,IAAI;gBACX,KAAK,EAAE,cAAc,CAAC,OAAO;gBAC7B,QAAQ,EAAE,sBAAsB;gBAChC,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aAC5B,CAAC,CAAA;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,6BAA6B;QAC/B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,MAAmB;QACrD,MAAM,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,IAAI,CAAC,MAAM,CAAA;QAC1D,IAAI,CAAC,YAAY;YAAE,OAAM;QAEzB,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,YAAY,CAAC,OAAO,CAAC,CAAA;YAClE,IAAI,oBAAoB,EAAE,CAAC;gBACzB,MAAM,MAAM,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,CAAA;YACrD,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAE5B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;gBACnB,KAAK,EAAE,IAAI;gBACX,KAAK,EAAE,YAAY,CAAC,OAAO;gBAC3B,QAAQ,EAAE,oBAAoB;gBAC9B,MAAM,EAAE,eAAe;aACxB,CAAC,CAAA;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,6BAA6B;QAC/B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import { EventEmitter } from "events";
|
|
2
|
+
import type { BridgeOptions, SessionStats } from "./types.js";
|
|
3
|
+
interface SweepResult {
|
|
4
|
+
service: string;
|
|
5
|
+
path: string;
|
|
6
|
+
success: boolean;
|
|
7
|
+
output?: string;
|
|
8
|
+
stats?: SessionStats;
|
|
9
|
+
error?: string;
|
|
10
|
+
durationMs: number;
|
|
11
|
+
}
|
|
12
|
+
interface EvalSweepOptions {
|
|
13
|
+
bridgeOptions?: Partial<BridgeOptions>;
|
|
14
|
+
concurrency?: number;
|
|
15
|
+
evalCommand?: string;
|
|
16
|
+
hubUrl?: string;
|
|
17
|
+
}
|
|
18
|
+
export declare class EvalSweep extends EventEmitter {
|
|
19
|
+
private options;
|
|
20
|
+
constructor(options?: EvalSweepOptions);
|
|
21
|
+
sweep(projectRoot: string): Promise<SweepResult[]>;
|
|
22
|
+
private loadServices;
|
|
23
|
+
private evalService;
|
|
24
|
+
private postResults;
|
|
25
|
+
}
|
|
26
|
+
export {};
|
|
27
|
+
//# sourceMappingURL=eval-sweep.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"eval-sweep.d.ts","sourceRoot":"","sources":["../../../src/lib/pi-sky/eval-sweep.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AAIrC,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAQ7D,UAAU,WAAW;IACnB,OAAO,EAAE,MAAM,CAAA;IACf,IAAI,EAAE,MAAM,CAAA;IACZ,OAAO,EAAE,OAAO,CAAA;IAChB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,KAAK,CAAC,EAAE,YAAY,CAAA;IACpB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,UAAU,EAAE,MAAM,CAAA;CACnB;AAED,UAAU,gBAAgB;IACxB,aAAa,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAA;IACtC,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,MAAM,CAAC,EAAE,MAAM,CAAA;CAChB;AAED,qBAAa,SAAU,SAAQ,YAAY;IACzC,OAAO,CAAC,OAAO,CAAkB;gBAErB,OAAO,GAAE,gBAAqB;IASpC,KAAK,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IA6BxD,OAAO,CAAC,YAAY;YAqBN,WAAW;YAkEX,WAAW;CA0B1B"}
|