@amaster.ai/pi-storage 0.1.0-beta.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 (73) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +56 -0
  3. package/dist/artifact-stores.d.ts +13 -0
  4. package/dist/artifact-stores.d.ts.map +1 -0
  5. package/dist/artifact-stores.js +84 -0
  6. package/dist/artifact-stores.js.map +1 -0
  7. package/dist/db-migrations.d.ts +12 -0
  8. package/dist/db-migrations.d.ts.map +1 -0
  9. package/dist/db-migrations.js +151 -0
  10. package/dist/db-migrations.js.map +1 -0
  11. package/dist/db-runtime-storage.d.ts +15 -0
  12. package/dist/db-runtime-storage.d.ts.map +1 -0
  13. package/dist/db-runtime-storage.js +1051 -0
  14. package/dist/db-runtime-storage.js.map +1 -0
  15. package/dist/db.d.ts +3 -0
  16. package/dist/db.d.ts.map +1 -0
  17. package/dist/db.js +3 -0
  18. package/dist/db.js.map +1 -0
  19. package/dist/event-stores.d.ts +69 -0
  20. package/dist/event-stores.d.ts.map +1 -0
  21. package/dist/event-stores.js +196 -0
  22. package/dist/event-stores.js.map +1 -0
  23. package/dist/index.d.ts +3 -0
  24. package/dist/index.d.ts.map +1 -0
  25. package/dist/index.js +2 -0
  26. package/dist/index.js.map +1 -0
  27. package/dist/internal.d.ts +4 -0
  28. package/dist/internal.d.ts.map +1 -0
  29. package/dist/internal.js +4 -0
  30. package/dist/internal.js.map +1 -0
  31. package/dist/json-file.d.ts +4 -0
  32. package/dist/json-file.d.ts.map +1 -0
  33. package/dist/json-file.js +59 -0
  34. package/dist/json-file.js.map +1 -0
  35. package/dist/json.d.ts +5 -0
  36. package/dist/json.d.ts.map +1 -0
  37. package/dist/json.js +5 -0
  38. package/dist/json.js.map +1 -0
  39. package/dist/migration-main.d.ts +2 -0
  40. package/dist/migration-main.d.ts.map +1 -0
  41. package/dist/migration-main.js +12 -0
  42. package/dist/migration-main.js.map +1 -0
  43. package/dist/redis-locks.d.ts +18 -0
  44. package/dist/redis-locks.d.ts.map +1 -0
  45. package/dist/redis-locks.js +76 -0
  46. package/dist/redis-locks.js.map +1 -0
  47. package/dist/runtime-scope.d.ts +2 -0
  48. package/dist/runtime-scope.d.ts.map +1 -0
  49. package/dist/runtime-scope.js +2 -0
  50. package/dist/runtime-scope.js.map +1 -0
  51. package/dist/runtime-storage.d.ts +30 -0
  52. package/dist/runtime-storage.d.ts.map +1 -0
  53. package/dist/runtime-storage.js +60 -0
  54. package/dist/runtime-storage.js.map +1 -0
  55. package/dist/scheduled-task-stores.d.ts +49 -0
  56. package/dist/scheduled-task-stores.d.ts.map +1 -0
  57. package/dist/scheduled-task-stores.js +381 -0
  58. package/dist/scheduled-task-stores.js.map +1 -0
  59. package/dist/scheduler.d.ts +2 -0
  60. package/dist/scheduler.d.ts.map +1 -0
  61. package/dist/scheduler.js +2 -0
  62. package/dist/scheduler.js.map +1 -0
  63. package/dist/session-stores.d.ts +47 -0
  64. package/dist/session-stores.d.ts.map +1 -0
  65. package/dist/session-stores.js +260 -0
  66. package/dist/session-stores.js.map +1 -0
  67. package/dist/subagent-store.d.ts +26 -0
  68. package/dist/subagent-store.d.ts.map +1 -0
  69. package/dist/subagent-store.js +167 -0
  70. package/dist/subagent-store.js.map +1 -0
  71. package/package.json +70 -0
  72. package/prisma/migrations/mysql/20260514113000_init_runtime/migration.sql +351 -0
  73. package/prisma/schema.prisma +351 -0
