@mastra/rag 2.5.0 → 2.6.0-alpha.0

package/dist/index.d.ts

@@ -3,6 +3,7 @@ export * from './document/types.js';
 export * from './rerank/index.js';
 export * from './rerank/relevance/index.js';
 export { GraphRAG } from './graph-rag/index.js';
+export type { GraphRAGSnapshot } from './graph-rag/index.js';
 export * from './tools/index.js';
 export * from './utils/default-settings.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AACzB,cAAc,oBAAoB,CAAC;AACnC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,cAAc,SAAS,CAAC;AACxB,cAAc,0BAA0B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AACzB,cAAc,oBAAoB,CAAC;AACnC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,YAAY,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AACpD,cAAc,SAAS,CAAC;AACxB,cAAc,0BAA0B,CAAC"}

package/dist/index.js

@@ -6222,7 +6222,8 @@ async function rerank(results, query, model, options) {
 }
 //#endregion
 //#region src/graph-rag/index.ts
-var GraphRAG = class {
+const GRAPH_RAG_SNAPSHOT_VERSION = 1;
+var GraphRAG = class GraphRAG {
 	nodes;
 	edges;
 	dimension;
@@ -6257,6 +6258,49 @@ var GraphRAG = class {
 	getEdgesByType(type) {
 		return this.edges.filter((edge) => edge.type === type);
 	}
+	/**
+	* Produce a JSON-safe snapshot of the graph so it can be persisted and
+	* restored later instead of rebuilt with `createGraph`.
+	*
+	* The snapshot is a deep copy, including nested node metadata, so mutating it
+	* does not affect this instance.
+	* Note that every node carries its full embedding, so snapshots of large
+	* graphs can be several megabytes of JSON.
+	*/
+	serialize() {
+		return {
+			version: GRAPH_RAG_SNAPSHOT_VERSION,
+			dimension: this.dimension,
+			threshold: this.threshold,
+			nodes: Array.from(this.nodes.values()).map((node) => ({
+				...node,
+				...node.embedding ? { embedding: [...node.embedding] } : {},
+				...node.metadata ? { metadata: structuredClone(node.metadata) } : {}
+			})),
+			edges: this.edges.map((edge) => ({ ...edge }))
+		};
+	}
+	/**
+	* Rebuild a GraphRAG instance from a snapshot produced by `serialize()`.
+	*
+	* @throws if the snapshot version is unsupported, a node embedding does not
+	* match the snapshot dimension, or an edge references an unknown node.
+	*/
+	static deserialize(snapshot) {
+		if (snapshot?.version !== GRAPH_RAG_SNAPSHOT_VERSION) throw new Error(`Unsupported GraphRAG snapshot version: ${snapshot?.version}`);
+		const graph = new GraphRAG(snapshot.dimension, snapshot.threshold);
+		for (const node of snapshot.nodes ?? []) graph.addNode({
+			...node,
+			...node.embedding ? { embedding: [...node.embedding] } : {},
+			...node.metadata ? { metadata: structuredClone(node.metadata) } : {}
+		});
+		for (const edge of snapshot.edges ?? []) {
+			if (!graph.nodes.has(edge.source)) throw new Error(`Edge references unknown node: ${edge.source}`);
+			if (!graph.nodes.has(edge.target)) throw new Error(`Edge references unknown node: ${edge.target}`);
+		}
+		graph.edges = (snapshot.edges ?? []).map((edge) => ({ ...edge }));
+		return graph;
+	}
 	clear() {
 		this.nodes.clear();
 		this.edges = [];