memo-grafter 0.2.7 → 0.3.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 (205) hide show
  1. package/README.md +26 -0
  2. package/USER_GUIDE.md +262 -17
  3. package/dist/MemoGrafter.d.ts +11 -5
  4. package/dist/MemoGrafter.d.ts.map +1 -1
  5. package/dist/MemoGrafter.js +64 -4
  6. package/dist/MemoGrafter.js.map +1 -1
  7. package/dist/MemoGrafterAgent.d.ts +8 -1
  8. package/dist/MemoGrafterAgent.d.ts.map +1 -1
  9. package/dist/MemoGrafterAgent.js +31 -2
  10. package/dist/MemoGrafterAgent.js.map +1 -1
  11. package/dist/adapters/AnthropicAdapter.d.ts +1 -1
  12. package/dist/adapters/AnthropicAdapter.d.ts.map +1 -1
  13. package/dist/adapters/GeminiAdapter.d.ts +1 -1
  14. package/dist/adapters/GeminiAdapter.d.ts.map +1 -1
  15. package/dist/adapters/OpenAIAdapter.d.ts +1 -1
  16. package/dist/adapters/OpenAIAdapter.d.ts.map +1 -1
  17. package/dist/adapters/types.d.ts +1 -1
  18. package/dist/adapters/types.d.ts.map +1 -1
  19. package/dist/agents/MemoGrafterAgent.d.ts +45 -0
  20. package/dist/agents/MemoGrafterAgent.d.ts.map +1 -0
  21. package/dist/agents/MemoGrafterAgent.js +278 -0
  22. package/dist/agents/MemoGrafterAgent.js.map +1 -0
  23. package/dist/agents/fleet/ConductorAgent.d.ts +15 -0
  24. package/dist/agents/fleet/ConductorAgent.d.ts.map +1 -0
  25. package/dist/agents/fleet/ConductorAgent.js +41 -0
  26. package/dist/agents/fleet/ConductorAgent.js.map +1 -0
  27. package/dist/agents/fleet/FleetStore.d.ts +12 -0
  28. package/dist/agents/fleet/FleetStore.d.ts.map +1 -0
  29. package/dist/agents/fleet/FleetStore.js +30 -0
  30. package/dist/agents/fleet/FleetStore.js.map +1 -0
  31. package/dist/agents/fleet/MemoGrafterFleet.d.ts +27 -0
  32. package/dist/agents/fleet/MemoGrafterFleet.d.ts.map +1 -0
  33. package/dist/agents/fleet/MemoGrafterFleet.js +87 -0
  34. package/dist/agents/fleet/MemoGrafterFleet.js.map +1 -0
  35. package/dist/agents/fleet/WorkerAgent.d.ts +36 -0
  36. package/dist/agents/fleet/WorkerAgent.d.ts.map +1 -0
  37. package/dist/agents/fleet/WorkerAgent.js +140 -0
  38. package/dist/agents/fleet/WorkerAgent.js.map +1 -0
  39. package/dist/agents/fleet/types.d.ts +58 -0
  40. package/dist/agents/fleet/types.d.ts.map +1 -0
  41. package/dist/agents/fleet/types.js +2 -0
  42. package/dist/agents/fleet/types.js.map +1 -0
  43. package/dist/cli/commands/init.d.ts +8 -0
  44. package/dist/cli/commands/init.d.ts.map +1 -0
  45. package/dist/cli/commands/init.js +124 -0
  46. package/dist/cli/commands/init.js.map +1 -0
  47. package/dist/cli/commands/migrate.d.ts +6 -0
  48. package/dist/cli/commands/migrate.d.ts.map +1 -0
  49. package/dist/cli/commands/migrate.js +34 -0
  50. package/dist/cli/commands/migrate.js.map +1 -0
  51. package/dist/cli/index.d.ts +3 -0
  52. package/dist/cli/index.d.ts.map +1 -0
  53. package/dist/cli/index.js +39 -0
  54. package/dist/cli/index.js.map +1 -0
  55. package/dist/cli/utils/config.d.ts +11 -0
  56. package/dist/cli/utils/config.d.ts.map +1 -0
  57. package/dist/cli/utils/config.js +92 -0
  58. package/dist/cli/utils/config.js.map +1 -0
  59. package/dist/cli/utils/logger.d.ts +7 -0
  60. package/dist/cli/utils/logger.d.ts.map +1 -0
  61. package/dist/cli/utils/logger.js +15 -0
  62. package/dist/cli/utils/logger.js.map +1 -0
  63. package/dist/core/MemoGrafter.d.ts +46 -0
  64. package/dist/core/MemoGrafter.d.ts.map +1 -0
  65. package/dist/core/MemoGrafter.js +236 -0
  66. package/dist/core/MemoGrafter.js.map +1 -0
  67. package/dist/core/types.d.ts +319 -0
  68. package/dist/core/types.d.ts.map +1 -0
  69. package/dist/core/types.js +2 -0
  70. package/dist/core/types.js.map +1 -0
  71. package/dist/crawler/DecayScoringPass.d.ts.map +1 -1
  72. package/dist/crawler/DecayScoringPass.js +6 -0
  73. package/dist/crawler/DecayScoringPass.js.map +1 -1
  74. package/dist/crawler/memoryMaintenance.d.ts +1 -0
  75. package/dist/crawler/memoryMaintenance.d.ts.map +1 -1
  76. package/dist/crawler/memoryMaintenance.js +2 -1
  77. package/dist/crawler/memoryMaintenance.js.map +1 -1
  78. package/dist/crawler/types.d.ts +1 -0
  79. package/dist/crawler/types.d.ts.map +1 -1
  80. package/dist/fleet/MemoGrafterFleet.d.ts +9 -2
  81. package/dist/fleet/MemoGrafterFleet.d.ts.map +1 -1
  82. package/dist/fleet/MemoGrafterFleet.js +39 -1
  83. package/dist/fleet/MemoGrafterFleet.js.map +1 -1
  84. package/dist/fleet/WorkerAgent.d.ts +9 -4
  85. package/dist/fleet/WorkerAgent.d.ts.map +1 -1
  86. package/dist/fleet/WorkerAgent.js +45 -3
  87. package/dist/fleet/WorkerAgent.js.map +1 -1
  88. package/dist/fleet/types.d.ts +21 -2
  89. package/dist/fleet/types.d.ts.map +1 -1
  90. package/dist/index.d.ts +13 -12
  91. package/dist/index.d.ts.map +1 -1
  92. package/dist/index.js +9 -9
  93. package/dist/index.js.map +1 -1
  94. package/dist/ingestion/IngestQueue.d.ts +17 -0
  95. package/dist/ingestion/IngestQueue.d.ts.map +1 -0
  96. package/dist/ingestion/IngestQueue.js +123 -0
  97. package/dist/ingestion/IngestQueue.js.map +1 -0
  98. package/dist/ingestion/conversation/IngestPipeline.d.ts +41 -0
  99. package/dist/ingestion/conversation/IngestPipeline.d.ts.map +1 -0
  100. package/dist/ingestion/conversation/IngestPipeline.js +189 -0
  101. package/dist/ingestion/conversation/IngestPipeline.js.map +1 -0
  102. package/dist/ingestion/conversation/SegmentProcessor.d.ts +20 -0
  103. package/dist/ingestion/conversation/SegmentProcessor.d.ts.map +1 -0
  104. package/dist/ingestion/conversation/SegmentProcessor.js +110 -0
  105. package/dist/ingestion/conversation/SegmentProcessor.js.map +1 -0
  106. package/dist/ingestion/conversation/TopicDriftDetector.d.ts +40 -0
  107. package/dist/ingestion/conversation/TopicDriftDetector.d.ts.map +1 -0
  108. package/dist/ingestion/conversation/TopicDriftDetector.js +222 -0
  109. package/dist/ingestion/conversation/TopicDriftDetector.js.map +1 -0
  110. package/dist/maintenance/ConflictDetectionPass.d.ts +6 -0
  111. package/dist/maintenance/ConflictDetectionPass.d.ts.map +1 -0
  112. package/dist/maintenance/ConflictDetectionPass.js +46 -0
  113. package/dist/maintenance/ConflictDetectionPass.js.map +1 -0
  114. package/dist/maintenance/DecayScoringPass.d.ts +17 -0
  115. package/dist/maintenance/DecayScoringPass.d.ts.map +1 -0
  116. package/dist/maintenance/DecayScoringPass.js +79 -0
  117. package/dist/maintenance/DecayScoringPass.js.map +1 -0
  118. package/dist/maintenance/MemoGrafterCrawler.d.ts +14 -0
  119. package/dist/maintenance/MemoGrafterCrawler.d.ts.map +1 -0
  120. package/dist/maintenance/MemoGrafterCrawler.js +108 -0
  121. package/dist/maintenance/MemoGrafterCrawler.js.map +1 -0
  122. package/dist/maintenance/VersioningPass.d.ts +6 -0
  123. package/dist/maintenance/VersioningPass.d.ts.map +1 -0
  124. package/dist/maintenance/VersioningPass.js +40 -0
  125. package/dist/maintenance/VersioningPass.js.map +1 -0
  126. package/dist/maintenance/decayScoring.d.ts +7 -0
  127. package/dist/maintenance/decayScoring.d.ts.map +1 -0
  128. package/dist/maintenance/decayScoring.js +9 -0
  129. package/dist/maintenance/decayScoring.js.map +1 -0
  130. package/dist/maintenance/index.d.ts +7 -0
  131. package/dist/maintenance/index.d.ts.map +1 -0
  132. package/dist/maintenance/index.js +5 -0
  133. package/dist/maintenance/index.js.map +1 -0
  134. package/dist/maintenance/memoryMaintenance.d.ts +23 -0
  135. package/dist/maintenance/memoryMaintenance.d.ts.map +1 -0
  136. package/dist/maintenance/memoryMaintenance.js +141 -0
  137. package/dist/maintenance/memoryMaintenance.js.map +1 -0
  138. package/dist/maintenance/types.d.ts +65 -0
  139. package/dist/maintenance/types.d.ts.map +1 -0
  140. package/dist/maintenance/types.js +2 -0
  141. package/dist/maintenance/types.js.map +1 -0
  142. package/dist/pipeline/GrafterPipeline.d.ts +2 -0
  143. package/dist/pipeline/GrafterPipeline.d.ts.map +1 -1
  144. package/dist/pipeline/GrafterPipeline.js +33 -7
  145. package/dist/pipeline/GrafterPipeline.js.map +1 -1
  146. package/dist/pipeline/RetrieverPipeline.d.ts +1 -0
  147. package/dist/pipeline/RetrieverPipeline.d.ts.map +1 -1
  148. package/dist/pipeline/RetrieverPipeline.js +16 -5
  149. package/dist/pipeline/RetrieverPipeline.js.map +1 -1
  150. package/dist/prompts/factRetrievalPrompt.d.ts +1 -1
  151. package/dist/prompts/factRetrievalPrompt.d.ts.map +1 -1
  152. package/dist/prompts/historyCompressionPrompt.d.ts +1 -1
  153. package/dist/prompts/historyCompressionPrompt.d.ts.map +1 -1
  154. package/dist/prompts/intentShiftPrompt.d.ts +1 -1
  155. package/dist/prompts/intentShiftPrompt.d.ts.map +1 -1
  156. package/dist/prompts/memoryInjectionPrompt.d.ts +1 -1
  157. package/dist/prompts/memoryInjectionPrompt.d.ts.map +1 -1
  158. package/dist/prompts/segmentExtractionPrompt.d.ts +1 -1
  159. package/dist/prompts/segmentExtractionPrompt.d.ts.map +1 -1
  160. package/dist/retrieval/GrafterPipeline.d.ts +26 -0
  161. package/dist/retrieval/GrafterPipeline.d.ts.map +1 -0
  162. package/dist/retrieval/GrafterPipeline.js +117 -0
  163. package/dist/retrieval/GrafterPipeline.js.map +1 -0
  164. package/dist/retrieval/RetrieverPipeline.d.ts +28 -0
  165. package/dist/retrieval/RetrieverPipeline.d.ts.map +1 -0
  166. package/dist/retrieval/RetrieverPipeline.js +160 -0
  167. package/dist/retrieval/RetrieverPipeline.js.map +1 -0
  168. package/dist/schema/builders.d.ts +40 -0
  169. package/dist/schema/builders.d.ts.map +1 -0
  170. package/dist/schema/builders.js +10 -0
  171. package/dist/schema/builders.js.map +1 -0
  172. package/dist/schema/index.d.ts +3 -0
  173. package/dist/schema/index.d.ts.map +1 -0
  174. package/dist/schema/index.js +3 -0
  175. package/dist/schema/index.js.map +1 -0
  176. package/dist/schema/mg-tables.d.ts +7 -0
  177. package/dist/schema/mg-tables.d.ts.map +1 -0
  178. package/dist/schema/mg-tables.js +180 -0
  179. package/dist/schema/mg-tables.js.map +1 -0
  180. package/dist/store/GraphStore.d.ts +19 -1
  181. package/dist/store/GraphStore.d.ts.map +1 -1
  182. package/dist/store/postgres-pgvector/GraphStore.d.ts +33 -1
  183. package/dist/store/postgres-pgvector/GraphStore.d.ts.map +1 -1
  184. package/dist/store/postgres-pgvector/GraphStore.js +491 -16
  185. package/dist/store/postgres-pgvector/GraphStore.js.map +1 -1
  186. package/dist/tsconfig.cli.tsbuildinfo +1 -0
  187. package/dist/types.d.ts +53 -0
  188. package/dist/types.d.ts.map +1 -1
  189. package/dist/utils/drift/adaptiveDriftSensitivity.d.ts +1 -1
  190. package/dist/utils/drift/adaptiveDriftSensitivity.d.ts.map +1 -1
  191. package/dist/utils/drift/driftScore.d.ts +1 -1
  192. package/dist/utils/drift/driftScore.d.ts.map +1 -1
  193. package/dist/utils/drift/driftThreshold.d.ts +1 -1
  194. package/dist/utils/drift/driftThreshold.d.ts.map +1 -1
  195. package/dist/utils/drift/reentryMatch.d.ts +1 -1
  196. package/dist/utils/drift/reentryMatch.d.ts.map +1 -1
  197. package/dist/utils/extraction/segmentExtraction.d.ts +1 -1
  198. package/dist/utils/extraction/segmentExtraction.d.ts.map +1 -1
  199. package/dist/utils/reentry/reentryEdges.d.ts +1 -1
  200. package/dist/utils/reentry/reentryEdges.d.ts.map +1 -1
  201. package/dist/utils/reentry/reentrySimilarity.d.ts +1 -1
  202. package/dist/utils/reentry/reentrySimilarity.d.ts.map +1 -1
  203. package/dist/utils/reentry/reentryText.d.ts +1 -1
  204. package/dist/utils/reentry/reentryText.d.ts.map +1 -1
  205. package/package.json +14 -1
