@zhivex-ai/core 1.18.0 → 1.20.0

package/dist/agent-store/memory.js

@@ -0,0 +1,198 @@
+import { normalizeAgentRunState } from "../agent-state.js";
+import { ConflictError, ValidationError } from "../errors.js";
+import { scopedKey, resolveScope, cloneState, assertLeaseOwner, assertExpectedRevision, nextStoredState, scopePrefix, listStates, validateLeaseOptions, cloneJournalEntry, assertJournalRevision, nextJournalEntry, defaultMemoryKey, defaultMemoryMessages } from "./shared.js";
+const cloneMessages = (messages) => JSON.parse(JSON.stringify(messages));
+export const createInMemoryAgentRunStore = (options = {}) => {
+    const states = new Map();
+    const idempotencyKeys = new Map();
+    const parentRunIds = new Map();
+    const leases = new Map();
+    const journal = new Map();
+    const runKey = (runId, scope) => scopedKey(resolveScope(options.scope, scope), runId);
+    const idempotencyKey = (key, scope) => scopedKey(resolveScope(options.scope, scope), key);
+    const journalKey = (runId, toolCallId, scope) => `${runKey(runId, scope)}:${toolCallId}`;
+    const removeParentIndex = (state) => {
+        if (!state?.parentRunId) {
+            return;
+        }
+        const parentKey = scopedKey(resolveScope(options.scope, state.scope), state.parentRunId);
+        const children = parentRunIds.get(parentKey);
+        children?.delete(runKey(state.runId, state.scope));
+        if (children?.size === 0) {
+            parentRunIds.delete(parentKey);
+        }
+    };
+    return {
+        reconciliationFencing: true,
+        load(runId, scope) {
+            const state = states.get(runKey(runId, scope));
+            return state ? cloneState(normalizeAgentRunState(state)) : undefined;
+        },
+        findByIdempotencyKey(idempotencyKeyValue, scope) {
+            const runId = idempotencyKeys.get(idempotencyKey(idempotencyKeyValue, scope));
+            if (!runId) {
+                return undefined;
+            }
+            const state = states.get(runId);
+            return state ? cloneState(normalizeAgentRunState(state)) : undefined;
+        },
+        findByParentRunId(parentRunId, scope) {
+            return [...(parentRunIds.get(scopedKey(resolveScope(options.scope, scope), parentRunId)) ?? [])].flatMap((runId) => {
+                const state = states.get(runId);
+                return state ? [cloneState(normalizeAgentRunState(state))] : [];
+            });
+        },
+        claimIdempotencyKey(state) {
+            const scope = resolveScope(options.scope, state.scope);
+            const existingRunId = idempotencyKeys.get(idempotencyKey(state.idempotencyKey, scope));
+            const existing = existingRunId ? states.get(existingRunId) : undefined;
+            if (existing) {
+                return { claimed: false, state: cloneState(existing) };
+            }
+            const normalized = normalizeAgentRunState(state);
+            removeParentIndex(states.get(runKey(normalized.runId, scope)));
+            const stored = cloneState({ ...normalized, ...(scope ? { scope } : {}) });
+            states.set(runKey(normalized.runId, scope), stored);
+            idempotencyKeys.set(idempotencyKey(state.idempotencyKey, scope), runKey(normalized.runId, scope));
+            if (normalized.parentRunId) {
+                const parentKey = scopedKey(scope, normalized.parentRunId);
+                const children = parentRunIds.get(parentKey) ?? new Set();
+                children.add(runKey(normalized.runId, scope));
+                parentRunIds.set(parentKey, children);
+            }
+            return { claimed: true, state: cloneState(stored) };
+        },
+        save(state, saveOptions) {
+            const scope = resolveScope(options.scope, state.scope);
+            assertLeaseOwner(leases.get(runKey(state.runId, scope)), saveOptions?.leaseOwnerId);
+            const current = states.get(runKey(state.runId, scope));
+            assertExpectedRevision(current, saveOptions?.expectedRevision);
+            const normalized = nextStoredState(state, saveOptions);
+            removeParentIndex(current);
+            states.set(runKey(normalized.runId, scope), cloneState({ ...normalized, ...(scope ? { scope } : {}) }));
+            if (normalized.idempotencyKey) {
+                const owner = idempotencyKeys.get(idempotencyKey(normalized.idempotencyKey, scope));
+                if (owner && owner !== runKey(normalized.runId, scope)) {
+                    throw new ConflictError("AgentRunState idempotency key conflict.");
+                }
+                idempotencyKeys.set(idempotencyKey(normalized.idempotencyKey, scope), runKey(normalized.runId, scope));
+            }
+            if (normalized.parentRunId) {
+                const parentKey = scopedKey(scope, normalized.parentRunId);
+                const children = parentRunIds.get(parentKey) ?? new Set();
+                children.add(runKey(normalized.runId, scope));
+                parentRunIds.set(parentKey, children);
+            }
+        },
+        delete(runId, scope) {
+            const key = runKey(runId, scope);
+            const state = states.get(key);
+            if (state?.idempotencyKey) {
+                idempotencyKeys.delete(idempotencyKey(state.idempotencyKey, state.scope));
+            }
+            removeParentIndex(state);
+            states.delete(key);
+            leases.delete(key);
+            for (const journalEntryKey of journal.keys()) {
+                if (journalEntryKey.startsWith(`${key}:`))
+                    journal.delete(journalEntryKey);
+            }
+        },
+        list(listOptions, scope) {
+            const prefix = scopePrefix(resolveScope(options.scope, scope));
+            return listStates([...states.entries()].filter(([key]) => key.startsWith(prefix)).map(([, state]) => state), listOptions);
+        },
+        deleteExpired(retention, scope) {
+            const prefix = scopePrefix(resolveScope(options.scope, scope));
+            const candidates = listStates([...states.entries()].filter(([key]) => key.startsWith(prefix)).map(([, state]) => state), {
+                statuses: retention.statuses,
+                updatedBefore: retention.before,
+                limit: retention.limit ?? 1_000
+            }).items;
+            for (const state of candidates)
+                this.delete?.(state.runId, state.scope);
+            return candidates.length;
+        },
+        acquireLease(runId, leaseOptions, scope) {
+            validateLeaseOptions(leaseOptions);
+            const key = runKey(runId, scope);
+            if (!states.has(key))
+                return undefined;
+            const now = leaseOptions.now ?? Date.now();
+            const current = leases.get(key);
+            if (current && current.expiresAt > now && current.ownerId !== leaseOptions.ownerId)
+                return undefined;
+            const lease = { runId, ownerId: leaseOptions.ownerId, expiresAt: now + leaseOptions.ttlMs };
+            leases.set(key, lease);
+            return { ...lease };
+        },
+        renewLease(runId, leaseOptions, scope) {
+            validateLeaseOptions(leaseOptions);
+            const key = runKey(runId, scope);
+            const now = leaseOptions.now ?? Date.now();
+            const current = leases.get(key);
+            if (!current || current.ownerId !== leaseOptions.ownerId || current.expiresAt <= now)
+                return undefined;
+            const lease = { runId, ownerId: leaseOptions.ownerId, expiresAt: now + leaseOptions.ttlMs };
+            leases.set(key, lease);
+            return { ...lease };
+        },
+        releaseLease(runId, ownerId, scope) {
+            const key = runKey(runId, scope);
+            if (leases.get(key)?.ownerId !== ownerId)
+                return false;
+            return leases.delete(key);
+        },
+        loadToolCall(runId, toolCallId, scope) {
+            const entry = journal.get(journalKey(runId, toolCallId, scope));
+            return entry ? cloneJournalEntry(entry) : undefined;
+        },
+        loadToolExecution(runId, toolCallId, scope) {
+            return this.loadToolCall?.(runId, toolCallId, scope);
+        },
+        listToolCalls(runId, scope) {
+            const prefix = `${runKey(runId, scope)}:`;
+            return [...journal.entries()]
+                .filter(([key]) => key.startsWith(prefix))
+                .map(([, entry]) => cloneJournalEntry(entry))
+                .sort((left, right) => left.updatedAt - right.updatedAt || left.toolCallId.localeCompare(right.toolCallId));
+        },
+        saveToolCall(entry, journalOptions) {
+            const scope = resolveScope(options.scope, entry.scope);
+            if (!states.has(runKey(entry.runId, scope)))
+                throw new ValidationError("Cannot journal a tool call for an unknown run.");
+            assertLeaseOwner(leases.get(runKey(entry.runId, scope)), journalOptions?.leaseOwnerId);
+            const key = journalKey(entry.runId, entry.toolCallId, scope);
+            const current = journal.get(key);
+            assertJournalRevision(current, journalOptions?.expectedRevision);
+            const next = nextJournalEntry(entry, journalOptions);
+            journal.set(key, next);
+            return cloneJournalEntry(next);
+        },
+        claimToolExecution(entry) {
+            const key = journalKey(entry.runId, entry.toolCallId, entry.scope);
+            const existing = journal.get(key);
+            if (existing)
+                return { claimed: false, entry: cloneJournalEntry(existing) };
+            const claimed = this.saveToolCall?.({ ...entry, status: "running", revision: 0 });
+            return { claimed: true, entry: claimed };
+        },
+        completeToolExecution(entry, journalOptions) {
+            return this.saveToolCall?.({ ...entry, status: entry.status === "failed" ? "failed" : "completed" }, journalOptions);
+        }
+    };
+};
+export const createInMemoryAgentMemoryStore = (options = {}) => {
+    const keyFor = (context) => (options.key ?? defaultMemoryKey)({ ...context, scope: resolveScope(options.scope, context.scope) });
+    const selectMessages = options.selectMessages ?? defaultMemoryMessages;
+    const memories = new Map(Object.entries(options.initialMessages ?? {}).map(([key, messages]) => [key, cloneMessages(messages)]));
+    return {
+        load(context) {
+            return cloneMessages(memories.get(keyFor(context)) ?? (context.agentId ? options.initialMessages?.[context.agentId] : undefined) ?? []);
+        },
+        save(context) {
+            memories.set(keyFor(context), cloneMessages(selectMessages(context.state)));
+        }
+    };
+};
+//# sourceMappingURL=memory.js.map

