@prometheus-ags/entity-graph-core 4.3.0 → 4.4.0

package/dist/index.mjs

@@ -1,5 +1,6 @@
-import { graphStore, useGraphStore, subscribeDevtoolsEvent, getGraphDevtoolsCompatibilityDelegate, subscribeGraphDevtoolsCompatibility, matchesFilter, matchesSearch, compareEntities, emitLegacyDevtoolsEvent, __require } from './chunk-64YZRB56.mjs';
-export { EMPTY_ENTITY_STATE, EMPTY_IDS, EMPTY_LIST_STATE, EMPTY_SYNC_METADATA, applyView, attachGlobalListeners, cascadeInvalidation, checkCompleteness, compareEntities, configureEngine, createGraphStore, dedupe, fetchEntity, fetchList, findInsertionIndex, getActiveSubscriberCount, getEngineOptions, getMergeStrategy, getSchema, graphStore, hasMergeStrategy, lwwStrategy, matchesFilter, matchesSearch, readRelations, registerMergeStrategy, registerSchema, registerSubscriber, serializeKey, setDefaultMergeStrategy, startGarbageCollector, stopGarbageCollector, subscribeDevtoolsEvent, subscribeSubscriberStats, unregisterSubscriber, useGraphStore } from './chunk-64YZRB56.mjs';
+import { graphStore, useGraphStore, subscribeDevtoolsEvent, getGraphDevtoolsCompatibilityDelegate, subscribeGraphDevtoolsCompatibility, matchesFilter, matchesSearch, compareEntities, __require, setListIdsDraft, removeIdFromAllListsDraft, insertIdInListDraft, setEntitySyncMetadataDraft, clearPatchDraft, unpatchEntityDraft, patchEntityDraft, removeEntityDraft, replaceEntityDraft, upsertEntitiesDraft, upsertEntityDraft, emitLegacyDevtoolsEvent } from './chunk-T5WSFBXH.mjs';
+export { EMPTY_ENTITY_STATE, EMPTY_IDS, EMPTY_LIST_STATE, EMPTY_SYNC_METADATA, applyView, attachGlobalListeners, cascadeInvalidation, checkCompleteness, compareEntities, configureEngine, createGraphStore, dedupe, fetchEntity, fetchList, findInsertionIndex, getActiveSubscriberCount, getEngineOptions, getMergeStrategy, getSchema, graphStore, hasMergeStrategy, lwwStrategy, matchesFilter, matchesSearch, readRelations, registerMergeStrategy, registerSchema, registerSubscriber, serializeKey, setDefaultMergeStrategy, startGarbageCollector, stopGarbageCollector, subscribeDevtoolsEvent, subscribeSubscriberStats, unregisterSubscriber, useGraphStore } from './chunk-T5WSFBXH.mjs';
+import { isDraft, current } from 'immer';
 import { createStore } from 'zustand/vanilla';
 
 // src/errors.ts
