ds4-context-core 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (151) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +80 -0
  3. package/dist/artifacts/artifact-manager.d.ts +58 -0
  4. package/dist/artifacts/artifact-manager.d.ts.map +1 -0
  5. package/dist/artifacts/artifact-manager.js +445 -0
  6. package/dist/artifacts/artifact-manager.js.map +1 -0
  7. package/dist/artifacts/artifact-store.d.ts +33 -0
  8. package/dist/artifacts/artifact-store.d.ts.map +1 -0
  9. package/dist/artifacts/artifact-store.js +116 -0
  10. package/dist/artifacts/artifact-store.js.map +1 -0
  11. package/dist/compaction/compaction-record.d.ts +74 -0
  12. package/dist/compaction/compaction-record.d.ts.map +1 -0
  13. package/dist/compaction/compaction-record.js +113 -0
  14. package/dist/compaction/compaction-record.js.map +1 -0
  15. package/dist/compaction/summary-contract.d.ts +68 -0
  16. package/dist/compaction/summary-contract.d.ts.map +1 -0
  17. package/dist/compaction/summary-contract.js +315 -0
  18. package/dist/compaction/summary-contract.js.map +1 -0
  19. package/dist/compaction/summary-graph.d.ts +48 -0
  20. package/dist/compaction/summary-graph.d.ts.map +1 -0
  21. package/dist/compaction/summary-graph.js +123 -0
  22. package/dist/compaction/summary-graph.js.map +1 -0
  23. package/dist/config/config-loader.d.ts +18 -0
  24. package/dist/config/config-loader.d.ts.map +1 -0
  25. package/dist/config/config-loader.js +305 -0
  26. package/dist/config/config-loader.js.map +1 -0
  27. package/dist/config/config.d.ts +113 -0
  28. package/dist/config/config.d.ts.map +1 -0
  29. package/dist/config/config.js +89 -0
  30. package/dist/config/config.js.map +1 -0
  31. package/dist/continuation/native-continuation.d.ts +107 -0
  32. package/dist/continuation/native-continuation.d.ts.map +1 -0
  33. package/dist/continuation/native-continuation.js +374 -0
  34. package/dist/continuation/native-continuation.js.map +1 -0
  35. package/dist/core/budget-manager.d.ts +26 -0
  36. package/dist/core/budget-manager.d.ts.map +1 -0
  37. package/dist/core/budget-manager.js +42 -0
  38. package/dist/core/budget-manager.js.map +1 -0
  39. package/dist/core/canonical-message.d.ts +69 -0
  40. package/dist/core/canonical-message.d.ts.map +1 -0
  41. package/dist/core/canonical-message.js +2 -0
  42. package/dist/core/canonical-message.js.map +1 -0
  43. package/dist/core/model-awareness.d.ts +59 -0
  44. package/dist/core/model-awareness.d.ts.map +1 -0
  45. package/dist/core/model-awareness.js +177 -0
  46. package/dist/core/model-awareness.js.map +1 -0
  47. package/dist/core/model-profile.d.ts +27 -0
  48. package/dist/core/model-profile.d.ts.map +1 -0
  49. package/dist/core/model-profile.js +29 -0
  50. package/dist/core/model-profile.js.map +1 -0
  51. package/dist/core/token-estimator.d.ts +4 -0
  52. package/dist/core/token-estimator.d.ts.map +1 -0
  53. package/dist/core/token-estimator.js +64 -0
  54. package/dist/core/token-estimator.js.map +1 -0
  55. package/dist/index.d.ts +37 -0
  56. package/dist/index.d.ts.map +1 -0
  57. package/dist/index.js +37 -0
  58. package/dist/index.js.map +1 -0
  59. package/dist/manifest/context-manifest.d.ts +163 -0
  60. package/dist/manifest/context-manifest.d.ts.map +1 -0
  61. package/dist/manifest/context-manifest.js +2 -0
  62. package/dist/manifest/context-manifest.js.map +1 -0
  63. package/dist/manifest/observer.d.ts +63 -0
  64. package/dist/manifest/observer.d.ts.map +1 -0
  65. package/dist/manifest/observer.js +150 -0
  66. package/dist/manifest/observer.js.map +1 -0
  67. package/dist/memory/memory-manager.d.ts +112 -0
  68. package/dist/memory/memory-manager.d.ts.map +1 -0
  69. package/dist/memory/memory-manager.js +499 -0
  70. package/dist/memory/memory-manager.js.map +1 -0
  71. package/dist/memory/memory-types.d.ts +126 -0
  72. package/dist/memory/memory-types.d.ts.map +1 -0
  73. package/dist/memory/memory-types.js +73 -0
  74. package/dist/memory/memory-types.js.map +1 -0
  75. package/dist/persistence/migrations.d.ts +17 -0
  76. package/dist/persistence/migrations.d.ts.map +1 -0
  77. package/dist/persistence/migrations.js +588 -0
  78. package/dist/persistence/migrations.js.map +1 -0
  79. package/dist/persistence/repositories/artifact-repository.d.ts +50 -0
  80. package/dist/persistence/repositories/artifact-repository.d.ts.map +1 -0
  81. package/dist/persistence/repositories/artifact-repository.js +186 -0
  82. package/dist/persistence/repositories/artifact-repository.js.map +1 -0
  83. package/dist/persistence/repositories/context-manifest-repository.d.ts +19 -0
  84. package/dist/persistence/repositories/context-manifest-repository.d.ts.map +1 -0
  85. package/dist/persistence/repositories/context-manifest-repository.js +144 -0
  86. package/dist/persistence/repositories/context-manifest-repository.js.map +1 -0
  87. package/dist/persistence/repositories/memory-repository.d.ts +33 -0
  88. package/dist/persistence/repositories/memory-repository.d.ts.map +1 -0
  89. package/dist/persistence/repositories/memory-repository.js +429 -0
  90. package/dist/persistence/repositories/memory-repository.js.map +1 -0
  91. package/dist/persistence/repositories/project-knowledge-repository.d.ts +65 -0
  92. package/dist/persistence/repositories/project-knowledge-repository.d.ts.map +1 -0
  93. package/dist/persistence/repositories/project-knowledge-repository.js +248 -0
  94. package/dist/persistence/repositories/project-knowledge-repository.js.map +1 -0
  95. package/dist/persistence/repositories/session-index-repository.d.ts +81 -0
  96. package/dist/persistence/repositories/session-index-repository.d.ts.map +1 -0
  97. package/dist/persistence/repositories/session-index-repository.js +220 -0
  98. package/dist/persistence/repositories/session-index-repository.js.map +1 -0
  99. package/dist/persistence/repositories/summary-repository.d.ts +16 -0
  100. package/dist/persistence/repositories/summary-repository.d.ts.map +1 -0
  101. package/dist/persistence/repositories/summary-repository.js +242 -0
  102. package/dist/persistence/repositories/summary-repository.js.map +1 -0
  103. package/dist/persistence/sqlite.d.ts +45 -0
  104. package/dist/persistence/sqlite.d.ts.map +1 -0
  105. package/dist/persistence/sqlite.js +137 -0
  106. package/dist/persistence/sqlite.js.map +1 -0
  107. package/dist/planner/atomic-groups.d.ts +15 -0
  108. package/dist/planner/atomic-groups.d.ts.map +1 -0
  109. package/dist/planner/atomic-groups.js +177 -0
  110. package/dist/planner/atomic-groups.js.map +1 -0
  111. package/dist/planner/context-planner.d.ts +49 -0
  112. package/dist/planner/context-planner.d.ts.map +1 -0
  113. package/dist/planner/context-planner.js +367 -0
  114. package/dist/planner/context-planner.js.map +1 -0
  115. package/dist/privacy/privacy-policy.d.ts +58 -0
  116. package/dist/privacy/privacy-policy.d.ts.map +1 -0
  117. package/dist/privacy/privacy-policy.js +412 -0
  118. package/dist/privacy/privacy-policy.js.map +1 -0
  119. package/dist/project/file-indexer.d.ts +32 -0
  120. package/dist/project/file-indexer.d.ts.map +1 -0
  121. package/dist/project/file-indexer.js +374 -0
  122. package/dist/project/file-indexer.js.map +1 -0
  123. package/dist/project/git-state.d.ts +12 -0
  124. package/dist/project/git-state.d.ts.map +1 -0
  125. package/dist/project/git-state.js +109 -0
  126. package/dist/project/git-state.js.map +1 -0
  127. package/dist/project/project-knowledge.d.ts +63 -0
  128. package/dist/project/project-knowledge.d.ts.map +1 -0
  129. package/dist/project/project-knowledge.js +332 -0
  130. package/dist/project/project-knowledge.js.map +1 -0
  131. package/dist/retrieval/retrieval-engine.d.ts +55 -0
  132. package/dist/retrieval/retrieval-engine.d.ts.map +1 -0
  133. package/dist/retrieval/retrieval-engine.js +290 -0
  134. package/dist/retrieval/retrieval-engine.js.map +1 -0
  135. package/dist/retrieval/task-descriptor.d.ts +17 -0
  136. package/dist/retrieval/task-descriptor.d.ts.map +1 -0
  137. package/dist/retrieval/task-descriptor.js +107 -0
  138. package/dist/retrieval/task-descriptor.js.map +1 -0
  139. package/dist/shared/hash.d.ts +2 -0
  140. package/dist/shared/hash.d.ts.map +1 -0
  141. package/dist/shared/hash.js +5 -0
  142. package/dist/shared/hash.js.map +1 -0
  143. package/dist/shared/logging.d.ts +27 -0
  144. package/dist/shared/logging.d.ts.map +1 -0
  145. package/dist/shared/logging.js +83 -0
  146. package/dist/shared/logging.js.map +1 -0
  147. package/dist/shared/stable-json.d.ts +2 -0
  148. package/dist/shared/stable-json.d.ts.map +1 -0
  149. package/dist/shared/stable-json.js +38 -0
  150. package/dist/shared/stable-json.js.map +1 -0
  151. package/package.json +56 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"migrations.js","sourceRoot":"","sources":["../../src/persistence/migrations.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAe3C,MAAM,CAAC,MAAM,UAAU,GAAyB;IAC9C;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,uBAAuB;QAC7B,GAAG,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAqIJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,iDAAiD;QACvD,GAAG,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAkIJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,gCAAgC;QACtC,GAAG,EAAE;;;KAGJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,mCAAmC;QACzC,GAAG,EAAE;;;;;;;;;KASJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,8BAA8B;QACpC,GAAG,EAAE;;;;;;;;;;;;;KAaJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,4BAA4B;QAClC,GAAG,EAAE;;;;;;;;;KASJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,yBAAyB;QAC/B,GAAG,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAwDJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,kCAAkC;QACxC,GAAG,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAuDJ;KACF;IACD;QACE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,+BAA+B;QACrC,GAAG,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAiDJ;KACF;IACD;QACE,OAAO,EAAE,EAAE;QACX,IAAI,EAAE,8CAA8C;QACpD,GAAG,EAAE;;;;;;;;;;;;;;;;;;;;;;;;KAwBJ;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,CAAC,CAAC;AAStE,SAAS,iBAAiB,CAAC,SAAoB;IAC7C,OAAO,MAAM,CAAC,GAAG,SAAS,CAAC,OAAO,KAAK,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC,GAAG,EAAE,CAAC,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,QAAsB;IAC1D,MAAM,IAAI,GAAG,QAAQ;SAClB,OAAO,CAAC,oFAAoF,CAAC;SAC7F,GAAG,EAA+B,CAAC;IAEtC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxB,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,QAAsB,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IACtE,QAAQ,CAAC,IAAI,CAAC;;;;;;;GAOb,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;IAC5G,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,KAAK,QAAQ,CAAC,OAAO,CAAC,CAAC;QACrF,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,OAAO,wCAAwC,CAAC,CAAC;QACvG,CAAC;QACD,IAAI,QAAQ,CAAC,QAAQ,KAAK,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,2CAA2C,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAC7B,wFAAwF,CACzF,CAAC;IAEF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC;YAAE,SAAS;QAE7C,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACjC,IAAI,CAAC;YACH,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAC7B,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,EAAE,iBAAiB,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC;YACjF,QAAQ,CAAC,IAAI,CAAC,yBAAyB,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5D,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,QAAQ,CAAC,aAAa;gBAAE,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACtD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,qBAAqB,CAAC,QAAQ,CAAC,CAAC;AACzC,CAAC"}