@@ -0,0 +1,41 @@
1
+ import { WorkerAgent } from "./WorkerAgent.js";
2
+ export class ConductorAgent {
3
+ core;
4
+ fleetId;
5
+ getWorkerByColor;
6
+ constructor(core, fleetId, getWorkerByColor) {
7
+ this.core = core;
8
+ this.fleetId = fleetId;
9
+ this.getWorkerByColor = getWorkerByColor;
10
+ }
11
+ async graftColorIntoAgent(sourceColor, targetAgent, options = {}) {
12
+ const nodes = await this.selectNodesByColor(sourceColor, options);
13
+ return targetAgent.ingestGraftedNodes(nodes);
14
+ }
15
+ async graftByPrompt(prompt, targetAgent, options = {}) {
16
+ const sourceNodes = await this.core.store.getSimilarNodesAcrossFleet(this.fleetId, await this.core.embedder.embed(prompt), {
17
+ k: options.limit ?? 5,
18
+ minSimilarity: options.minSimilarity ?? 0.6,
19
+ });
20
+ return targetAgent.ingestGraftedNodes(sourceNodes);
21
+ }
22
+ getWorker(color) {
23
+ return this.getWorkerByColor(color);
24
+ }
25
+ async selectNodesByColor(sourceColor, options) {
26
+ if (options.prompt) {
27
+ return this.core.store.getSimilarNodesAcrossFleet(this.fleetId, await this.core.embedder.embed(options.prompt), {
28
+ k: options.limit ?? 5,
29
+ minSimilarity: options.minSimilarity ?? 0.6,
30
+ agentColor: sourceColor,
31
+ });
32
+ }
33
+ const nodes = await this.core.store.getNodesByColor(this.fleetId, sourceColor);
34
+ if (options.topicIds && options.topicIds.length > 0) {
35
+ const topicIds = new Set(options.topicIds);
36
+ return nodes.filter((node) => topicIds.has(node.id));
37
+ }
38
+ return nodes.slice(0, options.limit ?? nodes.length);
39
+ }
40
+ }
41
+ //# sourceMappingURL=ConductorAgent.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ConductorAgent.js","sourceRoot":"","sources":["../../../src/agents/fleet/ConductorAgent.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C,MAAM,OAAO,cAAc;IAEN;IACA;IACA;IAHnB,YACmB,IAAiB,EACjB,OAAe,EACf,gBAA4D;QAF5D,SAAI,GAAJ,IAAI,CAAa;QACjB,YAAO,GAAP,OAAO,CAAQ;QACf,qBAAgB,GAAhB,gBAAgB,CAA4C;IAC5E,CAAC;IAEJ,KAAK,CAAC,mBAAmB,CAAC,WAAmB,EAAE,WAAwB,EAAE,UAAiC,EAAE;QAC1G,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAClE,OAAO,WAAW,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,WAAwB,EAAE,UAAiD,EAAE;QAC/G,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,0BAA0B,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;YACzH,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;YACrB,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,GAAG;SAC5C,CAAC,CAAC;QACH,OAAO,WAAW,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;IACrD,CAAC;IAED,SAAS,CAAC,KAAa;QACrB,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,WAAmB,EAAE,OAA8B;QAClF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,0BAA0B,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;gBAC9G,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;gBACrB,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,GAAG;gBAC3C,UAAU,EAAE,WAAW;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC/E,IAAI,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC3C,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;IACvD,CAAC;CACF"}
@@ -0,0 +1,12 @@
1
+ import type { GraphStore } from "../../store/index.js";
2
+ import type { FleetAgentInfo, FleetGraph } from "./types.js";
3
+ export declare class FleetStore {
4
+ private readonly store;
5
+ constructor(store: GraphStore);
6
+ initializeFleet(fleetId: string, name?: string): Promise<void>;
7
+ upsertAgent(agent: FleetAgentInfo & {
8
+ fleetId: string;
9
+ }): Promise<void>;
10
+ getFleetGraph(fleetId: string, name?: string): Promise<FleetGraph>;
11
+ }
12
+ //# sourceMappingURL=FleetStore.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"FleetStore.d.ts","sourceRoot":"","sources":["../../../src/agents/fleet/FleetStore.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7D,qBAAa,UAAU;IACT,OAAO,CAAC,QAAQ,CAAC,KAAK;gBAAL,KAAK,EAAE,UAAU;IAExC,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9D,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IASvE,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;CAYzE"}
@@ -0,0 +1,30 @@
1
+ export class FleetStore {
2
+ store;
3
+ constructor(store) {
4
+ this.store = store;
5
+ }
6
+ async initializeFleet(fleetId, name) {
7
+ await this.store.saveFleet(fleetId, name);
8
+ }
9
+ async upsertAgent(agent) {
10
+ await this.store.saveFleetAgent({
11
+ id: agent.id,
12
+ fleetId: agent.fleetId,
13
+ sessionId: agent.sessionId,
14
+ agentColor: agent.color,
15
+ });
16
+ }
17
+ async getFleetGraph(fleetId, name) {
18
+ const agents = await this.store.getFleetAgents(fleetId);
19
+ return {
20
+ id: fleetId,
21
+ ...(name ? { name } : {}),
22
+ agents: agents.map((agent) => ({
23
+ id: agent.id,
24
+ sessionId: agent.sessionId,
25
+ color: agent.agentColor,
26
+ })),
27
+ };
28
+ }
29
+ }
30
+ //# sourceMappingURL=FleetStore.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"FleetStore.js","sourceRoot":"","sources":["../../../src/agents/fleet/FleetStore.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,UAAU;IACQ;IAA7B,YAA6B,KAAiB;QAAjB,UAAK,GAAL,KAAK,CAAY;IAAG,CAAC;IAElD,KAAK,CAAC,eAAe,CAAC,OAAe,EAAE,IAAa;QAClD,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAA2C;QAC3D,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC;YAC9B,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,KAAK,CAAC,KAAK;SACxB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAe,EAAE,IAAa;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACxD,OAAO;YACL,EAAE,EAAE,OAAO;YACX,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzB,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC7B,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,KAAK,EAAE,KAAK,CAAC,UAAU;aACxB,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,27 @@
1
+ import { MemoGrafter } from "../../core/MemoGrafter.js";
2
+ import type { FleetMemoryMode, MemoGrafterConfig } from "../../core/types.js";
3
+ import { ConductorAgent } from "./ConductorAgent.js";
4
+ import type { FleetGraph, FleetMemoryOptions, FleetRetrievalOptions, MemoGrafterFleetOptions, SharedMemorySnapshot, WorkerAgentConfig } from "./types.js";
5
+ import { WorkerAgent } from "./WorkerAgent.js";
6
+ export declare const FLEET_SHARED_AGENT_COLOR = "fleet";
7
+ export declare class MemoGrafterFleet {
8
+ private readonly core;
9
+ private readonly fleetStore;
10
+ private readonly workersByColor;
11
+ private readonly sharedSessionId;
12
+ readonly id: string;
13
+ readonly name: string | undefined;
14
+ readonly defaultWorkerMemory: FleetMemoryMode;
15
+ constructor(configOrCore: MemoGrafterConfig | MemoGrafter, options?: MemoGrafterFleetOptions);
16
+ initialize(): Promise<void>;
17
+ createConductor(): ConductorAgent;
18
+ createWorker(config: WorkerAgentConfig): Promise<WorkerAgent>;
19
+ getWorker(color: string): WorkerAgent | undefined;
20
+ getGraph(): Promise<FleetGraph>;
21
+ getSharedSessionId(): string;
22
+ ingestToFleet(text: string, options?: FleetMemoryOptions): Promise<import("../../core/types.js").TopicNode[]>;
23
+ recallFromFleet(query: string, options?: FleetRetrievalOptions): Promise<import("../../core/types.js").RetrievalResult>;
24
+ getSharedMemory(): Promise<SharedMemorySnapshot>;
25
+ close(): Promise<void>;
26
+ }
27
+ //# sourceMappingURL=MemoGrafterFleet.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"MemoGrafterFleet.d.ts","sourceRoot":"","sources":["../../../src/agents/fleet/MemoGrafterFleet.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAExD,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,KAAK,EACV,UAAU,EACV,kBAAkB,EAClB,qBAAqB,EACrB,uBAAuB,EACvB,oBAAoB,EACpB,iBAAiB,EAClB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C,eAAO,MAAM,wBAAwB,UAAU,CAAC;AAEhD,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAc;IACnC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IACxC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAkC;IACjE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,QAAQ,CAAC,mBAAmB,EAAE,eAAe,CAAC;gBAElC,YAAY,EAAE,iBAAiB,GAAG,WAAW,EAAE,OAAO,GAAE,uBAA4B;IAS1F,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAKjC,eAAe,IAAI,cAAc;IAI3B,YAAY,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC;IAmBnE,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAI3C,QAAQ,IAAI,OAAO,CAAC,UAAU,CAAC;IAIrC,kBAAkB,IAAI,MAAM;IAItB,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,GAAE,kBAAuB;IAU5D,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,qBAA0B;IAalE,eAAe,IAAI,OAAO,CAAC,oBAAoB,CAAC;IAWtD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAGvB"}
@@ -0,0 +1,87 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { MemoGrafter } from "../../core/MemoGrafter.js";
3
+ import { RetrieverPipeline } from "../../retrieval/RetrieverPipeline.js";
4
+ import { ConductorAgent } from "./ConductorAgent.js";
5
+ import { FleetStore } from "./FleetStore.js";
6
+ import { WorkerAgent } from "./WorkerAgent.js";
7
+ export const FLEET_SHARED_AGENT_COLOR = "fleet";
8
+ export class MemoGrafterFleet {
9
+ core;
10
+ fleetStore;
11
+ workersByColor = new Map();
12
+ sharedSessionId;
13
+ id;
14
+ name;
15
+ defaultWorkerMemory;
16
+ constructor(configOrCore, options = {}) {
17
+ this.core = configOrCore instanceof MemoGrafter ? configOrCore : new MemoGrafter(configOrCore);
18
+ this.fleetStore = new FleetStore(this.core.store);
19
+ this.id = options.id ?? randomUUID();
20
+ this.name = options.name;
21
+ this.defaultWorkerMemory = options.defaultWorkerMemory ?? "local";
22
+ this.sharedSessionId = `fleet:${this.id}:shared`;
23
+ }
24
+ async initialize() {
25
+ await this.core.initialize();
26
+ await this.fleetStore.initializeFleet(this.id, this.name);
27
+ }
28
+ createConductor() {
29
+ return new ConductorAgent(this.core, this.id, (color) => this.workersByColor.get(color));
30
+ }
31
+ async createWorker(config) {
32
+ if (config.color === "conductor") {
33
+ throw new Error("Worker color 'conductor' is reserved.");
34
+ }
35
+ const worker = new WorkerAgent(this.core, this.id, this.sharedSessionId, {
36
+ ...config,
37
+ memory: config.memory ?? this.defaultWorkerMemory,
38
+ });
39
+ await this.fleetStore.upsertAgent({
40
+ id: worker.getAgentId(),
41
+ fleetId: this.id,
42
+ sessionId: worker.getSessionId(),
43
+ color: worker.getColor(),
44
+ });
45
+ this.workersByColor.set(worker.getColor(), worker);
46
+ return worker;
47
+ }
48
+ getWorker(color) {
49
+ return this.workersByColor.get(color);
50
+ }
51
+ async getGraph() {
52
+ return this.fleetStore.getFleetGraph(this.id, this.name);
53
+ }
54
+ getSharedSessionId() {
55
+ return this.sharedSessionId;
56
+ }
57
+ async ingestToFleet(text, options = {}) {
58
+ const nodes = await this.core.ingestText(text, this.sharedSessionId, options);
59
+ await this.core.store.tagSessionNodes(this.sharedSessionId, {
60
+ fleetId: this.id,
61
+ agentId: null,
62
+ agentColor: FLEET_SHARED_AGENT_COLOR,
63
+ });
64
+ return nodes;
65
+ }
66
+ async recallFromFleet(query, options = {}) {
67
+ const pipeline = new RetrieverPipeline(this.core.store, this.core.embedder, {
68
+ ...options,
69
+ sessionIds: [this.sharedSessionId],
70
+ }, this.core.recallCache);
71
+ return pipeline.run(query, this.sharedSessionId);
72
+ }
73
+ async getSharedMemory() {
74
+ const { nodes, segments } = await this.core.getTopics(this.sharedSessionId, { includeSuppressed: true });
75
+ const memories = await this.core.store.getMemoriesBySession(this.sharedSessionId);
76
+ return {
77
+ sessionId: this.sharedSessionId,
78
+ nodes,
79
+ segments,
80
+ memories,
81
+ };
82
+ }
83
+ close() {
84
+ return this.core.close();
85
+ }
86
+ }
87
+ //# sourceMappingURL=MemoGrafterFleet.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"MemoGrafterFleet.js","sourceRoot":"","sources":["../../../src/agents/fleet/MemoGrafterFleet.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAEzE,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAS7C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C,MAAM,CAAC,MAAM,wBAAwB,GAAG,OAAO,CAAC;AAEhD,MAAM,OAAO,gBAAgB;IACV,IAAI,CAAc;IAClB,UAAU,CAAa;IACvB,cAAc,GAAG,IAAI,GAAG,EAAuB,CAAC;IAChD,eAAe,CAAS;IAChC,EAAE,CAAS;IACX,IAAI,CAAqB;IACzB,mBAAmB,CAAkB;IAE9C,YAAY,YAA6C,EAAE,UAAmC,EAAE;QAC9F,IAAI,CAAC,IAAI,GAAG,YAAY,YAAY,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,YAAY,CAAC,CAAC;QAC/F,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,IAAI,UAAU,EAAE,CAAC;QACrC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QACzB,IAAI,CAAC,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,IAAI,OAAO,CAAC;QAClE,IAAI,CAAC,eAAe,GAAG,SAAS,IAAI,CAAC,EAAE,SAAS,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5D,CAAC;IAED,eAAe;QACb,OAAO,IAAI,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAyB;QAC1C,IAAI,MAAM,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,eAAe,EAAE;YACvE,GAAG,MAAM;YACT,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,mBAAmB;SAClD,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;YAChC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,SAAS,EAAE,MAAM,CAAC,YAAY,EAAE;YAChC,KAAK,EAAE,MAAM,CAAC,QAAQ,EAAE;SACzB,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,MAAM,CAAC,CAAC;QACnD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS,CAAC,KAAa;QACrB,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3D,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,IAAY,EAAE,UAA8B,EAAE;QAChE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QAC9E,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,EAAE;YAC1D,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,wBAAwB;SACrC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,KAAa,EAAE,UAAiC,EAAE;QACtE,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CACpC,IAAI,CAAC,IAAI,CAAC,KAAK,EACf,IAAI,CAAC,IAAI,CAAC,QAAQ,EAClB;YACE,GAAG,OAAO;YACV,UAAU,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC;SACnC,EACD,IAAI,CAAC,IAAI,CAAC,WAAW,CACtB,CAAC;QACF,OAAO,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;QACzG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAClF,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,eAAe;YAC/B,KAAK;YACL,QAAQ;YACR,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK;QACH,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;CACF"}
@@ -0,0 +1,36 @@
1
+ import type { MemoGrafter } from "../../core/MemoGrafter.js";
2
+ import type { InjectionResult, Message, RetrievalResult, TopicNode, TopicSegment } from "../../core/types.js";
3
+ import type { FleetGraftByRelevanceOptions, FleetRetrievalOptions, WorkerAgentConfig } from "./types.js";
4
+ export declare class WorkerAgent {
5
+ private readonly core;
6
+ private readonly fleetId;
7
+ private readonly sharedSessionId;
8
+ private readonly agentId;
9
+ private readonly sessionId;
10
+ private readonly memory;
11
+ private readonly history;
12
+ readonly color: string;
13
+ constructor(core: MemoGrafter, fleetId: string, sharedSessionId: string, config: WorkerAgentConfig);
14
+ getAgentId(): string;
15
+ getSessionId(): string;
16
+ getColor(): string;
17
+ invoke(userMessage: string): Promise<string>;
18
+ getHistory(): Message[];
19
+ getActiveNodes(): Promise<TopicNode[]>;
20
+ getActiveSegments(): Promise<TopicSegment[]>;
21
+ graft(topicIds?: string[]): Promise<InjectionResult>;
22
+ graftByRelevance(query: string, options?: FleetGraftByRelevanceOptions): Promise<InjectionResult>;
23
+ recall(query: string, options?: FleetRetrievalOptions): Promise<RetrievalResult>;
24
+ ingestGraftedNodes(nodes: TopicNode[]): Promise<TopicNode[]>;
25
+ absorbFromWorker(sourceWorker: WorkerAgent, options?: {
26
+ topicIds?: string[];
27
+ prompt?: string;
28
+ minSimilarity?: number;
29
+ limit?: number;
30
+ }): Promise<TopicNode[]>;
31
+ tagNodes(): Promise<void>;
32
+ private selectNodesFromWorker;
33
+ private resolveMemorySessionIds;
34
+ private buildFleetMemoryContext;
35
+ }
36
+ //# sourceMappingURL=WorkerAgent.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"WorkerAgent.d.ts","sourceRoot":"","sources":["../../../src/agents/fleet/WorkerAgent.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAE7D,OAAO,KAAK,EAEV,eAAe,EACf,OAAO,EACP,eAAe,EACf,SAAS,EACT,YAAY,EACb,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,4BAA4B,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEzG,qBAAa,WAAW;IAQpB,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,eAAe;IATlC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkB;IACzC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiB;IACzC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;gBAGJ,IAAI,EAAE,WAAW,EACjB,OAAO,EAAE,MAAM,EACf,eAAe,EAAE,MAAM,EACxC,MAAM,EAAE,iBAAiB;IAY3B,UAAU,IAAI,MAAM;IAIpB,YAAY,IAAI,MAAM;IAItB,QAAQ,IAAI,MAAM;IAIZ,MAAM,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAqBlD,UAAU,IAAI,OAAO,EAAE;IAIjB,cAAc,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAKtC,iBAAiB,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;IAK5C,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,CAAC;IAM1D,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,4BAAiC,GAAG,OAAO,CAAC,eAAe,CAAC;IAQrG,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,eAAe,CAAC;IAc9E,kBAAkB,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAU5D,gBAAgB,CAAC,YAAY,EAAE,WAAW,EAAE,OAAO,GAAE;QACzD,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QACpB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,KAAK,CAAC,EAAE,MAAM,CAAC;KACX,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAKvB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;YAQjB,qBAAqB;IAwBnC,OAAO,CAAC,uBAAuB;YAMjB,uBAAuB;CAetC"}
@@ -0,0 +1,140 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { RetrieverPipeline } from "../../retrieval/RetrieverPipeline.js";
3
+ export class WorkerAgent {
4
+ core;
5
+ fleetId;
6
+ sharedSessionId;
7
+ agentId;
8
+ sessionId;
9
+ memory;
10
+ history = [];
11
+ color;
12
+ constructor(core, fleetId, sharedSessionId, config) {
13
+ this.core = core;
14
+ this.fleetId = fleetId;
15
+ this.sharedSessionId = sharedSessionId;
16
+ if (config.color === "conductor") {
17
+ throw new Error("Worker color 'conductor' is reserved.");
18
+ }
19
+ this.color = config.color;
20
+ this.agentId = config.id ?? randomUUID();
21
+ this.sessionId = config.sessionId ?? randomUUID();
22
+ this.memory = config.memory ?? "local";
23
+ }
24
+ getAgentId() {
25
+ return this.agentId;
26
+ }
27
+ getSessionId() {
28
+ return this.sessionId;
29
+ }
30
+ getColor() {
31
+ return this.color;
32
+ }
33
+ async invoke(userMessage) {
34
+ this.history.push({ role: "user", content: userMessage });
35
+ const { nodes } = await this.core.getTopics(this.sessionId);
36
+ const topicIds = nodes.map((node) => node.id);
37
+ const { systemPrompt } = await this.core.inject(this.sessionId, topicIds);
38
+ const memoryPrompt = await this.buildFleetMemoryContext(userMessage);
39
+ const combinedPrompt = [systemPrompt, memoryPrompt].filter(Boolean).join("\n\n");
40
+ const response = await this.core.llm.complete(this.history, combinedPrompt);
41
+ this.history.push({ role: "assistant", content: response });
42
+ await this.core.ingestNow(this.history, this.sessionId).catch((error) => {
43
+ console.warn("MemoGrafter worker ingest warning:", error);
44
+ });
45
+ await this.tagNodes().catch((error) => {
46
+ console.warn("MemoGrafter worker node tagging warning:", error);
47
+ });
48
+ return response;
49
+ }
50
+ getHistory() {
51
+ return [...this.history];
52
+ }
53
+ async getActiveNodes() {
54
+ const { nodes } = await this.core.getTopics(this.sessionId);
55
+ return nodes;
56
+ }
57
+ async getActiveSegments() {
58
+ const { segments } = await this.core.getTopics(this.sessionId);
59
+ return segments;
60
+ }
61
+ async graft(topicIds) {
62
+ const { nodes } = await this.core.getTopics(this.sessionId);
63
+ const selectedTopicIds = topicIds ?? nodes.map((node) => node.id);
64
+ return this.core.inject(this.sessionId, selectedTopicIds);
65
+ }
66
+ graftByRelevance(query, options = {}) {
67
+ const { memory: mode, ...graftOptions } = options;
68
+ return this.core.graftByRelevance(this.sessionId, query, {
69
+ ...graftOptions,
70
+ sessionIds: this.resolveMemorySessionIds(mode),
71
+ });
72
+ }
73
+ recall(query, options = {}) {
74
+ const { memory: mode, ...retrievalOptions } = options;
75
+ const pipeline = new RetrieverPipeline(this.core.store, this.core.embedder, {
76
+ ...retrievalOptions,
77
+ sessionIds: this.resolveMemorySessionIds(mode),
78
+ }, this.core.recallCache);
79
+ return pipeline.run(query, this.sessionId);
80
+ }
81
+ async ingestGraftedNodes(nodes) {
82
+ const copiedNodes = await this.core.store.absorbNodes(nodes, this.sessionId, {
83
+ fleetId: this.fleetId,
84
+ agentId: this.agentId,
85
+ agentColor: this.color,
86
+ });
87
+ await this.core.store.rebuildEdgesForSession(this.sessionId);
88
+ return copiedNodes;
89
+ }
90
+ async absorbFromWorker(sourceWorker, options = {}) {
91
+ const nodes = await this.selectNodesFromWorker(sourceWorker, options);
92
+ return this.ingestGraftedNodes(nodes);
93
+ }
94
+ async tagNodes() {
95
+ await this.core.store.tagSessionNodes(this.sessionId, {
96
+ fleetId: this.fleetId,
97
+ agentId: this.agentId,
98
+ agentColor: this.color,
99
+ });
100
+ }
101
+ async selectNodesFromWorker(sourceWorker, options) {
102
+ const sourceNodes = await sourceWorker.getActiveNodes();
103
+ if (options.topicIds && options.topicIds.length > 0) {
104
+ const topicIds = new Set(options.topicIds);
105
+ return sourceNodes.filter((node) => topicIds.has(node.id));
106
+ }
107
+ if (options.prompt) {
108
+ const embedding = await this.core.embedder.embed(options.prompt);
109
+ return this.core.store.getSimilarNodes(embedding, sourceWorker.getSessionId(), {
110
+ k: options.limit ?? 5,
111
+ minSimilarity: options.minSimilarity ?? 0.6,
112
+ });
113
+ }
114
+ return sourceNodes;
115
+ }
116
+ resolveMemorySessionIds(mode = this.memory) {
117
+ if (mode === "fleet")
118
+ return [this.sharedSessionId];
119
+ if (mode === "both")
120
+ return [this.sessionId, this.sharedSessionId];
121
+ return [this.sessionId];
122
+ }
123
+ async buildFleetMemoryContext(query) {
124
+ if (this.memory === "local")
125
+ return "";
126
+ try {
127
+ const result = await this.recall(query, {
128
+ memory: this.memory,
129
+ limit: 6,
130
+ minSimilarity: 0.55,
131
+ });
132
+ return result.facts.length > 0 ? result.systemPrompt : "";
133
+ }
134
+ catch (error) {
135
+ console.warn("MemoGrafter worker fleet recall warning:", error);
136
+ return "";
137
+ }
138
+ }
139
+ }
140
+ //# sourceMappingURL=WorkerAgent.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"WorkerAgent.js","sourceRoot":"","sources":["../../../src/agents/fleet/WorkerAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAWzE,MAAM,OAAO,WAAW;IAQH;IACA;IACA;IATF,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,MAAM,CAAkB;IACxB,OAAO,GAAc,EAAE,CAAC;IAChC,KAAK,CAAS;IAEvB,YACmB,IAAiB,EACjB,OAAe,EACf,eAAuB,EACxC,MAAyB;QAHR,SAAI,GAAJ,IAAI,CAAa;QACjB,YAAO,GAAP,OAAO,CAAQ;QACf,oBAAe,GAAf,eAAe,CAAQ;QAGxC,IAAI,MAAM,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,EAAE,IAAI,UAAU,EAAE,CAAC;QACzC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,UAAU,EAAE,CAAC;QAClD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC;IACzC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,WAAmB;QAC9B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAE1D,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5D,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC1E,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;QACrE,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QAE5E,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC5D,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC/E,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC7C,OAAO,CAAC,IAAI,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/D,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAmB;QAC7B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5D,MAAM,gBAAgB,GAAG,QAAQ,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAC5D,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,UAAwC,EAAE;QACxE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC;QAClD,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YACvD,GAAG,YAAY;YACf,UAAU,EAAE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC;SAC/C,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,UAAiC,EAAE;QACvD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,gBAAgB,EAAE,GAAG,OAAO,CAAC;QACtD,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CACpC,IAAI,CAAC,IAAI,CAAC,KAAK,EACf,IAAI,CAAC,IAAI,CAAC,QAAQ,EAClB;YACE,GAAG,gBAAgB;YACnB,UAAU,EAAE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC;SAC/C,EACD,IAAI,CAAC,IAAI,CAAC,WAAW,CACtB,CAAC;QACF,OAAO,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,KAAkB;QACzC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE;YAC3E,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,KAAK;SACvB,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7D,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,YAAyB,EAAE,UAK9C,EAAE;QACJ,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE;YACpD,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,KAAK;SACvB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,YAAyB,EAAE,OAK9D;QACC,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,cAAc,EAAE,CAAC;QAExD,IAAI,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC3C,OAAO,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACjE,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,SAAS,EAAE,YAAY,CAAC,YAAY,EAAE,EAAE;gBAC7E,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;gBACrB,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,GAAG;aAC5C,CAAC,CAAC;QACL,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,uBAAuB,CAAC,OAAwB,IAAI,CAAC,MAAM;QACjE,IAAI,IAAI,KAAK,OAAO;YAAE,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACpD,IAAI,IAAI,KAAK,MAAM;YAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,KAAa;QACjD,IAAI,IAAI,CAAC,MAAM,KAAK,OAAO;YAAE,OAAO,EAAE,CAAC;QAEvC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE;gBACtC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,CAAC;gBACR,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,OAAO,CAAC,IAAI,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;YAChE,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,58 @@
1
+ import type { AbsorbFromAgentOptions, GraftByRelevanceOptions, FleetMemoryMode, InjectionResult, IngestTextOptions, MemoryNode, MemoGrafterConfig, Message, RetrievalResult, RetrieverConfig, TopicNode, TopicSegment } from "../../core/types.js";
2
+ export interface FleetGraph {
3
+ id: string;
4
+ name?: string;
5
+ agents: FleetAgentInfo[];
6
+ }
7
+ export interface FleetAgentInfo {
8
+ id: string;
9
+ sessionId: string;
10
+ color: string;
11
+ }
12
+ export interface MemoGrafterFleetOptions {
13
+ id?: string;
14
+ name?: string;
15
+ defaultWorkerMemory?: FleetMemoryMode;
16
+ }
17
+ export interface WorkerAgentConfig {
18
+ color: string;
19
+ sessionId?: string;
20
+ id?: string;
21
+ memory?: FleetMemoryMode;
22
+ }
23
+ export interface FleetMemoryOptions extends IngestTextOptions {
24
+ tags?: string[];
25
+ replace?: boolean;
26
+ }
27
+ export interface SharedMemorySnapshot {
28
+ sessionId: string;
29
+ nodes: TopicNode[];
30
+ segments: TopicSegment[];
31
+ memories: MemoryNode[];
32
+ }
33
+ export interface FleetRetrievalOptions extends RetrieverConfig {
34
+ memory?: FleetMemoryMode;
35
+ }
36
+ export interface FleetGraftByRelevanceOptions extends GraftByRelevanceOptions {
37
+ memory?: FleetMemoryMode;
38
+ }
39
+ export interface ConductorGraftOptions {
40
+ topicIds?: string[];
41
+ prompt?: string;
42
+ minSimilarity?: number;
43
+ limit?: number;
44
+ }
45
+ export type FleetAbsorbOptions = AbsorbFromAgentOptions;
46
+ export interface FleetWorker {
47
+ getAgentId(): string;
48
+ getSessionId(): string;
49
+ getColor(): string;
50
+ invoke(userMessage: string): Promise<string>;
51
+ getHistory(): Message[];
52
+ getActiveNodes(): Promise<TopicNode[]>;
53
+ graft(topicIds?: string[]): Promise<InjectionResult>;
54
+ graftByRelevance(query: string, options?: FleetGraftByRelevanceOptions): Promise<InjectionResult>;
55
+ recall(query: string, options?: FleetRetrievalOptions): Promise<RetrievalResult>;
56
+ }
57
+ export type { MemoGrafterConfig };
58
+ //# sourceMappingURL=types.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/agents/fleet/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,sBAAsB,EACtB,uBAAuB,EACvB,eAAe,EACf,eAAe,EACf,iBAAiB,EACjB,UAAU,EACV,iBAAiB,EACjB,OAAO,EACP,eAAe,EACf,eAAe,EACf,SAAS,EACT,YAAY,EACb,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,cAAc,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,uBAAuB;IACtC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,mBAAmB,CAAC,EAAE,eAAe,CAAC;CACvC;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,WAAW,kBAAmB,SAAQ,iBAAiB;IAC3D,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,QAAQ,EAAE,UAAU,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,qBAAsB,SAAQ,eAAe;IAC5D,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,WAAW,4BAA6B,SAAQ,uBAAuB;IAC3E,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,MAAM,kBAAkB,GAAG,sBAAsB,CAAC;AAExD,MAAM,WAAW,WAAW;IAC1B,UAAU,IAAI,MAAM,CAAC;IACrB,YAAY,IAAI,MAAM,CAAC;IACvB,QAAQ,IAAI,MAAM,CAAC;IACnB,MAAM,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7C,UAAU,IAAI,OAAO,EAAE,CAAC;IACxB,cAAc,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACvC,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;IACrD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,4BAA4B,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;IAClG,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,qBAAqB,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;CAClF;AAED,YAAY,EAAE,iBAAiB,EAAE,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/agents/fleet/types.ts"],"names":[],"mappings":""}
@@ -0,0 +1,8 @@
1
+ export interface InitResult {
2
+ generated: string[];
3
+ created: string[];
4
+ skipped: string[];
5
+ updated: string[];
6
+ }
7
+ export declare function runInit(cwd?: string): Promise<InitResult>;
8
+ //# sourceMappingURL=init.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../../cli/commands/init.ts"],"names":[],"mappings":"AAsBA,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,EAAE,CAAC;CACnB;AAED,wBAAsB,OAAO,CAAC,GAAG,SAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,CAyBtE"}
@@ -0,0 +1,124 @@
1
+ import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
2
+ import path from "node:path";
3
+ import { logger } from "../utils/logger.js";
4
+ export async function runInit(cwd = process.cwd()) {
5
+ const result = {
6
+ generated: [],
7
+ created: [],
8
+ skipped: [],
9
+ updated: [],
10
+ };
11
+ const memoGrafterDir = path.join(cwd, "src", "memo-grafter");
12
+ mkdirSync(memoGrafterDir, { recursive: true });
13
+ writeFileSync(path.join(memoGrafterDir, "mg-schema.ts"), await renderMgSchema(), "utf8");
14
+ result.generated.push("src/memo-grafter/mg-schema.ts");
15
+ writeIfMissing(path.join(memoGrafterDir, "schema.ts"), renderSchemaTs(), "src/memo-grafter/schema.ts", result);
16
+ writeIfMissing(path.join(memoGrafterDir, "mg.config.ts"), renderConfigTs(), "src/memo-grafter/mg.config.ts", result);
17
+ ensureEnvExample(cwd, result);
18
+ logger.info("MemoGrafter project files initialized");
19
+ for (const file of result.generated)
20
+ logger.success(`${file} regenerated`);
21
+ for (const file of result.created)
22
+ logger.success(`${file} created`);
23
+ for (const file of result.updated)
24
+ logger.success(`${file} updated`);
25
+ for (const file of result.skipped)
26
+ logger.info(`- ${file} already exists, skipped`);
27
+ return result;
28
+ }
29
+ function writeIfMissing(filePath, content, label, result) {
30
+ if (existsSync(filePath)) {
31
+ result.skipped.push(label);
32
+ return;
33
+ }
34
+ writeFileSync(filePath, content, "utf8");
35
+ result.created.push(label);
36
+ }
37
+ function ensureEnvExample(cwd, result) {
38
+ const envExamplePath = path.join(cwd, ".env.example");
39
+ const databaseLine = "DATABASE_URL=postgres://user:password@localhost:5432/memo_grafter";
40
+ if (!existsSync(envExamplePath)) {
41
+ writeFileSync(envExamplePath, `${databaseLine}\n`, "utf8");
42
+ result.created.push(".env.example");
43
+ return;
44
+ }
45
+ const existing = readFileSync(envExamplePath, "utf8");
46
+ if (/^DATABASE_URL=/m.test(existing)) {
47
+ result.skipped.push(".env.example DATABASE_URL");
48
+ return;
49
+ }
50
+ const next = existing.endsWith("\n")
51
+ ? `${existing}${databaseLine}\n`
52
+ : `${existing}\n${databaseLine}\n`;
53
+ writeFileSync(envExamplePath, next, "utf8");
54
+ result.updated.push(".env.example");
55
+ }
56
+ async function renderMgSchema() {
57
+ const schema = await loadSchemaModule();
58
+ const tableBlocks = schema.memoGrafterTables.map((table) => {
59
+ const columns = table.columns.map((column) => ` ${column.name}: ${JSON.stringify(column)},`).join("\n");
60
+ const constraints = table.constraints
61
+ ? `,\n constraints: ${JSON.stringify(table.constraints)}`
62
+ : "";
63
+ return ` ${table.name}: {
64
+ name: "${table.name}",
65
+ description: "${table.description}",
66
+ columns: {
67
+ ${columns}
68
+ }${constraints}
69
+ },`;
70
+ }).join("\n");
71
+ return `// Auto-generated by memo-grafter init.
72
+ // Do not edit this file manually. Re-run \`npx memo-grafter init\` to regenerate it.
73
+
74
+ export const memoGrafterExtensions = ${JSON.stringify(schema.memoGrafterExtensions, null, 2)} as const;
75
+
76
+ export const memoGrafterTables = {
77
+ ${tableBlocks}
78
+ } as const;
79
+
80
+ export const memoGrafterIndexes = ${JSON.stringify(schema.memoGrafterIndexes, null, 2)} as const;
81
+ `;
82
+ }
83
+ async function loadSchemaModule() {
84
+ const packageName = "memo-grafter/schema";
85
+ return await import(packageName);
86
+ }
87
+ function renderSchemaTs() {
88
+ return `// Your application schema entrypoint.
89
+ //
90
+ // MemoGrafter owns and regenerates mg-schema.ts.
91
+ // This file is user-owned and will not be overwritten.
92
+ //
93
+ // Today, \`memo-grafter migrate\` manages only MemoGrafter \`mg_*\` tables.
94
+ // If you use Prisma, keep app models in prisma/schema.prisma.
95
+ // If you use another migration tool, keep using it for app tables.
96
+
97
+ export * from "./mg-schema.js";
98
+
99
+ /*
100
+ Example app table SQL for your own migration tool:
101
+
102
+ CREATE TABLE users (
103
+ id TEXT PRIMARY KEY,
104
+ email TEXT NOT NULL UNIQUE,
105
+ created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
106
+ );
107
+ */
108
+ `;
109
+ }
110
+ function renderConfigTs() {
111
+ return `declare const process: {
112
+ env: {
113
+ DATABASE_URL?: string;
114
+ };
115
+ };
116
+
117
+ export default {
118
+ db: {
119
+ connectionString: process.env.DATABASE_URL,
120
+ },
121
+ };
122
+ `;
123
+ }
124
+ //# sourceMappingURL=init.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"init.js","sourceRoot":"","sources":["../../../cli/commands/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AA2B5C,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE;IAC/C,MAAM,MAAM,GAAe;QACzB,SAAS,EAAE,EAAE;QACb,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;KACZ,CAAC;IAEF,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC;IAC7D,SAAS,CAAC,cAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/C,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC,EAAE,MAAM,cAAc,EAAE,EAAE,MAAM,CAAC,CAAC;IACzF,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;IAEvD,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,EAAE,cAAc,EAAE,EAAE,4BAA4B,EAAE,MAAM,CAAC,CAAC;IAC/G,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC,EAAE,cAAc,EAAE,EAAE,+BAA+B,EAAE,MAAM,CAAC,CAAC;IACrH,gBAAgB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAE9B,MAAM,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;IACrD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,SAAS;QAAE,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,cAAc,CAAC,CAAC;IAC3E,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO;QAAE,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IACrE,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO;QAAE,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IACrE,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO;QAAE,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,0BAA0B,CAAC,CAAC;IAEpF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,cAAc,CAAC,QAAgB,EAAE,OAAe,EAAE,KAAa,EAAE,MAAkB;IAC1F,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3B,OAAO;IACT,CAAC;IAED,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAW,EAAE,MAAkB;IACvD,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;IACtD,MAAM,YAAY,GAAG,mEAAmE,CAAC;IAEzF,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QAChC,aAAa,CAAC,cAAc,EAAE,GAAG,YAAY,IAAI,EAAE,MAAM,CAAC,CAAC;QAC3D,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpC,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAAG,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACtD,IAAI,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACjD,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QAClC,CAAC,CAAC,GAAG,QAAQ,GAAG,YAAY,IAAI;QAChC,CAAC,CAAC,GAAG,QAAQ,KAAK,YAAY,IAAI,CAAC;IACrC,aAAa,CAAC,cAAc,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IAC5C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AACtC,CAAC;AAED,KAAK,UAAU,cAAc;IAC3B,MAAM,MAAM,GAAG,MAAM,gBAAgB,EAAE,CAAC;IACxC,MAAM,WAAW,GAAG,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,KAAkB,EAAE,EAAE;QACtE,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAoB,EAAE,EAAE,CAAC,SAAS,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3H,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW;YACnC,CAAC,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE;YAC5D,CAAC,CAAC,EAAE,CAAC;QACP,OAAO,KAAK,KAAK,CAAC,IAAI;aACb,KAAK,CAAC,IAAI;oBACH,KAAK,CAAC,WAAW;;EAEnC,OAAO;OACF,WAAW;KACb,CAAC;IACJ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO;;;uCAG8B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;;;EAG1F,WAAW;;;oCAGuB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;CACrF,CAAC;AACF,CAAC;AAED,KAAK,UAAU,gBAAgB;IAC7B,MAAM,WAAW,GAAG,qBAAqB,CAAC;IAC1C,OAAO,MAAM,MAAM,CAAC,WAAW,CAAiB,CAAC;AACnD,CAAC;AAED,SAAS,cAAc;IACrB,OAAO;;;;;;;;;;;;;;;;;;;;CAoBR,CAAC;AACF,CAAC;AAED,SAAS,cAAc;IACrB,OAAO;;;;;;;;;;;CAWR,CAAC;AACF,CAAC"}
@@ -0,0 +1,6 @@
1
+ export interface MigrateOptions {
2
+ cwd?: string;
3
+ db?: string;
4
+ }
5
+ export declare function runMigrate(options?: MigrateOptions): Promise<void>;
6
+ //# sourceMappingURL=migrate.d.ts.map