peerllm-host-cli 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +19 -0
- package/README.md +83 -0
- package/dist/cli/commands/agreement.d.ts +3 -0
- package/dist/cli/commands/agreement.d.ts.map +1 -0
- package/dist/cli/commands/agreement.js +151 -0
- package/dist/cli/commands/agreement.js.map +1 -0
- package/dist/cli/commands/config.d.ts +3 -0
- package/dist/cli/commands/config.d.ts.map +1 -0
- package/dist/cli/commands/config.js +163 -0
- package/dist/cli/commands/config.js.map +1 -0
- package/dist/cli/commands/diagnostics.d.ts +3 -0
- package/dist/cli/commands/diagnostics.d.ts.map +1 -0
- package/dist/cli/commands/diagnostics.js +131 -0
- package/dist/cli/commands/diagnostics.js.map +1 -0
- package/dist/cli/commands/doctor.d.ts +3 -0
- package/dist/cli/commands/doctor.d.ts.map +1 -0
- package/dist/cli/commands/doctor.js +349 -0
- package/dist/cli/commands/doctor.js.map +1 -0
- package/dist/cli/commands/events.d.ts +3 -0
- package/dist/cli/commands/events.d.ts.map +1 -0
- package/dist/cli/commands/events.js +51 -0
- package/dist/cli/commands/events.js.map +1 -0
- package/dist/cli/commands/host.d.ts +3 -0
- package/dist/cli/commands/host.d.ts.map +1 -0
- package/dist/cli/commands/host.js +252 -0
- package/dist/cli/commands/host.js.map +1 -0
- package/dist/cli/commands/login.d.ts +3 -0
- package/dist/cli/commands/login.d.ts.map +1 -0
- package/dist/cli/commands/login.js +51 -0
- package/dist/cli/commands/login.js.map +1 -0
- package/dist/cli/commands/logout.d.ts +3 -0
- package/dist/cli/commands/logout.d.ts.map +1 -0
- package/dist/cli/commands/logout.js +28 -0
- package/dist/cli/commands/logout.js.map +1 -0
- package/dist/cli/commands/logs.d.ts +3 -0
- package/dist/cli/commands/logs.d.ts.map +1 -0
- package/dist/cli/commands/logs.js +157 -0
- package/dist/cli/commands/logs.js.map +1 -0
- package/dist/cli/commands/models.d.ts +3 -0
- package/dist/cli/commands/models.d.ts.map +1 -0
- package/dist/cli/commands/models.js +227 -0
- package/dist/cli/commands/models.js.map +1 -0
- package/dist/cli/commands/register.d.ts +3 -0
- package/dist/cli/commands/register.d.ts.map +1 -0
- package/dist/cli/commands/register.js +48 -0
- package/dist/cli/commands/register.js.map +1 -0
- package/dist/cli/commands/restart.d.ts +3 -0
- package/dist/cli/commands/restart.d.ts.map +1 -0
- package/dist/cli/commands/restart.js +66 -0
- package/dist/cli/commands/restart.js.map +1 -0
- package/dist/cli/commands/service.d.ts +3 -0
- package/dist/cli/commands/service.d.ts.map +1 -0
- package/dist/cli/commands/service.js +194 -0
- package/dist/cli/commands/service.js.map +1 -0
- package/dist/cli/commands/setup.d.ts +3 -0
- package/dist/cli/commands/setup.d.ts.map +1 -0
- package/dist/cli/commands/setup.js +242 -0
- package/dist/cli/commands/setup.js.map +1 -0
- package/dist/cli/commands/start.d.ts +3 -0
- package/dist/cli/commands/start.d.ts.map +1 -0
- package/dist/cli/commands/start.js +125 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/commands/status.d.ts +3 -0
- package/dist/cli/commands/status.d.ts.map +1 -0
- package/dist/cli/commands/status.js +76 -0
- package/dist/cli/commands/status.js.map +1 -0
- package/dist/cli/commands/stop.d.ts +3 -0
- package/dist/cli/commands/stop.d.ts.map +1 -0
- package/dist/cli/commands/stop.js +46 -0
- package/dist/cli/commands/stop.js.map +1 -0
- package/dist/cli/commands/subscription.d.ts +3 -0
- package/dist/cli/commands/subscription.d.ts.map +1 -0
- package/dist/cli/commands/subscription.js +342 -0
- package/dist/cli/commands/subscription.js.map +1 -0
- package/dist/cli/exit-codes.d.ts +12 -0
- package/dist/cli/exit-codes.d.ts.map +1 -0
- package/dist/cli/exit-codes.js +11 -0
- package/dist/cli/exit-codes.js.map +1 -0
- package/dist/cli/format.d.ts +8 -0
- package/dist/cli/format.d.ts.map +1 -0
- package/dist/cli/format.js +22 -0
- package/dist/cli/format.js.map +1 -0
- package/dist/cli/index.d.ts +3 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +88 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/ipc-client.d.ts +20 -0
- package/dist/cli/ipc-client.d.ts.map +1 -0
- package/dist/cli/ipc-client.js +101 -0
- package/dist/cli/ipc-client.js.map +1 -0
- package/dist/cli/prompts.d.ts +5 -0
- package/dist/cli/prompts.d.ts.map +1 -0
- package/dist/cli/prompts.js +68 -0
- package/dist/cli/prompts.js.map +1 -0
- package/dist/cli/services.d.ts +18 -0
- package/dist/cli/services.d.ts.map +1 -0
- package/dist/cli/services.js +16 -0
- package/dist/cli/services.js.map +1 -0
- package/dist/core/auth.d.ts +24 -0
- package/dist/core/auth.d.ts.map +1 -0
- package/dist/core/auth.js +111 -0
- package/dist/core/auth.js.map +1 -0
- package/dist/core/backend-client.d.ts +33 -0
- package/dist/core/backend-client.d.ts.map +1 -0
- package/dist/core/backend-client.js +79 -0
- package/dist/core/backend-client.js.map +1 -0
- package/dist/core/config.d.ts +21 -0
- package/dist/core/config.d.ts.map +1 -0
- package/dist/core/config.js +98 -0
- package/dist/core/config.js.map +1 -0
- package/dist/core/events.d.ts +23 -0
- package/dist/core/events.d.ts.map +1 -0
- package/dist/core/events.js +15 -0
- package/dist/core/events.js.map +1 -0
- package/dist/core/fs-lock.d.ts +13 -0
- package/dist/core/fs-lock.d.ts.map +1 -0
- package/dist/core/fs-lock.js +52 -0
- package/dist/core/fs-lock.js.map +1 -0
- package/dist/core/jsonrpc.d.ts +55 -0
- package/dist/core/jsonrpc.d.ts.map +1 -0
- package/dist/core/jsonrpc.js +59 -0
- package/dist/core/jsonrpc.js.map +1 -0
- package/dist/core/logger.d.ts +28 -0
- package/dist/core/logger.d.ts.map +1 -0
- package/dist/core/logger.js +80 -0
- package/dist/core/logger.js.map +1 -0
- package/dist/core/model-info.d.ts +11 -0
- package/dist/core/model-info.d.ts.map +1 -0
- package/dist/core/model-info.js +137 -0
- package/dist/core/model-info.js.map +1 -0
- package/dist/core/models-fs.d.ts +6 -0
- package/dist/core/models-fs.d.ts.map +1 -0
- package/dist/core/models-fs.js +21 -0
- package/dist/core/models-fs.js.map +1 -0
- package/dist/core/orchestrator.d.ts +42 -0
- package/dist/core/orchestrator.d.ts.map +1 -0
- package/dist/core/orchestrator.js +304 -0
- package/dist/core/orchestrator.js.map +1 -0
- package/dist/core/paths.d.ts +17 -0
- package/dist/core/paths.d.ts.map +1 -0
- package/dist/core/paths.js +51 -0
- package/dist/core/paths.js.map +1 -0
- package/dist/core/redact.d.ts +3 -0
- package/dist/core/redact.d.ts.map +1 -0
- package/dist/core/redact.js +27 -0
- package/dist/core/redact.js.map +1 -0
- package/dist/core/secrets.d.ts +12 -0
- package/dist/core/secrets.d.ts.map +1 -0
- package/dist/core/secrets.js +66 -0
- package/dist/core/secrets.js.map +1 -0
- package/dist/core/shared-runner.d.ts +70 -0
- package/dist/core/shared-runner.d.ts.map +1 -0
- package/dist/core/shared-runner.js +450 -0
- package/dist/core/shared-runner.js.map +1 -0
- package/dist/core/subscription.d.ts +53 -0
- package/dist/core/subscription.d.ts.map +1 -0
- package/dist/core/subscription.js +117 -0
- package/dist/core/subscription.js.map +1 -0
- package/dist/daemon/index.d.ts +3 -0
- package/dist/daemon/index.d.ts.map +1 -0
- package/dist/daemon/index.js +22 -0
- package/dist/daemon/index.js.map +1 -0
- package/dist/daemon/ipc-server.d.ts +33 -0
- package/dist/daemon/ipc-server.d.ts.map +1 -0
- package/dist/daemon/ipc-server.js +202 -0
- package/dist/daemon/ipc-server.js.map +1 -0
- package/dist/daemon/methods.d.ts +17 -0
- package/dist/daemon/methods.d.ts.map +1 -0
- package/dist/daemon/methods.js +68 -0
- package/dist/daemon/methods.js.map +1 -0
- package/dist/daemon/pid-lock.d.ts +10 -0
- package/dist/daemon/pid-lock.d.ts.map +1 -0
- package/dist/daemon/pid-lock.js +45 -0
- package/dist/daemon/pid-lock.js.map +1 -0
- package/dist/daemon/run.d.ts +19 -0
- package/dist/daemon/run.d.ts.map +1 -0
- package/dist/daemon/run.js +182 -0
- package/dist/daemon/run.js.map +1 -0
- package/dist/shared/config-types.d.ts +27 -0
- package/dist/shared/config-types.d.ts.map +1 -0
- package/dist/shared/config-types.js +13 -0
- package/dist/shared/config-types.js.map +1 -0
- package/dist/shared/protocol.d.ts +43 -0
- package/dist/shared/protocol.d.ts.map +1 -0
- package/dist/shared/protocol.js +11 -0
- package/dist/shared/protocol.js.map +1 -0
- package/package.json +70 -0
- package/templates/launchd/com.peerllm.host.plist +25 -0
- package/templates/systemd/peerllm-host.service +17 -0
- package/templates/windows/install-service.ps1 +29 -0
|
@@ -0,0 +1,304 @@
|
|
|
1
|
+
import { arch as osArch, platform as osPlatform } from "node:os";
|
|
2
|
+
import { HttpTransportType, HubConnectionBuilder, LogLevel, } from "@microsoft/signalr";
|
|
3
|
+
import { getLogger } from "./logger.js";
|
|
4
|
+
import { listModelsSync } from "./models-fs.js";
|
|
5
|
+
import { SharedGGUFRunner } from "./shared-runner.js";
|
|
6
|
+
const DEFAULT_HUB_URL = "https://api.peerllm.com/hosthub";
|
|
7
|
+
const HEARTBEAT_INTERVAL_MS = 5 * 1000;
|
|
8
|
+
const CLEANUP_INTERVAL_MS = 60 * 1000;
|
|
9
|
+
const CONVERSATION_IDLE_LIMIT_MS = 5 * 60 * 1000;
|
|
10
|
+
const DEFAULT_MODEL_FALLBACK = "mistral-7b-instruct-v0.1.Q8_0";
|
|
11
|
+
export class OrchestratorService {
|
|
12
|
+
opts;
|
|
13
|
+
connection = null;
|
|
14
|
+
heartbeatInterval = null;
|
|
15
|
+
cleanupInterval = null;
|
|
16
|
+
intentionalDisconnect = false;
|
|
17
|
+
isConnecting = false;
|
|
18
|
+
status = "disconnected";
|
|
19
|
+
url;
|
|
20
|
+
sharedRunner;
|
|
21
|
+
sessions = new Map();
|
|
22
|
+
constructor(opts) {
|
|
23
|
+
this.opts = opts;
|
|
24
|
+
this.url = opts.hubUrl ?? DEFAULT_HUB_URL;
|
|
25
|
+
this.sharedRunner = new SharedGGUFRunner({
|
|
26
|
+
paths: opts.paths,
|
|
27
|
+
config: { schemaVersion: 1, lastWriter: "cli", schemaUpdatedAt: "" },
|
|
28
|
+
});
|
|
29
|
+
}
|
|
30
|
+
getStatus() {
|
|
31
|
+
return this.status;
|
|
32
|
+
}
|
|
33
|
+
getRunnerStats() {
|
|
34
|
+
return this.sharedRunner.getStats();
|
|
35
|
+
}
|
|
36
|
+
async updateHostName(hostName) {
|
|
37
|
+
if (this.connection?.state !== "Connected")
|
|
38
|
+
return;
|
|
39
|
+
const cfg = await this.opts.config.load();
|
|
40
|
+
if (!cfg.hostId)
|
|
41
|
+
return;
|
|
42
|
+
try {
|
|
43
|
+
await this.connection.invoke("UpdateHostLabel", {
|
|
44
|
+
hostId: cfg.hostId,
|
|
45
|
+
label: hostName,
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
catch (err) {
|
|
49
|
+
getLogger().warn(`orchestrator: updateHostLabel failed: ${err.message}`);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
async connect() {
|
|
53
|
+
if (this.isConnecting || this.connection)
|
|
54
|
+
return;
|
|
55
|
+
this.isConnecting = true;
|
|
56
|
+
this.intentionalDisconnect = false;
|
|
57
|
+
const cfg = await this.opts.config.load();
|
|
58
|
+
if (!cfg.hostId || !cfg.hostName) {
|
|
59
|
+
this.setStatus("error");
|
|
60
|
+
this.isConnecting = false;
|
|
61
|
+
throw new Error("hostId or hostName missing from config");
|
|
62
|
+
}
|
|
63
|
+
// Refresh the runner env with the current config (covers config changes since boot).
|
|
64
|
+
Object.assign(this.sharedRunner.env, {
|
|
65
|
+
config: cfg,
|
|
66
|
+
});
|
|
67
|
+
this.setStatus("connecting");
|
|
68
|
+
getLogger().info("orchestrator: connecting to hub", this.url);
|
|
69
|
+
const connection = new HubConnectionBuilder()
|
|
70
|
+
.withUrl(this.url, {
|
|
71
|
+
transport: HttpTransportType.WebSockets | HttpTransportType.LongPolling,
|
|
72
|
+
})
|
|
73
|
+
.withAutomaticReconnect()
|
|
74
|
+
.configureLogging(LogLevel.Warning)
|
|
75
|
+
.build();
|
|
76
|
+
connection.onreconnecting(() => {
|
|
77
|
+
getLogger().warn("orchestrator: reconnecting…");
|
|
78
|
+
this.setStatus("connecting");
|
|
79
|
+
});
|
|
80
|
+
connection.onreconnected(() => {
|
|
81
|
+
getLogger().info("orchestrator: reconnected");
|
|
82
|
+
this.setStatus("connected");
|
|
83
|
+
void this.registerAndStartHeartbeat();
|
|
84
|
+
});
|
|
85
|
+
connection.onclose(async () => {
|
|
86
|
+
getLogger().warn("orchestrator: connection closed");
|
|
87
|
+
this.setStatus("disconnected");
|
|
88
|
+
this.stopTimers();
|
|
89
|
+
if (!this.intentionalDisconnect) {
|
|
90
|
+
setTimeout(() => {
|
|
91
|
+
void this.connect().catch((err) => getLogger().error("orchestrator: reconnect failed:", err.message));
|
|
92
|
+
}, 5000);
|
|
93
|
+
}
|
|
94
|
+
});
|
|
95
|
+
connection.on("Registered", (payload) => {
|
|
96
|
+
const hostId = payload?.HostId ?? payload?.hostId ?? "(unknown)";
|
|
97
|
+
const approved = payload?.ApprovedModels ?? payload?.approvedModels ?? [];
|
|
98
|
+
getLogger().info(`orchestrator: registered as ${hostId}, ${approved.length} approved model(s)`);
|
|
99
|
+
});
|
|
100
|
+
connection.on("Warning", (message) => {
|
|
101
|
+
getLogger().warn(`orchestrator warning: ${message}`);
|
|
102
|
+
});
|
|
103
|
+
connection.on("Ping", async (requestId, msg) => {
|
|
104
|
+
await this.send("EmitToken", {
|
|
105
|
+
requestId,
|
|
106
|
+
content: `pong:${msg}`,
|
|
107
|
+
isFinal: false,
|
|
108
|
+
});
|
|
109
|
+
});
|
|
110
|
+
const handleStartConversation = (conversationId, llmName) => {
|
|
111
|
+
if (!this.sessions.has(conversationId)) {
|
|
112
|
+
this.sessions.set(conversationId, { modelName: llmName, lastActive: Date.now() });
|
|
113
|
+
}
|
|
114
|
+
};
|
|
115
|
+
connection.on("StartConversationV2", handleStartConversation);
|
|
116
|
+
connection.on("StartConversation", handleStartConversation);
|
|
117
|
+
connection.on("RunPrompt", (req) => void this.handlePrompt(req));
|
|
118
|
+
connection.on("RunPromptV2", (req) => void this.handlePrompt(req));
|
|
119
|
+
const handleClose = (conversationId) => {
|
|
120
|
+
this.cleanupConversation(conversationId);
|
|
121
|
+
};
|
|
122
|
+
connection.on("CloseConversation", handleClose);
|
|
123
|
+
connection.on("CloseConversationV2", handleClose);
|
|
124
|
+
connection.on("cancelrequest", (conversationId) => {
|
|
125
|
+
this.sharedRunner.cancel(conversationId);
|
|
126
|
+
});
|
|
127
|
+
connection.on("Suspended", (event) => {
|
|
128
|
+
getLogger().warn("orchestrator: host suspended", JSON.stringify(event));
|
|
129
|
+
this.intentionalDisconnect = true;
|
|
130
|
+
void this.disconnect();
|
|
131
|
+
});
|
|
132
|
+
connection.on("Error", (message) => {
|
|
133
|
+
getLogger().error(`orchestrator error: ${message}`);
|
|
134
|
+
});
|
|
135
|
+
try {
|
|
136
|
+
await connection.start();
|
|
137
|
+
this.connection = connection;
|
|
138
|
+
this.setStatus("connected");
|
|
139
|
+
getLogger().info("orchestrator: connected");
|
|
140
|
+
await this.registerAndStartHeartbeat();
|
|
141
|
+
this.startCleanupLoop();
|
|
142
|
+
}
|
|
143
|
+
catch (err) {
|
|
144
|
+
getLogger().error("orchestrator: initial connect failed:", err.message);
|
|
145
|
+
this.setStatus("error");
|
|
146
|
+
throw err;
|
|
147
|
+
}
|
|
148
|
+
finally {
|
|
149
|
+
this.isConnecting = false;
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
async disconnect() {
|
|
153
|
+
this.intentionalDisconnect = true;
|
|
154
|
+
this.stopTimers();
|
|
155
|
+
try {
|
|
156
|
+
await this.sharedRunner.dispose("shutdown");
|
|
157
|
+
}
|
|
158
|
+
catch (err) {
|
|
159
|
+
getLogger().warn("orchestrator: runner dispose error:", err.message);
|
|
160
|
+
}
|
|
161
|
+
if (this.connection) {
|
|
162
|
+
try {
|
|
163
|
+
await this.connection.stop();
|
|
164
|
+
}
|
|
165
|
+
catch (err) {
|
|
166
|
+
getLogger().warn("orchestrator: stop error:", err.message);
|
|
167
|
+
}
|
|
168
|
+
this.connection = null;
|
|
169
|
+
}
|
|
170
|
+
this.setStatus("disconnected");
|
|
171
|
+
}
|
|
172
|
+
async registerAndStartHeartbeat() {
|
|
173
|
+
if (!this.connection)
|
|
174
|
+
return;
|
|
175
|
+
const cfg = await this.opts.config.load();
|
|
176
|
+
if (!cfg.hostId || !cfg.hostName)
|
|
177
|
+
return;
|
|
178
|
+
const capabilities = listModelsSync(this.opts.paths, cfg);
|
|
179
|
+
try {
|
|
180
|
+
await this.connection.invoke("RegisterHost", {
|
|
181
|
+
hostId: cfg.hostId,
|
|
182
|
+
label: cfg.hostName,
|
|
183
|
+
capabilities,
|
|
184
|
+
hostVersion: this.opts.cliVersion,
|
|
185
|
+
hostOS: osPlatform(),
|
|
186
|
+
hostArch: osArch(),
|
|
187
|
+
});
|
|
188
|
+
getLogger().info(`orchestrator: register sent (${capabilities.length} model(s))`);
|
|
189
|
+
}
|
|
190
|
+
catch (err) {
|
|
191
|
+
getLogger().error("orchestrator: register failed:", err.message);
|
|
192
|
+
return;
|
|
193
|
+
}
|
|
194
|
+
if (this.heartbeatInterval)
|
|
195
|
+
clearInterval(this.heartbeatInterval);
|
|
196
|
+
this.heartbeatInterval = setInterval(async () => {
|
|
197
|
+
if (this.connection?.state !== "Connected")
|
|
198
|
+
return;
|
|
199
|
+
try {
|
|
200
|
+
await this.connection.send("Heartbeat", cfg.hostId);
|
|
201
|
+
}
|
|
202
|
+
catch (err) {
|
|
203
|
+
getLogger().warn("orchestrator: heartbeat error:", err.message);
|
|
204
|
+
}
|
|
205
|
+
}, HEARTBEAT_INTERVAL_MS);
|
|
206
|
+
}
|
|
207
|
+
async handlePrompt(req) {
|
|
208
|
+
const conversationId = req.conversationId;
|
|
209
|
+
const requestId = req.requestId;
|
|
210
|
+
if (!conversationId || !requestId) {
|
|
211
|
+
getLogger().warn("orchestrator: prompt missing conversationId or requestId");
|
|
212
|
+
return;
|
|
213
|
+
}
|
|
214
|
+
let session = this.sessions.get(conversationId);
|
|
215
|
+
if (!session) {
|
|
216
|
+
session = {
|
|
217
|
+
modelName: req.modelName ?? DEFAULT_MODEL_FALLBACK,
|
|
218
|
+
lastActive: Date.now(),
|
|
219
|
+
};
|
|
220
|
+
this.sessions.set(conversationId, session);
|
|
221
|
+
}
|
|
222
|
+
const modelName = session.modelName;
|
|
223
|
+
try {
|
|
224
|
+
if (this.sharedRunner.getStats().modelName !== modelName) {
|
|
225
|
+
getLogger().info(`orchestrator: loading model ${modelName} for ${conversationId}`);
|
|
226
|
+
const cfg = await this.opts.config.load();
|
|
227
|
+
await this.sharedRunner.loadModel(modelName, cfg.selectedGpuIndex);
|
|
228
|
+
}
|
|
229
|
+
const promptText = req.prompt ?? req.content ?? "";
|
|
230
|
+
const promptOptions = {};
|
|
231
|
+
if (req.maxTokens != null)
|
|
232
|
+
promptOptions.maxTokens = req.maxTokens;
|
|
233
|
+
if (req.temperature != null)
|
|
234
|
+
promptOptions.temperature = req.temperature;
|
|
235
|
+
let tokenCount = 0;
|
|
236
|
+
const startTime = Date.now();
|
|
237
|
+
await this.sharedRunner.prompt(conversationId, promptText, async (chunk) => {
|
|
238
|
+
tokenCount++;
|
|
239
|
+
await this.send("EmitToken", {
|
|
240
|
+
requestId,
|
|
241
|
+
content: chunk,
|
|
242
|
+
isFinal: false,
|
|
243
|
+
});
|
|
244
|
+
}, promptOptions);
|
|
245
|
+
await this.send("EmitToken", { requestId, content: "", isFinal: true });
|
|
246
|
+
session.lastActive = Date.now();
|
|
247
|
+
getLogger().info(`orchestrator: prompt complete ${conversationId} (${tokenCount} tokens, ${Date.now() - startTime}ms)`);
|
|
248
|
+
}
|
|
249
|
+
catch (err) {
|
|
250
|
+
const message = err.message;
|
|
251
|
+
getLogger().error(`orchestrator: prompt failed: ${message}`);
|
|
252
|
+
await this.send("EmitToken", {
|
|
253
|
+
requestId,
|
|
254
|
+
content: "",
|
|
255
|
+
error: message,
|
|
256
|
+
isFinal: true,
|
|
257
|
+
});
|
|
258
|
+
}
|
|
259
|
+
}
|
|
260
|
+
cleanupConversation(conversationId) {
|
|
261
|
+
if (this.sessions.delete(conversationId)) {
|
|
262
|
+
this.sharedRunner.clearConversation(conversationId);
|
|
263
|
+
}
|
|
264
|
+
}
|
|
265
|
+
startCleanupLoop() {
|
|
266
|
+
if (this.cleanupInterval)
|
|
267
|
+
clearInterval(this.cleanupInterval);
|
|
268
|
+
this.cleanupInterval = setInterval(() => {
|
|
269
|
+
const now = Date.now();
|
|
270
|
+
for (const [id, session] of this.sessions.entries()) {
|
|
271
|
+
if (now - session.lastActive > CONVERSATION_IDLE_LIMIT_MS) {
|
|
272
|
+
this.cleanupConversation(id);
|
|
273
|
+
}
|
|
274
|
+
}
|
|
275
|
+
}, CLEANUP_INTERVAL_MS);
|
|
276
|
+
}
|
|
277
|
+
stopTimers() {
|
|
278
|
+
if (this.heartbeatInterval) {
|
|
279
|
+
clearInterval(this.heartbeatInterval);
|
|
280
|
+
this.heartbeatInterval = null;
|
|
281
|
+
}
|
|
282
|
+
if (this.cleanupInterval) {
|
|
283
|
+
clearInterval(this.cleanupInterval);
|
|
284
|
+
this.cleanupInterval = null;
|
|
285
|
+
}
|
|
286
|
+
}
|
|
287
|
+
async send(method, payload) {
|
|
288
|
+
if (this.connection?.state !== "Connected")
|
|
289
|
+
return;
|
|
290
|
+
try {
|
|
291
|
+
await this.connection.send(method, payload);
|
|
292
|
+
}
|
|
293
|
+
catch (err) {
|
|
294
|
+
getLogger().warn(`orchestrator: send(${method}) failed:`, err.message);
|
|
295
|
+
}
|
|
296
|
+
}
|
|
297
|
+
setStatus(status) {
|
|
298
|
+
if (this.status === status)
|
|
299
|
+
return;
|
|
300
|
+
this.status = status;
|
|
301
|
+
this.opts.events.emit("orchestratorStatus", { status });
|
|
302
|
+
}
|
|
303
|
+
}
|
|
304
|
+
//# sourceMappingURL=orchestrator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator.js","sourceRoot":"","sources":["../../src/core/orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,IAAI,MAAM,EAAE,QAAQ,IAAI,UAAU,EAAE,MAAM,SAAS,CAAC;AAEjE,OAAO,EACL,iBAAiB,EAEjB,oBAAoB,EACpB,QAAQ,GACT,MAAM,oBAAoB,CAAC;AAI5B,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAEhD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,eAAe,GAAG,iCAAiC,CAAC;AAC1D,MAAM,qBAAqB,GAAG,CAAC,GAAG,IAAI,CAAC;AACvC,MAAM,mBAAmB,GAAG,EAAE,GAAG,IAAI,CAAC;AACtC,MAAM,0BAA0B,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AACjD,MAAM,sBAAsB,GAAG,+BAA+B,CAAC;AA+B/D,MAAM,OAAO,mBAAmB;IAYD;IAXrB,UAAU,GAAyB,IAAI,CAAC;IACxC,iBAAiB,GAA0B,IAAI,CAAC;IAChD,eAAe,GAA0B,IAAI,CAAC;IAC9C,qBAAqB,GAAG,KAAK,CAAC;IAC9B,YAAY,GAAG,KAAK,CAAC;IACrB,MAAM,GAAuB,cAAc,CAAC;IAEnC,GAAG,CAAS;IACZ,YAAY,CAAmB;IAC/B,QAAQ,GAAG,IAAI,GAAG,EAAqD,CAAC;IAEzF,YAA6B,IAAyB;QAAzB,SAAI,GAAJ,IAAI,CAAqB;QACpD,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,IAAI,eAAe,CAAC;QAC1C,IAAI,CAAC,YAAY,GAAG,IAAI,gBAAgB,CAAC;YACvC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,EAAE;SACrE,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB;QACnC,IAAI,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,WAAW;YAAE,OAAO;QACnD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,EAAE;gBAC9C,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,EAAE,CAAC,IAAI,CACd,yCAA0C,GAAa,CAAC,OAAO,EAAE,CAClE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QACjD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC;QAEnC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACxB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,qFAAqF;QACrF,MAAM,CAAC,MAAM,CAAE,IAAI,CAAC,YAA2D,CAAC,GAAG,EAAE;YACnF,MAAM,EAAE,GAAG;SACZ,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC7B,SAAS,EAAE,CAAC,IAAI,CAAC,iCAAiC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QAE9D,MAAM,UAAU,GAAG,IAAI,oBAAoB,EAAE;aAC1C,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;YACjB,SAAS,EAAE,iBAAiB,CAAC,UAAU,GAAG,iBAAiB,CAAC,WAAW;SACxE,CAAC;aACD,sBAAsB,EAAE;aACxB,gBAAgB,CAAC,QAAQ,CAAC,OAAO,CAAC;aAClC,KAAK,EAAE,CAAC;QAEX,UAAU,CAAC,cAAc,CAAC,GAAG,EAAE;YAC7B,SAAS,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,aAAa,CAAC,GAAG,EAAE;YAC5B,SAAS,EAAE,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAC9C,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YAC5B,KAAK,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC5B,SAAS,EAAE,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACpD,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC/B,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAChC,UAAU,CAAC,GAAG,EAAE;oBACd,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CACvC,SAAS,EAAE,CAAC,KAAK,CAAC,iCAAiC,EAAE,GAAG,CAAC,OAAO,CAAC,CAClE,CAAC;gBACJ,CAAC,EAAE,IAAI,CAAC,CAAC;YACX,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,OAA0B,EAAE,EAAE;YACzD,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,EAAE,MAAM,IAAI,WAAW,CAAC;YACjE,MAAM,QAAQ,GAAG,OAAO,EAAE,cAAc,IAAI,OAAO,EAAE,cAAc,IAAI,EAAE,CAAC;YAC1E,SAAS,EAAE,CAAC,IAAI,CACd,+BAA+B,MAAM,KAAK,QAAQ,CAAC,MAAM,oBAAoB,CAC9E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAe,EAAE,EAAE;YAC3C,SAAS,EAAE,CAAC,IAAI,CAAC,yBAAyB,OAAO,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,SAAiB,EAAE,GAAW,EAAE,EAAE;YAC7D,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC3B,SAAS;gBACT,OAAO,EAAE,QAAQ,GAAG,EAAE;gBACtB,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,uBAAuB,GAAG,CAAC,cAAsB,EAAE,OAAe,EAAQ,EAAE;YAChF,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACpF,CAAC;QACH,CAAC,CAAC;QACF,UAAU,CAAC,EAAE,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,CAAC;QAC9D,UAAU,CAAC,EAAE,CAAC,mBAAmB,EAAE,uBAAuB,CAAC,CAAC;QAE5D,UAAU,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,GAAqB,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;QACnF,UAAU,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,GAAqB,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;QAErF,MAAM,WAAW,GAAG,CAAC,cAAsB,EAAQ,EAAE;YACnD,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;QAC3C,CAAC,CAAC;QACF,UAAU,CAAC,EAAE,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAC;QAChD,UAAU,CAAC,EAAE,CAAC,qBAAqB,EAAE,WAAW,CAAC,CAAC;QAElD,UAAU,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,cAAsB,EAAE,EAAE;YACxD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,KAAc,EAAE,EAAE;YAC5C,SAAS,EAAE,CAAC,IAAI,CAAC,8BAA8B,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;YACxE,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;YAClC,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,OAAe,EAAE,EAAE;YACzC,SAAS,EAAE,CAAC,KAAK,CAAC,uBAAuB,OAAO,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;YAC7B,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YAC5B,SAAS,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAC5C,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;YACvC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,EAAE,CAAC,KAAK,CAAC,uCAAuC,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;YACnF,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACxB,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QAClC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,EAAE,CAAC,IAAI,CAAC,qCAAqC,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;QAClF,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YAC/B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,SAAS,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;YACxE,CAAC;YACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,yBAAyB;QACrC,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO;QAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ;YAAE,OAAO;QACzC,MAAM,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAE1D,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,cAAc,EAAE;gBAC3C,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,KAAK,EAAE,GAAG,CAAC,QAAQ;gBACnB,YAAY;gBACZ,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;gBACjC,MAAM,EAAE,UAAU,EAAE;gBACpB,QAAQ,EAAE,MAAM,EAAE;aACnB,CAAC,CAAC;YACH,SAAS,EAAE,CAAC,IAAI,CACd,gCAAgC,YAAY,CAAC,MAAM,YAAY,CAChE,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,EAAE,CAAC,KAAK,CAAC,gCAAgC,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;YAC5E,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,iBAAiB;YAAE,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAClE,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;YAC9C,IAAI,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,WAAW;gBAAE,OAAO;YACnD,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;YACtD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,SAAS,EAAE,CAAC,IAAI,CAAC,gCAAgC,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC,EAAE,qBAAqB,CAAC,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,GAAqB;QAC9C,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,cAAc,IAAI,CAAC,SAAS,EAAE,CAAC;YAClC,SAAS,EAAE,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;YAC7E,OAAO;QACT,CAAC;QAED,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG;gBACR,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,sBAAsB;gBAClD,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;aACvB,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QAEpC,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACzD,SAAS,EAAE,CAAC,IAAI,CAAC,+BAA+B,SAAS,QAAQ,cAAc,EAAE,CAAC,CAAC;gBACnF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC1C,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACrE,CAAC;YAED,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;YACnD,MAAM,aAAa,GAAiD,EAAE,CAAC;YACvE,IAAI,GAAG,CAAC,SAAS,IAAI,IAAI;gBAAE,aAAa,CAAC,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC;YACnE,IAAI,GAAG,CAAC,WAAW,IAAI,IAAI;gBAAE,aAAa,CAAC,WAAW,GAAG,GAAG,CAAC,WAAW,CAAC;YAEzE,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAC5B,cAAc,EACd,UAAU,EACV,KAAK,EAAE,KAAa,EAAE,EAAE;gBACtB,UAAU,EAAE,CAAC;gBACb,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;oBAC3B,SAAS;oBACT,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,KAAK;iBACf,CAAC,CAAC;YACL,CAAC,EACD,aAAa,CACd,CAAC;YAEF,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACxE,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,SAAS,EAAE,CAAC,IAAI,CACd,iCAAiC,cAAc,KAAK,UAAU,YAAY,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,KAAK,CACtG,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAI,GAAa,CAAC,OAAO,CAAC;YACvC,SAAS,EAAE,CAAC,KAAK,CAAC,gCAAgC,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC3B,SAAS;gBACT,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,OAAO;gBACd,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,mBAAmB,CAAC,cAAsB;QAChD,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAEO,gBAAgB;QACtB,IAAI,IAAI,CAAC,eAAe;YAAE,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9D,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;gBACpD,IAAI,GAAG,GAAG,OAAO,CAAC,UAAU,GAAG,0BAA0B,EAAE,CAAC;oBAC1D,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC,EAAE,mBAAmB,CAAC,CAAC;IAC1B,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;QAChC,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,OAAgB;QACjD,IAAI,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,WAAW;YAAE,OAAO;QACnD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,EAAE,CAAC,IAAI,CAAC,sBAAsB,MAAM,WAAW,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,MAA0B;QAC1C,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM;YAAE,OAAO;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAC1D,CAAC;CACF"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
export interface ResolvedPaths {
|
|
2
|
+
configDir: string;
|
|
3
|
+
configFile: string;
|
|
4
|
+
runtimeFile: string;
|
|
5
|
+
secretsFile: string;
|
|
6
|
+
lockFile: string;
|
|
7
|
+
logsDir: string;
|
|
8
|
+
modelsDir: string;
|
|
9
|
+
ipcEndpoint: string;
|
|
10
|
+
pidFile: string;
|
|
11
|
+
}
|
|
12
|
+
export interface PathOptions {
|
|
13
|
+
configDirOverride?: string | undefined;
|
|
14
|
+
envHome?: string | undefined;
|
|
15
|
+
}
|
|
16
|
+
export declare function resolvePaths(opts?: PathOptions): ResolvedPaths;
|
|
17
|
+
//# sourceMappingURL=paths.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"paths.d.ts","sourceRoot":"","sources":["../../src/core/paths.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,WAAW;IAC1B,iBAAiB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACvC,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAsCD,wBAAgB,YAAY,CAAC,IAAI,GAAE,WAAgB,GAAG,aAAa,CAgBlE"}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
import { createHash } from "node:crypto";
|
|
2
|
+
import { homedir, platform, userInfo } from "node:os";
|
|
3
|
+
import { join, resolve } from "node:path";
|
|
4
|
+
function platformConfigDir() {
|
|
5
|
+
const home = homedir();
|
|
6
|
+
switch (platform()) {
|
|
7
|
+
case "darwin":
|
|
8
|
+
return join(home, "Library", "Application Support", "PeerLLM");
|
|
9
|
+
case "win32":
|
|
10
|
+
return join(process.env.APPDATA ?? join(home, "AppData", "Roaming"), "PeerLLM");
|
|
11
|
+
default:
|
|
12
|
+
return join(process.env.XDG_CONFIG_HOME ?? join(home, ".config"), "PeerLLM");
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
function platformLogsDir() {
|
|
16
|
+
return join(homedir(), "PeerLLM", "logs");
|
|
17
|
+
}
|
|
18
|
+
function platformRuntimeDir(configDir) {
|
|
19
|
+
if (platform() === "linux" && process.env.XDG_RUNTIME_DIR) {
|
|
20
|
+
return join(process.env.XDG_RUNTIME_DIR, "peerllm");
|
|
21
|
+
}
|
|
22
|
+
return configDir;
|
|
23
|
+
}
|
|
24
|
+
function ipcEndpointName(configDir, overridden) {
|
|
25
|
+
const username = userInfo().username.replace(/[^a-zA-Z0-9_-]/g, "_");
|
|
26
|
+
const salt = overridden
|
|
27
|
+
? "-" + createHash("sha1").update(configDir).digest("hex").slice(0, 6)
|
|
28
|
+
: "";
|
|
29
|
+
if (platform() === "win32") {
|
|
30
|
+
return `\\\\.\\pipe\\peerllm-host-${username}${salt}`;
|
|
31
|
+
}
|
|
32
|
+
const runtimeDir = platformRuntimeDir(configDir);
|
|
33
|
+
return join(runtimeDir, `host${salt}.sock`);
|
|
34
|
+
}
|
|
35
|
+
export function resolvePaths(opts = {}) {
|
|
36
|
+
const explicitOverride = opts.configDirOverride ?? opts.envHome ?? process.env["PEERLLM_HOME"];
|
|
37
|
+
const isOverridden = Boolean(explicitOverride);
|
|
38
|
+
const configDir = isOverridden ? resolve(explicitOverride) : platformConfigDir();
|
|
39
|
+
return {
|
|
40
|
+
configDir,
|
|
41
|
+
configFile: join(configDir, "peerllm-host-config.json"),
|
|
42
|
+
runtimeFile: join(configDir, "runtime.json"),
|
|
43
|
+
secretsFile: join(configDir, "secrets.json"),
|
|
44
|
+
lockFile: join(configDir, "peerllm-host-config.lock"),
|
|
45
|
+
logsDir: isOverridden ? join(configDir, "logs") : platformLogsDir(),
|
|
46
|
+
modelsDir: join(configDir, "LLMs"),
|
|
47
|
+
ipcEndpoint: ipcEndpointName(configDir, isOverridden),
|
|
48
|
+
pidFile: join(platformRuntimeDir(configDir), `host.pid`),
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=paths.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"paths.js","sourceRoot":"","sources":["../../src/core/paths.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACtD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAmB1C,SAAS,iBAAiB;IACxB,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,QAAQ,QAAQ,EAAE,EAAE,CAAC;QACnB,KAAK,QAAQ;YACX,OAAO,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,qBAAqB,EAAE,SAAS,CAAC,CAAC;QACjE,KAAK,OAAO;YACV,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;QAClF;YACE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;IACjF,CAAC;AACH,CAAC;AAED,SAAS,eAAe;IACtB,OAAO,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;AAC5C,CAAC;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,IAAI,QAAQ,EAAE,KAAK,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,SAAiB,EAAE,UAAmB;IAC7D,MAAM,QAAQ,GAAG,QAAQ,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;IACrE,MAAM,IAAI,GAAG,UAAU;QACrB,CAAC,CAAC,GAAG,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC,CAAC,EAAE,CAAC;IAEP,IAAI,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;QAC3B,OAAO,6BAA6B,QAAQ,GAAG,IAAI,EAAE,CAAC;IACxD,CAAC;IACD,MAAM,UAAU,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACjD,OAAO,IAAI,CAAC,UAAU,EAAE,OAAO,IAAI,OAAO,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,OAAoB,EAAE;IACjD,MAAM,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC/F,MAAM,YAAY,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAiB,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;IAElF,OAAO;QACL,SAAS;QACT,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,0BAA0B,CAAC;QACvD,WAAW,EAAE,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC;QAC5C,WAAW,EAAE,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC;QAC5C,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,0BAA0B,CAAC;QACrD,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,eAAe,EAAE;QACnE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC;QAClC,WAAW,EAAE,eAAe,CAAC,SAAS,EAAE,YAAY,CAAC;QACrD,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC;KACzD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"redact.d.ts","sourceRoot":"","sources":["../../src/core/redact.ts"],"names":[],"mappings":"AAGA,wBAAgB,MAAM,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAe9C;AAED,wBAAgB,YAAY,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAM9C"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
const SECRET_KEY_RE = /(token|password|api[_-]?key|authorization|secret|bearer)/i;
|
|
2
|
+
const REDACTED = "<redacted>";
|
|
3
|
+
export function redact(value) {
|
|
4
|
+
if (value === null || value === undefined)
|
|
5
|
+
return value;
|
|
6
|
+
if (typeof value === "string")
|
|
7
|
+
return value;
|
|
8
|
+
if (typeof value !== "object")
|
|
9
|
+
return value;
|
|
10
|
+
if (Array.isArray(value))
|
|
11
|
+
return value.map((v) => redact(v));
|
|
12
|
+
const out = {};
|
|
13
|
+
for (const [k, v] of Object.entries(value)) {
|
|
14
|
+
if (SECRET_KEY_RE.test(k)) {
|
|
15
|
+
out[k] = REDACTED;
|
|
16
|
+
}
|
|
17
|
+
else {
|
|
18
|
+
out[k] = redact(v);
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
return out;
|
|
22
|
+
}
|
|
23
|
+
export function redactString(s) {
|
|
24
|
+
// Best-effort redaction of `"refreshToken": "..."` style fragments.
|
|
25
|
+
return s.replace(/"(refresh[_-]?token|access[_-]?token|password|api[_-]?key|authorization|secret|bearer)"\s*:\s*"[^"]*"/gi, `"$1": "${REDACTED}"`);
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=redact.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"redact.js","sourceRoot":"","sources":["../../src/core/redact.ts"],"names":[],"mappings":"AAAA,MAAM,aAAa,GAAG,2DAA2D,CAAC;AAClF,MAAM,QAAQ,GAAG,YAAY,CAAC;AAE9B,MAAM,UAAU,MAAM,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IACxD,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7D,MAAM,GAAG,GAA4B,EAAE,CAAC;IACxC,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;QACtE,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC;QACpB,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,CAAS;IACpC,oEAAoE;IACpE,OAAO,CAAC,CAAC,OAAO,CACd,yGAAyG,EACzG,UAAU,QAAQ,GAAG,CACtB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { ResolvedPaths } from "./paths.js";
|
|
2
|
+
export type SecretKey = "refreshToken" | "restServerApiKey" | "networkApiKey";
|
|
3
|
+
export declare class SecretsService {
|
|
4
|
+
private readonly paths;
|
|
5
|
+
constructor(paths: ResolvedPaths);
|
|
6
|
+
get(key: SecretKey): Promise<string | undefined>;
|
|
7
|
+
set(key: SecretKey, value: string): Promise<void>;
|
|
8
|
+
clear(key: SecretKey): Promise<void>;
|
|
9
|
+
clearAll(): Promise<void>;
|
|
10
|
+
private load;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=secrets.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"secrets.d.ts","sourceRoot":"","sources":["../../src/core/secrets.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAEhD,MAAM,MAAM,SAAS,GAAG,cAAc,GAAG,kBAAkB,GAAG,eAAe,CAAC;AAW9E,qBAAa,cAAc;IACb,OAAO,CAAC,QAAQ,CAAC,KAAK;gBAAL,KAAK,EAAE,aAAa;IAE3C,GAAG,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAKhD,GAAG,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAYjD,KAAK,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAYpC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;YAUjB,IAAI;CAenB"}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { readFile } from "node:fs/promises";
|
|
2
|
+
import { acquireFileLock, atomicWriteFile } from "./fs-lock.js";
|
|
3
|
+
function emptyBlob() {
|
|
4
|
+
return { schemaVersion: 1, secrets: {} };
|
|
5
|
+
}
|
|
6
|
+
export class SecretsService {
|
|
7
|
+
paths;
|
|
8
|
+
constructor(paths) {
|
|
9
|
+
this.paths = paths;
|
|
10
|
+
}
|
|
11
|
+
async get(key) {
|
|
12
|
+
const blob = await this.load();
|
|
13
|
+
return blob.secrets[key];
|
|
14
|
+
}
|
|
15
|
+
async set(key, value) {
|
|
16
|
+
const lockPath = this.paths.lockFile + ".secrets";
|
|
17
|
+
const lock = await acquireFileLock(lockPath);
|
|
18
|
+
try {
|
|
19
|
+
const blob = await this.load();
|
|
20
|
+
blob.secrets[key] = value;
|
|
21
|
+
await atomicWriteFile(this.paths.secretsFile, JSON.stringify(blob, null, 2));
|
|
22
|
+
}
|
|
23
|
+
finally {
|
|
24
|
+
await lock.release();
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
async clear(key) {
|
|
28
|
+
const lockPath = this.paths.lockFile + ".secrets";
|
|
29
|
+
const lock = await acquireFileLock(lockPath);
|
|
30
|
+
try {
|
|
31
|
+
const blob = await this.load();
|
|
32
|
+
delete blob.secrets[key];
|
|
33
|
+
await atomicWriteFile(this.paths.secretsFile, JSON.stringify(blob, null, 2));
|
|
34
|
+
}
|
|
35
|
+
finally {
|
|
36
|
+
await lock.release();
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
async clearAll() {
|
|
40
|
+
const lockPath = this.paths.lockFile + ".secrets";
|
|
41
|
+
const lock = await acquireFileLock(lockPath);
|
|
42
|
+
try {
|
|
43
|
+
await atomicWriteFile(this.paths.secretsFile, JSON.stringify(emptyBlob(), null, 2));
|
|
44
|
+
}
|
|
45
|
+
finally {
|
|
46
|
+
await lock.release();
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
async load() {
|
|
50
|
+
try {
|
|
51
|
+
const raw = await readFile(this.paths.secretsFile, "utf8");
|
|
52
|
+
const parsed = JSON.parse(raw);
|
|
53
|
+
if (parsed.schemaVersion !== 1 || !parsed.secrets) {
|
|
54
|
+
return emptyBlob();
|
|
55
|
+
}
|
|
56
|
+
return parsed;
|
|
57
|
+
}
|
|
58
|
+
catch (err) {
|
|
59
|
+
if (err.code === "ENOENT") {
|
|
60
|
+
return emptyBlob();
|
|
61
|
+
}
|
|
62
|
+
throw err;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
//# sourceMappingURL=secrets.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"secrets.js","sourceRoot":"","sources":["../../src/core/secrets.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAE5C,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAUhE,SAAS,SAAS;IAChB,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,OAAO,cAAc;IACI;IAA7B,YAA6B,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;IAAG,CAAC;IAErD,KAAK,CAAC,GAAG,CAAC,GAAc;QACtB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAc,EAAE,KAAa;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,UAAU,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC1B,MAAM,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/E,CAAC;gBAAS,CAAC;YACT,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAc;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,UAAU,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACzB,MAAM,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/E,CAAC;gBAAS,CAAC;YACT,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,UAAU,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACtF,CAAC;gBAAS,CAAC;YACT,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAI;QAChB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAgB,CAAC;YAC9C,IAAI,MAAM,CAAC,aAAa,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClD,OAAO,SAAS,EAAE,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAK,GAA6B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrD,OAAO,SAAS,EAAE,CAAC;YACrB,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
import type { HostConfig } from "../shared/config-types.js";
|
|
2
|
+
import type { ResolvedPaths } from "./paths.js";
|
|
3
|
+
export interface ModelState {
|
|
4
|
+
modelId: string;
|
|
5
|
+
isLoaded: boolean;
|
|
6
|
+
loadType: "auto" | "manual";
|
|
7
|
+
lastUsedAt: number;
|
|
8
|
+
gpuLayers: number;
|
|
9
|
+
}
|
|
10
|
+
export type RunnerState = "cold" | "loading" | "loaded" | "disposing";
|
|
11
|
+
export type TransitionReason = "auto-load" | "manual-load" | "idle-timeout" | "manual-dispose" | "shutdown" | "load-failure" | "watchdog-evict";
|
|
12
|
+
export interface SharedRunnerEnv {
|
|
13
|
+
paths: ResolvedPaths;
|
|
14
|
+
config: HostConfig;
|
|
15
|
+
}
|
|
16
|
+
export declare class SharedGGUFRunner {
|
|
17
|
+
private readonly env;
|
|
18
|
+
private readonly idleTimeoutMs;
|
|
19
|
+
private llama;
|
|
20
|
+
private model;
|
|
21
|
+
private modelName;
|
|
22
|
+
private gpuLayers;
|
|
23
|
+
private readonly contexts;
|
|
24
|
+
private readonly maxConcurrentContexts;
|
|
25
|
+
private runnerState;
|
|
26
|
+
private modelState;
|
|
27
|
+
private activeRequestCount;
|
|
28
|
+
private loadingPromise;
|
|
29
|
+
private generation;
|
|
30
|
+
private idleTimer;
|
|
31
|
+
private watchdogTimer;
|
|
32
|
+
private onIdleDispose;
|
|
33
|
+
constructor(env: SharedRunnerEnv, idleTimeoutMs?: number);
|
|
34
|
+
setIdleDisposeCallback(cb: (modelName: string) => void): void;
|
|
35
|
+
getModelState(): ModelState | null;
|
|
36
|
+
getRunnerState(): RunnerState;
|
|
37
|
+
isModelInUse(): boolean;
|
|
38
|
+
getStats(): {
|
|
39
|
+
activeContexts: number;
|
|
40
|
+
maxContexts: number;
|
|
41
|
+
modelName: string;
|
|
42
|
+
gpuLayers: number;
|
|
43
|
+
};
|
|
44
|
+
loadModel(modelName: string, gpuIndex?: number, loadType?: "auto" | "manual"): Promise<void>;
|
|
45
|
+
private _doLoadModel;
|
|
46
|
+
private _cleanupPartialLoad;
|
|
47
|
+
prompt(conversationId: string, input: string, onToken?: (token: string) => void, options?: {
|
|
48
|
+
maxTokens?: number;
|
|
49
|
+
temperature?: number;
|
|
50
|
+
stop?: string[];
|
|
51
|
+
}): Promise<{
|
|
52
|
+
text: string;
|
|
53
|
+
promptTokens: number;
|
|
54
|
+
completionTokens: number;
|
|
55
|
+
}>;
|
|
56
|
+
cancel(conversationId: string): void;
|
|
57
|
+
cancelAll(): void;
|
|
58
|
+
clearConversation(conversationId: string): void;
|
|
59
|
+
dispose(reason?: TransitionReason): Promise<void>;
|
|
60
|
+
private scheduleIdleCheck;
|
|
61
|
+
private clearIdleTimer;
|
|
62
|
+
private startWatchdog;
|
|
63
|
+
private stopWatchdog;
|
|
64
|
+
private runWatchdogCheck;
|
|
65
|
+
private getOrCreateContext;
|
|
66
|
+
private validateModelFile;
|
|
67
|
+
private safeDispose;
|
|
68
|
+
private logTransition;
|
|
69
|
+
}
|
|
70
|
+
//# sourceMappingURL=shared-runner.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"shared-runner.d.ts","sourceRoot":"","sources":["../../src/core/shared-runner.ts"],"names":[],"mappings":"AAeA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAUhD,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,EAAE,MAAM,GAAG,QAAQ,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,SAAS,GAAG,QAAQ,GAAG,WAAW,CAAC;AAEtE,MAAM,MAAM,gBAAgB,GACxB,WAAW,GACX,aAAa,GACb,cAAc,GACd,gBAAgB,GAChB,UAAU,GACV,cAAc,GACd,gBAAgB,CAAC;AAyBrB,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,EAAE,UAAU,CAAC;CACpB;AAED,qBAAa,gBAAgB;IAqBzB,OAAO,CAAC,QAAQ,CAAC,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAC,aAAa;IArBhC,OAAO,CAAC,KAAK,CAAsB;IACnC,OAAO,CAAC,KAAK,CAA2B;IACxC,OAAO,CAAC,SAAS,CAAM;IACvB,OAAO,CAAC,SAAS,CAAK;IAEtB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAmC;IAC5D,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAM;IAE5C,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,UAAU,CAA2B;IAC7C,OAAO,CAAC,kBAAkB,CAAK;IAE/B,OAAO,CAAC,cAAc,CAA8B;IACpD,OAAO,CAAC,UAAU,CAAK;IAEvB,OAAO,CAAC,SAAS,CAA+B;IAChD,OAAO,CAAC,aAAa,CAA+B;IACpD,OAAO,CAAC,aAAa,CAA8C;gBAGhD,GAAG,EAAE,eAAe,EACpB,aAAa,GAAE,MAAgC;IAGlE,sBAAsB,CAAC,EAAE,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAI7D,aAAa,IAAI,UAAU,GAAG,IAAI;IAIlC,cAAc,IAAI,WAAW;IAI7B,YAAY,IAAI,OAAO;IAIvB,QAAQ,IAAI;QAAE,cAAc,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE;IAS3F,SAAS,CACb,SAAS,EAAE,MAAM,EACjB,QAAQ,CAAC,EAAE,MAAM,EACjB,QAAQ,GAAE,MAAM,GAAG,QAAiB,GACnC,OAAO,CAAC,IAAI,CAAC;YAyBF,YAAY;IA2F1B,OAAO,CAAC,mBAAmB;IAiBrB,MAAM,CACV,cAAc,EAAE,MAAM,EACtB,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EACjC,OAAO,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAA;KAAE,GACtE,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAA;KAAE,CAAC;IAgE5E,MAAM,CAAC,cAAc,EAAE,MAAM,GAAG,IAAI;IAWpC,SAAS,IAAI,IAAI;IAYjB,iBAAiB,CAAC,cAAc,EAAE,MAAM,GAAG,IAAI;IAezC,OAAO,CAAC,MAAM,GAAE,gBAAmC,GAAG,OAAO,CAAC,IAAI,CAAC;IA4CzE,OAAO,CAAC,iBAAiB;IAqBzB,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,gBAAgB;YAqBV,kBAAkB;YAiClB,iBAAiB;IAW/B,OAAO,CAAC,WAAW;IASnB,OAAO,CAAC,aAAa;CAqBtB"}
|