@@ -0,0 +1,50 @@
1
+ import type { DatabaseSync } from "node:sqlite";
2
+ export type ArtifactObjectStatus = "available" | "missing" | "corrupt";
3
+ export interface ArtifactObjectRecord {
4
+ sha256: string;
5
+ filePath: string;
6
+ mimeType: string;
7
+ sizeBytes: number;
8
+ createdAt: number;
9
+ lastVerifiedAt?: number;
10
+ status: ArtifactObjectStatus;
11
+ }
12
+ export interface ArtifactReferenceRecord {
13
+ artifactId: string;
14
+ sha256: string;
15
+ sourceSessionId: string;
16
+ sourceEntryKey: string;
17
+ sourceEntryId: string;
18
+ sourceToolCallId: string;
19
+ sourceToolName: string;
20
+ isError: boolean;
21
+ originalChars: number;
22
+ originalTokens: number;
23
+ condensedChars: number;
24
+ condensedTokens: number;
25
+ createdAt: number;
26
+ metadata: Record<string, unknown>;
27
+ }
28
+ export interface ArtifactRecord extends ArtifactReferenceRecord {
29
+ object: ArtifactObjectRecord;
30
+ }
31
+ export interface ArtifactStats {
32
+ objects: number;
33
+ references: number;
34
+ bytes: number;
35
+ missing: number;
36
+ corrupt: number;
37
+ }
38
+ export declare class ArtifactRepository {
39
+ private readonly database;
40
+ constructor(database: DatabaseSync);
41
+ save(record: ArtifactRecord): void;
42
+ getForSession(artifactId: string, sessionId: string): ArtifactRecord | undefined;
43
+ listForSession(sessionId: string, sourceEntryIds?: ReadonlySet<string>): ArtifactRecord[];
44
+ updateObjectStatus(sha256: string, status: ArtifactObjectStatus, verifiedAt: number): void;
45
+ deleteSessionReferencesExcept(sessionId: string, artifactIds: ReadonlySet<string>): number;
46
+ listOrphanObjects(): ArtifactObjectRecord[];
47
+ deleteObject(sha256: string): void;
48
+ stats(sessionId?: string): ArtifactStats;
49
+ }
50
+ //# sourceMappingURL=artifact-repository.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"artifact-repository.d.ts","sourceRoot":"","sources":["../../../src/persistence/repositories/artifact-repository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEhD,MAAM,MAAM,oBAAoB,GAAG,WAAW,GAAG,SAAS,GAAG,SAAS,CAAC;AAEvE,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,oBAAoB,CAAC;CAC9B;AAED,MAAM,WAAW,uBAAuB;IACtC,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE,MAAM,CAAC;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,gBAAgB,EAAE,MAAM,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,cAAe,SAAQ,uBAAuB;IAC7D,MAAM,EAAE,oBAAoB,CAAC;CAC9B;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB;AA4GD,qBAAa,kBAAkB;IACjB,OAAO,CAAC,QAAQ,CAAC,QAAQ;gBAAR,QAAQ,EAAE,YAAY;IAEnD,IAAI,CAAC,MAAM,EAAE,cAAc,GAAG,IAAI;IAuDlC,aAAa,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAOhF,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,cAAc,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,GAAG,cAAc,EAAE;IAWzF,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAM1F,6BAA6B,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,CAAC,MAAM,CAAC,GAAG,MAAM;IAoB1F,iBAAiB,IAAI,oBAAoB,EAAE;IAY3C,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAOlC,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,aAAa;CAwBzC"}
@@ -0,0 +1,186 @@
1
+ function metadata(value) {
2
+ try {
3
+ const parsed = JSON.parse(value);
4
+ return parsed && typeof parsed === "object" && !Array.isArray(parsed)
5
+ ? parsed
6
+ : {};
7
+ }
8
+ catch {
9
+ return {};
10
+ }
11
+ }
12
+ function mapObject(row) {
13
+ return {
14
+ sha256: row.sha256,
15
+ filePath: row.file_path,
16
+ mimeType: row.mime_type,
17
+ sizeBytes: row.size_bytes,
18
+ createdAt: row.created_at,
19
+ ...(row.last_verified_at !== null ? { lastVerifiedAt: row.last_verified_at } : {}),
20
+ status: row.status,
21
+ };
22
+ }
23
+ function mapReference(row) {
24
+ return {
25
+ artifactId: row.artifact_id,
26
+ sha256: row.sha256,
27
+ sourceSessionId: row.source_session_id,
28
+ sourceEntryKey: row.source_entry_key,
29
+ sourceEntryId: row.source_entry_id,
30
+ sourceToolCallId: row.source_tool_call_id,
31
+ sourceToolName: row.source_tool_name,
32
+ isError: row.is_error === 1,
33
+ originalChars: row.original_chars,
34
+ originalTokens: row.original_tokens,
35
+ condensedChars: row.condensed_chars,
36
+ condensedTokens: row.condensed_tokens,
37
+ createdAt: row.created_at,
38
+ metadata: metadata(row.metadata_json),
39
+ };
40
+ }
41
+ function mapJoined(row) {
42
+ return {
43
+ ...mapReference(row),
44
+ object: {
45
+ sha256: row.sha256,
46
+ filePath: row.file_path,
47
+ mimeType: row.mime_type,
48
+ sizeBytes: row.size_bytes,
49
+ createdAt: row.object_created_at,
50
+ ...(row.last_verified_at !== null ? { lastVerifiedAt: row.last_verified_at } : {}),
51
+ status: row.status,
52
+ },
53
+ };
54
+ }
55
+ const JOINED_SELECT = `
56
+ SELECT
57
+ reference.artifact_id, reference.sha256, reference.source_session_id,
58
+ reference.source_entry_key, reference.source_entry_id,
59
+ reference.source_tool_call_id, reference.source_tool_name, reference.is_error,
60
+ reference.original_chars, reference.original_tokens, reference.condensed_chars,
61
+ reference.condensed_tokens, reference.created_at, reference.metadata_json,
62
+ object.file_path, object.mime_type, object.size_bytes,
63
+ object.created_at AS object_created_at, object.last_verified_at, object.status
64
+ FROM artifacts AS reference
65
+ JOIN artifact_objects AS object ON object.sha256 = reference.sha256
66
+ `;
67
+ export class ArtifactRepository {
68
+ database;
69
+ constructor(database) {
70
+ this.database = database;
71
+ }
72
+ save(record) {
73
+ this.database.exec("BEGIN IMMEDIATE");
74
+ try {
75
+ this.database.prepare(`
76
+ INSERT INTO artifact_objects(
77
+ sha256, file_path, mime_type, size_bytes, created_at, last_verified_at, status
78
+ ) VALUES (?, ?, ?, ?, ?, ?, ?)
79
+ ON CONFLICT(sha256) DO UPDATE SET
80
+ file_path = excluded.file_path,
81
+ mime_type = excluded.mime_type,
82
+ size_bytes = excluded.size_bytes,
83
+ last_verified_at = excluded.last_verified_at,
84
+ status = excluded.status
85
+ `).run(record.object.sha256, record.object.filePath, record.object.mimeType, record.object.sizeBytes, record.object.createdAt, record.object.lastVerifiedAt ?? null, record.object.status);
86
+ this.database.prepare(`
87
+ INSERT INTO artifacts(
88
+ artifact_id, sha256, source_session_id, source_entry_key, source_entry_id,
89
+ source_tool_call_id, source_tool_name, is_error, original_chars,
90
+ original_tokens, condensed_chars, condensed_tokens, created_at, metadata_json
91
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
92
+ ON CONFLICT(artifact_id) DO UPDATE SET
93
+ condensed_chars = excluded.condensed_chars,
94
+ condensed_tokens = excluded.condensed_tokens,
95
+ metadata_json = excluded.metadata_json
96
+ `).run(record.artifactId, record.sha256, record.sourceSessionId, record.sourceEntryKey, record.sourceEntryId, record.sourceToolCallId, record.sourceToolName, record.isError ? 1 : 0, record.originalChars, record.originalTokens, record.condensedChars, record.condensedTokens, record.createdAt, JSON.stringify(record.metadata));
97
+ this.database.exec("COMMIT");
98
+ }
99
+ catch (error) {
100
+ if (this.database.isTransaction)
101
+ this.database.exec("ROLLBACK");
102
+ throw error;
103
+ }
104
+ }
105
+ getForSession(artifactId, sessionId) {
106
+ const row = this.database.prepare(`${JOINED_SELECT}
107
+ WHERE reference.artifact_id = ? AND reference.source_session_id = ?
108
+ `).get(artifactId, sessionId);
109
+ return row ? mapJoined(row) : undefined;
110
+ }
111
+ listForSession(sessionId, sourceEntryIds) {
112
+ const rows = this.database.prepare(`${JOINED_SELECT}
113
+ WHERE reference.source_session_id = ?
114
+ ORDER BY reference.created_at DESC, reference.artifact_id
115
+ `).all(sessionId);
116
+ const records = rows.map(mapJoined);
117
+ return sourceEntryIds
118
+ ? records.filter((record) => sourceEntryIds.has(record.sourceEntryId))
119
+ : records;
120
+ }
121
+ updateObjectStatus(sha256, status, verifiedAt) {
122
+ this.database.prepare(`
123
+ UPDATE artifact_objects SET status = ?, last_verified_at = ? WHERE sha256 = ?
124
+ `).run(status, verifiedAt, sha256);
125
+ }
126
+ deleteSessionReferencesExcept(sessionId, artifactIds) {
127
+ const rows = this.database.prepare("SELECT artifact_id FROM artifacts WHERE source_session_id = ?").all(sessionId);
128
+ const remove = this.database.prepare("DELETE FROM artifacts WHERE artifact_id = ?");
129
+ let changes = 0;
130
+ this.database.exec("BEGIN IMMEDIATE");
131
+ try {
132
+ for (const row of rows) {
133
+ if (artifactIds.has(row.artifact_id))
134
+ continue;
135
+ changes += Number(remove.run(row.artifact_id).changes);
136
+ }
137
+ this.database.exec("COMMIT");
138
+ }
139
+ catch (error) {
140
+ if (this.database.isTransaction)
141
+ this.database.exec("ROLLBACK");
142
+ throw error;
143
+ }
144
+ return changes;
145
+ }
146
+ listOrphanObjects() {
147
+ const rows = this.database.prepare(`
148
+ SELECT object.sha256, object.file_path, object.mime_type, object.size_bytes,
149
+ object.created_at, object.last_verified_at, object.status
150
+ FROM artifact_objects AS object
151
+ LEFT JOIN artifacts AS reference ON reference.sha256 = object.sha256
152
+ WHERE reference.artifact_id IS NULL
153
+ ORDER BY object.sha256
154
+ `).all();
155
+ return rows.map(mapObject);
156
+ }
157
+ deleteObject(sha256) {
158
+ this.database.prepare(`
159
+ DELETE FROM artifact_objects
160
+ WHERE sha256 = ? AND NOT EXISTS (SELECT 1 FROM artifacts WHERE artifacts.sha256 = artifact_objects.sha256)
161
+ `).run(sha256);
162
+ }
163
+ stats(sessionId) {
164
+ const referenceWhere = sessionId ? "WHERE source_session_id = ?" : "";
165
+ const referenceArgs = sessionId ? [sessionId] : [];
166
+ const references = this.database.prepare(`SELECT count(*) AS count FROM artifacts ${referenceWhere}`).get(...referenceArgs);
167
+ const objectWhere = sessionId
168
+ ? "WHERE EXISTS (SELECT 1 FROM artifacts AS reference WHERE reference.sha256 = object.sha256 AND reference.source_session_id = ?)"
169
+ : "";
170
+ const objectArgs = sessionId ? [sessionId] : [];
171
+ const objects = this.database.prepare(`
172
+ SELECT count(*) AS count, COALESCE(sum(size_bytes), 0) AS bytes,
173
+ COALESCE(sum(status = 'missing'), 0) AS missing,
174
+ COALESCE(sum(status = 'corrupt'), 0) AS corrupt
175
+ FROM artifact_objects AS object ${objectWhere}
176
+ `).get(...objectArgs);
177
+ return {
178
+ objects: objects.count,
179
+ references: references.count,
180
+ bytes: objects.bytes,
181
+ missing: objects.missing,
182
+ corrupt: objects.corrupt,
183
+ };
184
+ }
185
+ }
186
+ //# sourceMappingURL=artifact-repository.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"artifact-repository.js","sourceRoot":"","sources":["../../../src/persistence/repositories/artifact-repository.ts"],"names":[],"mappings":"AA+EA,SAAS,QAAQ,CAAC,KAAa;IAC7B,IAAI,CAAC;QACH,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1C,OAAO,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACnE,CAAC,CAAC,MAAiC;YACnC,CAAC,CAAC,EAAE,CAAC;IACT,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,GAAc;IAC/B,OAAO;QACL,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,GAAG,CAAC,GAAG,CAAC,gBAAgB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAClF,MAAM,EAAE,GAAG,CAAC,MAAM;KACnB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,GAAiB;IACrC,OAAO;QACL,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,eAAe,EAAE,GAAG,CAAC,iBAAiB;QACtC,cAAc,EAAE,GAAG,CAAC,gBAAgB;QACpC,aAAa,EAAE,GAAG,CAAC,eAAe;QAClC,gBAAgB,EAAE,GAAG,CAAC,mBAAmB;QACzC,cAAc,EAAE,GAAG,CAAC,gBAAgB;QACpC,OAAO,EAAE,GAAG,CAAC,QAAQ,KAAK,CAAC;QAC3B,aAAa,EAAE,GAAG,CAAC,cAAc;QACjC,cAAc,EAAE,GAAG,CAAC,eAAe;QACnC,cAAc,EAAE,GAAG,CAAC,eAAe;QACnC,eAAe,EAAE,GAAG,CAAC,gBAAgB;QACrC,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC;KACtC,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,GAAc;IAC/B,OAAO;QACL,GAAG,YAAY,CAAC,GAAG,CAAC;QACpB,MAAM,EAAE;YACN,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,QAAQ,EAAE,GAAG,CAAC,SAAS;YACvB,QAAQ,EAAE,GAAG,CAAC,SAAS;YACvB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,SAAS,EAAE,GAAG,CAAC,iBAAiB;YAChC,GAAG,CAAC,GAAG,CAAC,gBAAgB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClF,MAAM,EAAE,GAAG,CAAC,MAAM;SACnB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,aAAa,GAAG;;;;;;;;;;;CAWrB,CAAC;AAEF,MAAM,OAAO,kBAAkB;IACA;IAA7B,YAA6B,QAAsB;QAAtB,aAAQ,GAAR,QAAQ,CAAc;IAAG,CAAC;IAEvD,IAAI,CAAC,MAAsB;QACzB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACtC,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;;;;OAUrB,CAAC,CAAC,GAAG,CACJ,MAAM,CAAC,MAAM,CAAC,MAAM,EACpB,MAAM,CAAC,MAAM,CAAC,QAAQ,EACtB,MAAM,CAAC,MAAM,CAAC,QAAQ,EACtB,MAAM,CAAC,MAAM,CAAC,SAAS,EACvB,MAAM,CAAC,MAAM,CAAC,SAAS,EACvB,MAAM,CAAC,MAAM,CAAC,cAAc,IAAI,IAAI,EACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CACrB,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;;;;OAUrB,CAAC,CAAC,GAAG,CACJ,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,gBAAgB,EACvB,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACtB,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,SAAS,EAChB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAChC,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa;gBAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,aAAa,CAAC,UAAkB,EAAE,SAAiB;QACjD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,aAAa;;KAEjD,CAAC,CAAC,GAAG,CAAC,UAAU,EAAE,SAAS,CAA0B,CAAC;QACvD,OAAO,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1C,CAAC;IAED,cAAc,CAAC,SAAiB,EAAE,cAAoC;QACpE,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,aAAa;;;KAGlD,CAAC,CAAC,GAAG,CAAC,SAAS,CAA2B,CAAC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpC,OAAO,cAAc;YACnB,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YACtE,CAAC,CAAC,OAAO,CAAC;IACd,CAAC;IAED,kBAAkB,CAAC,MAAc,EAAE,MAA4B,EAAE,UAAkB;QACjF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;KAErB,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,6BAA6B,CAAC,SAAiB,EAAE,WAAgC;QAC/E,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAChC,+DAA+D,CAChE,CAAC,GAAG,CAAC,SAAS,CAA8C,CAAC;QAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC;QACpF,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACtC,IAAI,CAAC;YACH,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC;oBAAE,SAAS;gBAC/C,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa;gBAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,iBAAiB;QACf,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;KAOlC,CAAC,CAAC,GAAG,EAA4B,CAAC;QACnC,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;KAGrB,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,MAAM,cAAc,GAAG,SAAS,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,MAAM,aAAa,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CACtC,2CAA2C,cAAc,EAAE,CAC5D,CAAC,GAAG,CAAC,GAAG,aAAa,CAAsB,CAAC;QAC7C,MAAM,WAAW,GAAG,SAAS;YAC3B,CAAC,CAAC,gIAAgI;YAClI,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,UAAU,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;wCAIF,WAAW;KAC9C,CAAC,CAAC,GAAG,CAAC,GAAG,UAAU,CAAuE,CAAC;QAC5F,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,KAAK;YACtB,UAAU,EAAE,UAAU,CAAC,KAAK;YAC5B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,19 @@
1
+ import type { DatabaseSync } from "node:sqlite";
2
+ import type { TokenCalibrationSample } from "../../core/model-awareness.ts";
3
+ import type { ContextManifest } from "../../manifest/context-manifest.ts";
4
+ export interface ProviderTokenUsage {
5
+ inputTokens: number;
6
+ cacheReadTokens: number;
7
+ cacheWriteTokens: number;
8
+ }
9
+ export declare class ContextManifestRepository {
10
+ private readonly database;
11
+ constructor(database: DatabaseSync);
12
+ save(manifest: ContextManifest): void;
13
+ get(manifestId: string): ContextManifest | undefined;
14
+ getLatest(sessionId: string): ContextManifest | undefined;
15
+ recordProviderUsage(manifestId: string, usage: ProviderTokenUsage, createdAt: number): ContextManifest | undefined;
16
+ recordActualInput(manifestId: string, actualInputTokens: number, createdAt: number): ContextManifest | undefined;
17
+ listCalibrationSamples(provider: string, model: string, limit: number, estimatorVersion?: string): TokenCalibrationSample[];
18
+ }
19
+ //# sourceMappingURL=context-manifest-repository.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"context-manifest-repository.d.ts","sourceRoot":"","sources":["../../../src/persistence/repositories/context-manifest-repository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAC5E,OAAO,KAAK,EACV,eAAe,EAEhB,MAAM,oCAAoC,CAAC;AAgB5C,MAAM,WAAW,kBAAkB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAiBD,qBAAa,yBAAyB;IACxB,OAAO,CAAC,QAAQ,CAAC,QAAQ;gBAAR,QAAQ,EAAE,YAAY;IAEnD,IAAI,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IA2CrC,GAAG,CAAC,UAAU,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAOpD,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAWzD,mBAAmB,CACjB,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,kBAAkB,EACzB,SAAS,EAAE,MAAM,GAChB,eAAe,GAAG,SAAS;IA6E9B,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAQhH,sBAAsB,CACpB,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,gBAAgB,SAAa,GAC5B,sBAAsB,EAAE;CAmB5B"}
@@ -0,0 +1,144 @@
1
+ function validTokenCount(value) {
2
+ return Number.isSafeInteger(value) && value >= 0;
3
+ }
4
+ function providerUsage(input) {
5
+ const totalInputTokens = input.inputTokens + input.cacheReadTokens + input.cacheWriteTokens;
6
+ const rounded = (value) => Math.round(value * 1_000_000) / 1_000_000;
7
+ return {
8
+ ...input,
9
+ totalInputTokens,
10
+ cacheReadShare: totalInputTokens > 0 ? rounded(input.cacheReadTokens / totalInputTokens) : 0,
11
+ cacheWriteShare: totalInputTokens > 0 ? rounded(input.cacheWriteTokens / totalInputTokens) : 0,
12
+ };
13
+ }
14
+ export class ContextManifestRepository {
15
+ database;
16
+ constructor(database) {
17
+ this.database = database;
18
+ }
19
+ save(manifest) {
20
+ this.database.prepare(`
21
+ INSERT INTO context_manifests(
22
+ manifest_id, session_id, created_at, provider, model,
23
+ estimated_tokens, actual_tokens, manifest_json,
24
+ prompt_hash, policy_version, planner_version,
25
+ branch_leaf_id, context_window, output_reserve,
26
+ hard_input_limit, target_input_tokens, pi_reported_tokens,
27
+ input_tokens, cache_read_tokens, cache_write_tokens
28
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
29
+ ON CONFLICT(manifest_id) DO UPDATE SET
30
+ estimated_tokens = excluded.estimated_tokens,
31
+ actual_tokens = excluded.actual_tokens,
32
+ manifest_json = excluded.manifest_json,
33
+ prompt_hash = excluded.prompt_hash,
34
+ pi_reported_tokens = excluded.pi_reported_tokens,
35
+ input_tokens = excluded.input_tokens,
36
+ cache_read_tokens = excluded.cache_read_tokens,
37
+ cache_write_tokens = excluded.cache_write_tokens
38
+ `).run(manifest.id, manifest.sessionId, manifest.createdAt, manifest.provider, manifest.model, manifest.estimatedInputTokens, manifest.actualInputTokens ?? null, JSON.stringify(manifest), manifest.promptHash, manifest.policyVersion, manifest.plannerVersion, manifest.branchLeafId ?? null, manifest.contextWindow, manifest.outputReserve, manifest.hardInputLimit, manifest.targetInputTokens, manifest.piReportedContextTokens ?? null, manifest.providerUsage?.inputTokens ?? null, manifest.providerUsage?.cacheReadTokens ?? null, manifest.providerUsage?.cacheWriteTokens ?? null);
39
+ }
40
+ get(manifestId) {
41
+ const row = this.database.prepare("SELECT manifest_json, actual_tokens FROM context_manifests WHERE manifest_id = ?").get(manifestId);
42
+ return row ? parseManifest(row.manifest_json) : undefined;
43
+ }
44
+ getLatest(sessionId) {
45
+ const row = this.database.prepare(`
46
+ SELECT manifest_json, actual_tokens
47
+ FROM context_manifests
48
+ WHERE session_id = ?
49
+ ORDER BY created_at DESC, rowid DESC
50
+ LIMIT 1
51
+ `).get(sessionId);
52
+ return row ? parseManifest(row.manifest_json) : undefined;
53
+ }
54
+ recordProviderUsage(manifestId, usage, createdAt) {
55
+ if (!validTokenCount(usage.inputTokens)
56
+ || !validTokenCount(usage.cacheReadTokens)
57
+ || !validTokenCount(usage.cacheWriteTokens)
58
+ || !Number.isSafeInteger(createdAt)
59
+ || createdAt < 0) {
60
+ return this.get(manifestId);
61
+ }
62
+ const recordedUsage = providerUsage(usage);
63
+ if (recordedUsage.totalInputTokens <= 0)
64
+ return this.get(manifestId);
65
+ this.database.exec("BEGIN IMMEDIATE");
66
+ try {
67
+ const row = this.database.prepare(`
68
+ SELECT manifest_json, actual_tokens
69
+ FROM context_manifests
70
+ WHERE manifest_id = ?
71
+ `).get(manifestId);
72
+ if (!row) {
73
+ this.database.exec("COMMIT");
74
+ return undefined;
75
+ }
76
+ const manifest = parseManifest(row.manifest_json);
77
+ if (row.actual_tokens !== null) {
78
+ this.database.exec("COMMIT");
79
+ return manifest;
80
+ }
81
+ const updated = {
82
+ ...manifest,
83
+ actualInputTokens: recordedUsage.totalInputTokens,
84
+ providerUsage: recordedUsage,
85
+ };
86
+ this.database.prepare(`
87
+ UPDATE context_manifests
88
+ SET actual_tokens = ?, input_tokens = ?, cache_read_tokens = ?,
89
+ cache_write_tokens = ?, manifest_json = ?
90
+ WHERE manifest_id = ?
91
+ `).run(recordedUsage.totalInputTokens, recordedUsage.inputTokens, recordedUsage.cacheReadTokens, recordedUsage.cacheWriteTokens, JSON.stringify(updated), manifestId);
92
+ if (manifest.estimatedInputTokens > 0) {
93
+ this.database.prepare(`
94
+ INSERT INTO token_calibration(
95
+ provider, model, estimated, actual, ratio, created_at,
96
+ manifest_id, estimator_version, input_tokens, cache_read_tokens, cache_write_tokens
97
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
98
+ `).run(manifest.provider, manifest.model, manifest.estimatedInputTokens, recordedUsage.totalInputTokens, recordedUsage.totalInputTokens / manifest.estimatedInputTokens, createdAt, manifestId, manifest.modelAwareness?.calibration.estimator ?? "chars-v1", recordedUsage.inputTokens, recordedUsage.cacheReadTokens, recordedUsage.cacheWriteTokens);
99
+ }
100
+ this.database.exec("COMMIT");
101
+ return updated;
102
+ }
103
+ catch (error) {
104
+ if (this.database.isTransaction)
105
+ this.database.exec("ROLLBACK");
106
+ throw error;
107
+ }
108
+ }
109
+ recordActualInput(manifestId, actualInputTokens, createdAt) {
110
+ return this.recordProviderUsage(manifestId, {
111
+ inputTokens: actualInputTokens,
112
+ cacheReadTokens: 0,
113
+ cacheWriteTokens: 0,
114
+ }, createdAt);
115
+ }
116
+ listCalibrationSamples(provider, model, limit, estimatorVersion = "chars-v1") {
117
+ if (!Number.isInteger(limit) || limit <= 0)
118
+ return [];
119
+ const rows = this.database.prepare(`
120
+ SELECT estimated, actual, input_tokens, cache_read_tokens, cache_write_tokens, created_at
121
+ FROM token_calibration
122
+ WHERE provider = ? AND model = ? AND estimator_version = ?
123
+ AND estimated > 0 AND actual > 0
124
+ ORDER BY created_at DESC, calibration_id DESC
125
+ LIMIT ?
126
+ `).all(provider, model, estimatorVersion, limit);
127
+ return rows.map((row) => ({
128
+ estimatedTokens: row.estimated,
129
+ actualInputTokens: row.actual,
130
+ inputTokens: row.input_tokens ?? row.actual,
131
+ cacheReadTokens: row.cache_read_tokens ?? 0,
132
+ cacheWriteTokens: row.cache_write_tokens ?? 0,
133
+ createdAt: row.created_at,
134
+ }));
135
+ }
136
+ }
137
+ function parseManifest(value) {
138
+ const parsed = JSON.parse(value);
139
+ if (!parsed || typeof parsed !== "object" || !("id" in parsed) || typeof parsed.id !== "string") {
140
+ throw new Error("Stored Context Manifest is invalid");
141
+ }
142
+ return parsed;
143
+ }
144
+ //# sourceMappingURL=context-manifest-repository.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"context-manifest-repository.js","sourceRoot":"","sources":["../../../src/persistence/repositories/context-manifest-repository.ts"],"names":[],"mappings":"AA2BA,SAAS,eAAe,CAAC,KAAa;IACpC,OAAO,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,aAAa,CAAC,KAAyB;IAC9C,MAAM,gBAAgB,GAAG,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,eAAe,GAAG,KAAK,CAAC,gBAAgB,CAAC;IAC5F,MAAM,OAAO,GAAG,CAAC,KAAa,EAAU,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,SAAS,CAAC;IACrF,OAAO;QACL,GAAG,KAAK;QACR,gBAAgB;QAChB,cAAc,EAAE,gBAAgB,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5F,eAAe,EAAE,gBAAgB,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;KAC/F,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,yBAAyB;IACP;IAA7B,YAA6B,QAAsB;QAAtB,aAAQ,GAAR,QAAQ,CAAc;IAAG,CAAC;IAEvD,IAAI,CAAC,QAAyB;QAC5B,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;;;;;;;;;;;;KAkBrB,CAAC,CAAC,GAAG,CACJ,QAAQ,CAAC,EAAE,EACX,QAAQ,CAAC,SAAS,EAClB,QAAQ,CAAC,SAAS,EAClB,QAAQ,CAAC,QAAQ,EACjB,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,oBAAoB,EAC7B,QAAQ,CAAC,iBAAiB,IAAI,IAAI,EAClC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EACxB,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,aAAa,EACtB,QAAQ,CAAC,cAAc,EACvB,QAAQ,CAAC,YAAY,IAAI,IAAI,EAC7B,QAAQ,CAAC,aAAa,EACtB,QAAQ,CAAC,aAAa,EACtB,QAAQ,CAAC,cAAc,EACvB,QAAQ,CAAC,iBAAiB,EAC1B,QAAQ,CAAC,uBAAuB,IAAI,IAAI,EACxC,QAAQ,CAAC,aAAa,EAAE,WAAW,IAAI,IAAI,EAC3C,QAAQ,CAAC,aAAa,EAAE,eAAe,IAAI,IAAI,EAC/C,QAAQ,CAAC,aAAa,EAAE,gBAAgB,IAAI,IAAI,CACjD,CAAC;IACJ,CAAC;IAED,GAAG,CAAC,UAAkB;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAC/B,kFAAkF,CACnF,CAAC,GAAG,CAAC,UAAU,CAAuC,CAAC;QACxD,OAAO,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,CAAC;IAED,SAAS,CAAC,SAAiB;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;KAMjC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAuC,CAAC;QACxD,OAAO,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,CAAC;IAED,mBAAmB,CACjB,UAAkB,EAClB,KAAyB,EACzB,SAAiB;QAEjB,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,WAAW,CAAC;eAClC,CAAC,eAAe,CAAC,KAAK,CAAC,eAAe,CAAC;eACvC,CAAC,eAAe,CAAC,KAAK,CAAC,gBAAgB,CAAC;eACxC,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC;eAChC,SAAS,GAAG,CAAC,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;QACD,MAAM,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,aAAa,CAAC,gBAAgB,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAErE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACtC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;OAIjC,CAAC,CAAC,GAAG,CAAC,UAAU,CAAuC,CAAC;YACzD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC7B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YAClD,IAAI,GAAG,CAAC,aAAa,KAAK,IAAI,EAAE,CAAC;gBAC/B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC7B,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,MAAM,OAAO,GAAoB;gBAC/B,GAAG,QAAQ;gBACX,iBAAiB,EAAE,aAAa,CAAC,gBAAgB;gBACjD,aAAa,EAAE,aAAa;aAC7B,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;OAKrB,CAAC,CAAC,GAAG,CACJ,aAAa,CAAC,gBAAgB,EAC9B,aAAa,CAAC,WAAW,EACzB,aAAa,CAAC,eAAe,EAC7B,aAAa,CAAC,gBAAgB,EAC9B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,UAAU,CACX,CAAC;YAEF,IAAI,QAAQ,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;SAKrB,CAAC,CAAC,GAAG,CACJ,QAAQ,CAAC,QAAQ,EACjB,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,oBAAoB,EAC7B,aAAa,CAAC,gBAAgB,EAC9B,aAAa,CAAC,gBAAgB,GAAG,QAAQ,CAAC,oBAAoB,EAC9D,SAAS,EACT,UAAU,EACV,QAAQ,CAAC,cAAc,EAAE,WAAW,CAAC,SAAS,IAAI,UAAU,EAC5D,aAAa,CAAC,WAAW,EACzB,aAAa,CAAC,eAAe,EAC7B,aAAa,CAAC,gBAAgB,CAC/B,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7B,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa;gBAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,UAAkB,EAAE,iBAAyB,EAAE,SAAiB;QAChF,OAAO,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE;YAC1C,WAAW,EAAE,iBAAiB;YAC9B,eAAe,EAAE,CAAC;YAClB,gBAAgB,EAAE,CAAC;SACpB,EAAE,SAAS,CAAC,CAAC;IAChB,CAAC;IAED,sBAAsB,CACpB,QAAgB,EAChB,KAAa,EACb,KAAa,EACb,gBAAgB,GAAG,UAAU;QAE7B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC;YAAE,OAAO,EAAE,CAAC;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;KAOlC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,CAAgC,CAAC;QAChF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,eAAe,EAAE,GAAG,CAAC,SAAS;YAC9B,iBAAiB,EAAE,GAAG,CAAC,MAAM;YAC7B,WAAW,EAAE,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,MAAM;YAC3C,eAAe,EAAE,GAAG,CAAC,iBAAiB,IAAI,CAAC;YAC3C,gBAAgB,EAAE,GAAG,CAAC,kBAAkB,IAAI,CAAC;YAC7C,SAAS,EAAE,GAAG,CAAC,UAAU;SAC1B,CAAC,CAAC,CAAC;IACN,CAAC;CACF;AAED,SAAS,aAAa,CAAC,KAAa;IAClC,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;QAChG,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACxD,CAAC;IACD,OAAO,MAAyB,CAAC;AACnC,CAAC"}
@@ -0,0 +1,33 @@
1
+ import type { DatabaseSync } from "node:sqlite";
2
+ import { type MemoryItem, type MemoryMaterializationResult, type PinItem, type StoredMemoryMutation, type StoredPinMutation } from "../../memory/memory-types.ts";
3
+ export declare class MemoryRepository {
4
+ private readonly database;
5
+ constructor(database: DatabaseSync);
6
+ reconcileSession(sessionId: string, memoryMutations: readonly StoredMemoryMutation[], pinMutations: readonly StoredPinMutation[]): MemoryMaterializationResult;
7
+ getMemory(id: string): MemoryItem | undefined;
8
+ listMemories(options: {
9
+ sessionId: string;
10
+ projectPath: string;
11
+ includeProject: boolean;
12
+ activeOnly?: boolean;
13
+ }): MemoryItem[];
14
+ getPin(id: string): PinItem | undefined;
15
+ listPins(options: {
16
+ sessionId: string;
17
+ projectPath: string;
18
+ includeProject: boolean;
19
+ activeOnly?: boolean;
20
+ }): PinItem[];
21
+ stats(sessionId: string, projectPath: string): {
22
+ activeMemories: number;
23
+ inactiveMemories: number;
24
+ activePins: number;
25
+ inactivePins: number;
26
+ };
27
+ private materialize;
28
+ private applyMemoryMutation;
29
+ private applyPinMutation;
30
+ private memorySourceIds;
31
+ private memorySources;
32
+ }
33
+ //# sourceMappingURL=memory-repository.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memory-repository.d.ts","sourceRoot":"","sources":["../../../src/persistence/repositories/memory-repository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,EAGL,KAAK,UAAU,EACf,KAAK,2BAA2B,EAGhC,KAAK,OAAO,EAGZ,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACvB,MAAM,8BAA8B,CAAC;AAqItC,qBAAa,gBAAgB;IACf,OAAO,CAAC,QAAQ,CAAC,QAAQ;gBAAR,QAAQ,EAAE,YAAY;IAEnD,gBAAgB,CACd,SAAS,EAAE,MAAM,EACjB,eAAe,EAAE,SAAS,oBAAoB,EAAE,EAChD,YAAY,EAAE,SAAS,iBAAiB,EAAE,GACzC,2BAA2B;IAsC9B,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,UAAU,GAAG,SAAS;IAK7C,YAAY,CAAC,OAAO,EAAE;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,cAAc,EAAE,OAAO,CAAC;QACxB,UAAU,CAAC,EAAE,OAAO,CAAC;KACtB,GAAG,UAAU,EAAE;IAchB,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS;IAKvC,QAAQ,CAAC,OAAO,EAAE;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,cAAc,EAAE,OAAO,CAAC;QACxB,UAAU,CAAC,EAAE,OAAO,CAAC;KACtB,GAAG,OAAO,EAAE;IAab,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG;QAC7C,cAAc,EAAE,MAAM,CAAC;QACvB,gBAAgB,EAAE,MAAM,CAAC;QACzB,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,EAAE,MAAM,CAAC;KACtB;IAwBD,OAAO,CAAC,WAAW;IA8JnB,OAAO,CAAC,mBAAmB;IAoE3B,OAAO,CAAC,gBAAgB;IAsExB,OAAO,CAAC,eAAe;IAIvB,OAAO,CAAC,aAAa;CAiBtB"}