package/dist/agent-store/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../src/agent-store/memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAY9D,OAAO,EACL,SAAS,EACT,YAAY,EACZ,UAAU,EACV,gBAAgB,EAChB,sBAAsB,EACtB,eAAe,EACf,WAAW,EACX,UAAU,EACV,oBAAoB,EACpB,iBAAiB,EACjB,qBAAqB,EACrB,gBAAgB,EAChB,gBAAgB,EAChB,qBAAqB,EACtB,MAAM,aAAa,CAAC;AAErB,MAAM,aAAa,GAAG,CAAC,QAAwB,EAAkB,EAAE,CACjE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAmB,CAAC;AAEzD,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,OAAO,GAA8B,EAAE,EAAiB,EAAE;IACpG,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;IAChD,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;IAClD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAuB,CAAC;IACpD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;IAChD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAqC,CAAC;IAC7D,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,KAAuB,EAAE,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;IAChH,MAAM,cAAc,GAAG,CAAC,GAAW,EAAE,KAAuB,EAAE,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;IACpH,MAAM,UAAU,GAAG,CAAC,KAAa,EAAE,UAAkB,EAAE,KAAuB,EAAE,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,UAAU,EAAE,CAAC;IAE3H,MAAM,iBAAiB,GAAG,CAAC,KAAgC,EAAE,EAAE;QAC7D,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;QACzF,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACnD,IAAI,QAAQ,EAAE,IAAI,KAAK,CAAC,EAAE,CAAC;YACzB,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;IACH,CAAC,CAAC;IAEF,OAAO;QACL,qBAAqB,EAAE,IAAI;QAC3B,IAAI,CAAC,KAAK,EAAE,KAAK;YACf,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAC/C,OAAO,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACvE,CAAC;QACD,oBAAoB,CAAC,mBAAmB,EAAE,KAAK;YAC7C,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC,CAAC;YAC9E,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAChC,OAAO,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACvE,CAAC;QACD,iBAAiB,CAAC,WAAW,EAAE,KAAK;YAClC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjH,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAChC,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC;QACD,mBAAmB,CAAC,KAAK;YACvB,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACvD,MAAM,aAAa,GAAG,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAAC;YACvF,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACvE,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzD,CAAC;YAED,MAAM,UAAU,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;YACjD,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,UAAU,CAAC,EAAE,GAAG,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,CAAC;YACpD,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAClG,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC3B,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,EAAE,UAAU,CAAC,WAAW,CAAC,CAAC;gBAC3D,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,GAAG,EAAU,CAAC;gBAClE,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC9C,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACxC,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,KAAK,EAAE,WAAW;YACrB,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACvD,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;YACpF,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YACvD,sBAAsB,CAAC,OAAO,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YACvD,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC3B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,UAAU,CAAC,EAAE,GAAG,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YACxG,IAAI,UAAU,CAAC,cAAc,EAAE,CAAC;gBAC9B,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAAC;gBACpF,IAAI,KAAK,IAAI,KAAK,KAAK,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;oBACvD,MAAM,IAAI,aAAa,CAAC,yCAAyC,CAAC,CAAC;gBACrE,CAAC;gBACD,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YACzG,CAAC;YACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC3B,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,EAAE,UAAU,CAAC,WAAW,CAAC,CAAC;gBAC3D,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,GAAG,EAAU,CAAC;gBAClE,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC9C,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,MAAM,CAAC,KAAK,EAAE,KAAK;YACjB,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACjC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC9B,IAAI,KAAK,EAAE,cAAc,EAAE,CAAC;gBAC1B,eAAe,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5E,CAAC;YACD,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnB,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnB,KAAK,MAAM,eAAe,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC7C,IAAI,eAAe,CAAC,UAAU,CAAC,GAAG,GAAG,GAAG,CAAC;oBAAE,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,KAAK;YACrB,MAAM,MAAM,GAAG,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAC/D,OAAO,UAAU,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,WAAW,CAAC,CAAC;QAC5H,CAAC;QACD,aAAa,CAAC,SAAS,EAAE,KAAK;YAC5B,MAAM,MAAM,GAAG,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE;gBACvH,QAAQ,EAAE,SAAS,CAAC,QAAQ;gBAC5B,aAAa,EAAE,SAAS,CAAC,MAAM;gBAC/B,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,KAAK;aAChC,CAAC,CAAC,KAAK,CAAC;YACT,KAAK,MAAM,KAAK,IAAI,UAAU;gBAAE,IAAI,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACxE,OAAO,UAAU,CAAC,MAAM,CAAC;QAC3B,CAAC;QACD,YAAY,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK;YACrC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACnC,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO,SAAS,CAAC;YACvC,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChC,IAAI,OAAO,IAAI,OAAO,CAAC,SAAS,GAAG,GAAG,IAAI,OAAO,CAAC,OAAO,KAAK,YAAY,CAAC,OAAO;gBAAE,OAAO,SAAS,CAAC;YACrG,MAAM,KAAK,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,GAAG,YAAY,CAAC,KAAK,EAAE,CAAC;YAC5F,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACvB,OAAO,EAAE,GAAG,KAAK,EAAE,CAAC;QACtB,CAAC;QACD,UAAU,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK;YACnC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACnC,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACjC,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,KAAK,YAAY,CAAC,OAAO,IAAI,OAAO,CAAC,SAAS,IAAI,GAAG;gBAAE,OAAO,SAAS,CAAC;YACvG,MAAM,KAAK,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,GAAG,YAAY,CAAC,KAAK,EAAE,CAAC;YAC5F,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACvB,OAAO,EAAE,GAAG,KAAK,EAAE,CAAC;QACtB,CAAC;QACD,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK;YAChC,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,OAAO,KAAK,OAAO;gBAAE,OAAO,KAAK,CAAC;YACvD,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;QACD,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK;YACnC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC;YAChE,OAAO,KAAK,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACtD,CAAC;QACD,iBAAiB,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK;YACxC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QACvD,CAAC;QACD,aAAa,CAAC,KAAK,EAAE,KAAK;YACxB,MAAM,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC;YAC1C,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;iBAC1B,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;iBACzC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;iBAC5C,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;QAChH,CAAC;QACD,YAAY,CAAC,KAAK,EAAE,cAAc;YAChC,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACvD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBAAE,MAAM,IAAI,eAAe,CAAC,gDAAgD,CAAC,CAAC;YACzH,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC;YACvF,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC7D,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjC,qBAAqB,CAAC,OAAO,EAAE,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACjE,MAAM,IAAI,GAAG,gBAAgB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;YACrD,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACvB,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QACD,kBAAkB,CAAC,KAAK;YACtB,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,QAAQ;gBAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5E,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAClF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,OAAoC,EAAE,CAAC;QACxE,CAAC;QACD,qBAAqB,CAAC,KAAK,EAAE,cAAc;YACzC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,EAAE,cAAc,CAA8B,CAAC;QACpJ,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,OAAO,GAKlD,EAAE,EAAoB,EAAE;IAC1B,MAAM,MAAM,GAAG,CAAC,OAA2B,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,IAAI,gBAAgB,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACrJ,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,qBAAqB,CAAC;IACvE,MAAM,QAAQ,GAAG,IAAI,GAAG,CACtB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,CACvG,CAAC;IAEF,OAAO;QACL,IAAI,CAAC,OAAO;YACV,OAAO,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1I,CAAC;QACD,IAAI,CAAC,OAAO;YACV,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9E,CAAC;KACF,CAAC;AACJ,CAAC,CAAC"}

