jfl 0.5.0 → 0.6.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/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 +6 -5
- 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 @@
|
|
|
1
|
+
{"version":3,"file":"swarm.d.ts","sourceRoot":"","sources":["../../../src/lib/pi-sky/swarm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AAGzC,OAAO,KAAK,EAAE,YAAY,EAAoB,QAAQ,EAAE,MAAM,YAAY,CAAA;AAW1E,qBAAa,OAAQ,SAAQ,YAAY;IACvC,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,MAAM,CAA2B;IACzC,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,QAAQ,CAAQ;gBAEZ,OAAO,EAAE,YAAY;IAKjC,IAAI,OAAO,IAAI,OAAO,CAAyB;IAC/C,IAAI,UAAU,IAAI,MAAM,EAAE,CAAmC;IAC7D,IAAI,UAAU,IAAI,MAAM,CAA4B;IAEpD,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAIzC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAgCtB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAezB,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAYzD,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMxD,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOxC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAM3D,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAOzB,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAM/D,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQzE,QAAQ,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAalD,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,KAAK,EAAE,QAAQ,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI;YASvE,UAAU;CAgFzB"}
|
|
@@ -0,0 +1,208 @@
|
|
|
1
|
+
import { EventEmitter } from "events";
|
|
2
|
+
import { PiRpcBridge } from "./bridge.js";
|
|
3
|
+
import { EventRouter } from "./event-router.js";
|
|
4
|
+
import { CostMonitor } from "./cost-monitor.js";
|
|
5
|
+
export class PiSwarm extends EventEmitter {
|
|
6
|
+
agents = new Map();
|
|
7
|
+
options;
|
|
8
|
+
router = null;
|
|
9
|
+
costMonitor = null;
|
|
10
|
+
_started = false;
|
|
11
|
+
constructor(options) {
|
|
12
|
+
super();
|
|
13
|
+
this.options = options;
|
|
14
|
+
}
|
|
15
|
+
get started() { return this._started; }
|
|
16
|
+
get agentNames() { return [...this.agents.keys()]; }
|
|
17
|
+
get agentCount() { return this.agents.size; }
|
|
18
|
+
getAgent(name) {
|
|
19
|
+
return this.agents.get(name)?.bridge;
|
|
20
|
+
}
|
|
21
|
+
async start() {
|
|
22
|
+
if (this._started)
|
|
23
|
+
throw new Error("Swarm already started");
|
|
24
|
+
if (this.options.hubUrl) {
|
|
25
|
+
this.router = new EventRouter({ hubUrl: this.options.hubUrl });
|
|
26
|
+
this.router.on("route", (data) => this.emit("event_routed", data));
|
|
27
|
+
this.router.on("error", (err) => this.emit("router_error", err));
|
|
28
|
+
}
|
|
29
|
+
if (this.options.costBudget) {
|
|
30
|
+
this.costMonitor = new CostMonitor({ maxCost: this.options.costBudget });
|
|
31
|
+
this.costMonitor.on("downgrade", (data) => this.emit("cost_downgrade", data));
|
|
32
|
+
this.costMonitor.on("upgrade", (data) => this.emit("cost_upgrade", data));
|
|
33
|
+
this.costMonitor.on("cost_update", (data) => this.emit("cost_update", data));
|
|
34
|
+
}
|
|
35
|
+
for (const config of this.options.agents) {
|
|
36
|
+
await this.spawnAgent(config);
|
|
37
|
+
}
|
|
38
|
+
if (this.router) {
|
|
39
|
+
try {
|
|
40
|
+
await this.router.startSse();
|
|
41
|
+
}
|
|
42
|
+
catch {
|
|
43
|
+
await this.router.startPolling();
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
this._started = true;
|
|
47
|
+
this.emit("started", { agents: this.agentNames });
|
|
48
|
+
}
|
|
49
|
+
async shutdown() {
|
|
50
|
+
this.router?.stop();
|
|
51
|
+
const shutdowns = [...this.agents.values()].map(async (agent) => {
|
|
52
|
+
try {
|
|
53
|
+
await agent.bridge.shutdown();
|
|
54
|
+
}
|
|
55
|
+
catch { }
|
|
56
|
+
});
|
|
57
|
+
await Promise.all(shutdowns);
|
|
58
|
+
this.agents.clear();
|
|
59
|
+
this._started = false;
|
|
60
|
+
this.emit("shutdown");
|
|
61
|
+
}
|
|
62
|
+
async prompt(agentName, message) {
|
|
63
|
+
const agent = this.agents.get(agentName);
|
|
64
|
+
if (!agent)
|
|
65
|
+
throw new Error(`Agent not found: ${agentName}`);
|
|
66
|
+
if (agent.bridge.exited)
|
|
67
|
+
throw new Error(`Agent exited: ${agentName}`);
|
|
68
|
+
if (this.costMonitor) {
|
|
69
|
+
await this.costMonitor.checkCriticalPath(agentName, message);
|
|
70
|
+
}
|
|
71
|
+
await agent.bridge.prompt(message);
|
|
72
|
+
}
|
|
73
|
+
async steer(agentName, message) {
|
|
74
|
+
const agent = this.agents.get(agentName);
|
|
75
|
+
if (!agent)
|
|
76
|
+
throw new Error(`Agent not found: ${agentName}`);
|
|
77
|
+
await agent.bridge.steer(message);
|
|
78
|
+
}
|
|
79
|
+
async steerAll(message) {
|
|
80
|
+
const promises = [...this.agents.values()]
|
|
81
|
+
.filter(a => !a.bridge.exited)
|
|
82
|
+
.map(a => a.bridge.steer(message).catch(() => { }));
|
|
83
|
+
await Promise.all(promises);
|
|
84
|
+
}
|
|
85
|
+
async followUp(agentName, message) {
|
|
86
|
+
const agent = this.agents.get(agentName);
|
|
87
|
+
if (!agent)
|
|
88
|
+
throw new Error(`Agent not found: ${agentName}`);
|
|
89
|
+
await agent.bridge.followUp(message);
|
|
90
|
+
}
|
|
91
|
+
async abortAll() {
|
|
92
|
+
const promises = [...this.agents.values()]
|
|
93
|
+
.filter(a => !a.bridge.exited)
|
|
94
|
+
.map(a => a.bridge.abort().catch(() => { }));
|
|
95
|
+
await Promise.all(promises);
|
|
96
|
+
}
|
|
97
|
+
async getLastAssistantText(agentName) {
|
|
98
|
+
const agent = this.agents.get(agentName);
|
|
99
|
+
if (!agent)
|
|
100
|
+
return null;
|
|
101
|
+
return agent.bridge.getLastAssistantText();
|
|
102
|
+
}
|
|
103
|
+
async relay(fromAgent, toAgent, prefix) {
|
|
104
|
+
const text = await this.getLastAssistantText(fromAgent);
|
|
105
|
+
if (!text)
|
|
106
|
+
throw new Error(`No response from ${fromAgent}`);
|
|
107
|
+
const message = prefix ? `${prefix}\n\n${text}` : text;
|
|
108
|
+
await this.prompt(toAgent, message);
|
|
109
|
+
}
|
|
110
|
+
async getStats() {
|
|
111
|
+
const stats = {};
|
|
112
|
+
for (const [name, agent] of this.agents) {
|
|
113
|
+
try {
|
|
114
|
+
const resp = await agent.bridge.getSessionStats();
|
|
115
|
+
if (resp.success)
|
|
116
|
+
stats[name] = resp.data;
|
|
117
|
+
}
|
|
118
|
+
catch {
|
|
119
|
+
stats[name] = { error: "unavailable" };
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
return stats;
|
|
123
|
+
}
|
|
124
|
+
onMapEvent(pattern, handler) {
|
|
125
|
+
if (!this.router)
|
|
126
|
+
throw new Error("No hub URL configured — cannot subscribe to MAP events");
|
|
127
|
+
this.router.on("route", async (data) => {
|
|
128
|
+
if (matchGlob(pattern, data.event.type)) {
|
|
129
|
+
await handler(data.event);
|
|
130
|
+
}
|
|
131
|
+
});
|
|
132
|
+
}
|
|
133
|
+
async spawnAgent(config) {
|
|
134
|
+
const bridge = new PiRpcBridge({
|
|
135
|
+
extensionPath: this.options.extensionPath,
|
|
136
|
+
skillsPath: this.options.skillsPath,
|
|
137
|
+
themePath: this.options.themePath,
|
|
138
|
+
yolo: this.options.yolo ?? true,
|
|
139
|
+
provider: config.provider,
|
|
140
|
+
model: config.model,
|
|
141
|
+
env: {
|
|
142
|
+
JFL_AGENT_NAME: config.name,
|
|
143
|
+
JFL_AGENT_ROLE: config.role,
|
|
144
|
+
JFL_PI_MODE: "1",
|
|
145
|
+
},
|
|
146
|
+
});
|
|
147
|
+
const state = {
|
|
148
|
+
name: config.name,
|
|
149
|
+
config,
|
|
150
|
+
bridge,
|
|
151
|
+
streaming: false,
|
|
152
|
+
turnCount: 0,
|
|
153
|
+
lastActivity: Date.now(),
|
|
154
|
+
};
|
|
155
|
+
bridge.on("agent_start", () => {
|
|
156
|
+
state.streaming = true;
|
|
157
|
+
state.lastActivity = Date.now();
|
|
158
|
+
this.emit("agent_streaming", { name: config.name, streaming: true });
|
|
159
|
+
});
|
|
160
|
+
bridge.on("agent_end", (event) => {
|
|
161
|
+
state.streaming = false;
|
|
162
|
+
state.lastActivity = Date.now();
|
|
163
|
+
this.emit("agent_done", { name: config.name, event });
|
|
164
|
+
});
|
|
165
|
+
bridge.on("turn_start", () => {
|
|
166
|
+
state.turnCount++;
|
|
167
|
+
this.emit("agent_turn", { name: config.name, turn: state.turnCount });
|
|
168
|
+
});
|
|
169
|
+
bridge.on("message_update", (event) => {
|
|
170
|
+
state.lastActivity = Date.now();
|
|
171
|
+
this.emit("agent_message", { name: config.name, event });
|
|
172
|
+
});
|
|
173
|
+
bridge.on("tool_execution_start", (event) => {
|
|
174
|
+
this.emit("agent_tool", { name: config.name, event, phase: "start" });
|
|
175
|
+
});
|
|
176
|
+
bridge.on("tool_execution_end", (event) => {
|
|
177
|
+
this.emit("agent_tool", { name: config.name, event, phase: "end" });
|
|
178
|
+
});
|
|
179
|
+
bridge.on("exit", (info) => {
|
|
180
|
+
this.emit("agent_exit", { name: config.name, ...info });
|
|
181
|
+
});
|
|
182
|
+
bridge.on("error", (err) => {
|
|
183
|
+
this.emit("agent_error", { name: config.name, error: err });
|
|
184
|
+
});
|
|
185
|
+
await bridge.start();
|
|
186
|
+
if (config.thinkingLevel) {
|
|
187
|
+
await bridge.setThinkingLevel(config.thinkingLevel);
|
|
188
|
+
}
|
|
189
|
+
this.agents.set(config.name, state);
|
|
190
|
+
if (this.router) {
|
|
191
|
+
this.router.registerBridge(config.name, bridge);
|
|
192
|
+
}
|
|
193
|
+
if (this.costMonitor) {
|
|
194
|
+
this.costMonitor.registerBridge(config.name, bridge);
|
|
195
|
+
}
|
|
196
|
+
this.emit("agent_spawned", { name: config.name, role: config.role, pid: bridge.pid });
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
function matchGlob(pattern, value) {
|
|
200
|
+
if (pattern === value)
|
|
201
|
+
return true;
|
|
202
|
+
if (pattern === "*")
|
|
203
|
+
return true;
|
|
204
|
+
if (pattern.endsWith("*"))
|
|
205
|
+
return value.startsWith(pattern.slice(0, -1));
|
|
206
|
+
return false;
|
|
207
|
+
}
|
|
208
|
+
//# sourceMappingURL=swarm.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"swarm.js","sourceRoot":"","sources":["../../../src/lib/pi-sky/swarm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAY/C,MAAM,OAAO,OAAQ,SAAQ,YAAY;IAC/B,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAA;IACtC,OAAO,CAAc;IACrB,MAAM,GAAuB,IAAI,CAAA;IACjC,WAAW,GAAuB,IAAI,CAAA;IACtC,QAAQ,GAAG,KAAK,CAAA;IAExB,YAAY,OAAqB;QAC/B,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,UAAU,KAAe,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA,CAAC,CAAC;IAC7D,IAAI,UAAU,KAAa,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAA,CAAC,CAAC;IAEpD,QAAQ,CAAC,IAAY;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAA;QAE3D,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;YAC9D,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,CAAA;YAClE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAA;QAClE,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAA;YACxE,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC,CAAA;YAC7E,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,CAAA;YACzE,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC,CAAA;QAC9E,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACzC,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;QAC/B,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAA;YAC9B,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAA;YAClC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAA;QACpB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAA;IACnD,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAA;QAEnB,MAAM,SAAS,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC9D,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAA;YAC/B,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC,CAAC,CAAA;QAEF,MAAM,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;QACnB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAA;QACrB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,OAAe;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QACxC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAA;QAC5D,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,SAAS,EAAE,CAAC,CAAA;QAEtE,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;QAC9D,CAAC;QAED,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB,EAAE,OAAe;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QACxC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAA;QAC5D,MAAM,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IACnC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAe;QAC5B,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;aACvC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;aAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAA;QACpD,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAC7B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAAiB,EAAE,OAAe;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QACxC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAA;QAC5D,MAAM,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;aACvC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;aAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAA;QAC7C,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAC7B,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,SAAiB;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QACxC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAA;QACvB,OAAO,KAAK,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAA;IAC5C,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB,EAAE,OAAe,EAAE,MAAe;QAC7D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAA;QACvD,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAA;QAE3D,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAA;QACtD,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;IACrC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,KAAK,GAA4B,EAAE,CAAA;QACzC,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,EAAE,CAAA;gBACjD,IAAI,IAAI,CAAC,OAAO;oBAAE,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAA;YAC3C,CAAC;YAAC,MAAM,CAAC;gBACP,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,aAAa,EAAE,CAAA;YACxC,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAA;IACd,CAAC;IAED,UAAU,CAAC,OAAe,EAAE,OAAkD;QAC5E,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAA;QAC3F,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,IAAyB,EAAE,EAAE;YAC1D,IAAI,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC3B,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,MAAwB;QAC/C,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;YAC7B,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;YACnC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YACjC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,IAAI;YAC/B,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,GAAG,EAAE;gBACH,cAAc,EAAE,MAAM,CAAC,IAAI;gBAC3B,cAAc,EAAE,MAAM,CAAC,IAAI;gBAC3B,WAAW,EAAE,GAAG;aACjB;SACF,CAAC,CAAA;QAEF,MAAM,KAAK,GAAe;YACxB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,MAAM;YACN,MAAM;YACN,SAAS,EAAE,KAAK;YAChB,SAAS,EAAE,CAAC;YACZ,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE;SACzB,CAAA;QAED,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;YAC5B,KAAK,CAAC,SAAS,GAAG,IAAI,CAAA;YACtB,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC/B,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QACtE,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE;YAC/B,KAAK,CAAC,SAAS,GAAG,KAAK,CAAA;YACvB,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC/B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;QACvD,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;YAC3B,KAAK,CAAC,SAAS,EAAE,CAAA;YACjB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE;YACpC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC/B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;QAC1D,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,sBAAsB,EAAE,CAAC,KAAK,EAAE,EAAE;YAC1C,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,KAAK,EAAE,EAAE;YACxC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,EAAE,CAAC,CAAA;QACzD,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACzB,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;QAEF,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QAEpB,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,MAAM,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;QACrD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAEnC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;QACjD,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;QACtD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAA;IACvF,CAAC;CACF;AAED,SAAS,SAAS,CAAC,OAAe,EAAE,KAAa;IAC/C,IAAI,OAAO,KAAK,KAAK;QAAE,OAAO,IAAI,CAAA;IAClC,IAAI,OAAO,KAAK,GAAG;QAAE,OAAO,IAAI,CAAA;IAChC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;IACxE,OAAO,KAAK,CAAA;AACd,CAAC"}
|
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
export interface PiRpcCommand {
|
|
2
|
+
id?: string;
|
|
3
|
+
type: string;
|
|
4
|
+
[key: string]: unknown;
|
|
5
|
+
}
|
|
6
|
+
export interface PiRpcResponse {
|
|
7
|
+
id?: string;
|
|
8
|
+
type: "response";
|
|
9
|
+
command: string;
|
|
10
|
+
success: boolean;
|
|
11
|
+
data?: unknown;
|
|
12
|
+
error?: string;
|
|
13
|
+
}
|
|
14
|
+
export interface PiRpcEvent {
|
|
15
|
+
type: string;
|
|
16
|
+
[key: string]: unknown;
|
|
17
|
+
}
|
|
18
|
+
export interface AssistantMessageEvent {
|
|
19
|
+
type: "start" | "text_start" | "text_delta" | "text_end" | "thinking_start" | "thinking_delta" | "thinking_end" | "toolcall_start" | "toolcall_delta" | "toolcall_end" | "done" | "error";
|
|
20
|
+
contentIndex?: number;
|
|
21
|
+
delta?: string;
|
|
22
|
+
content?: string;
|
|
23
|
+
partial?: unknown;
|
|
24
|
+
toolCall?: unknown;
|
|
25
|
+
reason?: string;
|
|
26
|
+
}
|
|
27
|
+
export interface MessageUpdateEvent extends PiRpcEvent {
|
|
28
|
+
type: "message_update";
|
|
29
|
+
message: unknown;
|
|
30
|
+
assistantMessageEvent: AssistantMessageEvent;
|
|
31
|
+
}
|
|
32
|
+
export interface AgentEndEvent extends PiRpcEvent {
|
|
33
|
+
type: "agent_end";
|
|
34
|
+
messages: unknown[];
|
|
35
|
+
}
|
|
36
|
+
export interface ToolExecutionEvent extends PiRpcEvent {
|
|
37
|
+
type: "tool_execution_start" | "tool_execution_update" | "tool_execution_end";
|
|
38
|
+
toolCallId: string;
|
|
39
|
+
toolName: string;
|
|
40
|
+
args?: Record<string, unknown>;
|
|
41
|
+
result?: unknown;
|
|
42
|
+
partialResult?: unknown;
|
|
43
|
+
isError?: boolean;
|
|
44
|
+
}
|
|
45
|
+
export interface SessionStats {
|
|
46
|
+
sessionFile: string;
|
|
47
|
+
sessionId: string;
|
|
48
|
+
userMessages: number;
|
|
49
|
+
assistantMessages: number;
|
|
50
|
+
toolCalls: number;
|
|
51
|
+
toolResults: number;
|
|
52
|
+
totalMessages: number;
|
|
53
|
+
tokens: {
|
|
54
|
+
input: number;
|
|
55
|
+
output: number;
|
|
56
|
+
cacheRead: number;
|
|
57
|
+
cacheWrite: number;
|
|
58
|
+
total: number;
|
|
59
|
+
};
|
|
60
|
+
cost: number;
|
|
61
|
+
}
|
|
62
|
+
export interface ModelInfo {
|
|
63
|
+
id: string;
|
|
64
|
+
name: string;
|
|
65
|
+
api: string;
|
|
66
|
+
provider: string;
|
|
67
|
+
reasoning: boolean;
|
|
68
|
+
contextWindow: number;
|
|
69
|
+
maxTokens: number;
|
|
70
|
+
cost: {
|
|
71
|
+
input: number;
|
|
72
|
+
output: number;
|
|
73
|
+
cacheRead: number;
|
|
74
|
+
cacheWrite: number;
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
export interface BridgeOptions {
|
|
78
|
+
extensionPath?: string;
|
|
79
|
+
skillsPath?: string;
|
|
80
|
+
themePath?: string;
|
|
81
|
+
yolo?: boolean;
|
|
82
|
+
noSession?: boolean;
|
|
83
|
+
sessionDir?: string;
|
|
84
|
+
provider?: string;
|
|
85
|
+
model?: string;
|
|
86
|
+
cwd?: string;
|
|
87
|
+
env?: Record<string, string>;
|
|
88
|
+
}
|
|
89
|
+
export interface SwarmAgentConfig {
|
|
90
|
+
name: string;
|
|
91
|
+
role: string;
|
|
92
|
+
description?: string;
|
|
93
|
+
model?: string;
|
|
94
|
+
provider?: string;
|
|
95
|
+
skills?: string[];
|
|
96
|
+
thinkingLevel?: "off" | "minimal" | "low" | "medium" | "high";
|
|
97
|
+
}
|
|
98
|
+
export interface SwarmOptions {
|
|
99
|
+
agents: SwarmAgentConfig[];
|
|
100
|
+
extensionPath?: string;
|
|
101
|
+
skillsPath?: string;
|
|
102
|
+
themePath?: string;
|
|
103
|
+
costBudget?: number;
|
|
104
|
+
hubUrl?: string;
|
|
105
|
+
yolo?: boolean;
|
|
106
|
+
}
|
|
107
|
+
export interface MapEvent {
|
|
108
|
+
type: string;
|
|
109
|
+
ts: string;
|
|
110
|
+
data?: Record<string, unknown>;
|
|
111
|
+
source?: string;
|
|
112
|
+
}
|
|
113
|
+
export interface CostBudgetConfig {
|
|
114
|
+
maxCost: number;
|
|
115
|
+
downgradeModel?: {
|
|
116
|
+
provider: string;
|
|
117
|
+
modelId: string;
|
|
118
|
+
};
|
|
119
|
+
downgradeThinkingLevel?: "off" | "minimal" | "low";
|
|
120
|
+
upgradeModel?: {
|
|
121
|
+
provider: string;
|
|
122
|
+
modelId: string;
|
|
123
|
+
};
|
|
124
|
+
upgradeThinkingLevel?: "medium" | "high";
|
|
125
|
+
criticalKeywords?: string[];
|
|
126
|
+
}
|
|
127
|
+
export interface ExperimentConfig {
|
|
128
|
+
basePrompt: string;
|
|
129
|
+
variants: string[];
|
|
130
|
+
evalPrompt?: string;
|
|
131
|
+
}
|
|
132
|
+
export interface ExperimentResult {
|
|
133
|
+
variant: string;
|
|
134
|
+
response: string;
|
|
135
|
+
score?: number;
|
|
136
|
+
stats?: SessionStats;
|
|
137
|
+
}
|
|
138
|
+
export type BridgeEventType = "agent_start" | "agent_end" | "turn_start" | "turn_end" | "message_start" | "message_update" | "message_end" | "tool_execution_start" | "tool_execution_update" | "tool_execution_end" | "auto_compaction_start" | "auto_compaction_end" | "auto_retry_start" | "auto_retry_end" | "extension_error" | "extension_ui_request" | "raw" | "exit" | "error";
|
|
139
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/lib/pi-sky/types.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,YAAY;IAC3B,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,IAAI,EAAE,MAAM,CAAA;IACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,IAAI,EAAE,UAAU,CAAA;IAChB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,OAAO,CAAA;IAChB,IAAI,CAAC,EAAE,OAAO,CAAA;IACd,KAAK,CAAC,EAAE,MAAM,CAAA;CACf;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAA;IACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,OAAO,GAAG,YAAY,GAAG,YAAY,GAAG,UAAU,GAAG,gBAAgB,GAAG,gBAAgB,GAAG,cAAc,GAAG,gBAAgB,GAAG,gBAAgB,GAAG,cAAc,GAAG,MAAM,GAAG,OAAO,CAAA;IACzL,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,QAAQ,CAAC,EAAE,OAAO,CAAA;IAClB,MAAM,CAAC,EAAE,MAAM,CAAA;CAChB;AAED,MAAM,WAAW,kBAAmB,SAAQ,UAAU;IACpD,IAAI,EAAE,gBAAgB,CAAA;IACtB,OAAO,EAAE,OAAO,CAAA;IAChB,qBAAqB,EAAE,qBAAqB,CAAA;CAC7C;AAED,MAAM,WAAW,aAAc,SAAQ,UAAU;IAC/C,IAAI,EAAE,WAAW,CAAA;IACjB,QAAQ,EAAE,OAAO,EAAE,CAAA;CACpB;AAED,MAAM,WAAW,kBAAmB,SAAQ,UAAU;IACpD,IAAI,EAAE,sBAAsB,GAAG,uBAAuB,GAAG,oBAAoB,CAAA;IAC7E,UAAU,EAAE,MAAM,CAAA;IAClB,QAAQ,EAAE,MAAM,CAAA;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC9B,MAAM,CAAC,EAAE,OAAO,CAAA;IAChB,aAAa,CAAC,EAAE,OAAO,CAAA;IACvB,OAAO,CAAC,EAAE,OAAO,CAAA;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,WAAW,EAAE,MAAM,CAAA;IACnB,SAAS,EAAE,MAAM,CAAA;IACjB,YAAY,EAAE,MAAM,CAAA;IACpB,iBAAiB,EAAE,MAAM,CAAA;IACzB,SAAS,EAAE,MAAM,CAAA;IACjB,WAAW,EAAE,MAAM,CAAA;IACnB,aAAa,EAAE,MAAM,CAAA;IACrB,MAAM,EAAE;QACN,KAAK,EAAE,MAAM,CAAA;QACb,MAAM,EAAE,MAAM,CAAA;QACd,SAAS,EAAE,MAAM,CAAA;QACjB,UAAU,EAAE,MAAM,CAAA;QAClB,KAAK,EAAE,MAAM,CAAA;KACd,CAAA;IACD,IAAI,EAAE,MAAM,CAAA;CACb;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,QAAQ,EAAE,MAAM,CAAA;IAChB,SAAS,EAAE,OAAO,CAAA;IAClB,aAAa,EAAE,MAAM,CAAA;IACrB,SAAS,EAAE,MAAM,CAAA;IACjB,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAA;QACb,MAAM,EAAE,MAAM,CAAA;QACd,SAAS,EAAE,MAAM,CAAA;QACjB,UAAU,EAAE,MAAM,CAAA;KACnB,CAAA;CACF;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,IAAI,CAAC,EAAE,OAAO,CAAA;IACd,SAAS,CAAC,EAAE,OAAO,CAAA;IACnB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,GAAG,CAAC,EAAE,MAAM,CAAA;IACZ,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CAC7B;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,MAAM,CAAA;IACZ,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAA;IACjB,aAAa,CAAC,EAAE,KAAK,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAA;CAC9D;AAED,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,gBAAgB,EAAE,CAAA;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,IAAI,CAAC,EAAE,OAAO,CAAA;CACf;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAA;IACZ,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAA;CAChB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAA;IACf,cAAc,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAA;IACtD,sBAAsB,CAAC,EAAE,KAAK,GAAG,SAAS,GAAG,KAAK,CAAA;IAClD,YAAY,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAA;IACpD,oBAAoB,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAA;IACxC,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAA;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAA;IAClB,QAAQ,EAAE,MAAM,EAAE,CAAA;IAClB,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAA;IACf,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,KAAK,CAAC,EAAE,YAAY,CAAA;CACrB;AAED,MAAM,MAAM,eAAe,GACvB,aAAa,GACb,WAAW,GACX,YAAY,GACZ,UAAU,GACV,eAAe,GACf,gBAAgB,GAChB,aAAa,GACb,sBAAsB,GACtB,uBAAuB,GACvB,oBAAoB,GACpB,uBAAuB,GACvB,qBAAqB,GACrB,kBAAkB,GAClB,gBAAgB,GAChB,iBAAiB,GACjB,sBAAsB,GACtB,KAAK,GACL,MAAM,GACN,OAAO,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/lib/pi-sky/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { EventEmitter } from "events";
|
|
2
|
+
import type { PiRpcBridge } from "./bridge.js";
|
|
3
|
+
interface VoiceBridgeOptions {
|
|
4
|
+
device?: string;
|
|
5
|
+
mode?: "steer" | "follow_up" | "prompt";
|
|
6
|
+
vadSilenceMs?: number;
|
|
7
|
+
}
|
|
8
|
+
export declare class VoiceBridge extends EventEmitter {
|
|
9
|
+
private bridge;
|
|
10
|
+
private options;
|
|
11
|
+
private voiceProc;
|
|
12
|
+
private running;
|
|
13
|
+
constructor(bridge: PiRpcBridge, options?: VoiceBridgeOptions);
|
|
14
|
+
get isRunning(): boolean;
|
|
15
|
+
start(): Promise<void>;
|
|
16
|
+
stop(): void;
|
|
17
|
+
private handleTranscription;
|
|
18
|
+
}
|
|
19
|
+
export {};
|
|
20
|
+
//# sourceMappingURL=voice-bridge.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"voice-bridge.d.ts","sourceRoot":"","sources":["../../../src/lib/pi-sky/voice-bridge.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AAE9C,UAAU,kBAAkB;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,IAAI,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,QAAQ,CAAA;IACvC,YAAY,CAAC,EAAE,MAAM,CAAA;CACtB;AAED,qBAAa,WAAY,SAAQ,YAAY;IAC3C,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,SAAS,CAA4B;IAC7C,OAAO,CAAC,OAAO,CAAQ;gBAEX,MAAM,EAAE,WAAW,EAAE,OAAO,GAAE,kBAAuB;IAUjE,IAAI,SAAS,IAAI,OAAO,CAAwB;IAE1C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA2C5B,IAAI,IAAI,IAAI;YAOE,mBAAmB;CA4BlC"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import { spawn } from "child_process";
|
|
2
|
+
import { EventEmitter } from "events";
|
|
3
|
+
export class VoiceBridge extends EventEmitter {
|
|
4
|
+
bridge;
|
|
5
|
+
options;
|
|
6
|
+
voiceProc = null;
|
|
7
|
+
running = false;
|
|
8
|
+
constructor(bridge, options = {}) {
|
|
9
|
+
super();
|
|
10
|
+
this.bridge = bridge;
|
|
11
|
+
this.options = {
|
|
12
|
+
mode: "steer",
|
|
13
|
+
vadSilenceMs: 1500,
|
|
14
|
+
...options,
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
get isRunning() { return this.running; }
|
|
18
|
+
async start() {
|
|
19
|
+
if (this.running)
|
|
20
|
+
return;
|
|
21
|
+
this.running = true;
|
|
22
|
+
const args = ["voice", "record"];
|
|
23
|
+
if (this.options.device)
|
|
24
|
+
args.push("-d", this.options.device);
|
|
25
|
+
this.voiceProc = spawn("jfl", args, {
|
|
26
|
+
stdio: ["pipe", "pipe", "pipe"],
|
|
27
|
+
});
|
|
28
|
+
let buffer = "";
|
|
29
|
+
this.voiceProc.stdout?.on("data", (chunk) => {
|
|
30
|
+
buffer += chunk.toString();
|
|
31
|
+
const lines = buffer.split("\n");
|
|
32
|
+
buffer = lines.pop() ?? "";
|
|
33
|
+
for (const line of lines) {
|
|
34
|
+
const trimmed = line.trim();
|
|
35
|
+
if (!trimmed)
|
|
36
|
+
continue;
|
|
37
|
+
try {
|
|
38
|
+
const data = JSON.parse(trimmed);
|
|
39
|
+
if (data.text) {
|
|
40
|
+
this.handleTranscription(data.text);
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
catch {
|
|
44
|
+
if (trimmed.length > 2 && !trimmed.startsWith("[")) {
|
|
45
|
+
this.handleTranscription(trimmed);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
});
|
|
50
|
+
this.voiceProc.on("exit", () => {
|
|
51
|
+
this.running = false;
|
|
52
|
+
this.emit("stopped");
|
|
53
|
+
});
|
|
54
|
+
this.emit("started");
|
|
55
|
+
}
|
|
56
|
+
stop() {
|
|
57
|
+
if (!this.running)
|
|
58
|
+
return;
|
|
59
|
+
this.voiceProc?.kill("SIGTERM");
|
|
60
|
+
this.voiceProc = null;
|
|
61
|
+
this.running = false;
|
|
62
|
+
}
|
|
63
|
+
async handleTranscription(text) {
|
|
64
|
+
const trimmed = text.trim();
|
|
65
|
+
if (!trimmed || trimmed.length < 3)
|
|
66
|
+
return;
|
|
67
|
+
this.emit("transcription", trimmed);
|
|
68
|
+
if (this.bridge.exited) {
|
|
69
|
+
this.emit("error", new Error("Bridge has exited"));
|
|
70
|
+
return;
|
|
71
|
+
}
|
|
72
|
+
try {
|
|
73
|
+
switch (this.options.mode) {
|
|
74
|
+
case "steer":
|
|
75
|
+
await this.bridge.steer(trimmed);
|
|
76
|
+
break;
|
|
77
|
+
case "follow_up":
|
|
78
|
+
await this.bridge.followUp(trimmed);
|
|
79
|
+
break;
|
|
80
|
+
case "prompt":
|
|
81
|
+
await this.bridge.prompt(trimmed, { streamingBehavior: "steer" });
|
|
82
|
+
break;
|
|
83
|
+
}
|
|
84
|
+
this.emit("sent", { text: trimmed, mode: this.options.mode });
|
|
85
|
+
}
|
|
86
|
+
catch (err) {
|
|
87
|
+
this.emit("error", err);
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=voice-bridge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"voice-bridge.js","sourceRoot":"","sources":["../../../src/lib/pi-sky/voice-bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,eAAe,CAAA;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AASrC,MAAM,OAAO,WAAY,SAAQ,YAAY;IACnC,MAAM,CAAa;IACnB,OAAO,CAAoB;IAC3B,SAAS,GAAwB,IAAI,CAAA;IACrC,OAAO,GAAG,KAAK,CAAA;IAEvB,YAAY,MAAmB,EAAE,UAA8B,EAAE;QAC/D,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,OAAO,GAAG;YACb,IAAI,EAAE,OAAO;YACb,YAAY,EAAE,IAAI;YAClB,GAAG,OAAO;SACX,CAAA;IACH,CAAC;IAED,IAAI,SAAS,KAAc,OAAO,IAAI,CAAC,OAAO,CAAA,CAAC,CAAC;IAEhD,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO;YAAE,OAAM;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QAEnB,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAA;QAChC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QAE7D,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE;YAClC,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SAChC,CAAC,CAAA;QAEF,IAAI,MAAM,GAAG,EAAE,CAAA;QAEf,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YAClD,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAA;YAC1B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAChC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAA;YAE1B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAA;gBAC3B,IAAI,CAAC,OAAO;oBAAE,SAAQ;gBAEtB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;oBAChC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACd,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBACrC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;wBACnD,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAA;oBACnC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YAC7B,IAAI,CAAC,OAAO,GAAG,KAAK,CAAA;YACpB,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QACtB,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACtB,CAAC;IAED,IAAI;QACF,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAM;QACzB,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;QAC/B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA;QACrB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAA;IACtB,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,IAAY;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAA;QAC3B,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,OAAM;QAE1C,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,CAAA;QAEnC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAA;YAClD,OAAM;QACR,CAAC;QAED,IAAI,CAAC;YACH,QAAQ,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBAC1B,KAAK,OAAO;oBACV,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;oBAChC,MAAK;gBACP,KAAK,WAAW;oBACd,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAA;oBACnC,MAAK;gBACP,KAAK,QAAQ;oBACX,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC,CAAA;oBACjE,MAAK;YACT,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAA;QAC/D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;QACzB,CAAC;IACH,CAAC;CACF"}
|
|
@@ -1,25 +1,40 @@
|
|
|
1
1
|
/**
|
|
2
|
-
* @purpose Policy head inference —
|
|
2
|
+
* @purpose Policy head inference — v1 MLP reward predictor + v2 transformer action selector bridge
|
|
3
3
|
*/
|
|
4
4
|
import type { RLState, RLAction } from "./training-buffer.js";
|
|
5
|
+
export interface ActionSelection {
|
|
6
|
+
action: string;
|
|
7
|
+
confidence: number;
|
|
8
|
+
alternatives: Array<{
|
|
9
|
+
action: string;
|
|
10
|
+
confidence: number;
|
|
11
|
+
}>;
|
|
12
|
+
}
|
|
5
13
|
export declare class PolicyHeadInference {
|
|
6
14
|
private weights;
|
|
15
|
+
private v2Meta;
|
|
7
16
|
private embedCache;
|
|
17
|
+
private projectRoot;
|
|
18
|
+
private useV2;
|
|
8
19
|
constructor(projectRoot?: string);
|
|
9
20
|
get isLoaded(): boolean;
|
|
21
|
+
get version(): number;
|
|
10
22
|
get stats(): {
|
|
11
23
|
trained_on: number;
|
|
12
24
|
direction_accuracy: number;
|
|
13
25
|
rank_correlation: number;
|
|
26
|
+
val_accuracy?: number;
|
|
14
27
|
} | null;
|
|
15
28
|
embedText(text: string): Promise<number[]>;
|
|
16
29
|
formatStateText(state: RLState): string;
|
|
17
30
|
formatActionText(action: RLAction): string;
|
|
31
|
+
selectAction(state: RLState, goal: string): Promise<ActionSelection>;
|
|
18
32
|
predictReward(state: RLState, action: RLAction): Promise<number>;
|
|
19
33
|
rankActions(state: RLState, actions: RLAction[]): Promise<Array<{
|
|
20
34
|
action: RLAction;
|
|
21
35
|
predictedReward: number;
|
|
22
36
|
rank: number;
|
|
23
37
|
}>>;
|
|
38
|
+
private findPython;
|
|
24
39
|
}
|
|
25
40
|
//# sourceMappingURL=policy-head.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"policy-head.d.ts","sourceRoot":"","sources":["../../src/lib/policy-head.ts"],"names":[],"mappings":"AAAA;;GAEG;
|
|
1
|
+
{"version":3,"file":"policy-head.d.ts","sourceRoot":"","sources":["../../src/lib/policy-head.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,OAAO,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAA;AAwC7D,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAA;IACd,UAAU,EAAE,MAAM,CAAA;IAClB,YAAY,EAAE,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAC5D;AAkCD,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,UAAU,CAA8B;IAChD,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,KAAK,CAAiB;gBAElB,WAAW,CAAC,EAAE,MAAM;IAqBhC,IAAI,QAAQ,IAAI,OAAO,CAEtB;IAED,IAAI,OAAO,IAAI,MAAM,CAEpB;IAED,IAAI,KAAK,IAAI;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,kBAAkB,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAetH;IAEK,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA0BhD,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM;IAevC,gBAAgB,CAAC,MAAM,EAAE,QAAQ,GAAG,MAAM;IAUpC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAuCpE,aAAa,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC;IAiChE,WAAW,CACf,KAAK,EAAE,OAAO,EACd,OAAO,EAAE,QAAQ,EAAE,GAClB,OAAO,CAAC,KAAK,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,eAAe,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAmC9E,OAAO,CAAC,UAAU;CASnB"}
|