@@ -494,37 +495,37 @@ function isObject(value) {
 function getValueAtPath(source, path) {
   if (!path) return source;
   const segments = path.split(".").filter(Boolean);
-  let current = source;
+  let current2 = source;
   for (const segment of segments) {
-    if (!isObject(current) && !Array.isArray(current)) return void 0;
-    current = current[segment];
+    if (!isObject(current2) && !Array.isArray(current2)) return void 0;
+    current2 = current2[segment];
   }
-  return current;
+  return current2;
 }
 function setValueAtPath(source, path, value) {
   const segments = path.split(".").filter(Boolean);
   if (segments.length === 0) return source;
   const clone2 = structuredClone(source);
-  let current = clone2;
+  let current2 = clone2;
   for (let index = 0; index < segments.length - 1; index += 1) {
     const segment = segments[index];
-    const next = current[segment];
-    if (!isObject(next)) current[segment] = {};
-    current = current[segment];
+    const next = current2[segment];
+    if (!isObject(next)) current2[segment] = {};
+    current2 = current2[segment];
   }
-  current[segments[segments.length - 1]] = value;
+  current2[segments[segments.length - 1]] = value;
   return clone2;
 }
-function collectDirtyPaths(current, original, prefix = "", acc = /* @__PURE__ */ new Set()) {
-  if (isObject(current) && isObject(original)) {
-    const keys = /* @__PURE__ */ new Set([...Object.keys(current), ...Object.keys(original)]);
+function collectDirtyPaths(current2, original, prefix = "", acc = /* @__PURE__ */ new Set()) {
+  if (isObject(current2) && isObject(original)) {
+    const keys = /* @__PURE__ */ new Set([...Object.keys(current2), ...Object.keys(original)]);
     for (const key of keys) {
       const nextPrefix = prefix ? `${prefix}.${key}` : key;
-      collectDirtyPaths(current[key], original[key], nextPrefix, acc);
+      collectDirtyPaths(current2[key], original[key], nextPrefix, acc);
     }
     return acc;
   }
-  if (JSON.stringify(current) !== JSON.stringify(original) && prefix) acc.add(prefix);
+  if (JSON.stringify(current2) !== JSON.stringify(original) && prefix) acc.add(prefix);
   return acc;
 }
 
@@ -642,18 +643,41 @@ function applySelection(row, select) {
   return picked;
 }
 
+// src/devtools-gate.ts
+function resolveDevtoolsEmission(env) {
+  if (env.buildFlag !== void 0) return env.buildFlag && env.override !== false;
+  if (env.override !== void 0) return env.override;
+  if (env.processEnv !== void 0) return env.processEnv.NODE_ENV !== "production";
+  return env.viteDev === true;
+}
+var override;
+function setGraphDevtoolsEmission(enabled) {
+  override = enabled;
+}
+function isGraphDevtoolsEmissionEnabled() {
+  return resolveDevtoolsEmission({
+    buildFlag: typeof __PROMETHEUS_ENTITY_GRAPH_DEVTOOLS__ === "boolean" ? __PROMETHEUS_ENTITY_GRAPH_DEVTOOLS__ : void 0,
+    override,
+    processEnv: typeof process !== "undefined" ? process.env ?? {} : void 0,
+    viteDev: import.meta.env?.DEV
+  });
+}
+function emitGatedDevtoolsEvent(emit, build) {
+  if ((typeof __PROMETHEUS_ENTITY_GRAPH_DEVTOOLS__ !== "boolean" || __PROMETHEUS_ENTITY_GRAPH_DEVTOOLS__) && isGraphDevtoolsEmissionEnabled()) {
+    emit(build());
+  }
+}
+
 // src/graph-actions.ts
 var graphActionListeners = /* @__PURE__ */ new Set();
 var graphActionReplayers = /* @__PURE__ */ new Map();
 function createGraphTransaction(storeApi = graphStore) {
-  const baseline = cloneGraphData(storeApi.getState());
+  const baseline = captureGraphData(storeApi.getState());
   let closed = false;
   const tx = {
     upsertEntity(type, id, data) {
       storeApi.getState().upsertEntity(type, id, data);
-      if (typeof process === "undefined" || process.env.NODE_ENV !== "production") {
-        emitLegacyDevtoolsEvent({ kind: "upsert", type, id, data, at: (/* @__PURE__ */ new Date()).toISOString() });
-      }
+      emitGatedDevtoolsEvent(emitLegacyDevtoolsEvent, () => ({ kind: "upsert", type, id, data, at: (/* @__PURE__ */ new Date()).toISOString() }));
       return tx;
     },
     replaceEntity(type, id, data) {
@@ -666,16 +690,12 @@ function createGraphTransaction(storeApi = graphStore) {
     },
     patchEntity(type, id, patch) {
       storeApi.getState().patchEntity(type, id, patch);
-      if (typeof process === "undefined" || process.env.NODE_ENV !== "production") {
-        emitLegacyDevtoolsEvent({ kind: "patch", type, id, patch, at: (/* @__PURE__ */ new Date()).toISOString() });
-      }
+      emitGatedDevtoolsEvent(emitLegacyDevtoolsEvent, () => ({ kind: "patch", type, id, patch, at: (/* @__PURE__ */ new Date()).toISOString() }));
       return tx;
     },
     clearPatch(type, id) {
       storeApi.getState().clearPatch(type, id);
-      if (typeof process === "undefined" || process.env.NODE_ENV !== "production") {
-        emitLegacyDevtoolsEvent({ kind: "clearPatch", type, id, at: (/* @__PURE__ */ new Date()).toISOString() });
-      }
+      emitGatedDevtoolsEvent(emitLegacyDevtoolsEvent, () => ({ kind: "clearPatch", type, id, at: (/* @__PURE__ */ new Date()).toISOString() }));
       return tx;
     },
     insertIdInList(key, id, position) {
@@ -711,7 +731,7 @@ function createGraphTransaction(storeApi = graphStore) {
     },
     rollback() {
       if (closed) return;
-      storeApi.setState(cloneGraphData(baseline));
+      storeApi.setState({ ...baseline });
       closed = true;
     },
     snapshot() {
@@ -777,9 +797,224 @@ function cloneGraphData(source = graphStore.getState()) {
     lists: structuredClone(source.lists)
   };
 }
+function captureGraphData(source) {
+  return {
+    entities: source.entities,
+    patches: source.patches,
+    entityStates: source.entityStates,
+    syncMetadata: source.syncMetadata,
+    lists: source.lists
+  };
+}
 function emitGraphActionEvent(event) {
   for (const listener of graphActionListeners) listener(event);
 }
+var openDrafts = /* @__PURE__ */ new WeakMap();
+function createDraftBatch(draft) {
+  const batch = {
+    upsertEntity: (type, id, data) => (upsertEntityDraft(draft, type, id, data), batch),
+    upsertEntities: (type, entries) => (upsertEntitiesDraft(draft, type, entries), batch),
+    replaceEntity: (type, id, data) => (replaceEntityDraft(draft, type, id, data), batch),
+    removeEntity: (type, id) => (removeEntityDraft(draft, type, id), batch),
+    patchEntity: (type, id, patch) => (patchEntityDraft(draft, type, id, patch), batch),
+    unpatchEntity: (type, id, keys) => (unpatchEntityDraft(draft, type, id, keys), batch),
+    clearPatch: (type, id) => (clearPatchDraft(draft, type, id), batch),
+    setEntitySyncMetadata: (type, id, metadata) => (setEntitySyncMetadataDraft(draft, type, id, metadata), batch),
+    insertIdInList: (key, id, position) => (insertIdInListDraft(draft, key, id, position), batch),
+    removeIdFromAllLists: (type, id) => (removeIdFromAllListsDraft(draft, type, id), batch),
+    setListIds: (key, ids, entityType) => (setListIdsDraft(draft, key, ids, entityType), batch),
+    readEntity: (type, id) => {
+      const base = draft.entities[type]?.[id];
+      if (!base) return null;
+      const patch = draft.patches[type]?.[id];
+      const plain = (value) => isDraft(value) ? current(value) : value;
+      return patch ? { ...plain(base), ...plain(patch) } : { ...plain(base) };
+    }
+  };
+  return batch;
+}
+function batchGraphWrites(fn, store = graphStore) {
+  const open = openDrafts.get(store);
+  if (open) return fn(createDraftBatch(open));
+  let result;
+  const before = store.getState();
+  store.setState((draft) => {
+    openDrafts.set(store, draft);
+    try {
+      result = fn(createDraftBatch(draft));
+    } finally {
+      openDrafts.delete(store);
+    }
+    if (typeof result?.then === "function") {
+      throw new TypeError("batchGraphWrites: the callback must be synchronous; nothing was written");
+    }
+    if (store.getState() !== before) {
+      throw new Error(
+        "batchGraphWrites: the store was written outside the batch handle while the batch was open; the batch was not applied"
+      );
+    }
+  });
+  return result;
+}
+function applyChange(batch, change) {
+  switch (change.op) {
+    case "upsert":
+      batch.upsertEntity(change.type, change.id, change.data);
+      return;
+    case "replace":
+      batch.replaceEntity(change.type, change.id, change.data);
+      return;
+    case "remove":
+      batch.removeEntity(change.type, change.id);
+      return;
+    case "patch":
+      batch.patchEntity(change.type, change.id, change.patch);
+      return;
+    case "unpatch":
+      batch.unpatchEntity(change.type, change.id, change.keys);
+      return;
+    case "clearPatch":
+      batch.clearPatch(change.type, change.id);
+      return;
+    case "syncMetadata":
+      batch.setEntitySyncMetadata(change.type, change.id, change.metadata);
+      return;
+    case "insertId":
+      batch.insertIdInList(change.key, change.id, change.position);
+      return;
+    case "removeIdFromAllLists":
+      batch.removeIdFromAllLists(change.type, change.id);
+      return;
+    case "setListIds":
+      batch.setListIds(change.key, change.ids, change.entityType);
+      return;
+    default:
+      throw new Error(`Unknown graph change op: ${change.op}`);
+  }
+}
+function applyGraphChanges(changes, store = graphStore) {
+  batchGraphWrites((batch) => {
+    for (const change of changes) applyChange(batch, change);
+  }, store);
+}
+
+// src/live-layer.ts
+function liveKey(type, id) {
+  return `${type}:${id}`;
+}
+function without(values, fields) {
+  const rest = { ...values };
+  for (const field of fields) delete rest[field];
+  return Object.keys(rest).length > 0 ? rest : void 0;
+}
+function pick(values, fields) {
+  if (!fields) return values;
+  const picked = {};
+  for (const field of fields) if (Object.hasOwn(values, field)) picked[field] = values[field];
+  return picked;
+}
+function createLiveLayer(options = {}) {
+  const store = options.store ?? graphStore;
+  const values = /* @__PURE__ */ new Map();
+  const listeners = /* @__PURE__ */ new Map();
+  const mergedCache = /* @__PURE__ */ new Map();
+  const muted = /* @__PURE__ */ new Set();
+  const notify = (key) => {
+    const set = listeners.get(key);
+    if (!set) return;
+    for (const listener of [...set]) listener();
+  };
+  const putLive = (type, id, fields) => {
+    const key = liveKey(type, id);
+    if (fields) values.set(key, { type, id, fields });
+    else values.delete(key);
+  };
+  const durable = (state, type, id) => state.readEntity(type, id);
+  const layer = {
+    set(type, id, next) {
+      const current2 = values.get(liveKey(type, id))?.fields;
+      putLive(type, id, { ...current2, ...next });
+      notify(liveKey(type, id));
+    },
+    append(type, id, field, chunk) {
+      const current2 = values.get(liveKey(type, id))?.fields;
+      let prior = current2 && Object.hasOwn(current2, field) ? current2[field] : void 0;
+      if (prior === void 0) prior = durable(store.getState(), type, id)?.[field];
+      layer.set(type, id, { [field]: (typeof prior === "string" ? prior : "") + chunk });
+    },
+    get(type, id) {
+      return values.get(liveKey(type, id))?.fields;
+    },
+    read(type, id) {
+      const key = liveKey(type, id);
+      const base = durable(store.getState(), type, id);
+      const live = values.get(key)?.fields;
+      const cached = mergedCache.get(key);
+      if (cached && cached.base === base && cached.live === live) return cached.merged;
+      const merged = live ? { ...base ?? {}, ...live } : base;
+      if (merged === null) mergedCache.delete(key);
+      else mergedCache.set(key, { base, live, merged });
+      return merged;
+    },
+    subscribe(type, id, listener) {
+      const key = liveKey(type, id);
+      let set = listeners.get(key);
+      if (!set) listeners.set(key, set = /* @__PURE__ */ new Set());
+      set.add(listener);
+      const unsubscribeGraph = store.subscribe(
+        (state) => [state.entities[type]?.[id], state.patches[type]?.[id]],
+        () => {
+          if (!muted.has(key)) listener();
+        },
+        { equalityFn: (a, b) => a[0] === b[0] && a[1] === b[1] }
+      );
+      let active = true;
+      return () => {
+        if (!active) return;
+        active = false;
+        unsubscribeGraph();
+        set.delete(listener);
+        if (set.size === 0 && listeners.get(key) === set) listeners.delete(key);
+      };
+    },
+    commit(type, id, commitOptions = {}) {
+      const key = liveKey(type, id);
+      const live = values.get(key)?.fields;
+      if (!live) return;
+      const data = pick(live, commitOptions.fields);
+      muted.add(key);
+      try {
+        if (Object.keys(data).length > 0) store.getState().upsertEntity(type, id, data);
+      } finally {
+        muted.delete(key);
+      }
+      putLive(type, id, commitOptions.fields ? without(live, commitOptions.fields) : void 0);
+      notify(key);
+    },
+    commitAll() {
+      const entries = [...values.entries()];
+      if (entries.length === 0) return;
+      for (const [key] of entries) muted.add(key);
+      try {
+        batchGraphWrites((batch) => {
+          for (const [, entry] of entries) batch.upsertEntity(entry.type, entry.id, entry.fields);
+        }, store);
+      } finally {
+        for (const [key] of entries) muted.delete(key);
+      }
+      for (const [key] of entries) values.delete(key);
+      for (const [key] of entries) notify(key);
+    },
+    discard(type, id, fields) {
+      const key = liveKey(type, id);
+      const live = values.get(key)?.fields;
+      if (!live) return;
+      putLive(type, id, fields ? without(live, fields) : void 0);
+      notify(key);
+    }
+  };
+  return layer;
+}
 
 // src/graph-effects.ts
 function createGraphEffect(opts) {
@@ -1626,34 +1861,34 @@ function removeAt(doc, segs) {
   return root;
 }
 function applyJsonPatch(doc, ops) {
-  let current = doc;
+  let current2 = doc;
   for (const op of ops) {
     const segs = parsePointer(op.path);
     switch (op.op) {
       case "add":
-        current = setAt(current, segs, op.value, "add");
+        current2 = setAt(current2, segs, clone(op.value), "add");
         break;
       case "replace":
-        current = setAt(current, segs, op.value, "replace");
+        current2 = setAt(current2, segs, clone(op.value), "replace");
         break;
       case "remove":
-        current = removeAt(current, segs);
+        current2 = removeAt(current2, segs);
         break;
       case "move": {
         const fromSegs = parsePointer(op.from ?? "");
-        const moved = getAt(current, fromSegs);
-        current = removeAt(current, fromSegs);
-        current = setAt(current, segs, moved, "add");
+        const moved = getAt(current2, fromSegs);
+        current2 = removeAt(current2, fromSegs);
+        current2 = setAt(current2, segs, moved, "add");
         break;
       }
       case "copy": {
         const fromSegs = parsePointer(op.from ?? "");
-        const copied = clone(getAt(current, fromSegs));
-        current = setAt(current, segs, copied, "add");
+        const copied = clone(getAt(current2, fromSegs));
+        current2 = setAt(current2, segs, copied, "add");
         break;
       }
       case "test": {
-        const actual = getAt(current, segs);
+        const actual = getAt(current2, segs);
         if (JSON.stringify(actual) !== JSON.stringify(op.value)) {
           throw new Error(`JSON Patch test failed at "${op.path}"`);
         }
@@ -1663,83 +1898,148 @@ function applyJsonPatch(doc, ops) {
         throw new Error(`Unsupported JSON Patch op: ${op.op}`);
     }
   }
-  return current;
+  return current2;
 }
 
 // src/agent/ag-ui-bridge.ts
+var AGUI_STATE_ENTITY_TYPE = "AgUiState";
+var DEFAULT_STATE_ID = "default";
 function isObject3(v) {
   return typeof v === "object" && v !== null && !Array.isArray(v);
 }
+function parsePointer2(pointer) {
+  if (pointer === "" || pointer === "/") return [];
+  return pointer.slice(1).split("/").map((s) => s.replace(/~1/g, "/").replace(/~0/g, "~"));
+}
 function getAtPointer(state, pointer) {
-  if (pointer === "" || pointer === "/") return state;
-  const segs = pointer.slice(1).split("/").map((s) => s.replace(/~1/g, "/").replace(/~0/g, "~"));
   let cur = state;
-  for (const s of segs) {
+  for (const s of parsePointer2(pointer)) {
     if (Array.isArray(cur)) cur = cur[Number(s)];
     else if (isObject3(cur)) cur = cur[s];
     else return void 0;
   }
   return cur;
 }
-function writeEntity(mapping, id, obj) {
-  const store = useGraphStore.getState();
-  if ((mapping.write ?? "upsert") === "replace") {
-    store.replaceEntity(mapping.entityType, id, obj);
-  } else {
-    store.upsertEntity(mapping.entityType, id, obj);
+function agUiListKey(mapping, stateId = DEFAULT_STATE_ID) {
+  return mapping.listKey ?? `agui:${stateId}:${mapping.entityType}:${mapping.pointer}`;
+}
+function mappedItems(state, mapping) {
+  const slice = getAtPointer(state, mapping.pointer);
+  if (slice === void 0) return [];
+  if (mapping.kind === "single") {
+    if (!isObject3(slice)) return [];
+    const id = mapping.constantId ?? String(slice[mapping.idField ?? "id"] ?? "");
+    return id ? [[id, slice]] : [];
+  }
+  const items = [];
+  if (Array.isArray(slice)) {
+    for (const item of slice) {
+      if (isObject3(item)) {
+        const id = String(item[mapping.idField ?? "id"] ?? "");
+        if (id) items.push([id, item]);
+      }
+    }
+  } else if (isObject3(slice)) {
+    for (const [key, item] of Object.entries(slice)) {
+      if (isObject3(item)) items.push([String(item[mapping.idField ?? "id"] ?? key), item]);
+    }
   }
+  return items;
 }
-function projectState(state, mappings) {
-  for (const mapping of mappings) {
-    const slice = getAtPointer(state, mapping.pointer);
-    if (slice === void 0) continue;
-    if (mapping.kind === "single") {
-      if (!isObject3(slice)) continue;
-      const id = mapping.constantId ?? String(slice[mapping.idField ?? "id"] ?? "");
-      if (!id) continue;
-      writeEntity(mapping, id, slice);
-    } else {
-      const items = [];
-      if (Array.isArray(slice)) {
-        for (const item of slice) {
-          if (isObject3(item)) {
-            const id = String(item[mapping.idField ?? "id"] ?? "");
-            if (id) items.push([id, item]);
-          }
-        }
-      } else if (isObject3(slice)) {
-        for (const [key, item] of Object.entries(slice)) {
-          if (isObject3(item)) {
-            const id = String(item[mapping.idField ?? "id"] ?? key);
-            items.push([id, item]);
-          }
-        }
+function projectState(batch, prev, next, mappings, stateId) {
+  const itemsByMapping = mappings.map((mapping) => mappedItems(next, mapping));
+  if (prev) {
+    const held = /* @__PURE__ */ new Map();
+    mappings.forEach((mapping, i) => {
+      let ids = held.get(mapping.entityType);
+      if (!ids) held.set(mapping.entityType, ids = /* @__PURE__ */ new Set());
+      for (const [id] of itemsByMapping[i]) ids.add(id);
+    });
+    const removed = /* @__PURE__ */ new Set();
+    for (const mapping of mappings) {
+      for (const [id] of mappedItems(prev, mapping)) {
+        const key = `${mapping.entityType}:${id}`;
+        if (held.get(mapping.entityType)?.has(id) || removed.has(key)) continue;
+        removed.add(key);
+        batch.removeEntity(mapping.entityType, id);
+        batch.removeIdFromAllLists(mapping.entityType, id);
       }
-      for (const [id, obj] of items) writeEntity(mapping, id, obj);
     }
   }
+  mappings.forEach((mapping, i) => {
+    const items = itemsByMapping[i];
+    const nextIds = new Set(items.map(([id]) => id));
+    for (const [id, obj] of items) {
+      if ((mapping.write ?? "upsert") === "replace") batch.replaceEntity(mapping.entityType, id, obj);
+      else batch.upsertEntity(mapping.entityType, id, obj);
+    }
+    if (mapping.kind === "collection") {
+      batch.setListIds(agUiListKey(mapping, stateId), [...nextIds], mapping.entityType);
+    }
+  });
+}
+function readRecord(store, stateId) {
+  const record = store.getState().entities[AGUI_STATE_ENTITY_TYPE]?.[stateId];
+  return record;
+}
+function readAgUiState(stateId = DEFAULT_STATE_ID, store = graphStore) {
+  return readRecord(store, stateId)?.state ?? null;
+}
+function applyState(options, nextOf) {
+  const store = options.store ?? graphStore;
+  const stateId = options.stateId ?? DEFAULT_STATE_ID;
+  const record = readRecord(store, stateId);
+  if (options.eventId !== void 0 && record?.appliedEventIds.includes(options.eventId)) {
+    return { applied: false, state: record.state };
+  }
+  const { prev, next } = nextOf(record);
+  if (!isObject3(next)) return { applied: false, state: record?.state ?? null };
+  const appliedEventIds = options.eventId === void 0 ? record?.appliedEventIds ?? [] : [...record?.appliedEventIds ?? [], options.eventId].slice(-256);
+  batchGraphWrites((batch) => {
+    projectState(batch, prev, next, options.mappings, stateId);
+    batch.replaceEntity(AGUI_STATE_ENTITY_TYPE, stateId, { id: stateId, state: next, appliedEventIds });
+  }, store);
+  return { applied: true, state: readAgUiState(stateId, store) };
 }
 function applyAgUiSnapshot(event, options) {
-  if (!isObject3(event.snapshot)) return;
-  projectState(event.snapshot, options.mappings);
+  if (!isObject3(event.snapshot)) return { applied: false, state: null };
+  return applyState(options, (record) => ({ prev: record?.state, next: structuredClone(event.snapshot) }));
 }
 function applyAgUiDelta(event, options, currentState) {
-  const base = currentState ?? reconstructState(options.mappings);
-  const next = applyJsonPatch(base, event.delta);
-  if (isObject3(next)) projectState(next, options.mappings);
+  const store = options.store ?? graphStore;
+  const stateId = options.stateId ?? DEFAULT_STATE_ID;
+  return applyState(options, (record) => {
+    const base = currentState !== void 0 ? structuredClone(currentState) : record?.state;
+    if (base) return { prev: base, next: applyJsonPatch(base, event.delta) };
+    const reconstructed = reconstructState(options.mappings, store, stateId);
+    return { prev: void 0, next: applyJsonPatch(reconstructed, event.delta) };
+  });
 }
-function reconstructState(mappings) {
-  const store = useGraphStore.getState();
-  const state = {};
+function setAtPointer(target, pointer, value) {
+  const segs = parsePointer2(pointer);
+  if (segs.length === 0) return isObject3(value) ? { ...value } : target;
+  const root = { ...target };
+  let cur = root;
+  for (const seg of segs.slice(0, -1)) {
+    const child = isObject3(cur[seg]) ? { ...cur[seg] } : {};
+    cur[seg] = child;
+    cur = child;
+  }
+  cur[segs[segs.length - 1]] = value;
+  return root;
+}
+function reconstructState(mappings, store, stateId) {
+  const s = store.getState();
+  let state = {};
   for (const mapping of mappings) {
-    const key = mapping.pointer.replace(/^\//, "").split("/")[0];
-    if (!key) continue;
-    const bucket = store.entities[mapping.entityType] ?? {};
+    const bucket = s.entities[mapping.entityType] ?? {};
     if (mapping.kind === "single") {
       const id = mapping.constantId;
-      if (id && bucket[id]) state[key] = bucket[id];
+      if (id && bucket[id]) state = setAtPointer(state, mapping.pointer, bucket[id]);
     } else {
-      state[key] = Object.values(bucket);
+      const order = s.lists[agUiListKey(mapping, stateId)]?.ids;
+      const items = order ? order.map((id) => bucket[id]).filter(Boolean) : Object.values(bucket);
+      state = setAtPointer(state, mapping.pointer, items);
     }
   }
   return state;
@@ -2838,18 +3138,18 @@ function parseCreateTable(sql) {
 function splitTopLevelCommas(body) {
   const parts = [];
   let depth = 0;
-  let current = "";
+  let current2 = "";
   for (const ch of body) {
     if (ch === "(") depth++;
     else if (ch === ")") depth--;
     if (ch === "," && depth === 0) {
-      parts.push(current);
-      current = "";
+      parts.push(current2);
+      current2 = "";
     } else {
-      current += ch;
+      current2 += ch;
     }
   }
-  if (current.trim()) parts.push(current);
+  if (current2.trim()) parts.push(current2);
   return parts;
 }
 function sqlTypeToJsonSchema(sqlType) {
@@ -2883,9 +3183,9 @@ function sqlTypeToJsonSchema(sqlType) {
     warning: `sql type "${sqlType}" not explicitly mapped; defaulting to string`
   };
 }
-function deepMergeUnknown(base, override) {
+function deepMergeUnknown(base, override2) {
   const out = { ...base };
-  for (const [key, value] of Object.entries(override)) {
+  for (const [key, value] of Object.entries(override2)) {
     const baseValue = base[key];
     if (value !== null && typeof value === "object" && !Array.isArray(value) && baseValue !== null && typeof baseValue === "object" && !Array.isArray(baseValue)) {
       out[key] = deepMergeUnknown(
@@ -3289,6 +3589,6 @@ function getFacetedRowModel(preFilteredRowModel, columnId, allFilteredRowModel)
   return preFilteredRowModel;
 }
 
-export { IncrementalView, RealtimeManager, TerminalError, TransientError, __resetActiveGraphStore, __resetEntityTransports, __resetStoreRegistry, applyAgUiDelta, applyAgUiSnapshot, applyJsonPatch, buildTenantWhere, collectDirtyPaths, configureTimeTravel, createConvexAdapter, createDevtoolsEventBus, createElectricAdapter, createFlintAdapter, createGraphAction, createGraphEffect, createGraphQLSubscriptionAdapter, createGraphSyncStatusStore, createGraphTool, createGraphTransaction, createLoroMergeStrategy, createPGlitePersistenceAdapter, createRow, createRuntimeScope, createSchemaGraphTool, createSupabaseRealtimeAdapter, createSurrealLiveAdapter, createTauriSqlPersistenceAdapter, createTenantScopedElectricAdapter, createWebSocketAdapter, escapeHtml, evaluateResume, exportGraphSnapshot, exportGraphSnapshotWithSchemas, flattenClauses, getCoreRowModel, getRealtimeManager as getElectricRealtimeManager, getEntityJsonSchema, getEntityTransport, getExpandedRowModel, getFacetedMinMaxValues, getFacetedRowModel, getFacetedUniqueValues, getFilteredRowModel, getGraphSyncStatus, getGroupedRowModel, getPaginatedRowModel, getRealtimeManager, getRegisteredEntityTypes, getRegisteredStores, getSelectedRowModel, getSortedRowModel, getTimeTravelState, getValueAtPath, graphSyncStatusStore, hasCustomPredicates, humanizeFieldName, hydrateGraphFromStorage, inferSchemaFieldType, makeRestTransport, parseCreateTable, persistGraphToStorage, prepareGraphStoreScope, publishFlintMutation, queryOnce, recordGraphSnapshot, registerEntityFromSql, registerEntityJsonSchema, registerEntityTransport, registerRuntimeSchema, registerStore, registryKey, renderMarkdownToHtml, replayActionWithRetry, resetRealtimeManager, resolveActiveGraphStore, restoreGraphSnapshot, restoreGraphSnapshotBySeq, runWithGraphStore, selectGraph, setActiveGraphStore, setValueAtPath, sqlTypeToJsonSchema, startLocalFirstGraph, stepTimeTravel, subscribeTimeTravel, toEntityError, toGraphQLVariables, toPrismaOrderBy, toPrismaWhere, toRestParams, toSQLClauses };
+export { AGUI_STATE_ENTITY_TYPE, IncrementalView, RealtimeManager, TerminalError, TransientError, __resetActiveGraphStore, __resetEntityTransports, __resetStoreRegistry, agUiListKey, applyAgUiDelta, applyAgUiSnapshot, applyGraphChanges, applyJsonPatch, batchGraphWrites, buildTenantWhere, collectDirtyPaths, configureTimeTravel, createConvexAdapter, createDevtoolsEventBus, createElectricAdapter, createFlintAdapter, createGraphAction, createGraphEffect, createGraphQLSubscriptionAdapter, createGraphSyncStatusStore, createGraphTool, createGraphTransaction, createLiveLayer, createLoroMergeStrategy, createPGlitePersistenceAdapter, createRow, createRuntimeScope, createSchemaGraphTool, createSupabaseRealtimeAdapter, createSurrealLiveAdapter, createTauriSqlPersistenceAdapter, createTenantScopedElectricAdapter, createWebSocketAdapter, escapeHtml, evaluateResume, exportGraphSnapshot, exportGraphSnapshotWithSchemas, flattenClauses, getCoreRowModel, getRealtimeManager as getElectricRealtimeManager, getEntityJsonSchema, getEntityTransport, getExpandedRowModel, getFacetedMinMaxValues, getFacetedRowModel, getFacetedUniqueValues, getFilteredRowModel, getGraphSyncStatus, getGroupedRowModel, getPaginatedRowModel, getRealtimeManager, getRegisteredEntityTypes, getRegisteredStores, getSelectedRowModel, getSortedRowModel, getTimeTravelState, getValueAtPath, graphSyncStatusStore, hasCustomPredicates, humanizeFieldName, hydrateGraphFromStorage, inferSchemaFieldType, isGraphDevtoolsEmissionEnabled, makeRestTransport, parseCreateTable, persistGraphToStorage, prepareGraphStoreScope, publishFlintMutation, queryOnce, readAgUiState, recordGraphSnapshot, registerEntityFromSql, registerEntityJsonSchema, registerEntityTransport, registerRuntimeSchema, registerStore, registryKey, renderMarkdownToHtml, replayActionWithRetry, resetRealtimeManager, resolveActiveGraphStore, restoreGraphSnapshot, restoreGraphSnapshotBySeq, runWithGraphStore, selectGraph, setActiveGraphStore, setGraphDevtoolsEmission, setValueAtPath, sqlTypeToJsonSchema, startLocalFirstGraph, stepTimeTravel, subscribeTimeTravel, toEntityError, toGraphQLVariables, toPrismaOrderBy, toPrismaWhere, toRestParams, toSQLClauses };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map