@elpapi42/pi-fleet-sdk 0.2.0-beta.0 → 0.2.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.
Files changed (47) hide show
  1. package/README.md +25 -1
  2. package/dist/agent.d.ts +13 -0
  3. package/dist/agent.d.ts.map +1 -0
  4. package/dist/agent.js +20 -0
  5. package/dist/agent.js.map +1 -0
  6. package/dist/client.d.ts +3 -0
  7. package/dist/client.d.ts.map +1 -0
  8. package/dist/client.js +163 -0
  9. package/dist/client.js.map +1 -0
  10. package/dist/index.d.ts +5 -0
  11. package/dist/index.d.ts.map +1 -0
  12. package/dist/index.js +4 -0
  13. package/dist/index.js.map +1 -0
  14. package/dist/internal/paths.d.ts +6 -0
  15. package/dist/internal/paths.d.ts.map +1 -0
  16. package/dist/internal/paths.js +26 -0
  17. package/dist/internal/paths.js.map +1 -0
  18. package/dist/internal/pi-rpc.d.ts +13 -0
  19. package/dist/internal/pi-rpc.d.ts.map +1 -0
  20. package/dist/internal/pi-rpc.js +105 -0
  21. package/dist/internal/pi-rpc.js.map +1 -0
  22. package/dist/internal/protocol.d.ts +23 -0
  23. package/dist/internal/protocol.d.ts.map +1 -0
  24. package/dist/internal/protocol.js +7 -0
  25. package/dist/internal/protocol.js.map +1 -0
  26. package/dist/internal/registry.d.ts +46 -0
  27. package/dist/internal/registry.d.ts.map +1 -0
  28. package/dist/internal/registry.js +118 -0
  29. package/dist/internal/registry.js.map +1 -0
  30. package/dist/internal/worker-client.d.ts +4 -0
  31. package/dist/internal/worker-client.d.ts.map +1 -0
  32. package/dist/internal/worker-client.js +60 -0
  33. package/dist/internal/worker-client.js.map +1 -0
  34. package/dist/internal/worker-launcher.d.ts +3 -0
  35. package/dist/internal/worker-launcher.d.ts.map +1 -0
  36. package/dist/internal/worker-launcher.js +13 -0
  37. package/dist/internal/worker-launcher.js.map +1 -0
  38. package/dist/internal/worker.d.ts +2 -0
  39. package/dist/internal/worker.d.ts.map +1 -0
  40. package/dist/internal/worker.js +114 -0
  41. package/dist/internal/worker.js.map +1 -0
  42. package/dist/types.d.ts +43 -0
  43. package/dist/types.d.ts.map +1 -0
  44. package/dist/types.js +19 -0
  45. package/dist/types.js.map +1 -0
  46. package/package.json +25 -8
  47. package/LICENSE +0 -21
package/README.md CHANGED
@@ -1,3 +1,27 @@
1
1
  # @elpapi42/pi-fleet-sdk
2
2
 