package/dist/agent-store/postgres.d.ts

@@ -0,0 +1,4 @@
+import type { AgentMemoryStore, AgentRunStore, PostgresAgentMemoryStoreOptions, PostgresAgentRunStoreOptions } from "../types.js";
+export declare const createPostgresAgentRunStore: (options: PostgresAgentRunStoreOptions) => AgentRunStore;
+export declare const createPostgresAgentMemoryStore: (options: PostgresAgentMemoryStoreOptions) => AgentMemoryStore;
+//# sourceMappingURL=postgres.d.ts.map

package/dist/agent-store/postgres.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../src/agent-store/postgres.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAEV,gBAAgB,EAGhB,aAAa,EAIb,+BAA+B,EAC/B,4BAA4B,EAE7B,MAAM,aAAa,CAAC;AA+DrB,eAAO,MAAM,2BAA2B,YAAa,4BAA4B,KAAG,aAuWnF,CAAC;AAEF,eAAO,MAAM,8BAA8B,YAAa,+BAA+B,KAAG,gBAoCzF,CAAC"}

package/dist/agent-store/postgres.js

@@ -0,0 +1,374 @@
+import { normalizeAgentRunState } from "../agent-state.js";
+import { ConflictError, ValidationError } from "../errors.js";
+import { assertPostgresClient } from "../postgres-client.js";
+import { validateIdentifier, scopedKey, resolveScope, getRecordField, nextStoredState, scopePrefix, listStates, validateLeaseOptions, cloneJournalEntry, nextJournalEntry, defaultMemoryKey, defaultMemoryMessages } from "./shared.js";
+const isConcurrentPostgresTableCreationConflict = (error) => {
+    if (!error || typeof error !== "object")
+        return false;
+    const record = error;
+    return record.code === "23505" &&
+        (record.constraint === "pg_type_typname_nsp_index" ||
+            record.constraint_name === "pg_type_typname_nsp_index");
+};
+const ensurePostgresTable = (() => {
+    const initializedTables = new WeakMap();
+    return async (client, tableName, createSql) => {
+        let tables = initializedTables.get(client);
+        if (!tables) {
+            tables = new Map();
+            initializedTables.set(client, tables);
+        }
+        let initialization = tables.get(tableName);
+        if (!initialization) {
+            initialization = (async () => {
+                for (let attempt = 0;; attempt += 1) {
+                    try {
+                        await client.query(createSql, []);
+                        return;
+                    }
+                    catch (error) {
+                        if (!isConcurrentPostgresTableCreationConflict(error) || attempt >= 2) {
+                            throw error;
+                        }
+                        await new Promise((resolve) => setTimeout(resolve, 10 * (attempt + 1)));
+                    }
+                }
+            })();
+            tables.set(tableName, initialization);
+        }
+        try {
+            await initialization;
+        }
+        catch (error) {
+            if (tables.get(tableName) === initialization) {
+                tables.delete(tableName);
+            }
+            throw error;
+        }
+    };
+})();
+export const createPostgresAgentRunStore = (options) => {
+    assertPostgresClient(options.client);
+    const tableName = validateIdentifier(options.tableName ?? "zhivex_agent_runs", "tableName");
+    const idempotencyTableName = `${tableName}_idempotency`;
+    const parentTableName = `${tableName}_parents`;
+    const leaseTableName = `${tableName}_leases`;
+    const journalTableName = `${tableName}_tool_journal`;
+    const dbKey = (value, scope) => scopedKey(resolveScope(options.scope, scope), value);
+    const createSql = `
+    CREATE TABLE IF NOT EXISTS ${tableName} (
+      run_id TEXT PRIMARY KEY,
+      state_json JSONB NOT NULL,
+      updated_at_ms BIGINT NOT NULL
+    )
+  `;
+    const createIdempotencySql = `
+    CREATE TABLE IF NOT EXISTS ${idempotencyTableName} (
+      idempotency_key TEXT PRIMARY KEY,
+      run_id TEXT NOT NULL,
+      updated_at_ms BIGINT NOT NULL
+    )
+  `;
+    const createParentSql = `
+    CREATE TABLE IF NOT EXISTS ${parentTableName} (
+      run_id TEXT PRIMARY KEY,
+      parent_run_id TEXT NOT NULL,
+      updated_at_ms BIGINT NOT NULL
+    )
+  `;
+    const createLeaseSql = `CREATE TABLE IF NOT EXISTS ${leaseTableName} (
+    run_key TEXT PRIMARY KEY,
+    run_id TEXT NOT NULL,
+    owner_id TEXT NOT NULL,
+    expires_at_ms BIGINT NOT NULL
+  )`;
+    const createJournalSql = `CREATE TABLE IF NOT EXISTS ${journalTableName} (
+    run_key TEXT NOT NULL,
+    tool_call_id TEXT NOT NULL,
+    entry_json JSONB NOT NULL,
+    revision BIGINT NOT NULL,
+    updated_at_ms BIGINT NOT NULL,
+    PRIMARY KEY (run_key, tool_call_id)
+  )`;
+    const createIndexesSql = `
+    CREATE INDEX IF NOT EXISTS ${tableName}_updated_idx ON ${tableName} (updated_at_ms DESC, run_id);
+    CREATE INDEX IF NOT EXISTS ${parentTableName}_parent_idx ON ${parentTableName} (parent_run_id, updated_at_ms DESC);
+    CREATE INDEX IF NOT EXISTS ${leaseTableName}_expiry_idx ON ${leaseTableName} (expires_at_ms);
+    CREATE INDEX IF NOT EXISTS ${journalTableName}_run_idx ON ${journalTableName} (run_key, updated_at_ms, tool_call_id)
+  `;
+    const ensureAllTables = async () => {
+        await ensurePostgresTable(options.client, tableName, createSql);
+        await ensurePostgresTable(options.client, idempotencyTableName, createIdempotencySql);
+        await ensurePostgresTable(options.client, parentTableName, createParentSql);
+        await ensurePostgresTable(options.client, leaseTableName, createLeaseSql);
+        await ensurePostgresTable(options.client, journalTableName, createJournalSql);
+        await ensurePostgresTable(options.client, `${tableName}:indexes`, createIndexesSql);
+    };
+    return {
+        async load(runId, scope) {
+            await ensureAllTables();
+            const result = await options.client.query(`SELECT state_json FROM ${tableName} WHERE run_id = $1`, [dbKey(runId, scope)]);
+            const state = result.rows[0] ? (getRecordField(result.rows[0], ["state_json", "stateJson"]) ?? undefined) : undefined;
+            return state ? normalizeAgentRunState(state) : undefined;
+        },
+        async findByIdempotencyKey(idempotencyKey, scope) {
+            await ensureAllTables();
+            const result = await options.client.query(`SELECT runs.state_json
+         FROM ${tableName} runs
+         INNER JOIN ${idempotencyTableName} keys ON keys.run_id = runs.run_id
+         WHERE keys.idempotency_key = $1`, [dbKey(idempotencyKey, scope)]);
+            const state = result.rows[0] ? (getRecordField(result.rows[0], ["state_json", "stateJson"]) ?? undefined) : undefined;
+            return state ? normalizeAgentRunState(state) : undefined;
+        },
+        async findByParentRunId(parentRunId, scope) {
+            await ensureAllTables();
+            const result = await options.client.query(`SELECT runs.state_json
+         FROM ${tableName} runs
+         INNER JOIN ${parentTableName} parents ON parents.run_id = runs.run_id
+         WHERE parents.parent_run_id = $1`, [dbKey(parentRunId, scope)]);
+            return result.rows.flatMap((row) => {
+                const state = getRecordField(row, ["state_json", "stateJson"]) ?? undefined;
+                return state ? [normalizeAgentRunState(state)] : [];
+            });
+        },
+        async claimIdempotencyKey(state) {
+            await ensureAllTables();
+            const scope = resolveScope(options.scope, state.scope);
+            const normalized = normalizeAgentRunState({ ...state, ...(scope ? { scope } : {}) });
+            const updatedAt = Date.now();
+            await options.client.query(`INSERT INTO ${tableName} (run_id, state_json, updated_at_ms)
+         VALUES ($1, $2::jsonb, $3)
+         ON CONFLICT(run_id) DO NOTHING`, [dbKey(normalized.runId, scope), normalized, updatedAt]);
+            const claim = await options.client.query(`INSERT INTO ${idempotencyTableName} (idempotency_key, run_id, updated_at_ms)
+         VALUES ($1, $2, $3)
+         ON CONFLICT(idempotency_key) DO NOTHING
+         RETURNING run_id`, [dbKey(state.idempotencyKey, scope), dbKey(normalized.runId, scope), updatedAt]);
+            const claimedRunId = getRecordField(claim.rows[0], ["run_id", "runId"]);
+            if (claimedRunId === dbKey(normalized.runId, scope)) {
+                if (normalized.parentRunId) {
+                    await options.client.query(`INSERT INTO ${parentTableName} (run_id, parent_run_id, updated_at_ms)
+             VALUES ($1, $2, $3)
+             ON CONFLICT(run_id) DO UPDATE SET
+               parent_run_id = EXCLUDED.parent_run_id,
+               updated_at_ms = EXCLUDED.updated_at_ms`, [dbKey(normalized.runId, scope), dbKey(normalized.parentRunId, scope), updatedAt]);
+                }
+                return { claimed: true, state: normalized };
+            }
+            const existing = await options.client.query(`SELECT runs.state_json
+         FROM ${tableName} runs
+         INNER JOIN ${idempotencyTableName} keys ON keys.run_id = runs.run_id
+         WHERE keys.idempotency_key = $1`, [dbKey(state.idempotencyKey, scope)]);
+            const existingState = existing.rows[0]
+                ? getRecordField(existing.rows[0], ["state_json", "stateJson"])
+                : undefined;
+            if (!existingState) {
+                throw new ConflictError("AgentRunState idempotency claim could not be loaded.");
+            }
+            if (existingState.runId !== normalized.runId) {
+                await options.client.query(`DELETE FROM ${tableName} WHERE run_id = $1`, [dbKey(normalized.runId, scope)]);
+            }
+            return { claimed: false, state: normalizeAgentRunState(existingState) };
+        },
+        async save(state, saveOptions) {
+            await ensureAllTables();
+            const scope = resolveScope(options.scope, state.scope);
+            const normalized = nextStoredState(state, saveOptions);
+            const stored = { ...normalized, ...(scope ? { scope } : {}) };
+            const updatedAt = Date.now();
+            if (normalized.idempotencyKey) {
+                const owner = await options.client.query(`SELECT run_id FROM ${idempotencyTableName} WHERE idempotency_key = $1`, [dbKey(normalized.idempotencyKey, scope)]);
+                const ownerRunId = getRecordField(owner.rows[0], ["run_id", "runId"]);
+                if (typeof ownerRunId === "string" && ownerRunId !== dbKey(normalized.runId, scope)) {
+                    throw new ConflictError("AgentRunState idempotency key conflict.");
+                }
+            }
+            if (saveOptions?.expectedRevision === undefined) {
+                await options.client.query(`INSERT INTO ${tableName} (run_id, state_json, updated_at_ms)
+           VALUES ($1, $2::jsonb, $3)
+           ON CONFLICT(run_id) DO UPDATE SET
+             state_json = EXCLUDED.state_json,
+             updated_at_ms = EXCLUDED.updated_at_ms`, [dbKey(normalized.runId, scope), stored, updatedAt]);
+            }
+            else {
+                const saved = await options.client.query(`INSERT INTO ${tableName} (run_id, state_json, updated_at_ms)
+           VALUES ($1, $2::jsonb, $3)
+           ON CONFLICT(run_id) DO UPDATE SET
+             state_json = EXCLUDED.state_json,
+             updated_at_ms = EXCLUDED.updated_at_ms
+           WHERE COALESCE((${tableName}.state_json->>'revision')::bigint, 0) = $4
+           RETURNING run_id`, [dbKey(normalized.runId, scope), stored, updatedAt, saveOptions.expectedRevision]);
+                if (getRecordField(saved.rows[0], ["run_id", "runId"]) !== dbKey(normalized.runId, scope)) {
+                    throw new ConflictError("AgentRunState revision conflict.");
+                }
+            }
+            if (normalized.idempotencyKey) {
+                await options.client.query(`INSERT INTO ${idempotencyTableName} (idempotency_key, run_id, updated_at_ms)
+           VALUES ($1, $2, $3)
+           ON CONFLICT(idempotency_key) DO NOTHING`, [dbKey(normalized.idempotencyKey, scope), dbKey(normalized.runId, scope), updatedAt]);
+            }
+            await options.client.query(`DELETE FROM ${parentTableName} WHERE run_id = $1`, [dbKey(normalized.runId, scope)]);
+            if (normalized.parentRunId) {
+                await options.client.query(`INSERT INTO ${parentTableName} (run_id, parent_run_id, updated_at_ms)
+           VALUES ($1, $2, $3)
+           ON CONFLICT(run_id) DO UPDATE SET
+             parent_run_id = EXCLUDED.parent_run_id,
+             updated_at_ms = EXCLUDED.updated_at_ms`, [dbKey(normalized.runId, scope), dbKey(normalized.parentRunId, scope), updatedAt]);
+            }
+        },
+        async delete(runId, scope) {
+            await ensureAllTables();
+            const key = dbKey(runId, scope);
+            await options.client.query(`WITH deleted_run AS (
+           DELETE FROM ${tableName} WHERE run_id = $1 RETURNING run_id
+         ), deleted_idempotency AS (
+           DELETE FROM ${idempotencyTableName} WHERE run_id = $1 RETURNING run_id
+         ), deleted_parent AS (
+           DELETE FROM ${parentTableName} WHERE run_id = $1 RETURNING run_id
+         ), deleted_lease AS (
+           DELETE FROM ${leaseTableName} WHERE run_key = $1 RETURNING run_key
+         )
+         DELETE FROM ${journalTableName} WHERE run_key = $1`, [key]);
+        },
+        async list(listOptions, scope) {
+            await ensureAllTables();
+            const prefix = scopePrefix(resolveScope(options.scope, scope));
+            const result = await options.client.query(`SELECT state_json FROM ${tableName} WHERE run_id >= $1 AND run_id < $2`, [prefix, `${prefix}\uffff`]);
+            const states = result.rows.flatMap((row) => {
+                const state = getRecordField(row, ["state_json", "stateJson"]);
+                return state ? [normalizeAgentRunState(state)] : [];
+            });
+            return listStates(states, listOptions);
+        },
+        async deleteExpired(retention, scope) {
+            const page = await this.list?.({ statuses: retention.statuses, updatedBefore: retention.before, limit: retention.limit ?? 1_000 }, scope);
+            for (const state of page.items)
+                await this.delete?.(state.runId, state.scope);
+            return page.items.length;
+        },
+        async acquireLease(runId, leaseOptions, scope) {
+            validateLeaseOptions(leaseOptions);
+            await ensureAllTables();
+            const now = leaseOptions.now ?? Date.now();
+            const key = dbKey(runId, scope);
+            const expiresAt = now + leaseOptions.ttlMs;
+            const result = await options.client.query(`INSERT INTO ${leaseTableName} (run_key, run_id, owner_id, expires_at_ms)
+         SELECT $1, $2, $3, $4
+         WHERE EXISTS (SELECT 1 FROM ${tableName} WHERE run_id = $1)
+         ON CONFLICT(run_key) DO UPDATE SET
+           owner_id = EXCLUDED.owner_id,
+           expires_at_ms = EXCLUDED.expires_at_ms
+         WHERE ${leaseTableName}.owner_id = EXCLUDED.owner_id OR ${leaseTableName}.expires_at_ms <= $5
+         RETURNING owner_id, expires_at_ms`, [key, runId, leaseOptions.ownerId, expiresAt, now]);
+            const owner = getRecordField(result.rows[0], ["owner_id", "ownerId"]);
+            return owner === leaseOptions.ownerId ? { runId, ownerId: leaseOptions.ownerId, expiresAt } : undefined;
+        },
+        async renewLease(runId, leaseOptions, scope) {
+            validateLeaseOptions(leaseOptions);
+            await ensureAllTables();
+            const now = leaseOptions.now ?? Date.now();
+            const expiresAt = now + leaseOptions.ttlMs;
+            const result = await options.client.query(`UPDATE ${leaseTableName}
+         SET expires_at_ms = $3
+         WHERE run_key = $1 AND owner_id = $2 AND expires_at_ms > $4
+         RETURNING owner_id`, [dbKey(runId, scope), leaseOptions.ownerId, expiresAt, now]);
+            return getRecordField(result.rows[0], ["owner_id", "ownerId"]) === leaseOptions.ownerId
+                ? { runId, ownerId: leaseOptions.ownerId, expiresAt }
+                : undefined;
+        },
+        async releaseLease(runId, ownerId, scope) {
+            await ensureAllTables();
+            const result = await options.client.query(`DELETE FROM ${leaseTableName} WHERE run_key = $1 AND owner_id = $2 RETURNING owner_id`, [dbKey(runId, scope), ownerId]);
+            return getRecordField(result.rows[0], ["owner_id", "ownerId"]) === ownerId;
+        },
+        async loadToolCall(runId, toolCallId, scope) {
+            await ensureAllTables();
+            const result = await options.client.query(`SELECT entry_json FROM ${journalTableName} WHERE run_key = $1 AND tool_call_id = $2`, [dbKey(runId, scope), toolCallId]);
+            const entry = getRecordField(result.rows[0], ["entry_json", "entryJson"]);
+            return entry && typeof entry === "object" ? cloneJournalEntry(entry) : undefined;
+        },
+        async loadToolExecution(runId, toolCallId, scope) {
+            return this.loadToolCall?.(runId, toolCallId, scope);
+        },
+        async listToolCalls(runId, scope) {
+            await ensureAllTables();
+            const result = await options.client.query(`SELECT entry_json FROM ${journalTableName} WHERE run_key = $1 ORDER BY updated_at_ms, tool_call_id`, [dbKey(runId, scope)]);
+            return result.rows.flatMap((row) => {
+                const entry = getRecordField(row, ["entry_json", "entryJson"]);
+                return entry && typeof entry === "object" ? [cloneJournalEntry(entry)] : [];
+            });
+        },
+        async saveToolCall(entry, journalOptions) {
+            await ensureAllTables();
+            const next = nextJournalEntry(entry, journalOptions);
+            const key = dbKey(entry.runId, entry.scope);
+            const result = journalOptions?.expectedRevision === undefined
+                ? await options.client.query(`INSERT INTO ${journalTableName} (run_key, tool_call_id, entry_json, revision, updated_at_ms)
+           VALUES ($1, $2, $3::jsonb, $4, $5)
+           ON CONFLICT(run_key, tool_call_id) DO UPDATE SET
+             entry_json = EXCLUDED.entry_json,
+             revision = EXCLUDED.revision,
+             updated_at_ms = EXCLUDED.updated_at_ms
+           RETURNING revision`, [key, entry.toolCallId, next, next.revision, next.updatedAt])
+                : await options.client.query(`UPDATE ${journalTableName}
+           SET entry_json = $3::jsonb,
+               revision = $4,
+               updated_at_ms = $5
+           WHERE run_key = $1 AND tool_call_id = $2 AND revision = $6
+           RETURNING revision`, [key, entry.toolCallId, next, next.revision, next.updatedAt, journalOptions.expectedRevision]);
+            if (getRecordField(result.rows[0], ["revision"]) === undefined) {
+                throw new ConflictError("Agent tool-call journal revision conflict.");
+            }
+            return next;
+        },
+        async claimToolExecution(entry) {
+            await ensureAllTables();
+            const next = nextJournalEntry({ ...entry, status: "running", revision: 0 });
+            const result = await options.client.query(`INSERT INTO ${journalTableName} (run_key, tool_call_id, entry_json, revision, updated_at_ms)
+         SELECT $1, $2, $3::jsonb, 0, $4
+         WHERE EXISTS (SELECT 1 FROM ${tableName} WHERE run_id = $1)
+         ON CONFLICT(run_key, tool_call_id) DO NOTHING
+         RETURNING entry_json`, [dbKey(entry.runId, entry.scope), entry.toolCallId, next, next.updatedAt]);
+            const claimed = getRecordField(result.rows[0], ["entry_json", "entryJson"]);
+            if (claimed && typeof claimed === "object")
+                return { claimed: true, entry: next };
+            const existing = await this.loadToolCall?.(entry.runId, entry.toolCallId, entry.scope);
+            if (!existing)
+                throw new ValidationError("Cannot journal a tool call for an unknown run.");
+            return { claimed: false, entry: existing };
+        },
+        async completeToolExecution(entry, journalOptions) {
+            return this.saveToolCall?.({ ...entry, status: entry.status === "failed" ? "failed" : "completed" }, journalOptions);
+        }
+    };
+};
+export const createPostgresAgentMemoryStore = (options) => {
+    assertPostgresClient(options.client);
+    const tableName = validateIdentifier(options.tableName ?? "zhivex_agent_memory", "tableName");
+    const keyFor = (context) => (options.key ?? defaultMemoryKey)({ ...context, scope: resolveScope(options.scope, context.scope) });
+    const selectMessages = options.selectMessages ?? defaultMemoryMessages;
+    const createSql = `
+    CREATE TABLE IF NOT EXISTS ${tableName} (
+      memory_key TEXT PRIMARY KEY,
+      messages_json JSONB NOT NULL,
+      updated_at_ms BIGINT NOT NULL
+    )
+  `;
+    return {
+        async load(context) {
+            await ensurePostgresTable(options.client, tableName, createSql);
+            const result = await options.client.query(`SELECT messages_json FROM ${tableName} WHERE memory_key = $1`, [keyFor(context)]);
+            return result.rows[0]
+                ? (getRecordField(result.rows[0], ["messages_json", "messagesJson"]) ?? [])
+                : [];
+        },
+        async save(context) {
+            await ensurePostgresTable(options.client, tableName, createSql);
+            await options.client.query(`INSERT INTO ${tableName} (memory_key, messages_json, updated_at_ms)
+         VALUES ($1, $2::jsonb, $3)
+         ON CONFLICT(memory_key) DO UPDATE SET
+           messages_json = EXCLUDED.messages_json,
+           updated_at_ms = EXCLUDED.updated_at_ms`, [keyFor(context), selectMessages(context.state), Date.now()]);
+        }
+    };
+};
+//# sourceMappingURL=postgres.js.map

