@onkernel/cua-agent 0.1.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 (123) hide show
  1. package/CHANGELOG.md +6 -0
  2. package/README.md +44 -19
  3. package/dist/agent.d.ts +83 -27
  4. package/dist/agent.d.ts.map +1 -1
  5. package/dist/agent.js +176 -74
  6. package/dist/agent.js.map +1 -1
  7. package/dist/index.d.ts +3 -3
  8. package/dist/index.d.ts.map +1 -1
  9. package/dist/index.js +2 -2
  10. package/dist/index.js.map +1 -1
  11. package/dist/tools.d.ts +1 -1
  12. package/dist/tools.d.ts.map +1 -1
  13. package/dist/tools.js.map +1 -1
  14. package/dist/translator/translator.js +63 -6
  15. package/dist/translator/translator.js.map +1 -1
  16. package/dist/vendor/pi-agent-core/agent-loop.d.ts +24 -0
  17. package/dist/vendor/pi-agent-core/agent-loop.d.ts.map +1 -0
  18. package/dist/vendor/pi-agent-core/agent-loop.js +479 -0
  19. package/dist/vendor/pi-agent-core/agent-loop.js.map +1 -0
  20. package/dist/vendor/pi-agent-core/agent.d.ts +125 -0
  21. package/dist/vendor/pi-agent-core/agent.d.ts.map +1 -0
  22. package/dist/vendor/pi-agent-core/agent.js +402 -0
  23. package/dist/vendor/pi-agent-core/agent.js.map +1 -0
  24. package/dist/vendor/pi-agent-core/harness/agent-harness.d.ts +85 -0
  25. package/dist/vendor/pi-agent-core/harness/agent-harness.d.ts.map +1 -0
  26. package/dist/vendor/pi-agent-core/harness/agent-harness.js +728 -0
  27. package/dist/vendor/pi-agent-core/harness/agent-harness.js.map +1 -0
  28. package/dist/vendor/pi-agent-core/harness/compaction/branch-summarization.d.ts +88 -0
  29. package/dist/vendor/pi-agent-core/harness/compaction/branch-summarization.d.ts.map +1 -0
  30. package/dist/vendor/pi-agent-core/harness/compaction/branch-summarization.js +243 -0
  31. package/dist/vendor/pi-agent-core/harness/compaction/branch-summarization.js.map +1 -0
  32. package/dist/vendor/pi-agent-core/harness/compaction/compaction.d.ts +122 -0
  33. package/dist/vendor/pi-agent-core/harness/compaction/compaction.d.ts.map +1 -0
  34. package/dist/vendor/pi-agent-core/harness/compaction/compaction.js +616 -0
  35. package/dist/vendor/pi-agent-core/harness/compaction/compaction.js.map +1 -0
  36. package/dist/vendor/pi-agent-core/harness/compaction/utils.d.ts +38 -0
  37. package/dist/vendor/pi-agent-core/harness/compaction/utils.d.ts.map +1 -0
  38. package/dist/vendor/pi-agent-core/harness/compaction/utils.js +153 -0
  39. package/dist/vendor/pi-agent-core/harness/compaction/utils.js.map +1 -0
  40. package/dist/vendor/pi-agent-core/harness/env/nodejs.d.ts +44 -0
  41. package/dist/vendor/pi-agent-core/harness/env/nodejs.d.ts.map +1 -0
  42. package/dist/vendor/pi-agent-core/harness/env/nodejs.js +348 -0
  43. package/dist/vendor/pi-agent-core/harness/env/nodejs.js.map +1 -0
  44. package/dist/vendor/pi-agent-core/harness/execution-env.d.ts +4 -0
  45. package/dist/vendor/pi-agent-core/harness/execution-env.d.ts.map +1 -0
  46. package/dist/vendor/pi-agent-core/harness/execution-env.js +3 -0
  47. package/dist/vendor/pi-agent-core/harness/execution-env.js.map +1 -0
  48. package/dist/vendor/pi-agent-core/harness/messages.d.ts +51 -0
  49. package/dist/vendor/pi-agent-core/harness/messages.d.ts.map +1 -0
  50. package/dist/vendor/pi-agent-core/harness/messages.js +102 -0
  51. package/dist/vendor/pi-agent-core/harness/messages.js.map +1 -0
  52. package/dist/vendor/pi-agent-core/harness/prompt-templates.d.ts +45 -0
  53. package/dist/vendor/pi-agent-core/harness/prompt-templates.d.ts.map +1 -0
  54. package/dist/vendor/pi-agent-core/harness/prompt-templates.js +200 -0
  55. package/dist/vendor/pi-agent-core/harness/prompt-templates.js.map +1 -0
  56. package/dist/vendor/pi-agent-core/harness/session/repo/jsonl.d.ts +20 -0
  57. package/dist/vendor/pi-agent-core/harness/session/repo/jsonl.d.ts.map +1 -0
  58. package/dist/vendor/pi-agent-core/harness/session/repo/jsonl.js +92 -0
  59. package/dist/vendor/pi-agent-core/harness/session/repo/jsonl.js.map +1 -0
  60. package/dist/vendor/pi-agent-core/harness/session/repo/memory.d.ts +18 -0
  61. package/dist/vendor/pi-agent-core/harness/session/repo/memory.d.ts.map +1 -0
  62. package/dist/vendor/pi-agent-core/harness/session/repo/memory.js +42 -0
  63. package/dist/vendor/pi-agent-core/harness/session/repo/memory.js.map +1 -0
  64. package/dist/vendor/pi-agent-core/harness/session/repo/shared.d.ts +10 -0
  65. package/dist/vendor/pi-agent-core/harness/session/repo/shared.d.ts.map +1 -0
  66. package/dist/vendor/pi-agent-core/harness/session/repo/shared.js +31 -0
  67. package/dist/vendor/pi-agent-core/harness/session/repo/shared.js.map +1 -0
  68. package/dist/vendor/pi-agent-core/harness/session/session.d.ts +32 -0
  69. package/dist/vendor/pi-agent-core/harness/session/session.d.ts.map +1 -0
  70. package/dist/vendor/pi-agent-core/harness/session/session.js +196 -0
  71. package/dist/vendor/pi-agent-core/harness/session/session.js.map +1 -0
  72. package/dist/vendor/pi-agent-core/harness/session/storage/jsonl.d.ts +30 -0
  73. package/dist/vendor/pi-agent-core/harness/session/storage/jsonl.d.ts.map +1 -0
  74. package/dist/vendor/pi-agent-core/harness/session/storage/jsonl.js +170 -0
  75. package/dist/vendor/pi-agent-core/harness/session/storage/jsonl.js.map +1 -0
  76. package/dist/vendor/pi-agent-core/harness/session/storage/memory.d.ts +26 -0
  77. package/dist/vendor/pi-agent-core/harness/session/storage/memory.d.ts.map +1 -0
  78. package/dist/vendor/pi-agent-core/harness/session/storage/memory.js +90 -0
  79. package/dist/vendor/pi-agent-core/harness/session/storage/memory.js.map +1 -0
  80. package/dist/vendor/pi-agent-core/harness/session/uuid.d.ts +2 -0
  81. package/dist/vendor/pi-agent-core/harness/session/uuid.d.ts.map +1 -0
  82. package/dist/vendor/pi-agent-core/harness/session/uuid.js +40 -0
  83. package/dist/vendor/pi-agent-core/harness/session/uuid.js.map +1 -0
  84. package/dist/vendor/pi-agent-core/harness/skills.d.ts +41 -0
  85. package/dist/vendor/pi-agent-core/harness/skills.d.ts.map +1 -0
  86. package/dist/vendor/pi-agent-core/harness/skills.js +259 -0
  87. package/dist/vendor/pi-agent-core/harness/skills.js.map +1 -0
  88. package/dist/vendor/pi-agent-core/harness/system-prompt.d.ts +3 -0
  89. package/dist/vendor/pi-agent-core/harness/system-prompt.d.ts.map +1 -0
  90. package/dist/vendor/pi-agent-core/harness/system-prompt.js +30 -0
  91. package/dist/vendor/pi-agent-core/harness/system-prompt.js.map +1 -0
  92. package/dist/vendor/pi-agent-core/harness/types.d.ts +525 -0
  93. package/dist/vendor/pi-agent-core/harness/types.d.ts.map +1 -0
  94. package/dist/vendor/pi-agent-core/harness/types.js +16 -0
  95. package/dist/vendor/pi-agent-core/harness/types.js.map +1 -0
  96. package/dist/vendor/pi-agent-core/harness/utils/shell-output.d.ts +14 -0
  97. package/dist/vendor/pi-agent-core/harness/utils/shell-output.d.ts.map +1 -0
  98. package/dist/vendor/pi-agent-core/harness/utils/shell-output.js +97 -0
  99. package/dist/vendor/pi-agent-core/harness/utils/shell-output.js.map +1 -0
  100. package/dist/vendor/pi-agent-core/harness/utils/truncate.d.ts +70 -0
  101. package/dist/vendor/pi-agent-core/harness/utils/truncate.d.ts.map +1 -0
  102. package/dist/vendor/pi-agent-core/harness/utils/truncate.js +205 -0
  103. package/dist/vendor/pi-agent-core/harness/utils/truncate.js.map +1 -0
  104. package/dist/vendor/pi-agent-core/index.d.ts +21 -0
  105. package/dist/vendor/pi-agent-core/index.d.ts.map +1 -0
  106. package/dist/vendor/pi-agent-core/index.js +26 -0
  107. package/dist/vendor/pi-agent-core/index.js.map +1 -0
  108. package/dist/vendor/pi-agent-core/proxy.d.ts +69 -0
  109. package/dist/vendor/pi-agent-core/proxy.d.ts.map +1 -0
  110. package/dist/vendor/pi-agent-core/proxy.js +278 -0
  111. package/dist/vendor/pi-agent-core/proxy.js.map +1 -0
  112. package/dist/vendor/pi-agent-core/types.d.ts +386 -0
  113. package/dist/vendor/pi-agent-core/types.d.ts.map +1 -0
  114. package/dist/vendor/pi-agent-core/types.js +2 -0
  115. package/dist/vendor/pi-agent-core/types.js.map +1 -0
  116. package/examples/agent-openai-smoke.ts +4 -10
  117. package/examples/agent-provider-matrix.ts +4 -0
  118. package/examples/harness-openai-smoke.ts +17 -16
  119. package/examples/harness-provider-matrix.ts +16 -4
  120. package/examples/shared/logging.ts +37 -0
  121. package/package.json +6 -4
  122. package/src/vendor/pi-agent-core/LICENSE +21 -0
  123. package/src/vendor/pi-agent-core/README.md +15 -0