3
- This version only reserves the official package name for pi-fleet trusted publishing. It contains no SDK implementation. Use `0.2.0-beta.1` or later.
3
+ Create and discover durable, host-local Pi agents.
4
+
5
+ ```bash
6
+ npm install @elpapi42/pi-fleet-sdk
7
+ ```
8
+
9
+ ```ts
10
+ import { connectPiFleet } from "@elpapi42/pi-fleet-sdk"
11
+
12
+ const client = await connectPiFleet()
13
+ const agent = await client.create({
14
+ name: "researcher",
15
+ cwd: process.cwd(),
16
+ instructions: "Give concise answers.",
17
+ })
18
+
19
+ console.log(await agent.status())
20
+ await client.close()
21
+ ```
22
+
23
+ `instructions` adds durable system instructions. It does not send initial work.
24
+
25
+ `client.get(name)` and `client.list()` discover agents created by another local client. `client.close()` only closes this SDK client. It does not stop an agent worker.
26
+
27
+ Slice 1 supports Unix-like hosts with ZeroMQ `ipc://` support. It provides create, get, list, and status. Sending work, event streaming, recovery, retirement, compact, and destroy come in later slices.
@@ -0,0 +1,13 @@
1
+ import type { Agent, AgentStatus } from "./types.js";
2
+ type AgentClient = {
3
+ status(id: string, name: string): Promise<AgentStatus>;
4
+ };
5
+ export declare class AgentHandle implements Agent {
6
+ #private;
7
+ constructor(client: AgentClient, id: string, name: string);
8
+ get id(): string;
9
+ get name(): string;
10
+ status(): Promise<AgentStatus>;
11
+ }
12
+ export {};
13
+ //# sourceMappingURL=agent.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AAEpD,KAAK,WAAW,GAAG;IACjB,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAA;CACvD,CAAA;AAED,qBAAa,WAAY,YAAW,KAAK;;gBAK3B,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;IAMzD,IAAI,EAAE,IAAI,MAAM,CAEf;IAED,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,MAAM,IAAI,OAAO,CAAC,WAAW,CAAC;CAG/B"}
package/dist/agent.js ADDED
@@ -0,0 +1,20 @@
1
+ export class AgentHandle {
2
+ #client;
3
+ #id;
4
+ #name;
5
+ constructor(client, id, name) {
6
+ this.#client = client;
7
+ this.#id = id;
8
+ this.#name = name;
9
+ }
10
+ get id() {
11
+ return this.#id;
12
+ }
13
+ get name() {
14
+ return this.#name;
15
+ }
16
+ status() {
17
+ return this.#client.status(this.#id, this.#name);
18
+ }
19
+ }
20
+ //# sourceMappingURL=agent.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,WAAW;IACb,OAAO,CAAa;IACpB,GAAG,CAAQ;IACX,KAAK,CAAQ;IAEtB,YAAY,MAAmB,EAAE,EAAU,EAAE,IAAY;QACvD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,GAAG,GAAG,EAAE,CAAA;QACb,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;IACnB,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,GAAG,CAAA;IACjB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAA;IACnB,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;IAClD,CAAC;CACF"}
@@ -0,0 +1,3 @@
1
+ import { type ConnectOptions, type PiFleetClient } from "./types.js";
2
+ export declare function connectPiFleet(options?: ConnectOptions): Promise<PiFleetClient>;
3
+ //# sourceMappingURL=client.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAMA,OAAO,EAAuE,KAAK,cAAc,EAA2B,KAAK,aAAa,EAAE,MAAM,YAAY,CAAA;AAsFlK,wBAAsB,cAAc,CAAC,OAAO,GAAE,cAAmB,GAAG,OAAO,CAAC,aAAa,CAAC,CAIzF"}
package/dist/client.js ADDED
@@ -0,0 +1,163 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { capability } from "zeromq";
3
+ import { stat } from "node:fs/promises";
4
+ import { resolve } from "node:path";
5
+ import { AgentHandle } from "./agent.js";
6
+ import { AgentNotFoundError } from "./types.js";
7
+ import { resolveStateDir, workerEndpoint } from "./internal/paths.js";
8
+ import { openRegistry } from "./internal/registry.js";
9
+ import { launchWorker } from "./internal/worker-launcher.js";
10
+ import { requestStatus } from "./internal/worker-client.js";
11
+ const STARTUP_TIMEOUT_MS = 10_000;
12
+ const OWNED_PI_OPTIONS = new Set(["--mode", "--session", "--session-id", "--no-session"]);
13
+ class PiFleetClientImpl {
14
+ #registry;
15
+ #stateDir;
16
+ #closed = false;
17
+ constructor(registry, stateDir) {
18
+ this.#registry = registry;
19
+ this.#stateDir = stateDir;
20
+ }
21
+ async create(options) {
22
+ this.assertOpen();
23
+ const input = await validateCreateOptions(options);
24
+ const id = randomUUID();
25
+ const generation = randomUUID();
26
+ const now = Date.now();
27
+ const record = {
28
+ id,
29
+ name: input.name,
30
+ cwd: input.cwd,
31
+ instructions: input.instructions,
32
+ piArgs: input.piArgs,
33
+ state: "starting",
34
+ runtime: {
35
+ generation,
36
+ state: "starting",
37
+ endpoint: workerEndpoint(this.#stateDir, id, generation),
38
+ },
39
+ lastEventSeq: 0,
40
+ createdAt: now,
41
+ updatedAt: now,
42
+ };
43
+ await this.#registry.create(record);
44
+ let worker;
45
+ try {
46
+ worker = launchWorker(this.#stateDir, id, generation);
47
+ await waitForWorkerReady(record, worker, STARTUP_TIMEOUT_MS);
48
+ return new AgentHandle(this, id, input.name);
49
+ }
50
+ catch (error) {
51
+ await stopWorker(worker);
52
+ await this.#registry.rollbackCreation(id, input.name, generation);
53
+ throw error;
54
+ }
55
+ }
56
+ async get(name) {
57
+ this.assertOpen();
58
+ const record = this.#registry.getByName(name);
59
+ if (!record)
60
+ throw new AgentNotFoundError(name);
61
+ return new AgentHandle(this, record.id, record.name);
62
+ }
63
+ async list() {
64
+ this.assertOpen();
65
+ return this.#registry.list();
66
+ }
67
+ async status(id, name) {
68
+ this.assertOpen();
69
+ const record = this.#registry.getById(id);
70
+ if (!record || record.name !== name)
71
+ throw new AgentNotFoundError(name);
72
+ const status = await requestStatus(record);
73
+ return { id: status.id, name: status.name, state: status.state };
74
+ }
75
+ close() {
76
+ if (this.#closed)
77
+ return Promise.resolve();
78
+ this.#closed = true;
79
+ return this.#registry.close();
80
+ }
81
+ assertOpen() {
82
+ if (this.#closed)
83
+ throw new Error("The pi-fleet client is closed");
84
+ }
85
+ }
86
+ export async function connectPiFleet(options = {}) {
87
+ if (!capability.ipc)
88
+ throw new Error("pi-fleet requires ZeroMQ ipc:// support on this host");
89
+ const stateDir = resolveStateDir(options);
90
+ return new PiFleetClientImpl(await openRegistry(stateDir), stateDir);
91
+ }
92
+ async function validateCreateOptions(options) {
93
+ const name = options.name?.trim();
94
+ if (!name)
95
+ throw new TypeError("Agent name must not be empty");
96
+ if (name.includes("\0"))
97
+ throw new TypeError("Agent name must not contain a null byte");
98
+ if (!options.cwd)
99
+ throw new TypeError("Agent cwd is required");
100
+ const cwd = resolve(options.cwd);
101
+ if (!(await stat(cwd)).isDirectory())
102
+ throw new TypeError(`Agent cwd is not a directory: ${cwd}`);
103
+ const piArgs = options.piArgs ? [...options.piArgs] : [];
104
+ for (const arg of piArgs) {
105
+ if (OWNED_PI_OPTIONS.has(arg) || [...OWNED_PI_OPTIONS].some((option) => arg.startsWith(`${option}=`))) {
106
+ throw new TypeError(`Pi argument ${JSON.stringify(arg)} is managed by pi-fleet`);
107
+ }
108
+ }
109
+ return { name, cwd, instructions: options.instructions, piArgs };
110
+ }
111
+ async function waitForWorkerReady(record, worker, timeoutMs) {
112
+ let onExit = () => { };
113
+ const exited = new Promise((_, reject) => {
114
+ onExit = (code, signal) => reject(new Error(`Worker exited before readiness (${signal ?? code ?? "unknown"})`));
115
+ worker.once("exit", onExit);
116
+ });
117
+ try {
118
+ await Promise.race([waitForStatus(record, timeoutMs), exited]);
119
+ }
120
+ finally {
121
+ worker.off("exit", onExit);
122
+ }
123
+ }
124
+ async function stopWorker(worker) {
125
+ if (!worker || worker.exitCode !== null || worker.signalCode !== null)
126
+ return;
127
+ worker.kill("SIGTERM");
128
+ if (await waitForExit(worker, 1_000))
129
+ return;
130
+ worker.kill("SIGKILL");
131
+ await waitForExit(worker, 1_000);
132
+ }
133
+ async function waitForExit(worker, timeoutMs) {
134
+ if (worker.exitCode !== null || worker.signalCode !== null)
135
+ return true;
136
+ return new Promise((resolve) => {
137
+ const timeout = setTimeout(() => {
138
+ worker.off("exit", onExit);
139
+ resolve(false);
140
+ }, timeoutMs);
141
+ const onExit = () => {
142
+ clearTimeout(timeout);
143
+ resolve(true);
144
+ };
145
+ worker.once("exit", onExit);
146
+ });
147
+ }
148
+ async function waitForStatus(record, timeoutMs) {
149
+ const end = Date.now() + timeoutMs;
150
+ let lastError;
151
+ while (Date.now() < end) {
152
+ try {
153
+ await requestStatus(record, Math.min(500, end - Date.now()));
154
+ return;
155
+ }
156
+ catch (error) {
157
+ lastError = error;
158
+ await new Promise((resolve) => setTimeout(resolve, 100));
159
+ }
160
+ }
161
+ throw lastError instanceof Error ? lastError : new Error("Worker did not become ready");
162
+ }
163
+ //# sourceMappingURL=client.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AAEnC,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AACvC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAA;AACnC,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AACxC,OAAO,EAAE,kBAAkB,EAAqH,MAAM,YAAY,CAAA;AAClK,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAA;AACrE,OAAO,EAAE,YAAY,EAAmC,MAAM,wBAAwB,CAAA;AACtF,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAA;AAC5D,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAA;AAE3D,MAAM,kBAAkB,GAAG,MAAM,CAAA;AACjC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC,CAAA;AAEzF,MAAM,iBAAiB;IACZ,SAAS,CAAU;IACnB,SAAS,CAAQ;IAC1B,OAAO,GAAG,KAAK,CAAA;IAEf,YAAY,QAAkB,EAAE,QAAgB;QAC9C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAA;QACzB,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAA;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA2B;QACtC,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,KAAK,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAA;QAClD,MAAM,EAAE,GAAG,UAAU,EAAE,CAAA;QACvB,MAAM,UAAU,GAAG,UAAU,EAAE,CAAA;QAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QACtB,MAAM,MAAM,GAAgB;YAC1B,EAAE;YACF,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,KAAK,EAAE,UAAU;YACjB,OAAO,EAAE;gBACP,UAAU;gBACV,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,UAAU,CAAC;aACzD;YACD,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAA;QAED,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QACnC,IAAI,MAAgC,CAAA;QACpC,IAAI,CAAC;YACH,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,UAAU,CAAC,CAAA;YACrD,MAAM,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,kBAAkB,CAAC,CAAA;YAC5D,OAAO,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,CAAA;QAC9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,UAAU,CAAC,MAAM,CAAC,CAAA;YACxB,MAAM,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;YACjE,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,IAAY;QACpB,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;QAC7C,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAA;QAC/C,OAAO,IAAI,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,IAAY;QACnC,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI;YAAE,MAAM,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAA;QACvE,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,CAAA;QAC1C,OAAO,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAA;IAClE,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO,OAAO,CAAC,OAAO,EAAE,CAAA;QAC1C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAA;IAC/B,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAA;IACpE,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,UAA0B,EAAE;IAC/D,IAAI,CAAC,UAAU,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAA;IAC5F,MAAM,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,CAAA;IACzC,OAAO,IAAI,iBAAiB,CAAC,MAAM,YAAY,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAA;AACtE,CAAC;AAED,KAAK,UAAU,qBAAqB,CAAC,OAA2B;IAC9D,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;IACjC,IAAI,CAAC,IAAI;QAAE,MAAM,IAAI,SAAS,CAAC,8BAA8B,CAAC,CAAA;IAC9D,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,yCAAyC,CAAC,CAAA;IACvF,IAAI,CAAC,OAAO,CAAC,GAAG;QAAE,MAAM,IAAI,SAAS,CAAC,uBAAuB,CAAC,CAAA;IAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;IAChC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE;QAAE,MAAM,IAAI,SAAS,CAAC,iCAAiC,GAAG,EAAE,CAAC,CAAA;IACjG,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;IACxD,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QACzB,IAAI,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC;YACtG,MAAM,IAAI,SAAS,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAA;QAClF,CAAC;IACH,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,YAAY,EAAE,OAAO,CAAC,YAAY,EAAE,MAAM,EAAE,CAAA;AAClE,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,MAAmB,EAAE,MAAoB,EAAE,SAAiB;IAC5F,IAAI,MAAM,GAAiE,GAAG,EAAE,GAAE,CAAC,CAAA;IACnF,MAAM,MAAM,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;QAC9C,MAAM,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,MAAM,IAAI,IAAI,IAAI,SAAS,GAAG,CAAC,CAAC,CAAA;QAC/G,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC7B,CAAC,CAAC,CAAA;IACF,IAAI,CAAC;QACH,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,CAAC,CAAC,CAAA;IAChE,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC5B,CAAC;AACH,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,MAAgC;IACxD,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,KAAK,IAAI,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI;QAAE,OAAM;IAC7E,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACtB,IAAI,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC;QAAE,OAAM;IAC5C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACtB,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;AAClC,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,MAAoB,EAAE,SAAiB;IAChE,IAAI,MAAM,CAAC,QAAQ,KAAK,IAAI,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IACvE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9B,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;YAC1B,OAAO,CAAC,KAAK,CAAC,CAAA;QAChB,CAAC,EAAE,SAAS,CAAC,CAAA;QACb,MAAM,MAAM,GAAG,GAAG,EAAE;YAClB,YAAY,CAAC,OAAO,CAAC,CAAA;YACrB,OAAO,CAAC,IAAI,CAAC,CAAA;QACf,CAAC,CAAA;QACD,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC7B,CAAC,CAAC,CAAA;AACJ,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,MAAmB,EAAE,SAAiB;IACjE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;IAClC,IAAI,SAAkB,CAAA;IACtB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;QACxB,IAAI,CAAC;YACH,MAAM,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA;YAC5D,OAAM;QACR,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,SAAS,GAAG,KAAK,CAAA;YACjB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;QAC1D,CAAC;IACH,CAAC;IACD,MAAM,SAAS,YAAY,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAA;AACzF,CAAC"}
@@ -0,0 +1,5 @@
1
+ export declare const version = "0.2.0";
2
+ export { connectPiFleet } from "./client.js";
3
+ export { AgentNameTakenError, AgentNotFoundError, AgentUnavailableError, } from "./types.js";
4
+ export type { Agent, AgentState, AgentStatus, AgentSummary, ConnectOptions, CreateAgentOptions, PiFleetClient, } from "./types.js";
5
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,UAAU,CAAA;AAE9B,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAC5C,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,YAAY,CAAA;AAEnB,YAAY,EACV,KAAK,EACL,UAAU,EACV,WAAW,EACX,YAAY,EACZ,cAAc,EACd,kBAAkB,EAClB,aAAa,GACd,MAAM,YAAY,CAAA"}
package/dist/index.js ADDED
@@ -0,0 +1,4 @@
1
+ export const version = "0.2.0";
2
+ export { connectPiFleet } from "./client.js";
3
+ export { AgentNameTakenError, AgentNotFoundError, AgentUnavailableError, } from "./types.js";
4
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAA;AAE9B,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAC5C,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,YAAY,CAAA"}
@@ -0,0 +1,6 @@
1
+ import type { ConnectOptions } from "../types.js";
2
+ export declare function resolveStateDir(options?: ConnectOptions): string;
3
+ export declare function workerEndpoint(stateDir: string, agentId: string, generation: string): string;
4
+ export declare function prepareStateDir(stateDir: string): Promise<string>;
5
+ export declare function createStateDirectories(stateDir: string): Promise<void>;
6
+ //# sourceMappingURL=paths.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"paths.d.ts","sourceRoot":"","sources":["../../src/internal/paths.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAEjD,wBAAgB,eAAe,CAAC,OAAO,GAAE,cAAmB,GAAG,MAAM,CAIpE;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM,CAG5F;AAED,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAIvE;AAED,wBAAsB,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAK5E"}
@@ -0,0 +1,26 @@
1
+ import { createHash } from "node:crypto";
2
+ import { chmod, mkdir, realpath } from "node:fs/promises";
3
+ import { homedir } from "node:os";
4
+ import { join, resolve } from "node:path";
5
+ export function resolveStateDir(options = {}) {
6
+ if (options.stateDir)
7
+ return resolve(options.stateDir);
8
+ const stateHome = process.env.XDG_STATE_HOME ?? resolve(homedir(), ".local", "state");
9
+ return resolve(stateHome, "pi-fleet");
10
+ }
11
+ export function workerEndpoint(stateDir, agentId, generation) {
12
+ const identity = createHash("sha256").update(`${stateDir}\0${agentId}\0${generation}`).digest("hex").slice(0, 24);
13
+ return `ipc://${join(stateDir, "ipc", `${identity}.sock`)}`;
14
+ }
15
+ export async function prepareStateDir(stateDir) {
16
+ await mkdir(stateDir, { recursive: true, mode: 0o700 });
17
+ await chmod(stateDir, 0o700);
18
+ return realpath(stateDir);
19
+ }
20
+ export async function createStateDirectories(stateDir) {
21
+ const canonicalStateDir = await prepareStateDir(stateDir);
22
+ const ipcDir = join(canonicalStateDir, "ipc");
23
+ await mkdir(ipcDir, { recursive: true, mode: 0o700 });
24
+ await chmod(ipcDir, 0o700);
25
+ }
26
+ //# sourceMappingURL=paths.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"paths.js","sourceRoot":"","sources":["../../src/internal/paths.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AACxC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AACzD,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAA;AACjC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAA;AAGzC,MAAM,UAAU,eAAe,CAAC,UAA0B,EAAE;IAC1D,IAAI,OAAO,CAAC,QAAQ;QAAE,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;IACtD,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAA;IACrF,OAAO,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAA;AACvC,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,QAAgB,EAAE,OAAe,EAAE,UAAkB;IAClF,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,GAAG,QAAQ,KAAK,OAAO,KAAK,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;IACjH,OAAO,SAAS,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,GAAG,QAAQ,OAAO,CAAC,EAAE,CAAA;AAC7D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,QAAgB;IACpD,MAAM,KAAK,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;IACvD,MAAM,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;IAC5B,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAA;AAC3B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,QAAgB;IAC3D,MAAM,iBAAiB,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAA;IACzD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAA;IAC7C,MAAM,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;IACrD,MAAM,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;AAC5B,CAAC"}
@@ -0,0 +1,13 @@
1
+ import { type ChildProcessWithoutNullStreams } from "node:child_process";
2
+ import type { AgentRecord } from "./registry.js";
3
+ export type PiState = {
4
+ sessionFile: string;
5
+ sessionId: string;
6
+ };
7
+ export declare class PiStartupError extends Error {
8
+ }
9
+ export declare function startPi(record: AgentRecord, timeoutMs?: number): Promise<{
10
+ process: ChildProcessWithoutNullStreams;
11
+ state: PiState;
12
+ }>;
13
+ //# sourceMappingURL=pi-rpc.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pi-rpc.d.ts","sourceRoot":"","sources":["../../src/internal/pi-rpc.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,KAAK,8BAA8B,EAAE,MAAM,oBAAoB,CAAA;AAE/E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAEhD,MAAM,MAAM,OAAO,GAAG;IACpB,WAAW,EAAE,MAAM,CAAA;IACnB,SAAS,EAAE,MAAM,CAAA;CAClB,CAAA;AAED,qBAAa,cAAe,SAAQ,KAAK;CAAG;AAE5C,wBAAsB,OAAO,CAAC,MAAM,EAAE,WAAW,EAAE,SAAS,SAAS,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,8BAA8B,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,CAAC,CAqB3I"}
@@ -0,0 +1,105 @@
1
+ import { spawn } from "node:child_process";
2
+ import { StringDecoder } from "node:string_decoder";
3
+ export class PiStartupError extends Error {
4
+ }
5
+ export async function startPi(record, timeoutMs = 10_000) {
6
+ const command = process.env.PI_FLEET_PI_COMMAND ?? "pi";
7
+ const args = ["--mode", "rpc", ...record.piArgs];
8
+ if (record.instructions)
9
+ args.push("--append-system-prompt", record.instructions);
10
+ if (record.sessionPath)
11
+ args.push("--session", record.sessionPath);
12
+ const child = spawn(command, args, { cwd: record.cwd, stdio: "pipe" });
13
+ let stderr = "";
14
+ child.stderr.on("data", (chunk) => {
15
+ stderr = `${stderr}${chunk.toString("utf8")}`.slice(-4_096);
16
+ });
17
+ try {
18
+ const state = await getState(child, timeoutMs);
19
+ child.stdout.resume();
20
+ return { process: child, state };
21
+ }
22
+ catch (error) {
23
+ await terminate(child);
24
+ const detail = stderr.trim();
25
+ if (error instanceof PiStartupError && detail)
26
+ throw new PiStartupError(`${error.message}: ${detail}`);
27
+ throw error;
28
+ }
29
+ }
30
+ async function terminate(child) {
31
+ if (child.exitCode !== null || child.signalCode !== null)
32
+ return;
33
+ child.kill("SIGTERM");
34
+ if (await waitForExit(child, 1_000))
35
+ return;
36
+ child.kill("SIGKILL");
37
+ await waitForExit(child, 1_000);
38
+ }
39
+ async function waitForExit(child, timeoutMs) {
40
+ if (child.exitCode !== null || child.signalCode !== null)
41
+ return true;
42
+ return new Promise((resolve) => {
43
+ const timeout = setTimeout(() => {
44
+ child.off("exit", onExit);
45
+ resolve(false);
46
+ }, timeoutMs);
47
+ const onExit = () => {
48
+ clearTimeout(timeout);
49
+ resolve(true);
50
+ };
51
+ child.once("exit", onExit);
52
+ });
53
+ }
54
+ async function getState(child, timeoutMs) {
55
+ const id = "ready";
56
+ const decoder = new StringDecoder("utf8");
57
+ let pending = "";
58
+ return new Promise((resolve, reject) => {
59
+ const timeout = setTimeout(() => fail(new PiStartupError("Pi did not answer get_state before the startup deadline")), timeoutMs);
60
+ const onError = (error) => fail(new PiStartupError(`Pi failed to start: ${error.message}`));
61
+ const onExit = (code) => fail(new PiStartupError(`Pi exited before readiness (code ${code ?? "signal"})`));
62
+ const onData = (chunk) => {
63
+ pending += decoder.write(chunk);
64
+ while (true) {
65
+ const newline = pending.indexOf("\n");
66
+ if (newline < 0)
67
+ return;
68
+ const line = pending.slice(0, newline).replace(/\r$/, "");
69
+ pending = pending.slice(newline + 1);
70
+ if (!line)
71
+ continue;
72
+ try {
73
+ const message = JSON.parse(line);
74
+ if (message.type === "response" && message.id === id) {
75
+ if (!message.success)
76
+ return fail(new PiStartupError("Pi rejected get_state"));
77
+ const { sessionFile, sessionId } = message.data ?? {};
78
+ if (typeof sessionFile !== "string" || typeof sessionId !== "string")
79
+ return fail(new PiStartupError("Pi returned an invalid get_state response"));
80
+ cleanup();
81
+ resolve({ sessionFile, sessionId });
82
+ }
83
+ }
84
+ catch {
85
+ // Pi events and diagnostics may share stdout. Only a matching response matters here.
86
+ }
87
+ }
88
+ };
89
+ const cleanup = () => {
90
+ clearTimeout(timeout);
91
+ child.stdout.off("data", onData);
92
+ child.off("error", onError);
93
+ child.off("exit", onExit);
94
+ };
95
+ const fail = (error) => {
96
+ cleanup();
97
+ reject(error);
98
+ };
99
+ child.stdout.on("data", onData);
100
+ child.once("error", onError);
101
+ child.once("exit", onExit);
102
+ child.stdin.write(`${JSON.stringify({ id, type: "get_state" })}\n`);
103
+ });
104
+ }
105
+ //# sourceMappingURL=pi-rpc.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pi-rpc.js","sourceRoot":"","sources":["../../src/internal/pi-rpc.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAuC,MAAM,oBAAoB,CAAA;AAC/E,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AAQnD,MAAM,OAAO,cAAe,SAAQ,KAAK;CAAG;AAE5C,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,MAAmB,EAAE,SAAS,GAAG,MAAM;IACnE,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,IAAI,CAAA;IACvD,MAAM,IAAI,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,CAAA;IAChD,IAAI,MAAM,CAAC,YAAY;QAAE,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE,MAAM,CAAC,YAAY,CAAC,CAAA;IACjF,IAAI,MAAM,CAAC,WAAW;QAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,CAAA;IAElE,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;IACtE,IAAI,MAAM,GAAG,EAAE,CAAA;IACf,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;QACxC,MAAM,GAAG,GAAG,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAA;IAC7D,CAAC,CAAC,CAAA;IACF,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,SAAS,CAAC,CAAA;QAC9C,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAA;QACrB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAA;IAClC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,SAAS,CAAC,KAAK,CAAC,CAAA;QACtB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAA;QAC5B,IAAI,KAAK,YAAY,cAAc,IAAI,MAAM;YAAE,MAAM,IAAI,cAAc,CAAC,GAAG,KAAK,CAAC,OAAO,KAAK,MAAM,EAAE,CAAC,CAAA;QACtG,MAAM,KAAK,CAAA;IACb,CAAC;AACH,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,KAAqC;IAC5D,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI;QAAE,OAAM;IAChE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACrB,IAAI,MAAM,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,OAAM;IAC3C,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACrB,MAAM,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;AACjC,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,KAAqC,EAAE,SAAiB;IACjF,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IACrE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9B,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;YACzB,OAAO,CAAC,KAAK,CAAC,CAAA;QAChB,CAAC,EAAE,SAAS,CAAC,CAAA;QACb,MAAM,MAAM,GAAG,GAAG,EAAE;YAClB,YAAY,CAAC,OAAO,CAAC,CAAA;YACrB,OAAO,CAAC,IAAI,CAAC,CAAA;QACf,CAAC,CAAA;QACD,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;AACJ,CAAC;AAED,KAAK,UAAU,QAAQ,CAAC,KAAqC,EAAE,SAAiB;IAC9E,MAAM,EAAE,GAAG,OAAO,CAAA;IAClB,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAA;IACzC,IAAI,OAAO,GAAG,EAAE,CAAA;IAEhB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,yDAAyD,CAAC,CAAC,EAAE,SAAS,CAAC,CAAA;QAChI,MAAM,OAAO,GAAG,CAAC,KAAY,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,uBAAuB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;QAClG,MAAM,MAAM,GAAG,CAAC,IAAmB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,oCAAoC,IAAI,IAAI,QAAQ,GAAG,CAAC,CAAC,CAAA;QACzH,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,EAAE;YAC/B,OAAO,IAAI,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;YAC/B,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;gBACrC,IAAI,OAAO,GAAG,CAAC;oBAAE,OAAM;gBACvB,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;gBACzD,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAA;gBACpC,IAAI,CAAC,IAAI;oBAAE,SAAQ;gBACnB,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;oBAChC,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,IAAI,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;wBACrD,IAAI,CAAC,OAAO,CAAC,OAAO;4BAAE,OAAO,IAAI,CAAC,IAAI,cAAc,CAAC,uBAAuB,CAAC,CAAC,CAAA;wBAC9E,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAA;wBACrD,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,OAAO,SAAS,KAAK,QAAQ;4BAAE,OAAO,IAAI,CAAC,IAAI,cAAc,CAAC,2CAA2C,CAAC,CAAC,CAAA;wBAClJ,OAAO,EAAE,CAAA;wBACT,OAAO,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAA;oBACrC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,qFAAqF;gBACvF,CAAC;YACH,CAAC;QACH,CAAC,CAAA;QACD,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,OAAO,CAAC,CAAA;YACrB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;YAChC,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;YAC3B,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC3B,CAAC,CAAA;QACD,MAAM,IAAI,GAAG,CAAC,KAAY,EAAE,EAAE;YAC5B,OAAO,EAAE,CAAA;YACT,MAAM,CAAC,KAAK,CAAC,CAAA;QACf,CAAC,CAAA;QAED,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC/B,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAC5B,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC1B,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,IAAI,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;AACJ,CAAC"}
@@ -0,0 +1,23 @@
1
+ import type { AgentState } from "../types.js";
2
+ export type StatusRequest = {
3
+ version: 1;
4
+ requestId: string;
5
+ command: "status";
6
+ agentId: string;
7
+ runtimeGeneration: string;
8
+ };
9
+ export type StatusResponse = {
10
+ version: 1;
11
+ requestId: string;
12
+ ok: boolean;
13
+ status?: {
14
+ id: string;
15
+ name: string;
16
+ runtimeGeneration: string;
17
+ state: AgentState;
18
+ };
19
+ error?: string;
20
+ };
21
+ export declare function encode(message: StatusRequest | StatusResponse): string;
22
+ export declare function decode(text: Buffer): unknown;
23
+ //# sourceMappingURL=protocol.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"protocol.d.ts","sourceRoot":"","sources":["../../src/internal/protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAE7C,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,CAAC,CAAA;IACV,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,QAAQ,CAAA;IACjB,OAAO,EAAE,MAAM,CAAA;IACf,iBAAiB,EAAE,MAAM,CAAA;CAC1B,CAAA;AAED,MAAM,MAAM,cAAc,GAAG;IAC3B,OAAO,EAAE,CAAC,CAAA;IACV,SAAS,EAAE,MAAM,CAAA;IACjB,EAAE,EAAE,OAAO,CAAA;IACX,MAAM,CAAC,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,iBAAiB,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,UAAU,CAAA;KAAE,CAAA;IACnF,KAAK,CAAC,EAAE,MAAM,CAAA;CACf,CAAA;AAED,wBAAgB,MAAM,CAAC,OAAO,EAAE,aAAa,GAAG,cAAc,GAAG,MAAM,CAEtE;AAED,wBAAgB,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAE5C"}
@@ -0,0 +1,7 @@
1
+ export function encode(message) {
2
+ return JSON.stringify(message);
3
+ }
4
+ export function decode(text) {
5
+ return JSON.parse(text.toString("utf8"));
6
+ }
7
+ //# sourceMappingURL=protocol.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"protocol.js","sourceRoot":"","sources":["../../src/internal/protocol.ts"],"names":[],"mappings":"AAkBA,MAAM,UAAU,MAAM,CAAC,OAAuC;IAC5D,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;AAChC,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,IAAY;IACjC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAA;AAC1C,CAAC"}
@@ -0,0 +1,46 @@
1
+ import { type Database, type RootDatabase } from "lmdb";
2
+ import type { AgentState, AgentSummary } from "../types.js";
3
+ export type AgentRecord = {
4
+ id: string;
5
+ name: string;
6
+ cwd: string;
7
+ instructions?: string;
8
+ piArgs: string[];
9
+ sessionPath?: string;
10
+ sessionId?: string;
11
+ state: AgentState;
12
+ runtime?: {
13
+ generation: string;
14
+ state: "starting" | "ready";
15
+ workerPid?: number;
16
+ endpoint?: string;
17
+ claimId?: string;
18
+ claimedAt?: number;
19
+ };
20
+ lastEventSeq: number;
21
+ createdAt: number;
22
+ updatedAt: number;
23
+ };
24
+ type Environment = RootDatabase;
25
+ type SharedEnvironment = {
26
+ root: Environment;
27
+ agents: Database<AgentRecord, string>;
28
+ names: Database<string, string>;
29
+ references: number;
30
+ closing?: Promise<void>;
31
+ };
32
+ export declare class Registry {
33
+ #private;
34
+ constructor(stateDir: string, shared: SharedEnvironment);
35
+ create(record: AgentRecord): Promise<void>;
36
+ getById(id: string): AgentRecord | undefined;
37
+ getByName(name: string): AgentRecord | undefined;
38
+ list(): AgentSummary[];
39
+ markReady(id: string, runtimeGeneration: string, ready: Pick<NonNullable<AgentRecord["runtime"]>, "workerPid" | "endpoint"> & Pick<AgentRecord, "sessionPath" | "sessionId">): Promise<boolean>;
40
+ rollbackCreation(id: string, name: string, runtimeGeneration: string): Promise<void>;
41
+ close(): Promise<void>;
42
+ private assertOpen;
43
+ }
44
+ export declare function openRegistry(stateDir: string): Promise<Registry>;
45
+ export {};
46
+ //# sourceMappingURL=registry.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/internal/registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,KAAK,QAAQ,EAAE,KAAK,YAAY,EAAE,MAAM,MAAM,CAAA;AAE7D,OAAO,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAG3D,MAAM,MAAM,WAAW,GAAG;IACxB,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,EAAE,MAAM,EAAE,CAAA;IAChB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,KAAK,EAAE,UAAU,CAAA;IACjB,OAAO,CAAC,EAAE;QACR,UAAU,EAAE,MAAM,CAAA;QAClB,KAAK,EAAE,UAAU,GAAG,OAAO,CAAA;QAC3B,SAAS,CAAC,EAAE,MAAM,CAAA;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,OAAO,CAAC,EAAE,MAAM,CAAA;QAChB,SAAS,CAAC,EAAE,MAAM,CAAA;KACnB,CAAA;IACD,YAAY,EAAE,MAAM,CAAA;IACpB,SAAS,EAAE,MAAM,CAAA;IACjB,SAAS,EAAE,MAAM,CAAA;CAClB,CAAA;AAED,KAAK,WAAW,GAAG,YAAY,CAAA;AAC/B,KAAK,iBAAiB,GAAG;IACvB,IAAI,EAAE,WAAW,CAAA;IACjB,MAAM,EAAE,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,CAAA;IACrC,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC/B,UAAU,EAAE,MAAM,CAAA;IAClB,OAAO,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,CAAA;CACxB,CAAA;AAID,qBAAa,QAAQ;;gBAOP,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,iBAAiB;IAOjD,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAUhD,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAK5C,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAMhD,IAAI,IAAI,YAAY,EAAE;IAUhB,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,WAAW,GAAG,UAAU,CAAC,GAAG,IAAI,CAAC,WAAW,EAAE,aAAa,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;IAuB/L,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAUpF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAe5B,OAAO,CAAC,UAAU;CAGnB;AAED,wBAAsB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAmBtE"}
@@ -0,0 +1,118 @@
1
+ import { open } from "lmdb";
2
+ import { prepareStateDir } from "./paths.js";
3
+ import { AgentNameTakenError } from "../types.js";
4
+ const environments = new Map();
5
+ export class Registry {
6
+ #stateDir;
7
+ #shared;
8
+ #agents;
9
+ #names;
10
+ #closed = false;
11
+ constructor(stateDir, shared) {
12
+ this.#stateDir = stateDir;
13
+ this.#shared = shared;
14
+ this.#agents = shared.agents;
15
+ this.#names = shared.names;
16
+ }
17
+ async create(record) {
18
+ this.assertOpen();
19
+ const created = await this.#names.ifNoExists(record.name, () => {
20
+ this.#agents.put(record.id, record, 1);
21
+ this.#names.put(record.name, record.id);
22
+ });
23
+ if (!created)
24
+ throw new AgentNameTakenError(record.name);
25
+ }
26
+ getById(id) {
27
+ this.assertOpen();
28
+ return this.#agents.get(id);
29
+ }
30
+ getByName(name) {
31
+ this.assertOpen();
32
+ const id = this.#names.get(name);
33
+ return id === undefined ? undefined : this.#agents.get(id);
34
+ }
35
+ list() {
36
+ this.assertOpen();
37
+ return [...this.#agents.getRange({})].map(({ value }) => ({
38
+ id: value.id,
39
+ name: value.name,
40
+ cwd: value.cwd,
41
+ state: value.state,
42
+ }));
43
+ }
44
+ async markReady(id, runtimeGeneration, ready) {
45
+ this.assertOpen();
46
+ const entry = this.#agents.getEntry(id);
47
+ if (!entry || entry.value.runtime?.generation !== runtimeGeneration)
48
+ return false;
49
+ const { endpoint, workerPid, sessionId, sessionPath } = ready;
50
+ const record = {
51
+ ...entry.value,
52
+ sessionId,
53
+ sessionPath,
54
+ state: "idle",
55
+ runtime: {
56
+ ...entry.value.runtime,
57
+ endpoint,
58
+ workerPid,
59
+ state: "ready",
60
+ },
61
+ updatedAt: Date.now(),
62
+ };
63
+ const version = entry.version ?? 0;
64
+ return this.#agents.put(id, record, version + 1, version);
65
+ }
66
+ async rollbackCreation(id, name, runtimeGeneration) {
67
+ this.assertOpen();
68
+ await this.#names.transaction(() => {
69
+ const record = this.#agents.get(id);
70
+ if (this.#names.get(name) !== id || record?.runtime?.generation !== runtimeGeneration)
71
+ return;
72
+ this.#agents.remove(id);
73
+ this.#names.remove(name);
74
+ });
75
+ }
76
+ async close() {
77
+ if (this.#closed)
78
+ return;
79
+ this.#closed = true;
80
+ this.#shared.references -= 1;
81
+ if (this.#shared.references > 0)
82
+ return;
83
+ const closing = this.#shared.root.close();
84
+ this.#shared.closing = closing;
85
+ try {
86
+ await closing;
87
+ }
88
+ finally {
89
+ if (environments.get(this.#stateDir) === this.#shared)
90
+ environments.delete(this.#stateDir);
91
+ }
92
+ }
93
+ assertOpen() {
94
+ if (this.#closed)
95
+ throw new Error("The pi-fleet client is closed");
96
+ }
97
+ }
98
+ export async function openRegistry(stateDir) {
99
+ const canonicalStateDir = await prepareStateDir(stateDir);
100
+ let shared = environments.get(canonicalStateDir);
101
+ if (shared?.closing) {
102
+ await shared.closing;
103
+ shared = environments.get(canonicalStateDir);
104
+ }
105
+ if (!shared) {
106
+ const root = open({ path: canonicalStateDir, maxDbs: 3 });
107
+ shared = {
108
+ root,
109
+ agents: root.openDB("agents", { encoding: "json", useVersions: true }),
110
+ names: root.openDB("names", { encoding: "string" }),
111
+ references: 0,
112
+ };
113
+ environments.set(canonicalStateDir, shared);
114
+ }
115
+ shared.references += 1;
116
+ return new Registry(canonicalStateDir, shared);
117
+ }
118
+ //# sourceMappingURL=registry.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/internal/registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAoC,MAAM,MAAM,CAAA;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AAE5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAA;AAiCjD,MAAM,YAAY,GAAG,IAAI,GAAG,EAA6B,CAAA;AAEzD,MAAM,OAAO,QAAQ;IACV,SAAS,CAAQ;IACjB,OAAO,CAAmB;IAC1B,OAAO,CAA+B;IACtC,MAAM,CAA0B;IACzC,OAAO,GAAG,KAAK,CAAA;IAEf,YAAY,QAAgB,EAAE,MAAyB;QACrD,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAA;QACzB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,CAAA;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAA;IAC5B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAmB;QAC9B,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;YAC7D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;YACtC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAC1D,CAAC;IAED,OAAO,CAAC,EAAU;QAChB,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAC7B,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QAChC,OAAO,EAAE,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAC5D,CAAC;IAED,IAAI;QACF,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACxD,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CAAC,CAAC,CAAA;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU,EAAE,iBAAyB,EAAE,KAA2H;QAChL,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAA;QACvC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,EAAE,UAAU,KAAK,iBAAiB;YAAE,OAAO,KAAK,CAAA;QAEjF,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,KAAK,CAAA;QAC7D,MAAM,MAAM,GAAgB;YAC1B,GAAG,KAAK,CAAC,KAAK;YACd,SAAS;YACT,WAAW;YACX,KAAK,EAAE,MAAM;YACb,OAAO,EAAE;gBACP,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO;gBACtB,QAAQ;gBACR,SAAS;gBACT,KAAK,EAAE,OAAO;aACf;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAA;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,CAAC,CAAA;QAClC,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,CAAC,CAAA;IAC3D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,EAAU,EAAE,IAAY,EAAE,iBAAyB;QACxE,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE;YACjC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YACnC,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,MAAM,EAAE,OAAO,EAAE,UAAU,KAAK,iBAAiB;gBAAE,OAAM;YAC7F,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACvB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QAC1B,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO;YAAE,OAAM;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,CAAC,CAAA;QAC5B,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,CAAC;YAAE,OAAM;QAEvC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAA;QACzC,IAAI,CAAC,OAAO,CAAC,OAAO,GAAG,OAAO,CAAA;QAC9B,IAAI,CAAC;YACH,MAAM,OAAO,CAAA;QACf,CAAC;gBAAS,CAAC;YACT,IAAI,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC,OAAO;gBAAE,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC5F,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAA;IACpE,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAAgB;IACjD,MAAM,iBAAiB,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAA;IACzD,IAAI,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;IAChD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;QACpB,MAAM,MAAM,CAAC,OAAO,CAAA;QACpB,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;IAC9C,CAAC;IACD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAA;QACzD,MAAM,GAAG;YACP,IAAI;YACJ,MAAM,EAAE,IAAI,CAAC,MAAM,CAAsB,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;YAC3F,KAAK,EAAE,IAAI,CAAC,MAAM,CAAiB,OAAO,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;YACnE,UAAU,EAAE,CAAC;SACd,CAAA;QACD,YAAY,CAAC,GAAG,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAA;IAC7C,CAAC;IACD,MAAM,CAAC,UAAU,IAAI,CAAC,CAAA;IACtB,OAAO,IAAI,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAA;AAChD,CAAC"}
@@ -0,0 +1,4 @@
1
+ import { type StatusResponse } from "./protocol.js";
2
+ import type { AgentRecord } from "./registry.js";
3
+ export declare function requestStatus(record: AgentRecord, timeoutMs?: number): Promise<NonNullable<StatusResponse["status"]>>;
4
+ //# sourceMappingURL=worker-client.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"worker-client.d.ts","sourceRoot":"","sources":["../../src/internal/worker-client.ts"],"names":[],"mappings":"AAGA,OAAO,EAAsC,KAAK,cAAc,EAAE,MAAM,eAAe,CAAA;AACvF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAIhD,wBAAsB,aAAa,CAAC,MAAM,EAAE,WAAW,EAAE,SAAS,SAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAqC1H"}
@@ -0,0 +1,60 @@
1
+ import { Dealer } from "zeromq";
2
+ import { randomUUID } from "node:crypto";
3
+ import { AgentUnavailableError } from "../types.js";
4
+ import { decode, encode } from "./protocol.js";
5
+ const AGENT_STATES = new Set(["starting", "idle", "working", "stopped", "failed"]);
6
+ export async function requestStatus(record, timeoutMs = 1_000) {
7
+ const runtime = record.runtime;
8
+ if (!runtime?.endpoint)
9
+ throw new AgentUnavailableError(record.name);
10
+ const socket = new Dealer({ routingId: randomUUID(), immediate: true, linger: 0, sendTimeout: timeoutMs });
11
+ try {
12
+ socket.connect(runtime.endpoint);
13
+ const request = {
14
+ version: 1,
15
+ requestId: randomUUID(),
16
+ command: "status",
17
+ agentId: record.id,
18
+ runtimeGeneration: runtime.generation,
19
+ };
20
+ await socket.send(encode(request));
21
+ const frames = await withTimeout(socket.receive(), timeoutMs, record.name);
22
+ const response = decode(frames[0]);
23
+ if (!response ||
24
+ response.version !== 1 ||
25
+ response.requestId !== request.requestId ||
26
+ !response.ok ||
27
+ !response.status ||
28
+ response.status.id !== record.id ||
29
+ response.status.name !== record.name ||
30
+ response.status.runtimeGeneration !== runtime.generation ||
31
+ !AGENT_STATES.has(response.status.state)) {
32
+ throw new AgentUnavailableError(record.name);
33
+ }
34
+ return response.status;
35
+ }
36
+ catch (error) {
37
+ if (error instanceof AgentUnavailableError)
38
+ throw error;
39
+ throw new AgentUnavailableError(record.name);
40
+ }
41
+ finally {
42
+ socket.close();
43
+ }
44
+ }
45
+ async function withTimeout(promise, timeoutMs, name) {
46
+ let timeout;
47
+ try {
48
+ return await Promise.race([
49
+ promise,
50
+ new Promise((_, reject) => {
51
+ timeout = setTimeout(() => reject(new AgentUnavailableError(name)), timeoutMs);
52
+ }),
53
+ ]);
54
+ }
55
+ finally {
56
+ if (timeout)
57
+ clearTimeout(timeout);
58
+ }
59
+ }
60
+ //# sourceMappingURL=worker-client.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"worker-client.js","sourceRoot":"","sources":["../../src/internal/worker-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC/B,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AACxC,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,EAA2C,MAAM,eAAe,CAAA;AAGvF,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAA;AAElF,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,MAAmB,EAAE,SAAS,GAAG,KAAK;IACxE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAA;IAC9B,IAAI,CAAC,OAAO,EAAE,QAAQ;QAAE,MAAM,IAAI,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAEpE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAA;IAC1G,IAAI,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QAChC,MAAM,OAAO,GAAkB;YAC7B,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,UAAU,EAAE;YACvB,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,MAAM,CAAC,EAAE;YAClB,iBAAiB,EAAE,OAAO,CAAC,UAAU;SACtC,CAAA;QACD,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;QAClC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,CAAA;QAC1E,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAmB,CAAA;QACpD,IACE,CAAC,QAAQ;YACT,QAAQ,CAAC,OAAO,KAAK,CAAC;YACtB,QAAQ,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS;YACxC,CAAC,QAAQ,CAAC,EAAE;YACZ,CAAC,QAAQ,CAAC,MAAM;YAChB,QAAQ,CAAC,MAAM,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE;YAChC,QAAQ,CAAC,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI;YACpC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,KAAK,OAAO,CAAC,UAAU;YACxD,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,EACxC,CAAC;YACD,MAAM,IAAI,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QAC9C,CAAC;QACD,OAAO,QAAQ,CAAC,MAAM,CAAA;IACxB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,qBAAqB;YAAE,MAAM,KAAK,CAAA;QACvD,MAAM,IAAI,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAC9C,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,KAAK,EAAE,CAAA;IAChB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,WAAW,CAAI,OAAmB,EAAE,SAAiB,EAAE,IAAY;IAChF,IAAI,OAAmC,CAAA;IACvC,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC;YACxB,OAAO;YACP,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAC/B,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC,EAAE,SAAS,CAAC,CAAA;YAChF,CAAC,CAAC;SACH,CAAC,CAAA;IACJ,CAAC;YAAS,CAAC;QACT,IAAI,OAAO;YAAE,YAAY,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC;AACH,CAAC"}
@@ -0,0 +1,3 @@
1
+ import { type ChildProcess } from "node:child_process";
2
+ export declare function launchWorker(stateDir: string, agentId: string, generation: string): ChildProcess;
3
+ //# sourceMappingURL=worker-launcher.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"worker-launcher.d.ts","sourceRoot":"","sources":["../../src/internal/worker-launcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,KAAK,YAAY,EAAE,MAAM,oBAAoB,CAAA;AAG7D,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,YAAY,CAShG"}
@@ -0,0 +1,13 @@
1
+ import { spawn } from "node:child_process";
2
+ import { fileURLToPath } from "node:url";
3
+ export function launchWorker(stateDir, agentId, generation) {
4
+ const workerPath = fileURLToPath(new URL("./worker.js", import.meta.url));
5
+ const child = spawn(process.execPath, [workerPath, "--state-dir", stateDir, "--agent", agentId, "--generation", generation], {
6
+ detached: true,
7
+ stdio: "ignore",
8
+ env: process.env,
9
+ });
10
+ child.unref();
11
+ return child;
12
+ }
13
+ //# sourceMappingURL=worker-launcher.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"worker-launcher.js","sourceRoot":"","sources":["../../src/internal/worker-launcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,oBAAoB,CAAA;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AAExC,MAAM,UAAU,YAAY,CAAC,QAAgB,EAAE,OAAe,EAAE,UAAkB;IAChF,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;IACzE,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,CAAC,EAAE;QAC3H,QAAQ,EAAE,IAAI;QACd,KAAK,EAAE,QAAQ;QACf,GAAG,EAAE,OAAO,CAAC,GAAG;KACjB,CAAC,CAAA;IACF,KAAK,CAAC,KAAK,EAAE,CAAA;IACb,OAAO,KAAK,CAAA;AACd,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=worker.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../src/internal/worker.ts"],"names":[],"mappings":""}
@@ -0,0 +1,114 @@
1
+ import { Router, capability } from "zeromq";
2
+ import { createStateDirectories } from "./paths.js";
3
+ import { decode, encode } from "./protocol.js";
4
+ import { startPi } from "./pi-rpc.js";
5
+ import { openRegistry } from "./registry.js";
6
+ async function main() {
7
+ const { stateDir, agentId, generation } = parseArguments(process.argv.slice(2));
8
+ if (!capability.ipc)
9
+ throw new Error("pi-fleet requires ZeroMQ ipc:// support on this host");
10
+ const registry = await openRegistry(stateDir);
11
+ const record = registry.getById(agentId);
12
+ if (!record || record.runtime?.generation !== generation || !record.runtime.endpoint)
13
+ throw new Error("Worker claim is no longer current");
14
+ await createStateDirectories(stateDir);
15
+ const router = new Router({ mandatory: true, immediate: true, linger: 0, sendTimeout: 1_000 });
16
+ let routerClosed = false;
17
+ const closeRouter = () => {
18
+ if (routerClosed)
19
+ return;
20
+ routerClosed = true;
21
+ router.close();
22
+ };
23
+ let pi;
24
+ const stop = () => closeRouter();
25
+ process.once("SIGTERM", stop);
26
+ process.once("SIGINT", stop);
27
+ try {
28
+ await router.bind(record.runtime.endpoint);
29
+ pi = await startPi(record);
30
+ pi.process.once("exit", closeRouter);
31
+ const markedReady = await registry.markReady(agentId, generation, {
32
+ workerPid: process.pid,
33
+ endpoint: record.runtime.endpoint,
34
+ sessionPath: pi.state.sessionFile,
35
+ sessionId: pi.state.sessionId,
36
+ });
37
+ if (!markedReady)
38
+ throw new Error("Worker claim is no longer current");
39
+ for await (const [route, frame] of router) {
40
+ const response = statusResponse(frame, agentId, generation, record.name);
41
+ try {
42
+ await router.send([route, encode(response)]);
43
+ }
44
+ catch (error) {
45
+ if (!isUnreachablePeer(error))
46
+ throw error;
47
+ }
48
+ }
49
+ }
50
+ finally {
51
+ process.off("SIGTERM", stop);
52
+ process.off("SIGINT", stop);
53
+ pi?.process.off("exit", closeRouter);
54
+ closeRouter();
55
+ await stopPi(pi?.process);
56
+ await registry.close();
57
+ }
58
+ }
59
+ async function stopPi(pi) {
60
+ if (!pi || pi.exitCode !== null || pi.signalCode !== null)
61
+ return;
62
+ pi.kill("SIGTERM");
63
+ if (await waitForExit(pi, 1_000))
64
+ return;
65
+ pi.kill("SIGKILL");
66
+ await waitForExit(pi, 1_000);
67
+ }
68
+ async function waitForExit(pi, timeoutMs) {
69
+ if (pi.exitCode !== null || pi.signalCode !== null)
70
+ return true;
71
+ return new Promise((resolve) => {
72
+ const timeout = setTimeout(() => {
73
+ pi.off("exit", onExit);
74
+ resolve(false);
75
+ }, timeoutMs);
76
+ const onExit = () => {
77
+ clearTimeout(timeout);
78
+ resolve(true);
79
+ };
80
+ pi.once("exit", onExit);
81
+ });
82
+ }
83
+ function statusResponse(frame, agentId, generation, name) {
84
+ try {
85
+ const request = decode(frame);
86
+ if (request.version !== 1 || request.command !== "status" || request.agentId !== agentId || request.runtimeGeneration !== generation) {
87
+ return { version: 1, requestId: request?.requestId ?? "", ok: false, error: "Worker identity does not match the request" };
88
+ }
89
+ return { version: 1, requestId: request.requestId, ok: true, status: { id: agentId, name, runtimeGeneration: generation, state: "idle" } };
90
+ }
91
+ catch {
92
+ return { version: 1, requestId: "", ok: false, error: "Invalid request" };
93
+ }
94
+ }
95
+ function isUnreachablePeer(error) {
96
+ return typeof error === "object" && error !== null && "code" in error &&
97
+ (error.code === "EHOSTUNREACH" || error.code === "EAGAIN");
98
+ }
99
+ function parseArguments(args) {
100
+ const values = new Map();
101
+ for (let index = 0; index < args.length; index += 2)
102
+ values.set(args[index], args[index + 1]);
103
+ const stateDir = values.get("--state-dir");
104
+ const agentId = values.get("--agent");
105
+ const generation = values.get("--generation");
106
+ if (!stateDir || !agentId || !generation)
107
+ throw new Error("Worker requires --state-dir, --agent, and --generation");
108
+ return { stateDir, agentId, generation };
109
+ }
110
+ void main().catch((error) => {
111
+ process.stderr.write(`${error instanceof Error ? error.stack : String(error)}\n`);
112
+ process.exitCode = 1;
113
+ });
114
+ //# sourceMappingURL=worker.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"worker.js","sourceRoot":"","sources":["../../src/internal/worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AAC3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAA;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,EAA2C,MAAM,eAAe,CAAA;AACvF,OAAO,EAAE,OAAO,EAAE,MAAM,aAAa,CAAA;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA;AAI5C,KAAK,UAAU,IAAI;IACjB,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;IAC/E,IAAI,CAAC,UAAU,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAA;IAE5F,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,CAAA;IAC7C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;IACxC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,UAAU,KAAK,UAAU,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAA;IAE1I,MAAM,sBAAsB,CAAC,QAAQ,CAAC,CAAA;IACtC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAA;IAC9F,IAAI,YAAY,GAAG,KAAK,CAAA;IACxB,MAAM,WAAW,GAAG,GAAG,EAAE;QACvB,IAAI,YAAY;YAAE,OAAM;QACxB,YAAY,GAAG,IAAI,CAAA;QACnB,MAAM,CAAC,KAAK,EAAE,CAAA;IAChB,CAAC,CAAA;IACD,IAAI,EAAmD,CAAA;IACvD,MAAM,IAAI,GAAG,GAAG,EAAE,CAAC,WAAW,EAAE,CAAA;IAChC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAA;IAC7B,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;IAC5B,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QAC1C,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAA;QAC1B,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QACpC,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,UAAU,EAAE;YAChE,SAAS,EAAE,OAAO,CAAC,GAAG;YACtB,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ;YACjC,WAAW,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW;YACjC,SAAS,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS;SAC9B,CAAC,CAAA;QACF,IAAI,CAAC,WAAW;YAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAA;QAEtE,IAAI,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;YAC1C,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAA;YACxE,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;YAC9C,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;oBAAE,MAAM,KAAK,CAAA;YAC5C,CAAC;QACH,CAAC;IACH,CAAC;YAAS,CAAC;QACT,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAA;QAC5B,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QAC3B,EAAE,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QACpC,WAAW,EAAE,CAAA;QACb,MAAM,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,CAAA;QACzB,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAA;IACxB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,EAA8D;IAClF,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,QAAQ,KAAK,IAAI,IAAI,EAAE,CAAC,UAAU,KAAK,IAAI;QAAE,OAAM;IACjE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IAClB,IAAI,MAAM,WAAW,CAAC,EAAE,EAAE,KAAK,CAAC;QAAE,OAAM;IACxC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IAClB,MAAM,WAAW,CAAC,EAAE,EAAE,KAAK,CAAC,CAAA;AAC9B,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,EAAkD,EAAE,SAAiB;IAC9F,IAAI,EAAE,CAAC,QAAQ,KAAK,IAAI,IAAI,EAAE,CAAC,UAAU,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IAC/D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9B,EAAE,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;YACtB,OAAO,CAAC,KAAK,CAAC,CAAA;QAChB,CAAC,EAAE,SAAS,CAAC,CAAA;QACb,MAAM,MAAM,GAAG,GAAG,EAAE;YAClB,YAAY,CAAC,OAAO,CAAC,CAAA;YACrB,OAAO,CAAC,IAAI,CAAC,CAAA;QACf,CAAC,CAAA;QACD,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACzB,CAAC,CAAC,CAAA;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,KAAa,EAAE,OAAe,EAAE,UAAkB,EAAE,IAAY;IACtF,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAkB,CAAA;QAC9C,IAAI,OAAO,CAAC,OAAO,KAAK,CAAC,IAAI,OAAO,CAAC,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,OAAO,KAAK,OAAO,IAAI,OAAO,CAAC,iBAAiB,KAAK,UAAU,EAAE,CAAC;YACrI,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,4CAA4C,EAAE,CAAA;QAC5H,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAA;IAC5I,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAA;IAC3E,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK;QACnE,CAAE,KAA2B,CAAC,IAAI,KAAK,cAAc,IAAK,KAA2B,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAA;AAC5G,CAAC;AAED,SAAS,cAAc,CAAC,IAAc;IACpC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAA;IACxC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC;QAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAA;IAC7F,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAA;IAC1C,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;IACrC,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;IAC7C,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,IAAI,CAAC,UAAU;QAAE,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAA;IACnH,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,CAAA;AAC1C,CAAC;AAED,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IAC1B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IACjF,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAA;AACtB,CAAC,CAAC,CAAA"}
@@ -0,0 +1,43 @@
1
+ export type AgentState = "starting" | "idle" | "working" | "stopped" | "failed";
2
+ export type ConnectOptions = {
3
+ /** A private state directory. Intended for isolated tests and advanced local setups. */
4
+ stateDir?: string;
5
+ };
6
+ export type CreateAgentOptions = {
7
+ name: string;
8
+ cwd: string;
9
+ instructions?: string;
10
+ piArgs?: string[];
11
+ };
12
+ export type AgentSummary = {
13
+ id: string;
14
+ name: string;
15
+ cwd: string;
16
+ state: AgentState;
17
+ };
18
+ export type AgentStatus = {
19
+ id: string;
20
+ name: string;
21
+ state: AgentState;
22
+ };
23
+ export interface Agent {
24
+ readonly id: string;
25
+ readonly name: string;
26
+ status(): Promise<AgentStatus>;
27
+ }
28
+ export interface PiFleetClient {
29
+ create(options: CreateAgentOptions): Promise<Agent>;
30
+ get(name: string): Promise<Agent>;
31
+ list(): Promise<AgentSummary[]>;
32
+ close(): Promise<void>;
33
+ }
34
+ export declare class AgentNameTakenError extends Error {
35
+ constructor(name: string);
36
+ }
37
+ export declare class AgentNotFoundError extends Error {
38
+ constructor(name: string);
39
+ }
40
+ export declare class AgentUnavailableError extends Error {
41
+ constructor(name: string);
42
+ }
43
+ //# sourceMappingURL=types.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG,MAAM,GAAG,SAAS,GAAG,SAAS,GAAG,QAAQ,CAAA;AAE/E,MAAM,MAAM,cAAc,GAAG;IAC3B,wFAAwF;IACxF,QAAQ,CAAC,EAAE,MAAM,CAAA;CAClB,CAAA;AAED,MAAM,MAAM,kBAAkB,GAAG;IAC/B,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAA;CAClB,CAAA;AAED,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,KAAK,EAAE,UAAU,CAAA;CAClB,CAAA;AAED,MAAM,MAAM,WAAW,GAAG;IACxB,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,UAAU,CAAA;CAClB,CAAA;AAED,MAAM,WAAW,KAAK;IACpB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,MAAM,IAAI,OAAO,CAAC,WAAW,CAAC,CAAA;CAC/B;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,CAAA;IACnD,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAA;IACjC,IAAI,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC,CAAA;IAC/B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;CACvB;AAED,qBAAa,mBAAoB,SAAQ,KAAK;gBAChC,IAAI,EAAE,MAAM;CAIzB;AAED,qBAAa,kBAAmB,SAAQ,KAAK;gBAC/B,IAAI,EAAE,MAAM;CAIzB;AAED,qBAAa,qBAAsB,SAAQ,KAAK;gBAClC,IAAI,EAAE,MAAM;CAIzB"}
package/dist/types.js ADDED
@@ -0,0 +1,19 @@
1
+ export class AgentNameTakenError extends Error {
2
+ constructor(name) {
3
+ super(`An agent named ${JSON.stringify(name)} already exists`);
4
+ this.name = "AgentNameTakenError";
5
+ }
6
+ }
7
+ export class AgentNotFoundError extends Error {
8
+ constructor(name) {
9
+ super(`No agent named ${JSON.stringify(name)} exists`);
10
+ this.name = "AgentNotFoundError";
11
+ }
12
+ }
13
+ export class AgentUnavailableError extends Error {
14
+ constructor(name) {
15
+ super(`Agent ${JSON.stringify(name)} is unavailable`);
16
+ this.name = "AgentUnavailableError";
17
+ }
18
+ }
19
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAwCA,MAAM,OAAO,mBAAoB,SAAQ,KAAK;IAC5C,YAAY,IAAY;QACtB,KAAK,CAAC,kBAAkB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QAC9D,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAA;IACnC,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,KAAK;IAC3C,YAAY,IAAY;QACtB,KAAK,CAAC,kBAAkB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QACtD,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAA;IAClC,CAAC;CACF;AAED,MAAM,OAAO,qBAAsB,SAAQ,KAAK;IAC9C,YAAY,IAAY;QACtB,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QACrD,IAAI,CAAC,IAAI,GAAG,uBAAuB,CAAA;IACrC,CAAC;CACF"}
package/package.json CHANGED
@@ -1,17 +1,34 @@
1
1
  {
2
2
  "name": "@elpapi42/pi-fleet-sdk",
3
- "version": "0.2.0-beta.0",
4
- "description": "Reserved bootstrap version for the official pi-fleet SDK package",
3
+ "version": "0.2.0",
4
+ "description": "SDK for durable, host-local Pi agents",
5
5
  "license": "MIT",
6
- "private": false,
7
6
  "repository": {
8
7
  "type": "git",
9
- "url": "git+https://github.com/elpapi42/pi-fleet.git",
8
+ "url": "https://github.com/elpapi42/pi-fleet-v2.git",
10
9
  "directory": "packages/sdk"
11
10
  },
12
- "publishConfig": {
13
- "access": "public",
14
- "tag": "bootstrap"
11
+ "type": "module",
12
+ "engines": {
13
+ "node": ">=22"
15
14
  },
16
- "files": ["README.md", "LICENSE"]
15
+ "exports": {
16
+ ".": {
17
+ "types": "./dist/index.d.ts",
18
+ "import": "./dist/index.js"
19
+ }
20
+ },
21
+ "types": "./dist/index.d.ts",
22
+ "files": [
23
+ "dist",
24
+ "README.md"
25
+ ],
26
+ "scripts": {
27
+ "build": "tsc --project tsconfig.json",
28
+ "test": "node --test test/*.test.mjs"
29
+ },
30
+ "dependencies": {
31
+ "lmdb": "3.5.6",
32
+ "zeromq": "6.6.0"
33
+ }
17
34
  }
package/LICENSE DELETED
@@ -1,21 +0,0 @@
1
- MIT License
2
-
3
- Copyright (c) 2026 elpapi42
4
-
5
- Permission is hereby granted, free of charge, to any person obtaining a copy
6
- of this software and associated documentation files (the "Software"), to deal
7
- in the Software without restriction, including without limitation the rights
8
- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
- copies of the Software, and to permit persons to whom the Software is
10
- furnished to do so, subject to the following conditions:
11
-
12
- The above copyright notice and this permission notice shall be included in all
13
- copies or substantial portions of the Software.
14
-
15
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
- SOFTWARE.