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.
Files changed (135) hide show
  1. package/dist/commands/context-hub.d.ts +1 -0
  2. package/dist/commands/context-hub.d.ts.map +1 -1
  3. package/dist/commands/context-hub.js +246 -2
  4. package/dist/commands/context-hub.js.map +1 -1
  5. package/dist/commands/peter.d.ts +2 -0
  6. package/dist/commands/peter.d.ts.map +1 -1
  7. package/dist/commands/peter.js +242 -52
  8. package/dist/commands/peter.js.map +1 -1
  9. package/dist/commands/setup.d.ts +12 -0
  10. package/dist/commands/setup.d.ts.map +1 -0
  11. package/dist/commands/setup.js +322 -0
  12. package/dist/commands/setup.js.map +1 -0
  13. package/dist/commands/train.d.ts +33 -0
  14. package/dist/commands/train.d.ts.map +1 -0
  15. package/dist/commands/train.js +510 -0
  16. package/dist/commands/train.js.map +1 -0
  17. package/dist/commands/verify.d.ts +14 -0
  18. package/dist/commands/verify.d.ts.map +1 -0
  19. package/dist/commands/verify.js +276 -0
  20. package/dist/commands/verify.js.map +1 -0
  21. package/dist/dashboard-static/assets/index-CW9ZxqX8.css +1 -0
  22. package/dist/dashboard-static/assets/index-DNN__p4K.js +121 -0
  23. package/dist/dashboard-static/index.html +2 -2
  24. package/dist/index.js +99 -3
  25. package/dist/index.js.map +1 -1
  26. package/dist/lib/agent-session.d.ts.map +1 -1
  27. package/dist/lib/agent-session.js +12 -4
  28. package/dist/lib/agent-session.js.map +1 -1
  29. package/dist/lib/eval-snapshot.js +1 -1
  30. package/dist/lib/eval-snapshot.js.map +1 -1
  31. package/dist/lib/pi-sky/bridge.d.ts +55 -0
  32. package/dist/lib/pi-sky/bridge.d.ts.map +1 -0
  33. package/dist/lib/pi-sky/bridge.js +264 -0
  34. package/dist/lib/pi-sky/bridge.js.map +1 -0
  35. package/dist/lib/pi-sky/cost-monitor.d.ts +21 -0
  36. package/dist/lib/pi-sky/cost-monitor.d.ts.map +1 -0
  37. package/dist/lib/pi-sky/cost-monitor.js +126 -0
  38. package/dist/lib/pi-sky/cost-monitor.js.map +1 -0
  39. package/dist/lib/pi-sky/eval-sweep.d.ts +27 -0
  40. package/dist/lib/pi-sky/eval-sweep.d.ts.map +1 -0
  41. package/dist/lib/pi-sky/eval-sweep.js +141 -0
  42. package/dist/lib/pi-sky/eval-sweep.js.map +1 -0
  43. package/dist/lib/pi-sky/event-router.d.ts +32 -0
  44. package/dist/lib/pi-sky/event-router.d.ts.map +1 -0
  45. package/dist/lib/pi-sky/event-router.js +176 -0
  46. package/dist/lib/pi-sky/event-router.js.map +1 -0
  47. package/dist/lib/pi-sky/experiment.d.ts +9 -0
  48. package/dist/lib/pi-sky/experiment.d.ts.map +1 -0
  49. package/dist/lib/pi-sky/experiment.js +83 -0
  50. package/dist/lib/pi-sky/experiment.js.map +1 -0
  51. package/dist/lib/pi-sky/index.d.ts +16 -0
  52. package/dist/lib/pi-sky/index.d.ts.map +1 -0
  53. package/dist/lib/pi-sky/index.js +16 -0
  54. package/dist/lib/pi-sky/index.js.map +1 -0
  55. package/dist/lib/pi-sky/stratus-gate.d.ts +28 -0
  56. package/dist/lib/pi-sky/stratus-gate.d.ts.map +1 -0
  57. package/dist/lib/pi-sky/stratus-gate.js +61 -0
  58. package/dist/lib/pi-sky/stratus-gate.js.map +1 -0
  59. package/dist/lib/pi-sky/swarm.d.ts +28 -0
  60. package/dist/lib/pi-sky/swarm.d.ts.map +1 -0
  61. package/dist/lib/pi-sky/swarm.js +208 -0
  62. package/dist/lib/pi-sky/swarm.js.map +1 -0
  63. package/dist/lib/pi-sky/types.d.ts +139 -0
  64. package/dist/lib/pi-sky/types.d.ts.map +1 -0
  65. package/dist/lib/pi-sky/types.js +2 -0
  66. package/dist/lib/pi-sky/types.js.map +1 -0
  67. package/dist/lib/pi-sky/voice-bridge.d.ts +20 -0
  68. package/dist/lib/pi-sky/voice-bridge.d.ts.map +1 -0
  69. package/dist/lib/pi-sky/voice-bridge.js +91 -0
  70. package/dist/lib/pi-sky/voice-bridge.js.map +1 -0
  71. package/dist/lib/policy-head.d.ts +16 -1
  72. package/dist/lib/policy-head.d.ts.map +1 -1
  73. package/dist/lib/policy-head.js +117 -19
  74. package/dist/lib/policy-head.js.map +1 -1
  75. package/dist/lib/predictor.d.ts +10 -0
  76. package/dist/lib/predictor.d.ts.map +1 -1
  77. package/dist/lib/predictor.js +46 -7
  78. package/dist/lib/predictor.js.map +1 -1
  79. package/dist/lib/setup/agent-generator.d.ts +18 -0
  80. package/dist/lib/setup/agent-generator.d.ts.map +1 -0
  81. package/dist/lib/setup/agent-generator.js +114 -0
  82. package/dist/lib/setup/agent-generator.js.map +1 -0
  83. package/dist/lib/setup/context-analyzer.d.ts +16 -0
  84. package/dist/lib/setup/context-analyzer.d.ts.map +1 -0
  85. package/dist/lib/setup/context-analyzer.js +112 -0
  86. package/dist/lib/setup/context-analyzer.js.map +1 -0
  87. package/dist/lib/setup/doc-auditor.d.ts +54 -0
  88. package/dist/lib/setup/doc-auditor.d.ts.map +1 -0
  89. package/dist/lib/setup/doc-auditor.js +629 -0
  90. package/dist/lib/setup/doc-auditor.js.map +1 -0
  91. package/dist/lib/setup/domain-generator.d.ts +7 -0
  92. package/dist/lib/setup/domain-generator.d.ts.map +1 -0
  93. package/dist/lib/setup/domain-generator.js +58 -0
  94. package/dist/lib/setup/domain-generator.js.map +1 -0
  95. package/dist/lib/setup/smart-eval-generator.d.ts +38 -0
  96. package/dist/lib/setup/smart-eval-generator.d.ts.map +1 -0
  97. package/dist/lib/setup/smart-eval-generator.js +378 -0
  98. package/dist/lib/setup/smart-eval-generator.js.map +1 -0
  99. package/dist/lib/setup/smart-recommender.d.ts +63 -0
  100. package/dist/lib/setup/smart-recommender.d.ts.map +1 -0
  101. package/dist/lib/setup/smart-recommender.js +329 -0
  102. package/dist/lib/setup/smart-recommender.js.map +1 -0
  103. package/dist/lib/setup/spec-generator.d.ts +63 -0
  104. package/dist/lib/setup/spec-generator.d.ts.map +1 -0
  105. package/dist/lib/setup/spec-generator.js +310 -0
  106. package/dist/lib/setup/spec-generator.js.map +1 -0
  107. package/dist/lib/setup/violation-agent-generator.d.ts +32 -0
  108. package/dist/lib/setup/violation-agent-generator.d.ts.map +1 -0
  109. package/dist/lib/setup/violation-agent-generator.js +255 -0
  110. package/dist/lib/setup/violation-agent-generator.js.map +1 -0
  111. package/package.json +1 -1
  112. package/packages/pi/extensions/context.ts +88 -55
  113. package/packages/pi/extensions/hub-resolver.ts +63 -0
  114. package/packages/pi/extensions/index.ts +16 -3
  115. package/packages/pi/extensions/memory-tool.ts +9 -4
  116. package/packages/pi/extensions/session.ts +68 -16
  117. package/packages/pi/extensions/tool-renderers.ts +23 -8
  118. package/scripts/train/requirements.txt +5 -0
  119. package/scripts/train/train-policy-head.py +477 -0
  120. package/scripts/train/v2/dataset.py +81 -0
  121. package/scripts/train/v2/domain.json +18 -0
  122. package/scripts/train/v2/eval.py +196 -0
  123. package/scripts/train/v2/generate_data.py +219 -0
  124. package/scripts/train/v2/infer.py +188 -0
  125. package/scripts/train/v2/model.py +112 -0
  126. package/scripts/train/v2/precompute.py +132 -0
  127. package/scripts/train/v2/train.py +302 -0
  128. package/scripts/train/v2/transform_buffer.py +227 -0
  129. package/scripts/train/v2/validate_data.py +115 -0
  130. package/template/.claude/settings.json +2 -15
  131. package/template/scripts/session/session-cleanup.sh +2 -11
  132. package/template/scripts/session/session-end-hub.sh +72 -0
  133. package/template/scripts/session/session-start-hub.sh +105 -0
  134. package/dist/dashboard-static/assets/index-B6b867Pv.js +0 -121
  135. 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"}