@@ -0,0 +1,31 @@
1
+ import { Session } from "../session.js";
2
+ import { uuidv7 } from "../uuid.js";
3
+ export function createSessionId() {
4
+ return uuidv7();
5
+ }
6
+ export function createTimestamp() {
7
+ return new Date().toISOString();
8
+ }
9
+ export function toSession(storage) {
10
+ return new Session(storage);
11
+ }
12
+ export async function getEntriesToFork(storage, options) {
13
+ if (!options.entryId)
14
+ return storage.getEntries();
15
+ const target = await storage.getEntry(options.entryId);
16
+ if (!target) {
17
+ throw new Error(`Entry ${options.entryId} not found`);
18
+ }
19
+ let effectiveLeafId;
20
+ if ((options.position ?? "before") === "at") {
21
+ effectiveLeafId = target.id;
22
+ }
23
+ else {
24
+ if (target.type !== "message" || target.message.role !== "user") {
25
+ throw new Error(`Entry ${options.entryId} is not a user message`);
26
+ }
27
+ effectiveLeafId = target.parentId;
28
+ }
29
+ return storage.getPathToRoot(effectiveLeafId);
30
+ }
31
+ //# sourceMappingURL=shared.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../../../../src/vendor/pi-agent-core/harness/session/repo/shared.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEpC,MAAM,UAAU,eAAe;IAC9B,OAAO,MAAM,EAAE,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,eAAe;IAC9B,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,SAAS,CAAoC,OAAkC;IAC9F,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;AAC7B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACrC,OAAuB,EACvB,OAAyD;IAEzD,IAAI,CAAC,OAAO,CAAC,OAAO;QAAE,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;IAClD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACvD,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,SAAS,OAAO,CAAC,OAAO,YAAY,CAAC,CAAC;IACvD,CAAC;IACD,IAAI,eAA8B,CAAC;IACnC,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;QAC7C,eAAe,GAAG,MAAM,CAAC,EAAE,CAAC;IAC7B,CAAC;SAAM,CAAC;QACP,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,SAAS,OAAO,CAAC,OAAO,wBAAwB,CAAC,CAAC;QACnE,CAAC;QACD,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC;IACnC,CAAC;IACD,OAAO,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;AAC/C,CAAC"}
@@ -0,0 +1,32 @@
1
+ import type { ImageContent, TextContent } from "@earendil-works/pi-ai";
2
+ import type { AgentMessage } from "../../types.js";
3
+ import type { SessionContext, SessionMetadata, SessionStorage, SessionTreeEntry } from "../types.js";
4
+ export declare function buildSessionContext(pathEntries: SessionTreeEntry[]): SessionContext;
5
+ export declare class Session<TMetadata extends SessionMetadata = SessionMetadata> {
6
+ private storage;
7
+ constructor(storage: SessionStorage<TMetadata>);
8
+ getMetadata(): Promise<TMetadata>;
9
+ getStorage(): SessionStorage<TMetadata>;
10
+ getLeafId(): Promise<string | null>;
11
+ getEntry(id: string): Promise<SessionTreeEntry | undefined>;
12
+ getEntries(): Promise<SessionTreeEntry[]>;
13
+ getBranch(fromId?: string): Promise<SessionTreeEntry[]>;
14
+ buildContext(): Promise<SessionContext>;
15
+ getLabel(id: string): Promise<string | undefined>;
16
+ getSessionName(): Promise<string | undefined>;
17
+ private appendTypedEntry;
18
+ appendMessage(message: AgentMessage): Promise<string>;
19
+ appendThinkingLevelChange(thinkingLevel: string): Promise<string>;
20
+ appendModelChange(provider: string, modelId: string): Promise<string>;
21
+ appendCompaction<T = unknown>(summary: string, firstKeptEntryId: string, tokensBefore: number, details?: T, fromHook?: boolean): Promise<string>;
22
+ appendCustomEntry(customType: string, data?: unknown): Promise<string>;
23
+ appendCustomMessageEntry<T = unknown>(customType: string, content: string | (TextContent | ImageContent)[], display: boolean, details?: T): Promise<string>;
24
+ appendLabel(targetId: string, label: string | undefined): Promise<string>;
25
+ appendSessionName(name: string): Promise<string>;
26
+ moveTo(entryId: string | null, summary?: {
27
+ summary: string;
28
+ details?: unknown;
29
+ fromHook?: boolean;
30
+ }): Promise<string | undefined>;
31
+ }
32
+ //# sourceMappingURL=session.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../../../src/vendor/pi-agent-core/harness/session/session.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAEnD,OAAO,KAAK,EAQX,cAAc,EAEd,eAAe,EACf,cAAc,EACd,gBAAgB,EAEhB,MAAM,aAAa,CAAC;AAErB,wBAAgB,mBAAmB,CAAC,WAAW,EAAE,gBAAgB,EAAE,GAAG,cAAc,CAuDnF;AAED,qBAAa,OAAO,CAAC,SAAS,SAAS,eAAe,GAAG,eAAe;IACvE,OAAO,CAAC,OAAO,CAA4B;gBAE/B,OAAO,EAAE,cAAc,CAAC,SAAS,CAAC;IAI9C,WAAW,IAAI,OAAO,CAAC,SAAS,CAAC;IAIjC,UAAU,IAAI,cAAc,CAAC,SAAS,CAAC;IAIvC,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAInC,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAI3D,UAAU,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAInC,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAKvD,YAAY,IAAI,OAAO,CAAC,cAAc,CAAC;IAI7C,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAI3C,cAAc,IAAI,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;YAKrC,gBAAgB;IAKxB,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;IAUrD,yBAAyB,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAUjE,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAWrE,gBAAgB,CAAC,CAAC,GAAG,OAAO,EACjC,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,MAAM,EACxB,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,CAAC,EACX,QAAQ,CAAC,EAAE,OAAO,GAChB,OAAO,CAAC,MAAM,CAAC;IAcZ,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAWtE,wBAAwB,CAAC,CAAC,GAAG,OAAO,EACzC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,EAChD,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,CAAC,GACT,OAAO,CAAC,MAAM,CAAC;IAaZ,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC;IAczE,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAUhD,MAAM,CACX,OAAO,EAAE,MAAM,GAAG,IAAI,EACtB,OAAO,CAAC,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GAClE,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;CAiB9B"}
@@ -0,0 +1,196 @@
1
+ import { createBranchSummaryMessage, createCompactionSummaryMessage, createCustomMessage } from "../messages.js";
2
+ export function buildSessionContext(pathEntries) {
3
+ let thinkingLevel = "off";
4
+ let model = null;
5
+ let compaction = null;
6
+ for (const entry of pathEntries) {
7
+ if (entry.type === "thinking_level_change") {
8
+ thinkingLevel = entry.thinkingLevel;
9
+ }
10
+ else if (entry.type === "model_change") {
11
+ model = { provider: entry.provider, modelId: entry.modelId };
12
+ }
13
+ else if (entry.type === "message" && entry.message.role === "assistant") {
14
+ model = { provider: entry.message.provider, modelId: entry.message.model };
15
+ }
16
+ else if (entry.type === "compaction") {
17
+ compaction = entry;
18
+ }
19
+ }
20
+ const messages = [];
21
+ const appendMessage = (entry) => {
22
+ if (entry.type === "message") {
23
+ messages.push(entry.message);
24
+ }
25
+ else if (entry.type === "custom_message") {
26
+ messages.push(createCustomMessage(entry.customType, entry.content, entry.display, entry.details, entry.timestamp));
27
+ }
28
+ else if (entry.type === "branch_summary" && entry.summary) {
29
+ messages.push(createBranchSummaryMessage(entry.summary, entry.fromId, entry.timestamp));
30
+ }
31
+ };
32
+ if (compaction) {
33
+ messages.push(createCompactionSummaryMessage(compaction.summary, compaction.tokensBefore, compaction.timestamp));
34
+ const compactionIdx = pathEntries.findIndex((e) => e.type === "compaction" && e.id === compaction.id);
35
+ let foundFirstKept = false;
36
+ for (let i = 0; i < compactionIdx; i++) {
37
+ const entry = pathEntries[i];
38
+ if (entry.id === compaction.firstKeptEntryId)
39
+ foundFirstKept = true;
40
+ if (foundFirstKept)
41
+ appendMessage(entry);
42
+ }
43
+ for (let i = compactionIdx + 1; i < pathEntries.length; i++) {
44
+ appendMessage(pathEntries[i]);
45
+ }
46
+ }
47
+ else {
48
+ for (const entry of pathEntries) {
49
+ appendMessage(entry);
50
+ }
51
+ }
52
+ return { messages, thinkingLevel, model };
53
+ }
54
+ export class Session {
55
+ storage;
56
+ constructor(storage) {
57
+ this.storage = storage;
58
+ }
59
+ getMetadata() {
60
+ return this.storage.getMetadata();
61
+ }
62
+ getStorage() {
63
+ return this.storage;
64
+ }
65
+ getLeafId() {
66
+ return this.storage.getLeafId();
67
+ }
68
+ getEntry(id) {
69
+ return this.storage.getEntry(id);
70
+ }
71
+ getEntries() {
72
+ return this.storage.getEntries();
73
+ }
74
+ async getBranch(fromId) {
75
+ const leafId = fromId ?? (await this.storage.getLeafId());
76
+ return this.storage.getPathToRoot(leafId);
77
+ }
78
+ async buildContext() {
79
+ return buildSessionContext(await this.getBranch());
80
+ }
81
+ getLabel(id) {
82
+ return this.storage.getLabel(id);
83
+ }
84
+ async getSessionName() {
85
+ const entries = await this.storage.findEntries("session_info");
86
+ return entries[entries.length - 1]?.name?.trim() || undefined;
87
+ }
88
+ async appendTypedEntry(entry) {
89
+ await this.storage.appendEntry(entry);
90
+ return entry.id;
91
+ }
92
+ async appendMessage(message) {
93
+ return this.appendTypedEntry({
94
+ type: "message",
95
+ id: await this.storage.createEntryId(),
96
+ parentId: await this.storage.getLeafId(),
97
+ timestamp: new Date().toISOString(),
98
+ message,
99
+ });
100
+ }
101
+ async appendThinkingLevelChange(thinkingLevel) {
102
+ return this.appendTypedEntry({
103
+ type: "thinking_level_change",
104
+ id: await this.storage.createEntryId(),
105
+ parentId: await this.storage.getLeafId(),
106
+ timestamp: new Date().toISOString(),
107
+ thinkingLevel,
108
+ });
109
+ }
110
+ async appendModelChange(provider, modelId) {
111
+ return this.appendTypedEntry({
112
+ type: "model_change",
113
+ id: await this.storage.createEntryId(),
114
+ parentId: await this.storage.getLeafId(),
115
+ timestamp: new Date().toISOString(),
116
+ provider,
117
+ modelId,
118
+ });
119
+ }
120
+ async appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromHook) {
121
+ return this.appendTypedEntry({
122
+ type: "compaction",
123
+ id: await this.storage.createEntryId(),
124
+ parentId: await this.storage.getLeafId(),
125
+ timestamp: new Date().toISOString(),
126
+ summary,
127
+ firstKeptEntryId,
128
+ tokensBefore,
129
+ details,
130
+ fromHook,
131
+ });
132
+ }
133
+ async appendCustomEntry(customType, data) {
134
+ return this.appendTypedEntry({
135
+ type: "custom",
136
+ id: await this.storage.createEntryId(),
137
+ parentId: await this.storage.getLeafId(),
138
+ timestamp: new Date().toISOString(),
139
+ customType,
140
+ data,
141
+ });
142
+ }
143
+ async appendCustomMessageEntry(customType, content, display, details) {
144
+ return this.appendTypedEntry({
145
+ type: "custom_message",
146
+ id: await this.storage.createEntryId(),
147
+ parentId: await this.storage.getLeafId(),
148
+ timestamp: new Date().toISOString(),
149
+ customType,
150
+ content,
151
+ display,
152
+ details,
153
+ });
154
+ }
155
+ async appendLabel(targetId, label) {
156
+ if (!(await this.storage.getEntry(targetId))) {
157
+ throw new Error(`Entry ${targetId} not found`);
158
+ }
159
+ return this.appendTypedEntry({
160
+ type: "label",
161
+ id: await this.storage.createEntryId(),
162
+ parentId: await this.storage.getLeafId(),
163
+ timestamp: new Date().toISOString(),
164
+ targetId,
165
+ label,
166
+ });
167
+ }
168
+ async appendSessionName(name) {
169
+ return this.appendTypedEntry({
170
+ type: "session_info",
171
+ id: await this.storage.createEntryId(),
172
+ parentId: await this.storage.getLeafId(),
173
+ timestamp: new Date().toISOString(),
174
+ name: name.trim(),
175
+ });
176
+ }
177
+ async moveTo(entryId, summary) {
178
+ if (entryId !== null && !(await this.storage.getEntry(entryId))) {
179
+ throw new Error(`Entry ${entryId} not found`);
180
+ }
181
+ await this.storage.setLeafId(entryId);
182
+ if (!summary)
183
+ return undefined;
184
+ return this.appendTypedEntry({
185
+ type: "branch_summary",
186
+ id: await this.storage.createEntryId(),
187
+ parentId: entryId,
188
+ timestamp: new Date().toISOString(),
189
+ fromId: entryId ?? "root",
190
+ summary: summary.summary,
191
+ details: summary.details,
192
+ fromHook: summary.fromHook,
193
+ });
194
+ }
195
+ }
196
+ //# sourceMappingURL=session.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"session.js","sourceRoot":"","sources":["../../../../../src/vendor/pi-agent-core/harness/session/session.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,0BAA0B,EAAE,8BAA8B,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAiBjH,MAAM,UAAU,mBAAmB,CAAC,WAA+B;IAClE,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,KAAK,GAAiD,IAAI,CAAC;IAC/D,IAAI,UAAU,GAA2B,IAAI,CAAC;IAE9C,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;YAC5C,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;QACrC,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YAC1C,KAAK,GAAG,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;QAC9D,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC3E,KAAK,GAAG,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC5E,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACxC,UAAU,GAAG,KAAK,CAAC;QACpB,CAAC;IACF,CAAC;IAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,aAAa,GAAG,CAAC,KAAuB,EAAE,EAAE;QACjD,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC9B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAuB,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YAC5C,QAAQ,CAAC,IAAI,CACZ,mBAAmB,CAClB,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,OAAkD,EACxD,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,SAAS,CACf,CACD,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC7D,QAAQ,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;QACzF,CAAC;IACF,CAAC,CAAC;IAEF,IAAI,UAAU,EAAE,CAAC;QAChB,QAAQ,CAAC,IAAI,CAAC,8BAA8B,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,YAAY,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;QACjH,MAAM,aAAa,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,IAAI,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,EAAE,CAAC,CAAC;QACtG,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,CAAE,CAAC;YAC9B,IAAI,KAAK,CAAC,EAAE,KAAK,UAAU,CAAC,gBAAgB;gBAAE,cAAc,GAAG,IAAI,CAAC;YACpE,IAAI,cAAc;gBAAE,aAAa,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,aAAa,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7D,aAAa,CAAC,WAAW,CAAC,CAAC,CAAE,CAAC,CAAC;QAChC,CAAC;IACF,CAAC;SAAM,CAAC;QACP,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACjC,aAAa,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACF,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,OAAO,OAAO;IACX,OAAO,CAA4B;IAE3C,YAAY,OAAkC;QAC7C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACxB,CAAC;IAED,WAAW;QACV,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAED,UAAU;QACT,OAAO,IAAI,CAAC,OAAO,CAAC;IACrB,CAAC;IAED,SAAS;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC;IAED,QAAQ,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,UAAU;QACT,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAe;QAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,YAAY;QACjB,OAAO,mBAAmB,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,QAAQ,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,cAAc;QACnB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAC/D,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;IAC/D,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAkC,KAAa;QAC5E,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACtC,OAAO,KAAK,CAAC,EAAE,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAqB;QACxC,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,SAAS;YACf,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,OAAO;SACgB,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,aAAqB;QACpD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,uBAAuB;YAC7B,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,aAAa;SACsB,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,OAAe;QACxD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,cAAc;YACpB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,QAAQ;YACR,OAAO;SACoB,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,gBAAgB,CACrB,OAAe,EACf,gBAAwB,EACxB,YAAoB,EACpB,OAAW,EACX,QAAkB;QAElB,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,YAAY;YAClB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,OAAO;YACP,gBAAgB;YAChB,YAAY;YACZ,OAAO;YACP,QAAQ;SACqB,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,UAAkB,EAAE,IAAc;QACzD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,QAAQ;YACd,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU;YACV,IAAI;SACkB,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,wBAAwB,CAC7B,UAAkB,EAClB,OAAgD,EAChD,OAAgB,EAChB,OAAW;QAEX,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,gBAAgB;YACtB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU;YACV,OAAO;YACP,OAAO;YACP,OAAO;SACyB,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,KAAyB;QAC5D,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,SAAS,QAAQ,YAAY,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,OAAO;YACb,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,QAAQ;YACR,KAAK;SACgB,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAAY;QACnC,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,cAAc;YACpB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;SACU,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,MAAM,CACX,OAAsB,EACtB,OAAoE;QAEpE,IAAI,OAAO,KAAK,IAAI,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,SAAS,OAAO,YAAY,CAAC,CAAC;QAC/C,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC5B,IAAI,EAAE,gBAAgB;YACtB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,OAAO;YACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,MAAM,EAAE,OAAO,IAAI,MAAM;YACzB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;SACG,CAAC,CAAC;IACjC,CAAC;CACD"}
@@ -0,0 +1,30 @@
1
+ import type { JsonlSessionMetadata, SessionStorage, SessionTreeEntry } from "../../types.js";
2
+ export declare function loadJsonlSessionMetadata(filePath: string): Promise<JsonlSessionMetadata>;
3
+ export declare class JsonlSessionStorage implements SessionStorage<JsonlSessionMetadata> {
4
+ private readonly filePath;
5
+ private readonly metadata;
6
+ private entries;
7
+ private byId;
8
+ private labelsById;
9
+ private currentLeafId;
10
+ private constructor();
11
+ static open(filePath: string): Promise<JsonlSessionStorage>;
12
+ static create(filePath: string, options: {
13
+ cwd: string;
14
+ sessionId: string;
15
+ parentSessionPath?: string;
16
+ }): Promise<JsonlSessionStorage>;
17
+ getMetadata(): Promise<JsonlSessionMetadata>;
18
+ getLeafId(): Promise<string | null>;
19
+ setLeafId(leafId: string | null): Promise<void>;
20
+ createEntryId(): Promise<string>;
21
+ appendEntry(entry: SessionTreeEntry): Promise<void>;
22
+ getEntry(id: string): Promise<SessionTreeEntry | undefined>;
23
+ findEntries<TType extends SessionTreeEntry["type"]>(type: TType): Promise<Array<Extract<SessionTreeEntry, {
24
+ type: TType;
25
+ }>>>;
26
+ getLabel(id: string): Promise<string | undefined>;
27
+ getPathToRoot(leafId: string | null): Promise<SessionTreeEntry[]>;
28
+ getEntries(): Promise<SessionTreeEntry[]>;
29
+ }
30
+ //# sourceMappingURL=jsonl.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"jsonl.d.ts","sourceRoot":"","sources":["../../../../../../src/vendor/pi-agent-core/harness/session/storage/jsonl.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,oBAAoB,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AA+C7F,wBAAsB,wBAAwB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAkB9F;AAkCD,qBAAa,mBAAoB,YAAW,cAAc,CAAC,oBAAoB,CAAC;IAC/E,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAuB;IAChD,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,IAAI,CAAgC;IAC5C,OAAO,CAAC,UAAU,CAAsB;IACxC,OAAO,CAAC,aAAa,CAAgB;IAErC,OAAO;WASM,IAAI,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC;WAMpD,MAAM,CAClB,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE;QACR,GAAG,EAAE,MAAM,CAAC;QACZ,SAAS,EAAE,MAAM,CAAC;QAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC3B,GACC,OAAO,CAAC,mBAAmB,CAAC;IAezB,WAAW,IAAI,OAAO,CAAC,oBAAoB,CAAC;IAI5C,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAInC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAO/C,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;IAIhC,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAQnD,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAI3D,WAAW,CAAC,KAAK,SAAS,gBAAgB,CAAC,MAAM,CAAC,EACvD,IAAI,EAAE,KAAK,GACT,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,EAAE;QAAE,IAAI,EAAE,KAAK,CAAA;KAAE,CAAC,CAAC,CAAC;IAIvD,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAIjD,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAWjE,UAAU,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;CAG/C"}
@@ -0,0 +1,170 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { createReadStream } from "node:fs";
3
+ import { appendFile, mkdir, readFile, writeFile } from "node:fs/promises";
4
+ import { dirname, resolve } from "node:path";
5
+ import { createInterface } from "node:readline";
6
+ function updateLabelCache(labelsById, entry) {
7
+ if (entry.type !== "label")
8
+ return;
9
+ const label = entry.label?.trim();
10
+ if (label) {
11
+ labelsById.set(entry.targetId, label);
12
+ }
13
+ else {
14
+ labelsById.delete(entry.targetId);
15
+ }
16
+ }
17
+ function buildLabelsById(entries) {
18
+ const labelsById = new Map();
19
+ for (const entry of entries) {
20
+ updateLabelCache(labelsById, entry);
21
+ }
22
+ return labelsById;
23
+ }
24
+ function generateEntryId(byId) {
25
+ for (let i = 0; i < 100; i++) {
26
+ const id = randomUUID().slice(0, 8);
27
+ if (!byId.has(id))
28
+ return id;
29
+ }
30
+ return randomUUID();
31
+ }
32
+ function headerToSessionMetadata(header, path) {
33
+ return {
34
+ id: header.id,
35
+ createdAt: header.timestamp,
36
+ cwd: header.cwd,
37
+ path,
38
+ parentSessionPath: header.parentSession,
39
+ };
40
+ }
41
+ export async function loadJsonlSessionMetadata(filePath) {
42
+ const stream = createReadStream(filePath, { encoding: "utf8" });
43
+ const lines = createInterface({ input: stream, crlfDelay: Infinity });
44
+ try {
45
+ for await (const line of lines) {
46
+ if (!line.trim())
47
+ break;
48
+ try {
49
+ const header = JSON.parse(line);
50
+ return headerToSessionMetadata(header, resolve(filePath));
51
+ }
52
+ catch {
53
+ throw new Error(`Invalid JSONL session file ${filePath}: first line is not a valid session header`);
54
+ }
55
+ }
56
+ throw new Error(`Invalid JSONL session file ${filePath}: missing session header`);
57
+ }
58
+ finally {
59
+ lines.close();
60
+ stream.destroy();
61
+ }
62
+ }
63
+ async function loadJsonlStorage(filePath) {
64
+ const content = await readFile(filePath, "utf8");
65
+ const lines = content.split("\n").filter((line) => line.trim());
66
+ if (lines.length === 0) {
67
+ throw new Error(`Invalid JSONL session file ${filePath}: missing session header`);
68
+ }
69
+ let header;
70
+ try {
71
+ header = JSON.parse(lines[0]);
72
+ }
73
+ catch {
74
+ throw new Error(`Invalid JSONL session file ${filePath}: first line is not a valid session header`);
75
+ }
76
+ const entries = [];
77
+ let leafId = null;
78
+ for (const line of lines.slice(1)) {
79
+ try {
80
+ const entry = JSON.parse(line);
81
+ entries.push(entry);
82
+ leafId = entry.id;
83
+ }
84
+ catch {
85
+ // ignore malformed entry lines
86
+ }
87
+ }
88
+ return { header, entries, leafId };
89
+ }
90
+ export class JsonlSessionStorage {
91
+ filePath;
92
+ metadata;
93
+ entries;
94
+ byId;
95
+ labelsById;
96
+ currentLeafId;
97
+ constructor(filePath, header, entries, leafId) {
98
+ this.filePath = resolve(filePath);
99
+ this.metadata = headerToSessionMetadata(header, this.filePath);
100
+ this.entries = entries;
101
+ this.byId = new Map(entries.map((entry) => [entry.id, entry]));
102
+ this.labelsById = buildLabelsById(entries);
103
+ this.currentLeafId = leafId;
104
+ }
105
+ static async open(filePath) {
106
+ const resolvedPath = resolve(filePath);
107
+ const loaded = await loadJsonlStorage(resolvedPath);
108
+ return new JsonlSessionStorage(resolvedPath, loaded.header, loaded.entries, loaded.leafId);
109
+ }
110
+ static async create(filePath, options) {
111
+ const resolvedPath = resolve(filePath);
112
+ const header = {
113
+ type: "session",
114
+ version: 3,
115
+ id: options.sessionId,
116
+ timestamp: new Date().toISOString(),
117
+ cwd: options.cwd,
118
+ parentSession: options.parentSessionPath,
119
+ };
120
+ await mkdir(dirname(resolvedPath), { recursive: true });
121
+ await writeFile(resolvedPath, `${JSON.stringify(header)}\n`);
122
+ return new JsonlSessionStorage(resolvedPath, header, [], null);
123
+ }
124
+ async getMetadata() {
125
+ return this.metadata;
126
+ }
127
+ async getLeafId() {
128
+ return this.currentLeafId;
129
+ }
130
+ async setLeafId(leafId) {
131
+ if (leafId !== null && !this.byId.has(leafId)) {
132
+ throw new Error(`Entry ${leafId} not found`);
133
+ }
134
+ this.currentLeafId = leafId;
135
+ }
136
+ async createEntryId() {
137
+ return generateEntryId(this.byId);
138
+ }
139
+ async appendEntry(entry) {
140
+ await appendFile(this.filePath, `${JSON.stringify(entry)}\n`);
141
+ this.entries.push(entry);
142
+ this.byId.set(entry.id, entry);
143
+ updateLabelCache(this.labelsById, entry);
144
+ this.currentLeafId = entry.id;
145
+ }
146
+ async getEntry(id) {
147
+ return this.byId.get(id);
148
+ }
149
+ async findEntries(type) {
150
+ return this.entries.filter((entry) => entry.type === type);
151
+ }
152
+ async getLabel(id) {
153
+ return this.labelsById.get(id);
154
+ }
155
+ async getPathToRoot(leafId) {
156
+ if (leafId === null)
157
+ return [];
158
+ const path = [];
159
+ let current = this.byId.get(leafId);
160
+ while (current) {
161
+ path.unshift(current);
162
+ current = current.parentId ? this.byId.get(current.parentId) : undefined;
163
+ }
164
+ return path;
165
+ }
166
+ async getEntries() {
167
+ return [...this.entries];
168
+ }
169
+ }
170
+ //# sourceMappingURL=jsonl.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"jsonl.js","sourceRoot":"","sources":["../../../../../../src/vendor/pi-agent-core/harness/session/storage/jsonl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC1E,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAYhD,SAAS,gBAAgB,CAAC,UAA+B,EAAE,KAAuB;IACjF,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;IAClC,IAAI,KAAK,EAAE,CAAC;QACX,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACP,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;AACF,CAAC;AAED,SAAS,eAAe,CAAC,OAA2B;IACnD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC7C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,gBAAgB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,UAAU,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,IAAkC;IAC1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,EAAE,CAAC;IAC9B,CAAC;IACD,OAAO,UAAU,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,uBAAuB,CAAC,MAAqB,EAAE,IAAY;IACnE,OAAO;QACN,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,IAAI;QACJ,iBAAiB,EAAE,MAAM,CAAC,aAAa;KACvC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,QAAgB;IAC9D,MAAM,MAAM,GAAG,gBAAgB,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IAChE,MAAM,KAAK,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;IACtE,IAAI,CAAC;QACJ,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,MAAM;YACxB,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAkB,CAAC;gBACjD,OAAO,uBAAuB,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC3D,CAAC;YAAC,MAAM,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,4CAA4C,CAAC,CAAC;YACrG,CAAC;QACF,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,0BAA0B,CAAC,CAAC;IACnF,CAAC;YAAS,CAAC;QACV,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,MAAM,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;AACF,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,QAAgB;IAK/C,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAChE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,0BAA0B,CAAC,CAAC;IACnF,CAAC;IAED,IAAI,MAAqB,CAAC;IAC1B,IAAI,CAAC;QACJ,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAkB,CAAC;IACjD,CAAC;IAAC,MAAM,CAAC;QACR,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,4CAA4C,CAAC,CAAC;IACrG,CAAC;IAED,MAAM,OAAO,GAAuB,EAAE,CAAC;IACvC,IAAI,MAAM,GAAkB,IAAI,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC;YACJ,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAqB,CAAC;YACnD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,GAAG,KAAK,CAAC,EAAE,CAAC;QACnB,CAAC;QAAC,MAAM,CAAC;YACR,+BAA+B;QAChC,CAAC;IACF,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AACpC,CAAC;AAED,MAAM,OAAO,mBAAmB;IACd,QAAQ,CAAS;IACjB,QAAQ,CAAuB;IACxC,OAAO,CAAqB;IAC5B,IAAI,CAAgC;IACpC,UAAU,CAAsB;IAChC,aAAa,CAAgB;IAErC,YAAoB,QAAgB,EAAE,MAAqB,EAAE,OAA2B,EAAE,MAAqB;QAC9G,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,QAAQ,GAAG,uBAAuB,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAgB;QACjC,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,YAAY,CAAC,CAAC;QACpD,OAAO,IAAI,mBAAmB,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5F,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAClB,QAAgB,EAChB,OAIC;QAED,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,MAAM,GAAkB;YAC7B,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,CAAC;YACV,EAAE,EAAE,OAAO,CAAC,SAAS;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,aAAa,EAAE,OAAO,CAAC,iBAAiB;SACxC,CAAC;QACF,MAAM,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,MAAM,SAAS,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7D,OAAO,IAAI,mBAAmB,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,OAAO,IAAI,CAAC,QAAQ,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,SAAS;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAqB;QACpC,IAAI,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,SAAS,MAAM,YAAY,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,aAAa;QAClB,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAuB;QACxC,MAAM,UAAU,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC9D,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAC/B,gBAAgB,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,EAAE,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,WAAW,CAChB,IAAW;QAEX,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAuD,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IACjH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAqB;QACxC,IAAI,MAAM,KAAK,IAAI;YAAE,OAAO,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAuB,EAAE,CAAC;QACpC,IAAI,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,OAAO,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,KAAK,CAAC,UAAU;QACf,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;CACD"}
@@ -0,0 +1,26 @@
1
+ import type { SessionMetadata, SessionStorage, SessionTreeEntry } from "../../types.js";
2
+ export declare class InMemorySessionStorage implements SessionStorage {
3
+ private readonly metadata;
4
+ private entries;
5
+ private byId;
6
+ private labelsById;
7
+ private leafId;
8
+ constructor(options?: {
9
+ entries?: SessionTreeEntry[];
10
+ leafId?: string | null;
11
+ metadata?: SessionMetadata;
12
+ });
13
+ getMetadata(): Promise<SessionMetadata>;
14
+ getLeafId(): Promise<string | null>;
15
+ setLeafId(leafId: string | null): Promise<void>;
16
+ createEntryId(): Promise<string>;
17
+ appendEntry(entry: SessionTreeEntry): Promise<void>;
18
+ getEntry(id: string): Promise<SessionTreeEntry | undefined>;
19
+ findEntries<TType extends SessionTreeEntry["type"]>(type: TType): Promise<Array<Extract<SessionTreeEntry, {
20
+ type: TType;
21
+ }>>>;
22
+ getLabel(id: string): Promise<string | undefined>;
23
+ getPathToRoot(leafId: string | null): Promise<SessionTreeEntry[]>;
24
+ getEntries(): Promise<SessionTreeEntry[]>;
25
+ }
26
+ //# sourceMappingURL=memory.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../../../../../src/vendor/pi-agent-core/harness/session/storage/memory.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AA6BxF,qBAAa,sBAAuB,YAAW,cAAc;IAC5D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkB;IAC3C,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,IAAI,CAAgC;IAC5C,OAAO,CAAC,UAAU,CAAsB;IACxC,OAAO,CAAC,MAAM,CAAgB;gBAElB,OAAO,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,gBAAgB,EAAE,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,QAAQ,CAAC,EAAE,eAAe,CAAA;KAAE;IAWpG,WAAW,IAAI,OAAO,CAAC,eAAe,CAAC;IAIvC,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAInC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAO/C,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;IAIhC,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAOnD,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAI3D,WAAW,CAAC,KAAK,SAAS,gBAAgB,CAAC,MAAM,CAAC,EACvD,IAAI,EAAE,KAAK,GACT,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,EAAE;QAAE,IAAI,EAAE,KAAK,CAAA;KAAE,CAAC,CAAC,CAAC;IAIvD,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAIjD,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAWjE,UAAU,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;CAG/C"}
@@ -0,0 +1,90 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { uuidv7 } from "../uuid.js";
3
+ function updateLabelCache(labelsById, entry) {
4
+ if (entry.type !== "label")
5
+ return;
6
+ const label = entry.label?.trim();
7
+ if (label) {
8
+ labelsById.set(entry.targetId, label);
9
+ }
10
+ else {
11
+ labelsById.delete(entry.targetId);
12
+ }
13
+ }
14
+ function buildLabelsById(entries) {
15
+ const labelsById = new Map();
16
+ for (const entry of entries) {
17
+ updateLabelCache(labelsById, entry);
18
+ }
19
+ return labelsById;
20
+ }
21
+ function generateEntryId(byId) {
22
+ for (let i = 0; i < 100; i++) {
23
+ const id = randomUUID().slice(0, 8);
24
+ if (!byId.has(id))
25
+ return id;
26
+ }
27
+ return randomUUID();
28
+ }
29
+ export class InMemorySessionStorage {
30
+ metadata;
31
+ entries;
32
+ byId;
33
+ labelsById;
34
+ leafId;
35
+ constructor(options) {
36
+ this.entries = options?.entries ? [...options.entries] : [];
37
+ this.byId = new Map(this.entries.map((entry) => [entry.id, entry]));
38
+ this.labelsById = buildLabelsById(this.entries);
39
+ this.leafId = options?.leafId ?? this.entries[this.entries.length - 1]?.id ?? null;
40
+ if (this.leafId !== null && !this.byId.has(this.leafId)) {
41
+ throw new Error(`Entry ${this.leafId} not found`);
42
+ }
43
+ this.metadata = options?.metadata ?? { id: uuidv7(), createdAt: new Date().toISOString() };
44
+ }
45
+ async getMetadata() {
46
+ return this.metadata;
47
+ }
48
+ async getLeafId() {
49
+ return this.leafId;
50
+ }
51
+ async setLeafId(leafId) {
52
+ if (leafId !== null && !this.byId.has(leafId)) {
53
+ throw new Error(`Entry ${leafId} not found`);
54
+ }
55
+ this.leafId = leafId;
56
+ }
57
+ async createEntryId() {
58
+ return generateEntryId(this.byId);
59
+ }
60
+ async appendEntry(entry) {
61
+ this.entries.push(entry);
62
+ this.byId.set(entry.id, entry);
63
+ updateLabelCache(this.labelsById, entry);
64
+ this.leafId = entry.id;
65
+ }
66
+ async getEntry(id) {
67
+ return this.byId.get(id);
68
+ }
69
+ async findEntries(type) {
70
+ return this.entries.filter((entry) => entry.type === type);
71
+ }
72
+ async getLabel(id) {
73
+ return this.labelsById.get(id);
74
+ }
75
+ async getPathToRoot(leafId) {
76
+ if (leafId === null)
77
+ return [];
78
+ const path = [];
79
+ let current = this.byId.get(leafId);
80
+ while (current) {
81
+ path.unshift(current);
82
+ current = current.parentId ? this.byId.get(current.parentId) : undefined;
83
+ }
84
+ return path;
85
+ }
86
+ async getEntries() {
87
+ return [...this.entries];
88
+ }
89
+ }
90
+ //# sourceMappingURL=memory.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../../../src/vendor/pi-agent-core/harness/session/storage/memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEpC,SAAS,gBAAgB,CAAC,UAA+B,EAAE,KAAuB;IACjF,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;IAClC,IAAI,KAAK,EAAE,CAAC;QACX,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACP,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;AACF,CAAC;AAED,SAAS,eAAe,CAAC,OAA2B;IACnD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC7C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,gBAAgB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,UAAU,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,IAAkC;IAC1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,EAAE,CAAC;IAC9B,CAAC;IACD,OAAO,UAAU,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,OAAO,sBAAsB;IACjB,QAAQ,CAAkB;IACnC,OAAO,CAAqB;IAC5B,IAAI,CAAgC;IACpC,UAAU,CAAsB;IAChC,MAAM,CAAgB;IAE9B,YAAY,OAA8F;QACzG,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,EAAE,IAAI,IAAI,CAAC;QACnF,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACzD,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,MAAM,YAAY,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;IAC5F,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,OAAO,IAAI,CAAC,QAAQ,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,SAAS;QACd,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAqB;QACpC,IAAI,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,SAAS,MAAM,YAAY,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,aAAa;QAClB,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAuB;QACxC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAC/B,gBAAgB,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,EAAE,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,WAAW,CAChB,IAAW;QAEX,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAuD,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IACjH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAqB;QACxC,IAAI,MAAM,KAAK,IAAI;YAAE,OAAO,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAuB,EAAE,CAAC;QACpC,IAAI,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,OAAO,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,KAAK,CAAC,UAAU;QACf,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;CACD"}
@@ -0,0 +1,2 @@
1
+ export declare function uuidv7(): string;
2
+ //# sourceMappingURL=uuid.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"uuid.d.ts","sourceRoot":"","sources":["../../../../../src/vendor/pi-agent-core/harness/session/uuid.ts"],"names":[],"mappings":"AAKA,wBAAgB,MAAM,IAAI,MAAM,CAiC/B"}