@@ -0,0 +1,2 @@
1
+ export { DbScheduledTaskStore, FileSchedulerLock, JsonScheduledTaskStore, RedisSchedulerLock, } from "./scheduled-task-stores.js";
2
+ //# sourceMappingURL=scheduler.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"scheduler.d.ts","sourceRoot":"","sources":["../src/scheduler.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,4BAA4B,CAAC"}
@@ -0,0 +1,2 @@
1
+ export { DbScheduledTaskStore, FileSchedulerLock, JsonScheduledTaskStore, RedisSchedulerLock, } from "./scheduled-task-stores.js";
2
+ //# sourceMappingURL=scheduler.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"scheduler.js","sourceRoot":"","sources":["../src/scheduler.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,4BAA4B,CAAC"}
@@ -0,0 +1,47 @@
1
+ import type { ConversationMessage, ConversationTurn, CopilotMemoryStore, JsonObject, MemoryRecord, RuntimeScope, RuntimeSession, RuntimeSessionStore, RuntimeSessionSummary, TranscriptStore } from "@amaster.ai/pi-types";
2
+ export declare class JsonFileConversationStore implements RuntimeSessionStore {
3
+ private readonly filePath;
4
+ private sessionsLoaded;
5
+ private readonly sessions;
6
+ private writeTail;
7
+ constructor(filePath: string);
8
+ getRuntimeSession(scope: RuntimeScope, sessionId: string): Promise<RuntimeSession | undefined>;
9
+ saveRuntimeSession(session: RuntimeSession): Promise<void>;
10
+ listRuntimeSessions(scope: RuntimeScope): Promise<RuntimeSession[]>;
11
+ private load;
12
+ private update;
13
+ }
14
+ export declare class JsonFileTranscriptStore implements TranscriptStore {
15
+ private readonly filePath;
16
+ private loaded;
17
+ private state;
18
+ private writeTail;
19
+ constructor(filePath: string);
20
+ appendTurn(turn: ConversationTurn): Promise<void>;
21
+ listTurns(scope: RuntimeScope, sessionId?: string): Promise<ConversationTurn[]>;
22
+ list(scope: RuntimeScope, sessionId?: string): Promise<ConversationTurn[]>;
23
+ listMessages(scope: RuntimeScope, sessionId: string): Promise<ConversationMessage[]>;
24
+ listSessionSummaries(scope: RuntimeScope, sessions: RuntimeSession[]): Promise<RuntimeSessionSummary[]>;
25
+ private update;
26
+ private load;
27
+ private save;
28
+ }
29
+ export declare class JsonFileMemoryStore implements CopilotMemoryStore {
30
+ private readonly filePath;
31
+ private writeTail;
32
+ constructor(filePath: string);
33
+ write(input: {
34
+ sessionId: string;
35
+ text: string;
36
+ tags?: string[];
37
+ metadata?: JsonObject;
38
+ }): Promise<MemoryRecord>;
39
+ search(input: {
40
+ sessionId: string;
41
+ query: string;
42
+ limit: number;
43
+ }): Promise<MemoryRecord[]>;
44
+ private load;
45
+ private update;
46
+ }
47
+ //# sourceMappingURL=session-stores.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"session-stores.d.ts","sourceRoot":"","sources":["../src/session-stores.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAGV,mBAAmB,EACnB,gBAAgB,EAChB,kBAAkB,EAClB,UAAU,EACV,YAAY,EAEZ,YAAY,EACZ,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,eAAe,EAChB,MAAM,sBAAsB,CAAC;AAY9B,qBAAa,yBAA0B,YAAW,mBAAmB;IAKvD,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAJrC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqC;IAC9D,OAAO,CAAC,SAAS,CAAuC;gBAE3B,QAAQ,EAAE,MAAM;IAEvC,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IAM9F,kBAAkB,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAM1D,mBAAmB,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YAK3D,IAAI;YAYJ,MAAM;CASrB;AAED,qBAAa,uBAAwB,YAAW,eAAe;IAKjD,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAJrC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAuE;IACpF,OAAO,CAAC,SAAS,CAAuC;gBAE3B,QAAQ,EAAE,MAAM;IAEvC,UAAU,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAQjD,SAAS,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAI/E,IAAI,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAO1E,YAAY,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAMpF,oBAAoB,CACxB,KAAK,EAAE,YAAY,EACnB,QAAQ,EAAE,cAAc,EAAE,GACzB,OAAO,CAAC,qBAAqB,EAAE,CAAC;YASrB,MAAM;YAUN,IAAI;YASJ,IAAI;CAGnB;AA2HD,qBAAa,mBAAoB,YAAW,kBAAkB;IAGhD,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAFrC,OAAO,CAAC,SAAS,CAAuC;gBAE3B,QAAQ,EAAE,MAAM;IAEvC,KAAK,CAAC,KAAK,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;QAChB,QAAQ,CAAC,EAAE,UAAU,CAAC;KACvB,GAAG,OAAO,CAAC,YAAY,CAAC;IAcnB,MAAM,CAAC,KAAK,EAAE;QAClB,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAmB3B,OAAO,CAAC,IAAI;YAIE,MAAM;CASrB"}
@@ -0,0 +1,260 @@
1
+ /**
2
+ * JSON-file stores for conversations, transcripts, and memory.
3
+ *
4
+ * Owns persistence adapters used by PiAgentRuntime and HTTP routes. Keep route
5
+ * validation, runtime event recording, and memory tool policy outside these
6
+ * stores.
7
+ */
8
+ import { randomUUID } from "node:crypto";
9
+ import { readJsonFile, writeJsonFile } from "./json-file.js";
10
+ export class JsonFileConversationStore {
11
+ filePath;
12
+ sessionsLoaded = false;
13
+ sessions = new Map();
14
+ writeTail = Promise.resolve();
15
+ constructor(filePath) {
16
+ this.filePath = filePath;
17
+ }
18
+ async getRuntimeSession(scope, sessionId) {
19
+ await this.load();
20
+ const session = this.sessions.get(sessionId);
21
+ return session && sessionMatchesScope(session, scope) ? session : undefined;
22
+ }
23
+ async saveRuntimeSession(session) {
24
+ await this.update(() => {
25
+ this.sessions.set(session.sessionId, session);
26
+ });
27
+ }
28
+ async listRuntimeSessions(scope) {
29
+ await this.load();
30
+ return [...this.sessions.values()].filter((session) => sessionMatchesScope(session, scope));
31
+ }
32
+ async load() {
33
+ if (this.sessionsLoaded) {
34
+ return;
35
+ }
36
+ const sessions = await readJsonFile(this.filePath, []);
37
+ this.sessions.clear();
38
+ for (const session of sessions) {
39
+ this.sessions.set(session.sessionId, session);
40
+ }
41
+ this.sessionsLoaded = true;
42
+ }
43
+ async update(mutator) {
44
+ const pending = this.writeTail.then(async () => {
45
+ await this.load();
46
+ mutator();
47
+ await writeJsonFile(this.filePath, [...this.sessions.values()]);
48
+ });
49
+ this.writeTail = pending.catch(() => undefined);
50
+ await pending;
51
+ }
52
+ }
53
+ export class JsonFileTranscriptStore {
54
+ filePath;
55
+ loaded = false;
56
+ state = createEmptyConversationHistoryState();
57
+ writeTail = Promise.resolve();
58
+ constructor(filePath) {
59
+ this.filePath = filePath;
60
+ }
61
+ async appendTurn(turn) {
62
+ await this.update((state) => {
63
+ state.turns.push(turn);
64
+ state.messages.push(...messagesForTurn(turn));
65
+ state.sessionSummaries[turn.sessionId] = nextSessionSummaryFields(state.sessionSummaries[turn.sessionId], turn);
66
+ });
67
+ }
68
+ async listTurns(scope, sessionId) {
69
+ return this.list(scope, sessionId);
70
+ }
71
+ async list(scope, sessionId) {
72
+ await this.load();
73
+ const turns = this.state.turns;
74
+ void scope;
75
+ return turns.filter((turn) => !sessionId || turn.sessionId === sessionId);
76
+ }
77
+ async listMessages(scope, sessionId) {
78
+ await this.load();
79
+ void scope;
80
+ return this.state.messages.filter((message) => message.sessionId === sessionId);
81
+ }
82
+ async listSessionSummaries(scope, sessions) {
83
+ await this.load();
84
+ return sessions.filter((session) => sessionMatchesScope(session, scope)).map((session) => ({
85
+ ...session,
86
+ turnCount: 0,
87
+ ...this.state.sessionSummaries[session.sessionId],
88
+ }));
89
+ }
90
+ async update(mutator) {
91
+ const pending = this.writeTail.then(async () => {
92
+ await this.load();
93
+ mutator(this.state);
94
+ await this.save();
95
+ });
96
+ this.writeTail = pending.catch(() => undefined);
97
+ await pending;
98
+ }
99
+ async load() {
100
+ if (this.loaded) {
101
+ return;
102
+ }
103
+ const raw = await readJsonFile(this.filePath, createEmptyConversationHistoryState());
104
+ this.state = isConversationHistoryState(raw) ? raw : createEmptyConversationHistoryState();
105
+ this.loaded = true;
106
+ }
107
+ async save() {
108
+ await writeJsonFile(this.filePath, this.state);
109
+ }
110
+ }
111
+ function sessionMatchesScope(session, scope) {
112
+ return runtimeTenantId(session.tenantId) === scope.tenantId &&
113
+ (!scope.userId || session.userId === scope.userId);
114
+ }
115
+ function runtimeTenantId(tenantId) {
116
+ return tenantId ?? "default";
117
+ }
118
+ function createEmptyConversationHistoryState() {
119
+ return {
120
+ version: 1,
121
+ turns: [],
122
+ messages: [],
123
+ sessionSummaries: {},
124
+ };
125
+ }
126
+ function isConversationHistoryState(raw) {
127
+ if (!isJsonObject(raw)) {
128
+ return false;
129
+ }
130
+ return (raw.version === 1 &&
131
+ Array.isArray(raw.turns) &&
132
+ raw.turns.every(isConversationTurn) &&
133
+ Array.isArray(raw.messages) &&
134
+ raw.messages.every(isConversationMessage) &&
135
+ isSummaryRecord(raw.sessionSummaries));
136
+ }
137
+ function messagesForTurn(turn) {
138
+ return [
139
+ {
140
+ id: `${turn.id}:user`,
141
+ sessionId: turn.sessionId,
142
+ conversationId: turn.conversationId,
143
+ turnId: turn.id,
144
+ role: "user",
145
+ text: turn.userMessage,
146
+ model: turn.model,
147
+ ...(turn.traceId ? { traceId: turn.traceId } : {}),
148
+ createdAt: turn.createdAt,
149
+ },
150
+ {
151
+ id: `${turn.id}:assistant`,
152
+ sessionId: turn.sessionId,
153
+ conversationId: turn.conversationId,
154
+ turnId: turn.id,
155
+ role: "assistant",
156
+ text: turn.assistantMessage,
157
+ model: turn.model,
158
+ ...(turn.traceId ? { traceId: turn.traceId } : {}),
159
+ createdAt: turn.createdAt,
160
+ },
161
+ ];
162
+ }
163
+ function nextSessionSummaryFields(previous, turn) {
164
+ const firstUserMessage = previous?.firstUserMessage ?? turn.userMessage;
165
+ const title = previous?.title ?? firstLine(firstUserMessage) ?? firstLine(turn.userMessage);
166
+ return {
167
+ turnCount: (previous?.turnCount ?? 0) + 1,
168
+ ...(title ? { title } : {}),
169
+ updatedAt: turn.createdAt,
170
+ firstUserMessage,
171
+ lastUserMessage: turn.userMessage,
172
+ lastAssistantMessage: turn.assistantMessage,
173
+ lastMessageAt: turn.createdAt,
174
+ };
175
+ }
176
+ function isConversationTurn(value) {
177
+ if (!isJsonObject(value)) {
178
+ return false;
179
+ }
180
+ return (typeof value.id === "string" &&
181
+ typeof value.sessionId === "string" &&
182
+ typeof value.conversationId === "string" &&
183
+ typeof value.userMessage === "string" &&
184
+ typeof value.assistantMessage === "string" &&
185
+ typeof value.createdAt === "string" &&
186
+ isJsonObject(value.model));
187
+ }
188
+ function isConversationMessage(value) {
189
+ if (!isJsonObject(value)) {
190
+ return false;
191
+ }
192
+ return (typeof value.id === "string" &&
193
+ typeof value.sessionId === "string" &&
194
+ typeof value.conversationId === "string" &&
195
+ typeof value.role === "string" &&
196
+ ["user", "assistant", "tool", "system"].includes(value.role) &&
197
+ typeof value.text === "string" &&
198
+ typeof value.createdAt === "string");
199
+ }
200
+ function isSummaryRecord(value) {
201
+ if (!isJsonObject(value)) {
202
+ return false;
203
+ }
204
+ return Object.values(value).every((entry) => isJsonObject(entry) && typeof entry.turnCount === "number");
205
+ }
206
+ function isJsonObject(value) {
207
+ return typeof value === "object" && value !== null && !Array.isArray(value);
208
+ }
209
+ function firstLine(value) {
210
+ return value?.split(/\r?\n/).find((line) => line.trim())?.trim().slice(0, 120) ?? "";
211
+ }
212
+ export class JsonFileMemoryStore {
213
+ filePath;
214
+ writeTail = Promise.resolve();
215
+ constructor(filePath) {
216
+ this.filePath = filePath;
217
+ }
218
+ async write(input) {
219
+ const record = {
220
+ id: randomUUID(),
221
+ text: input.text,
222
+ ...(input.tags ? { tags: input.tags } : {}),
223
+ createdAt: new Date().toISOString(),
224
+ ...(input.metadata ? { metadata: input.metadata } : {}),
225
+ };
226
+ await this.update((records) => {
227
+ records.push({ sessionId: input.sessionId, record });
228
+ });
229
+ return record;
230
+ }
231
+ async search(input) {
232
+ const query = input.query.trim().toLowerCase();
233
+ await this.writeTail.catch(() => undefined);
234
+ const records = await this.load();
235
+ return records
236
+ .filter((entry) => entry.sessionId === input.sessionId)
237
+ .map((entry) => entry.record)
238
+ .filter((record) => {
239
+ if (!query) {
240
+ return true;
241
+ }
242
+ return (record.text.toLowerCase().includes(query) ||
243
+ record.tags?.some((tag) => tag.toLowerCase().includes(query)));
244
+ })
245
+ .slice(0, input.limit);
246
+ }
247
+ load() {
248
+ return readJsonFile(this.filePath, []);
249
+ }
250
+ async update(mutator) {
251
+ const pending = this.writeTail.then(async () => {
252
+ const records = await this.load();
253
+ mutator(records);
254
+ await writeJsonFile(this.filePath, records);
255
+ });
256
+ this.writeTail = pending.catch(() => undefined);
257
+ await pending;
258
+ }
259
+ }
260
+ //# sourceMappingURL=session-stores.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"session-stores.js","sourceRoot":"","sources":["../src/session-stores.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAgBzC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAW7D,MAAM,OAAO,yBAAyB;IAKP;IAJrB,cAAc,GAAG,KAAK,CAAC;IACd,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IACtD,SAAS,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IAExD,YAA6B,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAEjD,KAAK,CAAC,iBAAiB,CAAC,KAAmB,EAAE,SAAiB;QAC5D,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,OAAO,OAAO,IAAI,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAAuB;QAC9C,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;YACrB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,KAAmB;QAC3C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IAC9F,CAAC;IAEO,KAAK,CAAC,IAAI;QAChB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAmB,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,OAAmB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,OAAO,EAAE,CAAC;YACV,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,OAAO,CAAC;IAChB,CAAC;CACF;AAED,MAAM,OAAO,uBAAuB;IAKL;IAJrB,MAAM,GAAG,KAAK,CAAC;IACf,KAAK,GAAiC,mCAAmC,EAAE,CAAC;IAC5E,SAAS,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IAExD,YAA6B,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAEjD,KAAK,CAAC,UAAU,CAAC,IAAsB;QACrC,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAC1B,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvB,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9C,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,wBAAwB,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;QAClH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,KAAmB,EAAE,SAAkB;QACrD,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAmB,EAAE,SAAkB;QAChD,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;QAC/B,KAAK,KAAK,CAAC;QACX,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAmB,EAAE,SAAiB;QACvD,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,KAAK,CAAC;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;IAClF,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,KAAmB,EACnB,QAA0B;QAE1B,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzF,GAAG,OAAO;YACV,SAAS,EAAE,CAAC;YACZ,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC;SAClD,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,OAAsD;QACzE,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,OAAO,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,IAAI;QAChB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,MAAM,YAAY,CAAU,IAAI,CAAC,QAAQ,EAAE,mCAAmC,EAAE,CAAC,CAAC;QAC9F,IAAI,CAAC,KAAK,GAAG,0BAA0B,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,mCAAmC,EAAE,CAAC;QAC3F,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,IAAI;QAChB,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;CACF;AAED,SAAS,mBAAmB,CAAC,OAAuB,EAAE,KAAmB;IACvE,OAAO,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,KAAK,CAAC,QAAQ;QACzD,CAAC,CAAC,KAAK,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,eAAe,CAAC,QAA4B;IACnD,OAAO,QAAQ,IAAI,SAAS,CAAC;AAC/B,CAAC;AAED,SAAS,mCAAmC;IAC1C,OAAO;QACL,OAAO,EAAE,CAAC;QACV,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,EAAE;QACZ,gBAAgB,EAAE,EAAE;KACrB,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CAAC,GAAY;IAC9C,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,CACL,GAAG,CAAC,OAAO,KAAK,CAAC;QACjB,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;QACxB,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,kBAAkB,CAAC;QACnC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;QAC3B,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,qBAAqB,CAAC;QACzC,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,CACtC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,IAAsB;IAC7C,OAAO;QACL;YACE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,IAAI,CAAC,WAAW;YACtB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClD,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B;QACD;YACE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,YAAY;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,IAAI,CAAC,gBAAgB;YAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClD,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B;KACF,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAC/B,QAAiD,EACjD,IAAsB;IAEtB,MAAM,gBAAgB,GAAG,QAAQ,EAAE,gBAAgB,IAAI,IAAI,CAAC,WAAW,CAAC;IACxE,MAAM,KAAK,GAAG,QAAQ,EAAE,KAAK,IAAI,SAAS,CAAC,gBAAgB,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC5F,OAAO;QACL,SAAS,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,CAAC,CAAC,GAAG,CAAC;QACzC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,gBAAgB;QAChB,eAAe,EAAE,IAAI,CAAC,WAAW;QACjC,oBAAoB,EAAE,IAAI,CAAC,gBAAgB;QAC3C,aAAa,EAAE,IAAI,CAAC,SAAS;KAC9B,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,CACL,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ;QAC5B,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;QACnC,OAAO,KAAK,CAAC,cAAc,KAAK,QAAQ;QACxC,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;QACrC,OAAO,KAAK,CAAC,gBAAgB,KAAK,QAAQ;QAC1C,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;QACnC,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAC1B,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,CACL,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ;QAC5B,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;QACnC,OAAO,KAAK,CAAC,cAAc,KAAK,QAAQ;QACxC,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAC9B,CAAC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;QAC5D,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAC9B,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,CACpC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC;AAC3G,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,SAAS,CAAC,KAAyB;IAC1C,OAAO,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;AACvF,CAAC;AAED,MAAM,OAAO,mBAAmB;IAGD;IAFrB,SAAS,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IAExD,YAA6B,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAEjD,KAAK,CAAC,KAAK,CAAC,KAKX;QACC,MAAM,MAAM,GAAiB;YAC3B,EAAE,EAAE,UAAU,EAAE;YAChB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACxD,CAAC;QACF,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YAC5B,OAAO,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAIZ;QACC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClC,OAAO,OAAO;aACX,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,KAAK,CAAC,SAAS,CAAC;aACtD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;aAC5B,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;YACjB,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,CACL,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAC9D,CAAC;QACJ,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAEO,IAAI;QACV,OAAO,YAAY,CAAqD,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC7F,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,OAA8E;QACjG,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClC,OAAO,CAAC,OAAO,CAAC,CAAC;YACjB,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,OAAO,CAAC;IAChB,CAAC;CACF"}
@@ -0,0 +1,26 @@
1
+ import type { RuntimeScope, SubagentRun, SubagentRunStatus, SubagentRunStore } from "@amaster.ai/pi-types";
2
+ type SubagentRunCreateInput = Parameters<SubagentRunStore["create"]>[0];
3
+ export declare class JsonFileSubagentRunStore implements SubagentRunStore {
4
+ private readonly filePath;
5
+ private loaded;
6
+ private readonly runs;
7
+ private writeTail;
8
+ constructor(filePath: string);
9
+ create(input: SubagentRunCreateInput): Promise<SubagentRun>;
10
+ list(_scope: RuntimeScope, parentSessionId?: string): Promise<SubagentRun[]>;
11
+ get(_scope: RuntimeScope, runId: string): Promise<SubagentRun | undefined>;
12
+ getDepthForSession(_scope: RuntimeScope, sessionId: string): Promise<number>;
13
+ countActiveChildren(_scope: RuntimeScope, parentSessionId: string): Promise<number>;
14
+ markRunning(_scope: RuntimeScope, runId: string): Promise<SubagentRun | undefined>;
15
+ markCompleted(_scope: RuntimeScope, runId: string, result: string): Promise<SubagentRun | undefined>;
16
+ markFailed(_scope: RuntimeScope, runId: string, error: string): Promise<SubagentRun | undefined>;
17
+ markCancelled(_scope: RuntimeScope, runId: string, reason?: string): Promise<SubagentRun | undefined>;
18
+ private patch;
19
+ private update;
20
+ private load;
21
+ private save;
22
+ }
23
+ export declare function isActiveSubagentStatus(status: SubagentRunStatus): boolean;
24
+ export declare function isTerminalSubagentStatus(status: SubagentRunStatus): boolean;
25
+ export {};
26
+ //# sourceMappingURL=subagent-store.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-store.d.ts","sourceRoot":"","sources":["../src/subagent-store.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EACV,YAAY,EAEZ,WAAW,EACX,iBAAiB,EACjB,gBAAgB,EACjB,MAAM,sBAAsB,CAAC;AAG9B,KAAK,sBAAsB,GAAG,UAAU,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAExE,qBAAa,wBAAyB,YAAW,gBAAgB;IAKnD,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAJrC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAkC;IACvD,OAAO,CAAC,SAAS,CAAuC;gBAE3B,QAAQ,EAAE,MAAM;IAEvC,MAAM,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAAC,WAAW,CAAC;IAgC3D,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAM5E,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;IAK1E,kBAAkB,CAAC,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAM5E,mBAAmB,CAAC,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOzF,WAAW,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;IAUlF,aAAa,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;IAgBpG,UAAU,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;IAgBhG,aAAa,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,SAAc,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;YAgB5F,KAAK;YAeL,MAAM;YAWN,IAAI;YAYJ,IAAI;CAGnB;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAEzE;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAE3E"}
@@ -0,0 +1,167 @@
1
+ /**
2
+ * JSON-file store for subagent run lifecycle state.
3
+ *
4
+ * Owns parent/child run records, status transitions, depth/count queries, and
5
+ * lifecycle event history. It does not execute subagent turns or enforce tool
6
+ * exposure policy.
7
+ */
8
+ import { randomUUID } from "node:crypto";
9
+ import { readJsonFile, writeJsonFile } from "./json-file.js";
10
+ export class JsonFileSubagentRunStore {
11
+ filePath;
12
+ loaded = false;
13
+ runs = new Map();
14
+ writeTail = Promise.resolve();
15
+ constructor(filePath) {
16
+ this.filePath = filePath;
17
+ }
18
+ async create(input) {
19
+ return await this.update(() => {
20
+ const now = new Date().toISOString();
21
+ const runId = randomUUID();
22
+ const run = {
23
+ runId,
24
+ taskRunId: input.taskRunId ?? runId,
25
+ ...(input.spawnBatchId ? { spawnBatchId: input.spawnBatchId } : {}),
26
+ ...(input.traceId ? { traceId: input.traceId } : {}),
27
+ parentSessionId: input.parentSessionId,
28
+ childSessionId: input.childSessionId,
29
+ ...(input.parentToolCallId ? { parentToolCallId: input.parentToolCallId } : {}),
30
+ task: input.task,
31
+ ...(input.agent ? { agent: input.agent } : {}),
32
+ ...(input.label ? { label: input.label } : {}),
33
+ status: "pending",
34
+ depth: input.depth,
35
+ model: input.model,
36
+ toolPolicyProfile: input.toolPolicyProfile,
37
+ context: input.context,
38
+ createdAt: now,
39
+ updatedAt: now,
40
+ events: [
41
+ { type: "subagent_spawning", at: now },
42
+ { type: "subagent_spawned", at: now },
43
+ ],
44
+ };
45
+ this.runs.set(run.runId, run);
46
+ return run;
47
+ });
48
+ }
49
+ async list(_scope, parentSessionId) {
50
+ await this.load();
51
+ const runs = [...this.runs.values()];
52
+ return parentSessionId ? runs.filter((run) => run.parentSessionId === parentSessionId) : runs;
53
+ }
54
+ async get(_scope, runId) {
55
+ await this.load();
56
+ return this.runs.get(runId);
57
+ }
58
+ async getDepthForSession(_scope, sessionId) {
59
+ await this.load();
60
+ const run = [...this.runs.values()].find((candidate) => candidate.childSessionId === sessionId);
61
+ return run?.depth ?? 0;
62
+ }
63
+ async countActiveChildren(_scope, parentSessionId) {
64
+ await this.load();
65
+ return [...this.runs.values()].filter((run) => run.parentSessionId === parentSessionId && isActiveSubagentStatus(run.status)).length;
66
+ }
67
+ markRunning(_scope, runId) {
68
+ return this.patch(runId, (run, now) => ({
69
+ ...run,
70
+ status: "running",
71
+ startedAt: run.startedAt ?? now,
72
+ updatedAt: now,
73
+ events: [...run.events, { type: "subagent_started", at: now }],
74
+ }));
75
+ }
76
+ markCompleted(_scope, runId, result) {
77
+ return this.patch(runId, (run, now) => {
78
+ if (run.status === "cancelled") {
79
+ return run;
80
+ }
81
+ return {
82
+ ...omitSubagentError(run),
83
+ status: "completed",
84
+ result,
85
+ endedAt: now,
86
+ updatedAt: now,
87
+ events: [...run.events, { type: "subagent_ended", at: now, reason: "completed" }],
88
+ };
89
+ });
90
+ }
91
+ markFailed(_scope, runId, error) {
92
+ return this.patch(runId, (run, now) => {
93
+ if (run.status === "cancelled") {
94
+ return run;
95
+ }
96
+ return {
97
+ ...run,
98
+ status: "failed",
99
+ error,
100
+ endedAt: now,
101
+ updatedAt: now,
102
+ events: [...run.events, { type: "subagent_ended", at: now, reason: "failed" }],
103
+ };
104
+ });
105
+ }
106
+ markCancelled(_scope, runId, reason = "cancelled") {
107
+ return this.patch(runId, (run, now) => {
108
+ if (isTerminalSubagentStatus(run.status)) {
109
+ return run;
110
+ }
111
+ return {
112
+ ...run,
113
+ status: "cancelled",
114
+ error: reason,
115
+ endedAt: now,
116
+ updatedAt: now,
117
+ events: [...run.events, { type: "subagent_ended", at: now, reason: "cancelled" }],
118
+ };
119
+ });
120
+ }
121
+ async patch(runId, mutator) {
122
+ return await this.update(() => {
123
+ const run = this.runs.get(runId);
124
+ if (!run) {
125
+ return undefined;
126
+ }
127
+ const next = mutator(run, new Date().toISOString());
128
+ this.runs.set(runId, next);
129
+ return next;
130
+ });
131
+ }
132
+ async update(mutator) {
133
+ const pending = this.writeTail.then(async () => {
134
+ await this.load();
135
+ const result = mutator();
136
+ await this.save();
137
+ return result;
138
+ });
139
+ this.writeTail = pending.catch(() => undefined);
140
+ return await pending;
141
+ }
142
+ async load() {
143
+ if (this.loaded) {
144
+ return;
145
+ }
146
+ const runs = await readJsonFile(this.filePath, []);
147
+ this.runs.clear();
148
+ for (const run of runs) {
149
+ this.runs.set(run.runId, run);
150
+ }
151
+ this.loaded = true;
152
+ }
153
+ async save() {
154
+ await writeJsonFile(this.filePath, [...this.runs.values()]);
155
+ }
156
+ }
157
+ export function isActiveSubagentStatus(status) {
158
+ return status === "pending" || status === "running";
159
+ }
160
+ export function isTerminalSubagentStatus(status) {
161
+ return status === "completed" || status === "failed" || status === "cancelled";
162
+ }
163
+ function omitSubagentError(run) {
164
+ const { error: _error, ...rest } = run;
165
+ return rest;
166
+ }
167
+ //# sourceMappingURL=subagent-store.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-store.js","sourceRoot":"","sources":["../src/subagent-store.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAQzC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAI7D,MAAM,OAAO,wBAAwB;IAKN;IAJrB,MAAM,GAAG,KAAK,CAAC;IACN,IAAI,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC/C,SAAS,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IAExD,YAA6B,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAEjD,KAAK,CAAC,MAAM,CAAC,KAA6B;QACxC,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;YAC5B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAgB;gBACvB,KAAK;gBACL,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,KAAK;gBACnC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpD,eAAe,EAAE,KAAK,CAAC,eAAe;gBACtC,cAAc,EAAE,KAAK,CAAC,cAAc;gBACpC,GAAG,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,KAAK,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/E,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9C,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9C,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;gBAC1C,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,SAAS,EAAE,GAAG;gBACd,SAAS,EAAE,GAAG;gBACd,MAAM,EAAE;oBACN,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,EAAE,GAAG,EAAE;oBACtC,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,GAAG,EAAE;iBACtC;aACF,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAC9B,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAoB,EAAE,eAAwB;QACvD,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACrC,OAAO,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,eAAe,KAAK,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChG,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAoB,EAAE,KAAa;QAC3C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAAoB,EAAE,SAAiB;QAC9D,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC;QAChG,OAAO,GAAG,EAAE,KAAK,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,MAAoB,EAAE,eAAuB;QACrE,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAC5C,GAAG,CAAC,eAAe,KAAK,eAAe,IAAI,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC,CAC9E,CAAC,MAAM,CAAC;IACX,CAAC;IAED,WAAW,CAAC,MAAoB,EAAE,KAAa;QAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,GAAG;YACN,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,GAAG;YAC/B,SAAS,EAAE,GAAG;YACd,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;SAC/D,CAAC,CAAC,CAAC;IACN,CAAC;IAED,aAAa,CAAC,MAAoB,EAAE,KAAa,EAAE,MAAc;QAC/D,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACpC,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAC/B,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO;gBACL,GAAG,iBAAiB,CAAC,GAAG,CAAC;gBACzB,MAAM,EAAE,WAAW;gBACnB,MAAM;gBACN,OAAO,EAAE,GAAG;gBACZ,SAAS,EAAE,GAAG;gBACd,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;aAClF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,MAAoB,EAAE,KAAa,EAAE,KAAa;QAC3D,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACpC,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAC/B,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO;gBACL,GAAG,GAAG;gBACN,MAAM,EAAE,QAAQ;gBAChB,KAAK;gBACL,OAAO,EAAE,GAAG;gBACZ,SAAS,EAAE,GAAG;gBACd,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;aAC/E,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,MAAoB,EAAE,KAAa,EAAE,MAAM,GAAG,WAAW;QACrE,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACpC,IAAI,wBAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzC,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO;gBACL,GAAG,GAAG;gBACN,MAAM,EAAE,WAAW;gBACnB,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,GAAG;gBACZ,SAAS,EAAE,GAAG;gBACd,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;aAClF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,KAAa,EACb,OAAuD;QAEvD,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;YAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;YACpD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAC3B,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,MAAM,CAAI,OAAgB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,MAAM,GAAG,OAAO,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAChD,OAAO,MAAM,OAAO,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,IAAI;QAChB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,YAAY,CAAgB,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,IAAI;QAChB,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAyB;IAC9D,OAAO,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAyB;IAChE,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,WAAW,CAAC;AACjF,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAgB;IACzC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,GAAG,CAAC;IACvC,OAAO,IAAI,CAAC;AACd,CAAC"}
package/package.json ADDED
@@ -0,0 +1,70 @@
1
+ {
2
+ "name": "@amaster.ai/pi-storage",
3
+ "version": "0.1.0-beta.0",
4
+ "license": "Apache-2.0",
5
+ "type": "module",
6
+ "sideEffects": false,
7
+ "main": "./dist/index.js",
8
+ "types": "./dist/index.d.ts",
9
+ "exports": {
10
+ ".": {
11
+ "types": "./dist/index.d.ts",
12
+ "default": "./dist/index.js"
13
+ },
14
+ "./json": {
15
+ "types": "./dist/json.d.ts",
16
+ "default": "./dist/json.js"
17
+ },
18
+ "./db": {
19
+ "types": "./dist/db.d.ts",
20
+ "default": "./dist/db.js"
21
+ },
22
+ "./scheduler": {
23
+ "types": "./dist/scheduler.d.ts",
24
+ "default": "./dist/scheduler.js"
25
+ },
26
+ "./internal": {
27
+ "types": "./dist/internal.d.ts",
28
+ "default": "./dist/internal.js"
29
+ },
30
+ "./prisma/schema.prisma": {
31
+ "default": "./prisma/schema.prisma"
32
+ },
33
+ "./package.json": {
34
+ "default": "./package.json"
35
+ }
36
+ },
37
+ "files": [
38
+ "dist",
39
+ "prisma",
40
+ "README.md"
41
+ ],
42
+ "publishConfig": {
43
+ "access": "public"
44
+ },
45
+ "repository": {
46
+ "type": "git",
47
+ "url": "https://github.com/TGYD-helige/pi.git",
48
+ "directory": "packages/storage"
49
+ },
50
+ "bin": {
51
+ "pi-storage-migrate": "./dist/migration-main.js"
52
+ },
53
+ "dependencies": {
54
+ "@prisma/client": "^6.19.3",
55
+ "ioredis": "^5.10.1",
56
+ "@amaster.ai/pi-task-scheduler": "0.1.0-beta.0",
57
+ "@amaster.ai/pi-types": "0.1.0-beta.0"
58
+ },
59
+ "devDependencies": {
60
+ "prisma": "^6.19.3",
61
+ "vitest": "^4.0.0"
62
+ },
63
+ "scripts": {
64
+ "build": "pnpm run prisma:generate && tsc -b",
65
+ "db:migrate": "node ./dist/migration-main.js",
66
+ "prisma:generate": "prisma generate --schema prisma/schema.prisma",
67
+ "typecheck": "pnpm run prisma:generate && tsc -b --pretty false",
68
+ "test": "vitest run"
69
+ }
70
+ }