package/dist/agent-store/postgres.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres.js","sourceRoot":"","sources":["../../src/agent-store/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAc7D,OAAO,EACL,kBAAkB,EAClB,SAAS,EACT,YAAY,EACZ,cAAc,EACd,eAAe,EACf,WAAW,EACX,UAAU,EACV,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,qBAAqB,EACtB,MAAM,aAAa,CAAC;AAErB,MAAM,yCAAyC,GAAG,CAAC,KAAc,EAAW,EAAE;IAC5E,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,MAAM,GAAG,KAA4E,CAAC;IAC5F,OAAO,MAAM,CAAC,IAAI,KAAK,OAAO;QAC5B,CAAC,MAAM,CAAC,UAAU,KAAK,2BAA2B;YAChD,MAAM,CAAC,eAAe,KAAK,2BAA2B,CAAC,CAAC;AAC9D,CAAC,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAAC,GAAG,EAAE;IAChC,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAkD,CAAC;IAExF,OAAO,KAAK,EAAE,MAA0B,EAAE,SAAiB,EAAE,SAAiB,EAAE,EAAE;QAChF,IAAI,MAAM,GAAG,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;YAC1C,iBAAiB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,cAAc,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC3B,KAAK,IAAI,OAAO,GAAG,CAAC,GAAI,OAAO,IAAI,CAAC,EAAE,CAAC;oBACrC,IAAI,CAAC;wBACH,MAAM,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,CAAC,yCAAyC,CAAC,KAAK,CAAC,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;4BACtE,MAAM,KAAK,CAAC;wBACd,CAAC;wBACD,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAChF,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;YACL,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,cAAc,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,cAAc,EAAE,CAAC;gBAC7C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC3B,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;AACJ,CAAC,CAAC,EAAE,CAAC;AAEL,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,OAAqC,EAAiB,EAAE;IAClG,oBAAoB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrC,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,SAAS,IAAI,mBAAmB,EAAE,WAAW,CAAC,CAAC;IAC5F,MAAM,oBAAoB,GAAG,GAAG,SAAS,cAAc,CAAC;IACxD,MAAM,eAAe,GAAG,GAAG,SAAS,UAAU,CAAC;IAC/C,MAAM,cAAc,GAAG,GAAG,SAAS,SAAS,CAAC;IAC7C,MAAM,gBAAgB,GAAG,GAAG,SAAS,eAAe,CAAC;IACrD,MAAM,KAAK,GAAG,CAAC,KAAa,EAAE,KAAuB,EAAE,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;IAC/G,MAAM,SAAS,GAAG;iCACa,SAAS;;;;;GAKvC,CAAC;IACF,MAAM,oBAAoB,GAAG;iCACE,oBAAoB;;;;;GAKlD,CAAC;IACF,MAAM,eAAe,GAAG;iCACO,eAAe;;;;;GAK7C,CAAC;IACF,MAAM,cAAc,GAAG,8BAA8B,cAAc;;;;;IAKjE,CAAC;IACH,MAAM,gBAAgB,GAAG,8BAA8B,gBAAgB;;;;;;;IAOrE,CAAC;IACH,MAAM,gBAAgB,GAAG;iCACM,SAAS,mBAAmB,SAAS;iCACrC,eAAe,kBAAkB,eAAe;iCAChD,cAAc,kBAAkB,cAAc;iCAC9C,gBAAgB,eAAe,gBAAgB;GAC7E,CAAC;IACF,MAAM,eAAe,GAAG,KAAK,IAAI,EAAE;QACjC,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;QAChE,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,oBAAoB,EAAE,oBAAoB,CAAC,CAAC;QACtF,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,eAAe,EAAE,eAAe,CAAC,CAAC;QAC5E,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC;QAC1E,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;QAC9E,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,UAAU,EAAE,gBAAgB,CAAC,CAAC;IACtF,CAAC,CAAC;IAEF,OAAO;QACL,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK;YACrB,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,0BAA0B,SAAS,oBAAoB,EACvD,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CACtB,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAA+B,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACrJ,OAAO,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,CAAC;QACD,KAAK,CAAC,oBAAoB,CAAC,cAAc,EAAE,KAAK;YAC9C,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC;gBACQ,SAAS;sBACH,oBAAoB;yCACD,EACjC,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAC/B,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAA+B,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACrJ,OAAO,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,CAAC;QACD,KAAK,CAAC,iBAAiB,CAAC,WAAW,EAAE,KAAK;YACxC,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC;gBACQ,SAAS;sBACH,eAAe;0CACK,EAClC,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,CAC5B,CAAC;YACF,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACjC,MAAM,KAAK,GAAI,cAAc,CAAC,GAAG,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAA+B,IAAI,SAAS,CAAC;gBAC3G,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,CAAC,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,mBAAmB,CAAC,KAAK;YAC7B,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,sBAAsB,CAAC,EAAE,GAAG,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACrF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB,eAAe,SAAS;;wCAEQ,EAChC,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,UAAU,EAAE,SAAS,CAAC,CACxD,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACtC,eAAe,oBAAoB;;;0BAGjB,EAClB,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,SAAS,CAAC,CAChF,CAAC;YACF,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YACxE,IAAI,YAAY,KAAK,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;gBACpD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;oBAC3B,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB,eAAe,eAAe;;;;sDAIY,EAC1C,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,SAAS,CAAC,CAClF,CAAC;gBACJ,CAAC;gBACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;YAC9C,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACzC;gBACQ,SAAS;sBACH,oBAAoB;yCACD,EACjC,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CACrC,CAAC;YACF,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpC,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAA8B;gBAC5F,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,aAAa,CAAC,sDAAsD,CAAC,CAAC;YAClF,CAAC;YACD,IAAI,aAAa,CAAC,KAAK,KAAK,UAAU,CAAC,KAAK,EAAE,CAAC;gBAC7C,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,SAAS,oBAAoB,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;YAC7G,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,sBAAsB,CAAC,aAAa,CAAC,EAAE,CAAC;QAC1E,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW;YAC3B,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,EAAE,GAAG,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC9D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,UAAU,CAAC,cAAc,EAAE,CAAC;gBAC9B,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACtC,sBAAsB,oBAAoB,6BAA6B,EACvE,CAAC,KAAK,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAC1C,CAAC;gBACF,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;gBACtE,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,UAAU,KAAK,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;oBACpF,MAAM,IAAI,aAAa,CAAC,yCAAyC,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;YAED,IAAI,WAAW,EAAE,gBAAgB,KAAK,SAAS,EAAE,CAAC;gBAChD,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB,eAAe,SAAS;;;;oDAIkB,EAC1C,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,CACpD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACtC,eAAe,SAAS;;;;;6BAKL,SAAS;4BACV,EAClB,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,CAAC,gBAAgB,CAAC,CAClF,CAAC;gBACF,IAAI,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,KAAK,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;oBAC1F,MAAM,IAAI,aAAa,CAAC,kCAAkC,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC;YACD,IAAI,UAAU,CAAC,cAAc,EAAE,CAAC;gBAC9B,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB,eAAe,oBAAoB;;mDAEM,EACzC,CAAC,KAAK,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,SAAS,CAAC,CACrF,CAAC;YACJ,CAAC;YACD,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,eAAe,oBAAoB,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;YACjH,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC3B,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB,eAAe,eAAe;;;;oDAIY,EAC1C,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,SAAS,CAAC,CAClF,CAAC;YACJ,CAAC;QACH,CAAC;QACD,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK;YACvB,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAChC,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB;yBACiB,SAAS;;yBAET,oBAAoB;;yBAEpB,eAAe;;yBAEf,cAAc;;uBAEhB,gBAAgB,qBAAqB,EACpD,CAAC,GAAG,CAAC,CACN,CAAC;QACJ,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK;YAC3B,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,0BAA0B,SAAS,qCAAqC,EACxE,CAAC,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,CAC5B,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACzC,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAA8B,CAAC;gBAC5F,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,CAAC,CAAC,CAAC;YACH,OAAO,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACzC,CAAC;QACD,KAAK,CAAC,aAAa,CAAC,SAAS,EAAE,KAAK;YAClC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,KAAK,EAAE,EAAE,KAAK,CAAiB,CAAC;YAC1J,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YAC9E,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAC3B,CAAC;QACD,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK;YAC3C,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACnC,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3C,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAChC,MAAM,SAAS,GAAG,GAAG,GAAG,YAAY,CAAC,KAAK,CAAC;YAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,eAAe,cAAc;;uCAEE,SAAS;;;;iBAI/B,cAAc,oCAAoC,cAAc;2CACtC,EACnC,CAAC,GAAG,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,CACnD,CAAC;YACF,MAAM,KAAK,GAAG,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;YACtE,OAAO,KAAK,KAAK,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1G,CAAC;QACD,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK;YACzC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACnC,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3C,MAAM,SAAS,GAAG,GAAG,GAAG,YAAY,CAAC,KAAK,CAAC;YAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,UAAU,cAAc;;;4BAGJ,EACpB,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,CAC5D,CAAC;YACF,OAAO,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,KAAK,YAAY,CAAC,OAAO;gBACrF,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE;gBACrD,CAAC,CAAC,SAAS,CAAC;QAChB,CAAC;QACD,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK;YACtC,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,eAAe,cAAc,0DAA0D,EACvF,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,CAC/B,CAAC;YACF,OAAO,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,KAAK,OAAO,CAAC;QAC7E,CAAC;QACD,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK;YACzC,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,0BAA0B,gBAAgB,2CAA2C,EACrF,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,UAAU,CAAC,CAClC,CAAC;YACF,MAAM,KAAK,GAAG,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;YAC1E,OAAO,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAkC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChH,CAAC;QACD,KAAK,CAAC,iBAAiB,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK;YAC9C,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QACvD,CAAC;QACD,KAAK,CAAC,aAAa,CAAC,KAAK,EAAE,KAAK;YAC9B,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,0BAA0B,gBAAgB,0DAA0D,EACpG,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CACtB,CAAC;YACF,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACjC,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;gBAC/D,OAAO,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAkC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3G,CAAC,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,cAAc;YACtC,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,IAAI,GAAG,gBAAgB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;YACrD,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,MAAM,GAAG,cAAc,EAAE,gBAAgB,KAAK,SAAS;gBAC3D,CAAC,CAAC,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CAC1B,eAAe,gBAAgB;;;;;;8BAMX,EACpB,CAAC,GAAG,EAAE,KAAK,CAAC,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,CAC7D;gBACD,CAAC,CAAC,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CAC1B,UAAU,gBAAgB;;;;;8BAKN,EACpB,CAAC,GAAG,EAAE,KAAK,CAAC,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,gBAAgB,CAAC,CAC9F,CAAC;YACJ,IAAI,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,KAAK,SAAS,EAAE,CAAC;gBAC/D,MAAM,IAAI,aAAa,CAAC,4CAA4C,CAAC,CAAC;YACxE,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,CAAC,kBAAkB,CAAC,KAAK;YAC5B,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,eAAe,gBAAgB;;uCAEA,SAAS;;8BAElB,EACtB,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAC1E,CAAC;YACF,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;YAC5E,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YAClF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACvF,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,eAAe,CAAC,gDAAgD,CAAC,CAAC;YAC3F,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;QAC7C,CAAC;QACD,KAAK,CAAC,qBAAqB,CAAC,KAAK,EAAE,cAAc;YAC/C,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,EAAE,cAAc,CAAuC,CAAC;QAC7J,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,OAAwC,EAAoB,EAAE;IAC3G,oBAAoB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrC,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,SAAS,IAAI,qBAAqB,EAAE,WAAW,CAAC,CAAC;IAC9F,MAAM,MAAM,GAAG,CAAC,OAA2B,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,IAAI,gBAAgB,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACrJ,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,qBAAqB,CAAC;IACvE,MAAM,SAAS,GAAG;iCACa,SAAS;;;;;GAKvC,CAAC;IAEF,OAAO;QACL,KAAK,CAAC,IAAI,CAAC,OAAO;YAChB,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACvC,6BAA6B,SAAS,wBAAwB,EAC9D,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAClB,CAAC;YACF,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;gBACnB,CAAC,CAAC,CAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,eAAe,EAAE,cAAc,CAAC,CAAgC,IAAI,EAAE,CAAC;gBAC3G,CAAC,CAAC,EAAE,CAAC;QACT,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,OAAO;YAChB,MAAM,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CACxB,eAAe,SAAS;;;;kDAIkB,EAC1C,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAC7D,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC,CAAC"}