@dbx-tools/appkit-memory 0.0.0-stage → 0.9.131

package/lib/src/plugin.js

@@ -0,0 +1,196 @@
+/**
+ * AppKit memory plugin with Databricks-first tools and lazy Graphiti fallback.
+ *
+ * @module
+ */
+import { getExecutionContext, toPlugin } from "@databricks/appkit";
+import { defineTool, executeFromRegistry, } from "@databricks/appkit/beta";
+import { appkit as dbxAppkit, toolkitEntries, ToolRegistryPlugin, } from "@dbx-tools/appkit";
+import { errorUtils, log } from "@dbx-tools/shared-core";
+import { RecallMemoryInputSchema, RememberMemoryInputSchema, } from "./_contracts.js";
+import { MEMORY_CONFIG_SCHEMA, resolveMemoryPluginConfig, } from "./config.js";
+import { DatabricksMemoryRuntime } from "./databricks/_runtime.js";
+/** AppKit memory plugin with Databricks managed memory and Graphiti fallback. */
+export class MemoryPlugin extends ToolRegistryPlugin {
+    static phase = "deferred";
+    static manifest = {
+        name: "memory",
+        displayName: "Memory",
+        description: "Publishes Databricks-first durable memory tools with a lazy Graphiti fallback.",
+        stability: "beta",
+        resources: { required: [], optional: [] },
+        config: { schema: MEMORY_CONFIG_SCHEMA },
+    };
+    logger = log.logger(this);
+    resolved;
+    databricks;
+    graphiti;
+    graphitiStarting;
+    graphitiStartupAbort = new AbortController();
+    closed = false;
+    get toolRegistry() {
+        return this.registry();
+    }
+    async setup() {
+        const resolved = resolveMemoryPluginConfig(this.config);
+        if (resolved.saveTurns === true && resolved.graphiti === false) {
+            throw new Error("AppKit memory saveTurns requires the Graphiti fallback");
+        }
+        this.resolved = resolved;
+        if (resolved.databricks) {
+            this.databricks = new DatabricksMemoryRuntime(getExecutionContext().client, resolved.databricks);
+        }
+        let mastra = false;
+        const context = this.context;
+        const shouldAttachTurnSaver = resolved.saveTurns !== false &&
+            resolved.graphiti !== false &&
+            context?.hasPlugin("mastra") === true;
+        if (shouldAttachTurnSaver && context) {
+            const { attachMastraTurnSaver } = await import("./_mastra.js");
+            mastra = await attachMastraTurnSaver(context, (turn, signal) => this.saveTurn(turn, signal));
+        }
+        if (resolved.saveTurns === true && !mastra) {
+            throw new Error("AppKit memory saveTurns requires a registered Mastra plugin");
+        }
+        this.logger.info("ready", {
+            databricks: resolved.databricks !== false,
+            graphiti: resolved.graphiti !== false ? "lazy" : false,
+            mastraTurnSaver: mastra,
+        });
+    }
+    async toolkit(options) {
+        return toolkitEntries.entries("memory", this.getAgentTools(), {
+            ...options,
+            prefix: options?.prefix ?? "",
+        });
+    }
+    executeAgentTool(name, args, signal, context) {
+        return executeFromRegistry(this.registry(context?.resourceId), name, args, signal);
+    }
+    abortActiveOperations() {
+        super.abortActiveOperations();
+        this.graphitiStartupAbort.abort(new Error("AppKit memory is shutting down"));
+    }
+    async shutdown() {
+        this.closed = true;
+        this.graphitiStartupAbort.abort(new Error("AppKit memory is shutting down"));
+        const starting = this.graphitiStarting;
+        const backend = this.graphiti;
+        this.graphitiStarting = undefined;
+        this.graphiti = undefined;
+        await starting?.catch(() => undefined);
+        await backend?.shutdown();
+    }
+    registry(resourceId) {
+        return {
+            remember: defineTool({
+                description: "Save a durable user preference, fact, decision, or instruction for future conversations.",
+                schema: RememberMemoryInputSchema,
+                annotations: { effect: "write", requiresUserContext: true },
+                autoInheritable: false,
+                execute: (input, signal) => this.remember(this.actorId(resourceId), input, signal),
+            }),
+            recall_memory: defineTool({
+                description: "Search durable actor-scoped memory for facts and preferences relevant to the current task.",
+                schema: RecallMemoryInputSchema,
+                annotations: { effect: "read", requiresUserContext: true },
+                autoInheritable: false,
+                execute: (input, signal) => this.recall(this.actorId(resourceId), input, signal),
+            }),
+        };
+    }
+    async remember(actorId, input, signal) {
+        const databricks = await this.databricksBackend(actorId, signal);
+        if (databricks) {
+            try {
+                return await databricks.remember(actorId, input, signal);
+            }
+            catch (error) {
+                if (!this.shouldFallbackFromDatabricks(error))
+                    throw error;
+            }
+        }
+        return (await this.graphitiBackend()).remember(actorId, input, signal);
+    }
+    async recall(actorId, input, signal) {
+        const parsed = RecallMemoryInputSchema.parse(input);
+        const databricks = await this.databricksBackend(actorId, signal);
+        if (databricks) {
+            try {
+                return await databricks.recall(actorId, parsed, signal);
+            }
+            catch (error) {
+                if (!this.shouldFallbackFromDatabricks(error))
+                    throw error;
+            }
+        }
+        return (await this.graphitiBackend()).recall(actorId, parsed, signal);
+    }
+    async saveTurn(turn, signal) {
+        if (await this.databricksBackend(turn.actorId, signal))
+            return;
+        await (await this.graphitiBackend()).saveTurn(turn, signal);
+    }
+    async databricksBackend(actorId, signal) {
+        const backend = this.databricks;
+        if (!backend)
+            return undefined;
+        if (await backend.available(actorId, signal))
+            return backend;
+        const configured = this.requireConfig().databricks;
+        if (configured && configured.mode === "required") {
+            throw new Error("Required Databricks managed memory is unavailable for the current actor");
+        }
+        return undefined;
+    }
+    shouldFallbackFromDatabricks(error) {
+        const configured = this.requireConfig().databricks;
+        return (configured !== false &&
+            configured.mode === "auto" &&
+            errorUtils.errorContext(error).hasStatusCode(403, 404));
+    }
+    async graphitiBackend() {
+        if (this.closed)
+            throw new Error("AppKit memory is closed");
+        if (this.graphiti?.running)
+            return this.graphiti;
+        if (this.graphiti) {
+            await this.graphiti.shutdown();
+            this.graphiti = undefined;
+        }
+        const options = this.requireConfig().graphiti;
+        if (!options)
+            throw new Error("Graphiti fallback is disabled");
+        this.graphitiStarting ??= import("./graphiti/_backend.js")
+            .then(({ startGraphitiBackend }) => startGraphitiBackend(options, this.graphitiStartupAbort.signal))
+            .then((backend) => {
+            if (this.closed) {
+                return backend.shutdown().then(() => {
+                    throw new Error("AppKit memory closed during Graphiti startup");
+                });
+            }
+            this.graphiti = backend;
+            return backend;
+        })
+            .finally(() => {
+            this.graphitiStarting = undefined;
+        });
+        return this.graphitiStarting;
+    }
+    actorId(resourceId) {
+        if (resourceId)
+            return resourceId;
+        const context = dbxAppkit.tryGetExecutionContext();
+        if (!context)
+            throw new Error("Memory tools require an AppKit execution context");
+        return "userId" in context ? context.userId : context.serviceUserId;
+    }
+    requireConfig() {
+        if (!this.resolved)
+            throw new Error("AppKit memory is not initialized");
+        return this.resolved;
+    }
+}
+/** AppKit registration factory for {@link MemoryPlugin}. */
+export const memory = toPlugin(MemoryPlugin);
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicGx1Z2luLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL3BsdWdpbi50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQTs7OztHQUlHO0FBRUgsT0FBTyxFQUFFLG1CQUFtQixFQUFFLFFBQVEsRUFBdUIsTUFBTSxvQkFBb0IsQ0FBQztBQUN4RixPQUFPLEVBQ0wsVUFBVSxFQUNWLG1CQUFtQixHQUlwQixNQUFNLHlCQUF5QixDQUFDO0FBQ2pDLE9BQU8sRUFDTCxNQUFNLElBQUksU0FBUyxFQUNuQixjQUFjLEVBQ2Qsa0JBQWtCLEdBRW5CLE1BQU0sbUJBQW1CLENBQUM7QUFDM0IsT0FBTyxFQUFFLFVBQVUsRUFBRSxHQUFHLEVBQUUsTUFBTSx3QkFBd0IsQ0FBQztBQUV6RCxPQUFPLEVBQ0wsdUJBQXVCLEVBQ3ZCLHlCQUF5QixHQU8xQixNQUFNLGlCQUFpQixDQUFDO0FBQ3pCLE9BQU8sRUFDTCxvQkFBb0IsRUFDcEIseUJBQXlCLEdBRzFCLE1BQU0sYUFBYSxDQUFDO0FBQ3JCLE9BQU8sRUFBRSx1QkFBdUIsRUFBRSxNQUFNLDBCQUEwQixDQUFDO0FBRW5FLGlGQUFpRjtBQUNqRixNQUFNLE9BQU8sWUFBYSxTQUFRLGtCQUFzQztJQUN0RSxNQUFNLENBQUMsS0FBSyxHQUFHLFVBQW1CLENBQUM7SUFDbkMsTUFBTSxDQUFDLFFBQVEsR0FBNkI7UUFDMUMsSUFBSSxFQUFFLFFBQVE7UUFDZCxXQUFXLEVBQUUsUUFBUTtRQUNyQixXQUFXLEVBQUUsZ0ZBQWdGO1FBQzdGLFNBQVMsRUFBRSxNQUFNO1FBQ2pCLFNBQVMsRUFBRSxFQUFFLFFBQVEsRUFBRSxFQUFFLEVBQUUsUUFBUSxFQUFFLEVBQUUsRUFBRTtRQUN6QyxNQUFNLEVBQUUsRUFBRSxNQUFNLEVBQUUsb0JBQW9CLEVBQUU7S0FDekMsQ0FBQztJQUVlLE1BQU0sR0FBRyxHQUFHLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxDQUFDO0lBQ25DLFFBQVEsQ0FBOEI7SUFDdEMsVUFBVSxDQUEyQjtJQUNyQyxRQUFRLENBQXlCO0lBQ2pDLGdCQUFnQixDQUFrQztJQUN6QyxvQkFBb0IsR0FBRyxJQUFJLGVBQWUsRUFBRSxDQUFDO0lBQ3RELE1BQU0sR0FBRyxLQUFLLENBQUM7SUFFdkIsSUFBYyxZQUFZO1FBQ3hCLE9BQU8sSUFBSSxDQUFDLFFBQVEsRUFBRSxDQUFDO0lBQ3pCLENBQUM7SUFFUSxLQUFLLENBQUMsS0FBSztRQUNsQixNQUFNLFFBQVEsR0FBRyx5QkFBeUIsQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLENBQUM7UUFDeEQsSUFBSSxRQUFRLENBQUMsU0FBUyxLQUFLLElBQUksSUFBSSxRQUFRLENBQUMsUUFBUSxLQUFLLEtBQUssRUFBRSxDQUFDO1lBQy9ELE1BQU0sSUFBSSxLQUFLLENBQUMsd0RBQXdELENBQUMsQ0FBQztRQUM1RSxDQUFDO1FBQ0QsSUFBSSxDQUFDLFFBQVEsR0FBRyxRQUFRLENBQUM7UUFDekIsSUFBSSxRQUFRLENBQUMsVUFBVSxFQUFFLENBQUM7WUFDeEIsSUFBSSxDQUFDLFVBQVUsR0FBRyxJQUFJLHVCQUF1QixDQUMzQyxtQkFBbUIsRUFBRSxDQUFDLE1BQU0sRUFDNUIsUUFBUSxDQUFDLFVBQVUsQ0FDcEIsQ0FBQztRQUNKLENBQUM7UUFFRCxJQUFJLE1BQU0sR0FBRyxLQUFLLENBQUM7UUFDbkIsTUFBTSxPQUFPLEdBQUcsSUFBSSxDQUFDLE9BQU8sQ0FBQztRQUM3QixNQUFNLHFCQUFxQixHQUN6QixRQUFRLENBQUMsU0FBUyxLQUFLLEtBQUs7WUFDNUIsUUFBUSxDQUFDLFFBQVEsS0FBSyxLQUFLO1lBQzNCLE9BQU8sRUFBRSxTQUFTLENBQUMsUUFBUSxDQUFDLEtBQUssSUFBSSxDQUFDO1FBQ3hDLElBQUkscUJBQXFCLElBQUksT0FBTyxFQUFFLENBQUM7WUFDckMsTUFBTSxFQUFFLHFCQUFxQixFQUFFLEdBQUcsTUFBTSxNQUFNLENBQUMsY0FBYyxDQUFDLENBQUM7WUFDL0QsTUFBTSxHQUFHLE1BQU0scUJBQXFCLENBQUMsT0FBTyxFQUFFLENBQUMsSUFBSSxFQUFFLE1BQU0sRUFBRSxFQUFFLENBQUMsSUFBSSxDQUFDLFFBQVEsQ0FBQyxJQUFJLEVBQUUsTUFBTSxDQUFDLENBQUMsQ0FBQztRQUMvRixDQUFDO1FBQ0QsSUFBSSxRQUFRLENBQUMsU0FBUyxLQUFLLElBQUksSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDO1lBQzNDLE1BQU0sSUFBSSxLQUFLLENBQUMsNkRBQTZELENBQUMsQ0FBQztRQUNqRixDQUFDO1FBQ0QsSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsT0FBTyxFQUFFO1lBQ3hCLFVBQVUsRUFBRSxRQUFRLENBQUMsVUFBVSxLQUFLLEtBQUs7WUFDekMsUUFBUSxFQUFFLFFBQVEsQ0FBQyxRQUFRLEtBQUssS0FBSyxDQUFDLENBQUMsQ0FBQyxNQUFNLENBQUMsQ0FBQyxDQUFDLEtBQUs7WUFDdEQsZUFBZSxFQUFFLE1BQU07U0FDeEIsQ0FBQyxDQUFDO0lBQ0wsQ0FBQztJQUVELEtBQUssQ0FBQyxPQUFPLENBQUMsT0FBd0I7UUFDcEMsT0FBTyxjQUFjLENBQUMsT0FBTyxDQUFDLFFBQVEsRUFBRSxJQUFJLENBQUMsYUFBYSxFQUFFLEVBQUU7WUFDNUQsR0FBRyxPQUFPO1lBQ1YsTUFBTSxFQUFFLE9BQU8sRUFBRSxNQUFNLElBQUksRUFBRTtTQUM5QixDQUFDLENBQUM7SUFDTCxDQUFDO0lBRVEsZ0JBQWdCLENBQ3ZCLElBQVksRUFDWixJQUFhLEVBQ2IsTUFBb0IsRUFDcEIsT0FBbUM7UUFFbkMsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsUUFBUSxDQUFDLE9BQU8sRUFBRSxVQUFVLENBQUMsRUFBRSxJQUFJLEVBQUUsSUFBSSxFQUFFLE1BQU0sQ0FBQyxDQUFDO0lBQ3JGLENBQUM7SUFFUSxxQkFBcUI7UUFDNUIsS0FBSyxDQUFDLHFCQUFxQixFQUFFLENBQUM7UUFDOUIsSUFBSSxDQUFDLG9CQUFvQixDQUFDLEtBQUssQ0FBQyxJQUFJLEtBQUssQ0FBQyxnQ0FBZ0MsQ0FBQyxDQUFDLENBQUM7SUFDL0UsQ0FBQztJQUVELEtBQUssQ0FBQyxRQUFRO1FBQ1osSUFBSSxDQUFDLE1BQU0sR0FBRyxJQUFJLENBQUM7UUFDbkIsSUFBSSxDQUFDLG9CQUFvQixDQUFDLEtBQUssQ0FBQyxJQUFJLEtBQUssQ0FBQyxnQ0FBZ0MsQ0FBQyxDQUFDLENBQUM7UUFDN0UsTUFBTSxRQUFRLEdBQUcsSUFBSSxDQUFDLGdCQUFnQixDQUFDO1FBQ3ZDLE1BQU0sT0FBTyxHQUFHLElBQUksQ0FBQyxRQUFRLENBQUM7UUFDOUIsSUFBSSxDQUFDLGdCQUFnQixHQUFHLFNBQVMsQ0FBQztRQUNsQyxJQUFJLENBQUMsUUFBUSxHQUFHLFNBQVMsQ0FBQztRQUMxQixNQUFNLFFBQVEsRUFBRSxLQUFLLENBQUMsR0FBRyxFQUFFLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDdkMsTUFBTSxPQUFPLEVBQUUsUUFBUSxFQUFFLENBQUM7SUFDNUIsQ0FBQztJQUVPLFFBQVEsQ0FBQyxVQUFtQjtRQUNsQyxPQUFPO1lBQ0wsUUFBUSxFQUFFLFVBQVUsQ0FBQztnQkFDbkIsV0FBVyxFQUNULDBGQUEwRjtnQkFDNUYsTUFBTSxFQUFFLHlCQUF5QjtnQkFDakMsV0FBVyxFQUFFLEVBQUUsTUFBTSxFQUFFLE9BQU8sRUFBRSxtQkFBbUIsRUFBRSxJQUFJLEVBQUU7Z0JBQzNELGVBQWUsRUFBRSxLQUFLO2dCQUN0QixPQUFPLEVBQUUsQ0FBQyxLQUFLLEVBQUUsTUFBTSxFQUFFLEVBQUUsQ0FBQyxJQUFJLENBQUMsUUFBUSxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsVUFBVSxDQUFDLEVBQUUsS0FBSyxFQUFFLE1BQU0sQ0FBQzthQUNuRixDQUFDO1lBQ0YsYUFBYSxFQUFFLFVBQVUsQ0FBQztnQkFDeEIsV0FBVyxFQUNULDRGQUE0RjtnQkFDOUYsTUFBTSxFQUFFLHVCQUF1QjtnQkFDL0IsV0FBVyxFQUFFLEVBQUUsTUFBTSxFQUFFLE1BQU0sRUFBRSxtQkFBbUIsRUFBRSxJQUFJLEVBQUU7Z0JBQzFELGVBQWUsRUFBRSxLQUFLO2dCQUN0QixPQUFPLEVBQUUsQ0FBQyxLQUFLLEVBQUUsTUFBTSxFQUFFLEVBQUUsQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsVUFBVSxDQUFDLEVBQUUsS0FBSyxFQUFFLE1BQU0sQ0FBQzthQUNqRixDQUFDO1NBQ0gsQ0FBQztJQUNKLENBQUM7SUFFTyxLQUFLLENBQUMsUUFBUSxDQUNwQixPQUFlLEVBQ2YsS0FBMEIsRUFDMUIsTUFBb0I7UUFFcEIsTUFBTSxVQUFVLEdBQUcsTUFBTSxJQUFJLENBQUMsaUJBQWlCLENBQUMsT0FBTyxFQUFFLE1BQU0sQ0FBQyxDQUFDO1FBQ2pFLElBQUksVUFBVSxFQUFFLENBQUM7WUFDZixJQUFJLENBQUM7Z0JBQ0gsT0FBTyxNQUFNLFVBQVUsQ0FBQyxRQUFRLENBQUMsT0FBTyxFQUFFLEtBQUssRUFBRSxNQUFNLENBQUMsQ0FBQztZQUMzRCxDQUFDO1lBQUMsT0FBTyxLQUFLLEVBQUUsQ0FBQztnQkFDZixJQUFJLENBQUMsSUFBSSxDQUFDLDRCQUE0QixDQUFDLEtBQUssQ0FBQztvQkFBRSxNQUFNLEtBQUssQ0FBQztZQUM3RCxDQUFDO1FBQ0gsQ0FBQztRQUNELE9BQU8sQ0FBQyxNQUFNLElBQUksQ0FBQyxlQUFlLEVBQUUsQ0FBQyxDQUFDLFFBQVEsQ0FBQyxPQUFPLEVBQUUsS0FBSyxFQUFFLE1BQU0sQ0FBQyxDQUFDO0lBQ3pFLENBQUM7SUFFTyxLQUFLLENBQUMsTUFBTSxDQUNsQixPQUFlLEVBQ2YsS0FBd0IsRUFDeEIsTUFBb0I7UUFFcEIsTUFBTSxNQUFNLEdBQUcsdUJBQXVCLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ3BELE1BQU0sVUFBVSxHQUFHLE1BQU0sSUFBSSxDQUFDLGlCQUFpQixDQUFDLE9BQU8sRUFBRSxNQUFNLENBQUMsQ0FBQztRQUNqRSxJQUFJLFVBQVUsRUFBRSxDQUFDO1lBQ2YsSUFBSSxDQUFDO2dCQUNILE9BQU8sTUFBTSxVQUFVLENBQUMsTUFBTSxDQUFDLE9BQU8sRUFBRSxNQUFNLEVBQUUsTUFBTSxDQUFDLENBQUM7WUFDMUQsQ0FBQztZQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7Z0JBQ2YsSUFBSSxDQUFDLElBQUksQ0FBQyw0QkFBNEIsQ0FBQyxLQUFLLENBQUM7b0JBQUUsTUFBTSxLQUFLLENBQUM7WUFDN0QsQ0FBQztRQUNILENBQUM7UUFDRCxPQUFPLENBQUMsTUFBTSxJQUFJLENBQUMsZUFBZSxFQUFFLENBQUMsQ0FBQyxNQUFNLENBQUMsT0FBTyxFQUFFLE1BQU0sRUFBRSxNQUFNLENBQUMsQ0FBQztJQUN4RSxDQUFDO0lBRU8sS0FBSyxDQUFDLFFBQVEsQ0FBQyxJQUFnQixFQUFFLE1BQW9CO1FBQzNELElBQUksTUFBTSxJQUFJLENBQUMsaUJBQWlCLENBQUMsSUFBSSxDQUFDLE9BQU8sRUFBRSxNQUFNLENBQUM7WUFBRSxPQUFPO1FBQy9ELE1BQU0sQ0FBQyxNQUFNLElBQUksQ0FBQyxlQUFlLEVBQUUsQ0FBQyxDQUFDLFFBQVEsQ0FBQyxJQUFJLEVBQUUsTUFBTSxDQUFDLENBQUM7SUFDOUQsQ0FBQztJQUVPLEtBQUssQ0FBQyxpQkFBaUIsQ0FDN0IsT0FBZSxFQUNmLE1BQW9CO1FBRXBCLE1BQU0sT0FBTyxHQUFHLElBQUksQ0FBQyxVQUFVLENBQUM7UUFDaEMsSUFBSSxDQUFDLE9BQU87WUFBRSxPQUFPLFNBQVMsQ0FBQztRQUMvQixJQUFJLE1BQU0sT0FBTyxDQUFDLFNBQVMsQ0FBQyxPQUFPLEVBQUUsTUFBTSxDQUFDO1lBQUUsT0FBTyxPQUFPLENBQUM7UUFDN0QsTUFBTSxVQUFVLEdBQUcsSUFBSSxDQUFDLGFBQWEsRUFBRSxDQUFDLFVBQVUsQ0FBQztRQUNuRCxJQUFJLFVBQVUsSUFBSSxVQUFVLENBQUMsSUFBSSxLQUFLLFVBQVUsRUFBRSxDQUFDO1lBQ2pELE1BQU0sSUFBSSxLQUFLLENBQUMseUVBQXlFLENBQUMsQ0FBQztRQUM3RixDQUFDO1FBQ0QsT0FBTyxTQUFTLENBQUM7SUFDbkIsQ0FBQztJQUVPLDRCQUE0QixDQUFDLEtBQWM7UUFDakQsTUFBTSxVQUFVLEdBQUcsSUFBSSxDQUFDLGFBQWEsRUFBRSxDQUFDLFVBQVUsQ0FBQztRQUNuRCxPQUFPLENBQ0wsVUFBVSxLQUFLLEtBQUs7WUFDcEIsVUFBVSxDQUFDLElBQUksS0FBSyxNQUFNO1lBQzFCLFVBQVUsQ0FBQyxZQUFZLENBQUMsS0FBSyxDQUFDLENBQUMsYUFBYSxDQUFDLEdBQUcsRUFBRSxHQUFHLENBQUMsQ0FDdkQsQ0FBQztJQUNKLENBQUM7SUFFTyxLQUFLLENBQUMsZUFBZTtRQUMzQixJQUFJLElBQUksQ0FBQyxNQUFNO1lBQUUsTUFBTSxJQUFJLEtBQUssQ0FBQyx5QkFBeUIsQ0FBQyxDQUFDO1FBQzVELElBQUksSUFBSSxDQUFDLFFBQVEsRUFBRSxPQUFPO1lBQUUsT0FBTyxJQUFJLENBQUMsUUFBUSxDQUFDO1FBQ2pELElBQUksSUFBSSxDQUFDLFFBQVEsRUFBRSxDQUFDO1lBQ2xCLE1BQU0sSUFBSSxDQUFDLFFBQVEsQ0FBQyxRQUFRLEVBQUUsQ0FBQztZQUMvQixJQUFJLENBQUMsUUFBUSxHQUFHLFNBQVMsQ0FBQztRQUM1QixDQUFDO1FBQ0QsTUFBTSxPQUFPLEdBQUcsSUFBSSxDQUFDLGFBQWEsRUFBRSxDQUFDLFFBQVEsQ0FBQztRQUM5QyxJQUFJLENBQUMsT0FBTztZQUFFLE1BQU0sSUFBSSxLQUFLLENBQUMsK0JBQStCLENBQUMsQ0FBQztRQUMvRCxJQUFJLENBQUMsZ0JBQWdCLEtBQUssTUFBTSxDQUFDLHdCQUF3QixDQUFDO2FBQ3ZELElBQUksQ0FBQyxDQUFDLEVBQUUsb0JBQW9CLEVBQUUsRUFBRSxFQUFFLENBQ2pDLG9CQUFvQixDQUFDLE9BQU8sRUFBRSxJQUFJLENBQUMsb0JBQW9CLENBQUMsTUFBTSxDQUFDLENBQ2hFO2FBQ0EsSUFBSSxDQUFDLENBQUMsT0FBTyxFQUFFLEVBQUU7WUFDaEIsSUFBSSxJQUFJLENBQUMsTUFBTSxFQUFFLENBQUM7Z0JBQ2hCLE9BQU8sT0FBTyxDQUFDLFFBQVEsRUFBRSxDQUFDLElBQUksQ0FBQyxHQUFHLEVBQUU7b0JBQ2xDLE1BQU0sSUFBSSxLQUFLLENBQUMsOENBQThDLENBQUMsQ0FBQztnQkFDbEUsQ0FBQyxDQUFDLENBQUM7WUFDTCxDQUFDO1lBQ0QsSUFBSSxDQUFDLFFBQVEsR0FBRyxPQUFPLENBQUM7WUFDeEIsT0FBTyxPQUFPLENBQUM7UUFDakIsQ0FBQyxDQUFDO2FBQ0QsT0FBTyxDQUFDLEdBQUcsRUFBRTtZQUNaLElBQUksQ0FBQyxnQkFBZ0IsR0FBRyxTQUFTLENBQUM7UUFDcEMsQ0FBQyxDQUFDLENBQUM7UUFDTCxPQUFPLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQztJQUMvQixDQUFDO0lBRU8sT0FBTyxDQUFDLFVBQW1CO1FBQ2pDLElBQUksVUFBVTtZQUFFLE9BQU8sVUFBVSxDQUFDO1FBQ2xDLE1BQU0sT0FBTyxHQUFHLFNBQVMsQ0FBQyxzQkFBc0IsRUFBRSxDQUFDO1FBQ25ELElBQUksQ0FBQyxPQUFPO1lBQUUsTUFBTSxJQUFJLEtBQUssQ0FBQyxrREFBa0QsQ0FBQyxDQUFDO1FBQ2xGLE9BQU8sUUFBUSxJQUFJLE9BQU8sQ0FBQyxDQUFDLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUFDLENBQUMsT0FBTyxDQUFDLGFBQWEsQ0FBQztJQUN0RSxDQUFDO0lBRU8sYUFBYTtRQUNuQixJQUFJLENBQUMsSUFBSSxDQUFDLFFBQVE7WUFBRSxNQUFNLElBQUksS0FBSyxDQUFDLGtDQUFrQyxDQUFDLENBQUM7UUFDeEUsT0FBTyxJQUFJLENBQUMsUUFBUSxDQUFDO0lBQ3ZCLENBQUM7O0FBR0gsNERBQTREO0FBQzVELE1BQU0sQ0FBQyxNQUFNLE1BQU0sR0FBRyxRQUFRLENBQUMsWUFBWSxDQUFDLENBQUMiLCJzb3VyY2VzQ29udGVudCI6WyIvKipcbiAqIEFwcEtpdCBtZW1vcnkgcGx1Z2luIHdpdGggRGF0YWJyaWNrcy1maXJzdCB0b29scyBhbmQgbGF6eSBHcmFwaGl0aSBmYWxsYmFjay5cbiAqXG4gKiBAbW9kdWxlXG4gKi9cblxuaW1wb3J0IHsgZ2V0RXhlY3V0aW9uQ29udGV4dCwgdG9QbHVnaW4sIHR5cGUgUGx1Z2luTWFuaWZlc3QgfSBmcm9tIFwiQGRhdGFicmlja3MvYXBwa2l0XCI7XG5pbXBvcnQge1xuICBkZWZpbmVUb29sLFxuICBleGVjdXRlRnJvbVJlZ2lzdHJ5LFxuICB0eXBlIFRvb2xraXRFbnRyeSxcbiAgdHlwZSBUb29sa2l0T3B0aW9ucyxcbiAgdHlwZSBUb29sUmVnaXN0cnksXG59IGZyb20gXCJAZGF0YWJyaWNrcy9hcHBraXQvYmV0YVwiO1xuaW1wb3J0IHtcbiAgYXBwa2l0IGFzIGRieEFwcGtpdCxcbiAgdG9vbGtpdEVudHJpZXMsXG4gIFRvb2xSZWdpc3RyeVBsdWdpbixcbiAgdHlwZSBBZ2VudFRvb2xFeGVjdXRpb25Db250ZXh0LFxufSBmcm9tIFwiQGRieC10b29scy9hcHBraXRcIjtcbmltcG9ydCB7IGVycm9yVXRpbHMsIGxvZyB9IGZyb20gXCJAZGJ4LXRvb2xzL3NoYXJlZC1jb3JlXCI7XG5cbmltcG9ydCB7XG4gIFJlY2FsbE1lbW9yeUlucHV0U2NoZW1hLFxuICBSZW1lbWJlck1lbW9yeUlucHV0U2NoZW1hLFxuICB0eXBlIEdyYXBoaXRpTWVtb3J5QmFja2VuZCxcbiAgdHlwZSBNZW1vcnlUdXJuLFxuICB0eXBlIFJlY2FsbE1lbW9yeUlucHV0LFxuICB0eXBlIFJlY2FsbE1lbW9yeVJlc3VsdCxcbiAgdHlwZSBSZW1lbWJlck1lbW9yeUlucHV0LFxuICB0eXBlIFJlbWVtYmVyTWVtb3J5UmVzdWx0LFxufSBmcm9tIFwiLi9fY29udHJhY3RzLnRzXCI7XG5pbXBvcnQge1xuICBNRU1PUllfQ09ORklHX1NDSEVNQSxcbiAgcmVzb2x2ZU1lbW9yeVBsdWdpbkNvbmZpZyxcbiAgdHlwZSBNZW1vcnlQbHVnaW5Db25maWcsXG4gIHR5cGUgUmVzb2x2ZWRNZW1vcnlQbHVnaW5Db25maWcsXG59IGZyb20gXCIuL2NvbmZpZy50c1wiO1xuaW1wb3J0IHsgRGF0YWJyaWNrc01lbW9yeVJ1bnRpbWUgfSBmcm9tIFwiLi9kYXRhYnJpY2tzL19ydW50aW1lLnRzXCI7XG5cbi8qKiBBcHBLaXQgbWVtb3J5IHBsdWdpbiB3aXRoIERhdGFicmlja3MgbWFuYWdlZCBtZW1vcnkgYW5kIEdyYXBoaXRpIGZhbGxiYWNrLiAqL1xuZXhwb3J0IGNsYXNzIE1lbW9yeVBsdWdpbiBleHRlbmRzIFRvb2xSZWdpc3RyeVBsdWdpbjxNZW1vcnlQbHVnaW5Db25maWc+IHtcbiAgc3RhdGljIHBoYXNlID0gXCJkZWZlcnJlZFwiIGFzIGNvbnN0O1xuICBzdGF0aWMgbWFuaWZlc3Q6IFBsdWdpbk1hbmlmZXN0PFwibWVtb3J5XCI+ID0ge1xuICAgIG5hbWU6IFwibWVtb3J5XCIsXG4gICAgZGlzcGxheU5hbWU6IFwiTWVtb3J5XCIsXG4gICAgZGVzY3JpcHRpb246IFwiUHVibGlzaGVzIERhdGFicmlja3MtZmlyc3QgZHVyYWJsZSBtZW1vcnkgdG9vbHMgd2l0aCBhIGxhenkgR3JhcGhpdGkgZmFsbGJhY2suXCIsXG4gICAgc3RhYmlsaXR5OiBcImJldGFcIixcbiAgICByZXNvdXJjZXM6IHsgcmVxdWlyZWQ6IFtdLCBvcHRpb25hbDogW10gfSxcbiAgICBjb25maWc6IHsgc2NoZW1hOiBNRU1PUllfQ09ORklHX1NDSEVNQSB9LFxuICB9O1xuXG4gIHByaXZhdGUgcmVhZG9ubHkgbG9nZ2VyID0gbG9nLmxvZ2dlcih0aGlzKTtcbiAgcHJpdmF0ZSByZXNvbHZlZD86IFJlc29sdmVkTWVtb3J5UGx1Z2luQ29uZmlnO1xuICBwcml2YXRlIGRhdGFicmlja3M/OiBEYXRhYnJpY2tzTWVtb3J5UnVudGltZTtcbiAgcHJpdmF0ZSBncmFwaGl0aT86IEdyYXBoaXRpTWVtb3J5QmFja2VuZDtcbiAgcHJpdmF0ZSBncmFwaGl0aVN0YXJ0aW5nPzogUHJvbWlzZTxHcmFwaGl0aU1lbW9yeUJhY2tlbmQ+O1xuICBwcml2YXRlIHJlYWRvbmx5IGdyYXBoaXRpU3RhcnR1cEFib3J0ID0gbmV3IEFib3J0Q29udHJvbGxlcigpO1xuICBwcml2YXRlIGNsb3NlZCA9IGZhbHNlO1xuXG4gIHByb3RlY3RlZCBnZXQgdG9vbFJlZ2lzdHJ5KCk6IFRvb2xSZWdpc3RyeSB7XG4gICAgcmV0dXJuIHRoaXMucmVnaXN0cnkoKTtcbiAgfVxuXG4gIG92ZXJyaWRlIGFzeW5jIHNldHVwKCk6IFByb21pc2U8dm9pZD4ge1xuICAgIGNvbnN0IHJlc29sdmVkID0gcmVzb2x2ZU1lbW9yeVBsdWdpbkNvbmZpZyh0aGlzLmNvbmZpZyk7XG4gICAgaWYgKHJlc29sdmVkLnNhdmVUdXJucyA9PT0gdHJ1ZSAmJiByZXNvbHZlZC5ncmFwaGl0aSA9PT0gZmFsc2UpIHtcbiAgICAgIHRocm93IG5ldyBFcnJvcihcIkFwcEtpdCBtZW1vcnkgc2F2ZVR1cm5zIHJlcXVpcmVzIHRoZSBHcmFwaGl0aSBmYWxsYmFja1wiKTtcbiAgICB9XG4gICAgdGhpcy5yZXNvbHZlZCA9IHJlc29sdmVkO1xuICAgIGlmIChyZXNvbHZlZC5kYXRhYnJpY2tzKSB7XG4gICAgICB0aGlzLmRhdGFicmlja3MgPSBuZXcgRGF0YWJyaWNrc01lbW9yeVJ1bnRpbWUoXG4gICAgICAgIGdldEV4ZWN1dGlvbkNvbnRleHQoKS5jbGllbnQsXG4gICAgICAgIHJlc29sdmVkLmRhdGFicmlja3MsXG4gICAgICApO1xuICAgIH1cblxuICAgIGxldCBtYXN0cmEgPSBmYWxzZTtcbiAgICBjb25zdCBjb250ZXh0ID0gdGhpcy5jb250ZXh0O1xuICAgIGNvbnN0IHNob3VsZEF0dGFjaFR1cm5TYXZlciA9XG4gICAgICByZXNvbHZlZC5zYXZlVHVybnMgIT09IGZhbHNlICYmXG4gICAgICByZXNvbHZlZC5ncmFwaGl0aSAhPT0gZmFsc2UgJiZcbiAgICAgIGNvbnRleHQ/Lmhhc1BsdWdpbihcIm1hc3RyYVwiKSA9PT0gdHJ1ZTtcbiAgICBpZiAoc2hvdWxkQXR0YWNoVHVyblNhdmVyICYmIGNvbnRleHQpIHtcbiAgICAgIGNvbnN0IHsgYXR0YWNoTWFzdHJhVHVyblNhdmVyIH0gPSBhd2FpdCBpbXBvcnQoXCIuL19tYXN0cmEudHNcIik7XG4gICAgICBtYXN0cmEgPSBhd2FpdCBhdHRhY2hNYXN0cmFUdXJuU2F2ZXIoY29udGV4dCwgKHR1cm4sIHNpZ25hbCkgPT4gdGhpcy5zYXZlVHVybih0dXJuLCBzaWduYWwpKTtcbiAgICB9XG4gICAgaWYgKHJlc29sdmVkLnNhdmVUdXJucyA9PT0gdHJ1ZSAmJiAhbWFzdHJhKSB7XG4gICAgICB0aHJvdyBuZXcgRXJyb3IoXCJBcHBLaXQgbWVtb3J5IHNhdmVUdXJucyByZXF1aXJlcyBhIHJlZ2lzdGVyZWQgTWFzdHJhIHBsdWdpblwiKTtcbiAgICB9XG4gICAgdGhpcy5sb2dnZXIuaW5mbyhcInJlYWR5XCIsIHtcbiAgICAgIGRhdGFicmlja3M6IHJlc29sdmVkLmRhdGFicmlja3MgIT09IGZhbHNlLFxuICAgICAgZ3JhcGhpdGk6IHJlc29sdmVkLmdyYXBoaXRpICE9PSBmYWxzZSA/IFwibGF6eVwiIDogZmFsc2UsXG4gICAgICBtYXN0cmFUdXJuU2F2ZXI6IG1hc3RyYSxcbiAgICB9KTtcbiAgfVxuXG4gIGFzeW5jIHRvb2xraXQob3B0aW9ucz86IFRvb2xraXRPcHRpb25zKTogUHJvbWlzZTxSZWNvcmQ8c3RyaW5nLCBUb29sa2l0RW50cnk+PiB7XG4gICAgcmV0dXJuIHRvb2xraXRFbnRyaWVzLmVudHJpZXMoXCJtZW1vcnlcIiwgdGhpcy5nZXRBZ2VudFRvb2xzKCksIHtcbiAgICAgIC4uLm9wdGlvbnMsXG4gICAgICBwcmVmaXg6IG9wdGlvbnM/LnByZWZpeCA/PyBcIlwiLFxuICAgIH0pO1xuICB9XG5cbiAgb3ZlcnJpZGUgZXhlY3V0ZUFnZW50VG9vbChcbiAgICBuYW1lOiBzdHJpbmcsXG4gICAgYXJnczogdW5rbm93bixcbiAgICBzaWduYWw/OiBBYm9ydFNpZ25hbCxcbiAgICBjb250ZXh0PzogQWdlbnRUb29sRXhlY3V0aW9uQ29udGV4dCxcbiAgKTogUHJvbWlzZTx1bmtub3duPiB7XG4gICAgcmV0dXJuIGV4ZWN1dGVGcm9tUmVnaXN0cnkodGhpcy5yZWdpc3RyeShjb250ZXh0Py5yZXNvdXJjZUlkKSwgbmFtZSwgYXJncywgc2lnbmFsKTtcbiAgfVxuXG4gIG92ZXJyaWRlIGFib3J0QWN0aXZlT3BlcmF0aW9ucygpOiB2b2lkIHtcbiAgICBzdXBlci5hYm9ydEFjdGl2ZU9wZXJhdGlvbnMoKTtcbiAgICB0aGlzLmdyYXBoaXRpU3RhcnR1cEFib3J0LmFib3J0KG5ldyBFcnJvcihcIkFwcEtpdCBtZW1vcnkgaXMgc2h1dHRpbmcgZG93blwiKSk7XG4gIH1cblxuICBhc3luYyBzaHV0ZG93bigpOiBQcm9taXNlPHZvaWQ+IHtcbiAgICB0aGlzLmNsb3NlZCA9IHRydWU7XG4gICAgdGhpcy5ncmFwaGl0aVN0YXJ0dXBBYm9ydC5hYm9ydChuZXcgRXJyb3IoXCJBcHBLaXQgbWVtb3J5IGlzIHNodXR0aW5nIGRvd25cIikpO1xuICAgIGNvbnN0IHN0YXJ0aW5nID0gdGhpcy5ncmFwaGl0aVN0YXJ0aW5nO1xuICAgIGNvbnN0IGJhY2tlbmQgPSB0aGlzLmdyYXBoaXRpO1xuICAgIHRoaXMuZ3JhcGhpdGlTdGFydGluZyA9IHVuZGVmaW5lZDtcbiAgICB0aGlzLmdyYXBoaXRpID0gdW5kZWZpbmVkO1xuICAgIGF3YWl0IHN0YXJ0aW5nPy5jYXRjaCgoKSA9PiB1bmRlZmluZWQpO1xuICAgIGF3YWl0IGJhY2tlbmQ/LnNodXRkb3duKCk7XG4gIH1cblxuICBwcml2YXRlIHJlZ2lzdHJ5KHJlc291cmNlSWQ/OiBzdHJpbmcpOiBUb29sUmVnaXN0cnkge1xuICAgIHJldHVybiB7XG4gICAgICByZW1lbWJlcjogZGVmaW5lVG9vbCh7XG4gICAgICAgIGRlc2NyaXB0aW9uOlxuICAgICAgICAgIFwiU2F2ZSBhIGR1cmFibGUgdXNlciBwcmVmZXJlbmNlLCBmYWN0LCBkZWNpc2lvbiwgb3IgaW5zdHJ1Y3Rpb24gZm9yIGZ1dHVyZSBjb252ZXJzYXRpb25zLlwiLFxuICAgICAgICBzY2hlbWE6IFJlbWVtYmVyTWVtb3J5SW5wdXRTY2hlbWEsXG4gICAgICAgIGFubm90YXRpb25zOiB7IGVmZmVjdDogXCJ3cml0ZVwiLCByZXF1aXJlc1VzZXJDb250ZXh0OiB0cnVlIH0sXG4gICAgICAgIGF1dG9Jbmhlcml0YWJsZTogZmFsc2UsXG4gICAgICAgIGV4ZWN1dGU6IChpbnB1dCwgc2lnbmFsKSA9PiB0aGlzLnJlbWVtYmVyKHRoaXMuYWN0b3JJZChyZXNvdXJjZUlkKSwgaW5wdXQsIHNpZ25hbCksXG4gICAgICB9KSxcbiAgICAgIHJlY2FsbF9tZW1vcnk6IGRlZmluZVRvb2woe1xuICAgICAgICBkZXNjcmlwdGlvbjpcbiAgICAgICAgICBcIlNlYXJjaCBkdXJhYmxlIGFjdG9yLXNjb3BlZCBtZW1vcnkgZm9yIGZhY3RzIGFuZCBwcmVmZXJlbmNlcyByZWxldmFudCB0byB0aGUgY3VycmVudCB0YXNrLlwiLFxuICAgICAgICBzY2hlbWE6IFJlY2FsbE1lbW9yeUlucHV0U2NoZW1hLFxuICAgICAgICBhbm5vdGF0aW9uczogeyBlZmZlY3Q6IFwicmVhZFwiLCByZXF1aXJlc1VzZXJDb250ZXh0OiB0cnVlIH0sXG4gICAgICAgIGF1dG9Jbmhlcml0YWJsZTogZmFsc2UsXG4gICAgICAgIGV4ZWN1dGU6IChpbnB1dCwgc2lnbmFsKSA9PiB0aGlzLnJlY2FsbCh0aGlzLmFjdG9ySWQocmVzb3VyY2VJZCksIGlucHV0LCBzaWduYWwpLFxuICAgICAgfSksXG4gICAgfTtcbiAgfVxuXG4gIHByaXZhdGUgYXN5bmMgcmVtZW1iZXIoXG4gICAgYWN0b3JJZDogc3RyaW5nLFxuICAgIGlucHV0OiBSZW1lbWJlck1lbW9yeUlucHV0LFxuICAgIHNpZ25hbD86IEFib3J0U2lnbmFsLFxuICApOiBQcm9taXNlPFJlbWVtYmVyTWVtb3J5UmVzdWx0PiB7XG4gICAgY29uc3QgZGF0YWJyaWNrcyA9IGF3YWl0IHRoaXMuZGF0YWJyaWNrc0JhY2tlbmQoYWN0b3JJZCwgc2lnbmFsKTtcbiAgICBpZiAoZGF0YWJyaWNrcykge1xuICAgICAgdHJ5IHtcbiAgICAgICAgcmV0dXJuIGF3YWl0IGRhdGFicmlja3MucmVtZW1iZXIoYWN0b3JJZCwgaW5wdXQsIHNpZ25hbCk7XG4gICAgICB9IGNhdGNoIChlcnJvcikge1xuICAgICAgICBpZiAoIXRoaXMuc2hvdWxkRmFsbGJhY2tGcm9tRGF0YWJyaWNrcyhlcnJvcikpIHRocm93IGVycm9yO1xuICAgICAgfVxuICAgIH1cbiAgICByZXR1cm4gKGF3YWl0IHRoaXMuZ3JhcGhpdGlCYWNrZW5kKCkpLnJlbWVtYmVyKGFjdG9ySWQsIGlucHV0LCBzaWduYWwpO1xuICB9XG5cbiAgcHJpdmF0ZSBhc3luYyByZWNhbGwoXG4gICAgYWN0b3JJZDogc3RyaW5nLFxuICAgIGlucHV0OiBSZWNhbGxNZW1vcnlJbnB1dCxcbiAgICBzaWduYWw/OiBBYm9ydFNpZ25hbCxcbiAgKTogUHJvbWlzZTxSZWNhbGxNZW1vcnlSZXN1bHQ+IHtcbiAgICBjb25zdCBwYXJzZWQgPSBSZWNhbGxNZW1vcnlJbnB1dFNjaGVtYS5wYXJzZShpbnB1dCk7XG4gICAgY29uc3QgZGF0YWJyaWNrcyA9IGF3YWl0IHRoaXMuZGF0YWJyaWNrc0JhY2tlbmQoYWN0b3JJZCwgc2lnbmFsKTtcbiAgICBpZiAoZGF0YWJyaWNrcykge1xuICAgICAgdHJ5IHtcbiAgICAgICAgcmV0dXJuIGF3YWl0IGRhdGFicmlja3MucmVjYWxsKGFjdG9ySWQsIHBhcnNlZCwgc2lnbmFsKTtcbiAgICAgIH0gY2F0Y2ggKGVycm9yKSB7XG4gICAgICAgIGlmICghdGhpcy5zaG91bGRGYWxsYmFja0Zyb21EYXRhYnJpY2tzKGVycm9yKSkgdGhyb3cgZXJyb3I7XG4gICAgICB9XG4gICAgfVxuICAgIHJldHVybiAoYXdhaXQgdGhpcy5ncmFwaGl0aUJhY2tlbmQoKSkucmVjYWxsKGFjdG9ySWQsIHBhcnNlZCwgc2lnbmFsKTtcbiAgfVxuXG4gIHByaXZhdGUgYXN5bmMgc2F2ZVR1cm4odHVybjogTWVtb3J5VHVybiwgc2lnbmFsPzogQWJvcnRTaWduYWwpOiBQcm9taXNlPHZvaWQ+IHtcbiAgICBpZiAoYXdhaXQgdGhpcy5kYXRhYnJpY2tzQmFja2VuZCh0dXJuLmFjdG9ySWQsIHNpZ25hbCkpIHJldHVybjtcbiAgICBhd2FpdCAoYXdhaXQgdGhpcy5ncmFwaGl0aUJhY2tlbmQoKSkuc2F2ZVR1cm4odHVybiwgc2lnbmFsKTtcbiAgfVxuXG4gIHByaXZhdGUgYXN5bmMgZGF0YWJyaWNrc0JhY2tlbmQoXG4gICAgYWN0b3JJZDogc3RyaW5nLFxuICAgIHNpZ25hbD86IEFib3J0U2lnbmFsLFxuICApOiBQcm9taXNlPERhdGFicmlja3NNZW1vcnlSdW50aW1lIHwgdW5kZWZpbmVkPiB7XG4gICAgY29uc3QgYmFja2VuZCA9IHRoaXMuZGF0YWJyaWNrcztcbiAgICBpZiAoIWJhY2tlbmQpIHJldHVybiB1bmRlZmluZWQ7XG4gICAgaWYgKGF3YWl0IGJhY2tlbmQuYXZhaWxhYmxlKGFjdG9ySWQsIHNpZ25hbCkpIHJldHVybiBiYWNrZW5kO1xuICAgIGNvbnN0IGNvbmZpZ3VyZWQgPSB0aGlzLnJlcXVpcmVDb25maWcoKS5kYXRhYnJpY2tzO1xuICAgIGlmIChjb25maWd1cmVkICYmIGNvbmZpZ3VyZWQubW9kZSA9PT0gXCJyZXF1aXJlZFwiKSB7XG4gICAgICB0aHJvdyBuZXcgRXJyb3IoXCJSZXF1aXJlZCBEYXRhYnJpY2tzIG1hbmFnZWQgbWVtb3J5IGlzIHVuYXZhaWxhYmxlIGZvciB0aGUgY3VycmVudCBhY3RvclwiKTtcbiAgICB9XG4gICAgcmV0dXJuIHVuZGVmaW5lZDtcbiAgfVxuXG4gIHByaXZhdGUgc2hvdWxkRmFsbGJhY2tGcm9tRGF0YWJyaWNrcyhlcnJvcjogdW5rbm93bik6IGJvb2xlYW4ge1xuICAgIGNvbnN0IGNvbmZpZ3VyZWQgPSB0aGlzLnJlcXVpcmVDb25maWcoKS5kYXRhYnJpY2tzO1xuICAgIHJldHVybiAoXG4gICAgICBjb25maWd1cmVkICE9PSBmYWxzZSAmJlxuICAgICAgY29uZmlndXJlZC5tb2RlID09PSBcImF1dG9cIiAmJlxuICAgICAgZXJyb3JVdGlscy5lcnJvckNvbnRleHQoZXJyb3IpLmhhc1N0YXR1c0NvZGUoNDAzLCA0MDQpXG4gICAgKTtcbiAgfVxuXG4gIHByaXZhdGUgYXN5bmMgZ3JhcGhpdGlCYWNrZW5kKCk6IFByb21pc2U8R3JhcGhpdGlNZW1vcnlCYWNrZW5kPiB7XG4gICAgaWYgKHRoaXMuY2xvc2VkKSB0aHJvdyBuZXcgRXJyb3IoXCJBcHBLaXQgbWVtb3J5IGlzIGNsb3NlZFwiKTtcbiAgICBpZiAodGhpcy5ncmFwaGl0aT8ucnVubmluZykgcmV0dXJuIHRoaXMuZ3JhcGhpdGk7XG4gICAgaWYgKHRoaXMuZ3JhcGhpdGkpIHtcbiAgICAgIGF3YWl0IHRoaXMuZ3JhcGhpdGkuc2h1dGRvd24oKTtcbiAgICAgIHRoaXMuZ3JhcGhpdGkgPSB1bmRlZmluZWQ7XG4gICAgfVxuICAgIGNvbnN0IG9wdGlvbnMgPSB0aGlzLnJlcXVpcmVDb25maWcoKS5ncmFwaGl0aTtcbiAgICBpZiAoIW9wdGlvbnMpIHRocm93IG5ldyBFcnJvcihcIkdyYXBoaXRpIGZhbGxiYWNrIGlzIGRpc2FibGVkXCIpO1xuICAgIHRoaXMuZ3JhcGhpdGlTdGFydGluZyA/Pz0gaW1wb3J0KFwiLi9ncmFwaGl0aS9fYmFja2VuZC50c1wiKVxuICAgICAgLnRoZW4oKHsgc3RhcnRHcmFwaGl0aUJhY2tlbmQgfSkgPT5cbiAgICAgICAgc3RhcnRHcmFwaGl0aUJhY2tlbmQob3B0aW9ucywgdGhpcy5ncmFwaGl0aVN0YXJ0dXBBYm9ydC5zaWduYWwpLFxuICAgICAgKVxuICAgICAgLnRoZW4oKGJhY2tlbmQpID0+IHtcbiAgICAgICAgaWYgKHRoaXMuY2xvc2VkKSB7XG4gICAgICAgICAgcmV0dXJuIGJhY2tlbmQuc2h1dGRvd24oKS50aGVuKCgpID0+IHtcbiAgICAgICAgICAgIHRocm93IG5ldyBFcnJvcihcIkFwcEtpdCBtZW1vcnkgY2xvc2VkIGR1cmluZyBHcmFwaGl0aSBzdGFydHVwXCIpO1xuICAgICAgICAgIH0pO1xuICAgICAgICB9XG4gICAgICAgIHRoaXMuZ3JhcGhpdGkgPSBiYWNrZW5kO1xuICAgICAgICByZXR1cm4gYmFja2VuZDtcbiAgICAgIH0pXG4gICAgICAuZmluYWxseSgoKSA9PiB7XG4gICAgICAgIHRoaXMuZ3JhcGhpdGlTdGFydGluZyA9IHVuZGVmaW5lZDtcbiAgICAgIH0pO1xuICAgIHJldHVybiB0aGlzLmdyYXBoaXRpU3RhcnRpbmc7XG4gIH1cblxuICBwcml2YXRlIGFjdG9ySWQocmVzb3VyY2VJZD86IHN0cmluZyk6IHN0cmluZyB7XG4gICAgaWYgKHJlc291cmNlSWQpIHJldHVybiByZXNvdXJjZUlkO1xuICAgIGNvbnN0IGNvbnRleHQgPSBkYnhBcHBraXQudHJ5R2V0RXhlY3V0aW9uQ29udGV4dCgpO1xuICAgIGlmICghY29udGV4dCkgdGhyb3cgbmV3IEVycm9yKFwiTWVtb3J5IHRvb2xzIHJlcXVpcmUgYW4gQXBwS2l0IGV4ZWN1dGlvbiBjb250ZXh0XCIpO1xuICAgIHJldHVybiBcInVzZXJJZFwiIGluIGNvbnRleHQgPyBjb250ZXh0LnVzZXJJZCA6IGNvbnRleHQuc2VydmljZVVzZXJJZDtcbiAgfVxuXG4gIHByaXZhdGUgcmVxdWlyZUNvbmZpZygpOiBSZXNvbHZlZE1lbW9yeVBsdWdpbkNvbmZpZyB7XG4gICAgaWYgKCF0aGlzLnJlc29sdmVkKSB0aHJvdyBuZXcgRXJyb3IoXCJBcHBLaXQgbWVtb3J5IGlzIG5vdCBpbml0aWFsaXplZFwiKTtcbiAgICByZXR1cm4gdGhpcy5yZXNvbHZlZDtcbiAgfVxufVxuXG4vKiogQXBwS2l0IHJlZ2lzdHJhdGlvbiBmYWN0b3J5IGZvciB7QGxpbmsgTWVtb3J5UGx1Z2lufS4gKi9cbmV4cG9ydCBjb25zdCBtZW1vcnkgPSB0b1BsdWdpbihNZW1vcnlQbHVnaW4pO1xuIl19

package/lib/tsconfig.tsbuildinfo

@@ -0,0 +1 @@
+{"root":["../src/_contracts.ts","../src/_mastra.ts","../src/config.ts","../src/plugin.ts","../src/databricks/_runtime.ts","../src/databricks/client.ts","../src/databricks/index.ts","../src/graphiti/_backend.ts","../src/graphiti/index.ts","../src/graphiti/options.ts","../src/graphiti/runtime.ts","../smoke/mastra-tool.ts","../index.ts"],"version":"5.9.3"}

package/package.json

@@ -1,6 +1,106 @@
 {
   "name": "@dbx-tools/appkit-memory",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "description": "AppKit durable memory tools with Databricks managed memory and lazy Graphiti fallback",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/reggie-db/dbx-tools.git",
+    "directory": "packages/js/node/appkit-memory"
+  },
+  "scripts": {
+    "build": "projen build",
+    "compile": "projen compile",
+    "default": "projen default",
+    "package": "projen package",
+    "post-compile": "projen post-compile",
+    "pre-compile": "projen pre-compile",
+    "prepack": "projen prepack",
+    "test": "projen test",
+    "watch": "projen watch",
+    "projen": "projen"
+  },
+  "devDependencies": {
+    "@dbx-tools/appkit-mastra": "^0.9.131",
+    "@types/json-schema": "^7",
+    "@types/node": "^24.6.0",
+    "typescript": "^5.9.3",
+    "vitest": "3.2.4"
+  },
+  "peerDependencies": {
+    "@dbx-tools/appkit-mastra": "^0.9.131"
+  },
+  "dependencies": {
+    "@databricks/appkit": "0.81.0",
+    "@dbx-tools/appkit": "^0.9.131",
+    "@dbx-tools/auth": "^0.9.131",
+    "@dbx-tools/cache": "^0.9.131",
+    "@dbx-tools/core": "^0.9.131",
+    "@dbx-tools/shared-core": "^0.9.131",
+    "@dbx-tools/shared-graphiti": "^0.9.131",
+    "semver": "^7.7.3",
+    "zod": "4.3.6"
+  },
+  "main": "./lib/index.js",
+  "license": "Apache-2.0",
+  "publishConfig": {
+    "access": "public",
+    "main": "./lib/index.js",
+    "types": "./lib/index.d.ts",
+    "exports": {
+      ".": {
+        "types": "./lib/index.d.ts",
+        "default": "./lib/index.js"
+      },
+      "./config": {
+        "types": "./lib/src/config.d.ts",
+        "default": "./lib/src/config.js"
+      },
+      "./plugin": {
+        "types": "./lib/src/plugin.d.ts",
+        "default": "./lib/src/plugin.js"
+      },
+      "./*": {
+        "types": "./lib/src/*/index.d.ts",
+        "default": "./lib/src/*/index.js"
+      },
+      "./package.json": "./package.json"
+    }
+  },
+  "version": "0.9.131",
+  "types": "./lib/index.d.ts",
+  "dbxToolsConfig": {
+    "tags": [
+      "node"
+    ]
+  },
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./lib/index.d.ts",
+      "default": "./lib/index.js"
+    },
+    "./config": {
+      "types": "./lib/src/config.d.ts",
+      "default": "./lib/src/config.js"
+    },
+    "./plugin": {
+      "types": "./lib/src/plugin.d.ts",
+      "default": "./lib/src/plugin.js"
+    },
+    "./*": {
+      "types": "./lib/src/*/index.d.ts",
+      "default": "./lib/src/*/index.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "index.ts",
+    "src",
+    "lib"
+  ],
+  "peerDependenciesMeta": {
+    "@dbx-tools/appkit-mastra": {
+      "optional": true
+    }
+  },
+  "//": "~~ Generated by projen. To modify, edit .projenrc.js and run \"bunx projen\"."
+}

package/src/_contracts.ts

@@ -0,0 +1,107 @@
+/**
+ * Backend-neutral memory tool and turn contracts.
+ *
+ * @module
+ */
+
+import { z } from "zod";
+
+export const RememberMemoryInputSchema = z
+  .object({
+    path: z
+      .string()
+      .trim()
+      .startsWith("/")
+      .describe("Stable absolute memory path, such as /preferences/response.md."),
+    content: z.string().trim().min(1).describe("Durable memory content."),
+    description: z
+      .string()
+      .trim()
+      .min(1)
+      .optional()
+      .describe("Short retrieval-oriented description."),
+  })
+  .strict()
+  .describe("Durable memory to save for the current actor.");
+
+export type RememberMemoryInput = z.infer<typeof RememberMemoryInputSchema>;
+
+export const RecallMemoryInputSchema = z
+  .object({
+    query: z.string().trim().min(1).describe("Question or phrase used to retrieve memory."),
+    limit: z.coerce
+      .number<number>()
+      .int()
+      .min(1)
+      .max(100)
+      .default(10)
+      .describe("Maximum number of matching memories to return."),
+  })
+  .strict()
+  .describe("Actor-scoped memory search.");
+
+export type RecallMemoryInput = z.input<typeof RecallMemoryInputSchema>;
+
+export const MemoryBackendSchema = z
+  .enum(["databricks", "graphiti"])
+  .describe("Memory backend that completed the operation.");
+
+export type MemoryBackend = z.infer<typeof MemoryBackendSchema>;
+
+export const MemoryRecordSchema = z
+  .object({
+    path: z.string().optional().describe("Stable memory path when the backend supports paths."),
+    content: z.string().describe("Retrieved or saved memory content."),
+    description: z.string().optional().describe("Retrieval-oriented memory description."),
+    score: z.number().optional().describe("Backend relevance score when available."),
+  })
+  .strict()
+  .describe("Backend-neutral durable memory record.");
+
+export type MemoryRecord = z.infer<typeof MemoryRecordSchema>;
+
+export const RememberMemoryResultSchema = z
+  .object({
+    backend: MemoryBackendSchema,
+    memory: MemoryRecordSchema,
+  })
+  .strict()
+  .describe("Result of saving one durable memory.");
+
+export type RememberMemoryResult = z.infer<typeof RememberMemoryResultSchema>;
+
+export const RecallMemoryResultSchema = z
+  .object({
+    backend: MemoryBackendSchema,
+    memories: z.array(MemoryRecordSchema).describe("Matching actor-scoped memories."),
+  })
+  .strict()
+  .describe("Result of one durable memory search.");
+
+export type RecallMemoryResult = z.infer<typeof RecallMemoryResultSchema>;
+
+/** Completed Mastra turn offered to the optional external saver. */
+export interface MemoryTurn {
+  readonly agentId: string;
+  readonly actorId: string;
+  readonly threadId: string;
+  readonly user: string;
+  readonly assistant: string;
+}
+
+/** Lazy Graphiti backend used only after managed memory is unavailable. */
+export interface GraphitiMemoryBackend {
+  readonly running: boolean;
+  remember(
+    actorId: string,
+    input: RememberMemoryInput,
+    signal?: AbortSignal,
+  ): Promise<RememberMemoryResult>;
+  recall(
+    actorId: string,
+    input: z.output<typeof RecallMemoryInputSchema>,
+    signal?: AbortSignal,
+  ): Promise<RecallMemoryResult>;
+  saveTurn(turn: MemoryTurn, signal?: AbortSignal): Promise<void>;
+  shutdown(): Promise<void>;
+}

package/src/_mastra.ts

@@ -0,0 +1,80 @@
+/**
+ * Optional Mastra output-processor attachment.
+ *
+ * Loaded only when the memory plugin finds a registered Mastra sibling.
+ *
+ * @module
+ */
+
+import { pluginRegistry } from "@dbx-tools/appkit";
+import { errorUtils, log, object } from "@dbx-tools/shared-core";
+
+import type { MemoryTurn } from "./_contracts.ts";
+
+const logger = log.logger("memory/mastra");
+
+interface MemoryMessage {
+  readonly role: string;
+  readonly content: { readonly parts: readonly unknown[] };
+}
+
+/** Attach a memory-owned turn saver through Mastra's generic processor extension. */
+export async function attachMastraTurnSaver(
+  context: pluginRegistry.PluginContextLike,
+  saveTurn: (turn: MemoryTurn, signal?: AbortSignal) => Promise<void>,
+): Promise<boolean> {
+  const mastraModule = await import("@dbx-tools/appkit-mastra");
+  const plugin = pluginRegistry.instance(context, mastraModule.mastra);
+  if (!plugin) return false;
+  plugin.exports().registerOutputProcessor((agentId) => ({
+    id: `appkit_memory_${agentId}`,
+    name: "AppKit memory turn saver",
+    description: "Saves completed turns to the configured fallback memory backend.",
+    processOutputResult: async ({ messageList, requestContext, abortSignal }) => {
+      const scope = mastraModule.memory.requestMemoryScope(requestContext);
+      if (!scope) return messageList;
+      const actorId = typeof scope.resource === "string" ? scope.resource : undefined;
+      const threadId = typeof scope.thread === "string" ? scope.thread : undefined;
+      if (!actorId || !threadId) return messageList;
+      const user = latestText(messageList.get.input.db(), "user");
+      const assistant = latestText(messageList.get.response.db(), "assistant");
+      if (!user || !assistant) return messageList;
+      try {
+        await saveTurn(
+          {
+            agentId,
+            actorId,
+            threadId,
+            user,
+            assistant,
+          },
+          abortSignal,
+        );
+      } catch (error) {
+        logger.warn("turn save failed", { error: errorUtils.toError(error), agentId });
+      }
+      return messageList;
+    },
+  }));
+  return true;
+}
+
+function latestText(
+  messages: readonly MemoryMessage[],
+  role: "user" | "assistant",
+): string | undefined {
+  for (let index = messages.length - 1; index >= 0; index -= 1) {
+    const message = messages[index];
+    if (message?.role !== role) continue;
+    const text = message.content.parts
+      .flatMap((part) =>
+        object.isRecord(part) && part.type === "text" && typeof part.text === "string"
+          ? [part.text]
+          : [],
+      )
+      .join("\n")
+      .trim();
+    if (text) return text;
+  }
+  return undefined;
+}

package/src/config.ts

@@ -0,0 +1,134 @@
+/**
+ * Top-level AppKit memory plugin configuration.
+ *
+ * @module
+ */
+
+import type { BasePluginConfig } from "@databricks/appkit";
+import { projectUtils } from "@dbx-tools/core";
+import { stringUtils } from "@dbx-tools/shared-core";
+import {
+  GraphitiOptionsSchema,
+  graphitiOptionOverrides,
+  type GraphitiOptions,
+} from "@dbx-tools/shared-graphiti";
+import type { JSONSchema7 } from "json-schema";
+import { z } from "zod";
+
+export const DatabricksMemoryOptionsSchema = z
+  .object({
+    memoryStore: z
+      .string()
+      .trim()
+      .min(1)
+      .optional()
+      .describe("Existing Databricks Agent Memory Store id or resource name."),
+    mode: z
+      .enum(["auto", "required"])
+      .default("auto")
+      .describe("Whether an inaccessible managed-memory store falls back or fails."),
+  })
+  .strict()
+  .describe("Databricks managed-memory backend configuration.");
+
+export type DatabricksMemoryOptions = z.input<typeof DatabricksMemoryOptionsSchema>;
+
+const GraphitiMemoryOptionsSchema = GraphitiOptionsSchema.partial().describe(
+  "Graphiti option overrides for the lazy fallback.",
+);
+
+export const MemoryPluginOptionsSchema = z
+  .object({
+    databricks: z
+      .union([z.literal("auto"), z.boolean(), DatabricksMemoryOptionsSchema])
+      .default("auto")
+      .describe(
+        "Databricks managed memory mode: auto falls back, true requires access, and false disables it.",
+      ),
+    graphiti: z
+      .union([z.boolean(), GraphitiMemoryOptionsSchema])
+      .default(true)
+      .describe("Lazy Graphiti fallback configuration, enabled by default."),
+    saveTurns: z
+      .union([z.literal("auto"), z.boolean()])
+      .default("auto")
+      .describe(
+        "Completed-turn saving mode: auto attaches when possible, true requires attachment, and false disables it.",
+      ),
+  })
+  .strict()
+  .describe("AppKit memory plugin options.");
+
+export type MemoryPluginOptions = z.input<typeof MemoryPluginOptionsSchema>;
+
+/** AppKit base configuration plus the memory plugin's serialized options. */
+export type MemoryPluginConfig = BasePluginConfig & MemoryPluginOptions;
+
+/** Resolved Databricks store identity and availability policy. */
+export interface ResolvedDatabricksMemoryOptions {
+  readonly memoryStore: string;
+  readonly mode: "auto" | "required";
+}
+
+/** Fully resolved top-level memory configuration. */
+export interface ResolvedMemoryPluginConfig {
+  readonly databricks: false | ResolvedDatabricksMemoryOptions;
+  readonly graphiti: false | GraphitiOptions;
+  readonly saveTurns: "auto" | boolean;
+}
+
+/** JSON Schema published through the AppKit memory plugin manifest. */
+export const MEMORY_CONFIG_SCHEMA = z.toJSONSchema(MemoryPluginOptionsSchema) as JSONSchema7;
+
+/** Resolve plugin defaults without parsing unrelated AppKit base fields. */
+export function resolveMemoryPluginConfig(
+  config: MemoryPluginConfig = {},
+): ResolvedMemoryPluginConfig {
+  const parsed = MemoryPluginOptionsSchema.parse({
+    databricks: config.databricks,
+    graphiti: config.graphiti,
+    saveTurns: config.saveTurns,
+  });
+  if (parsed.databricks === false && parsed.graphiti === false) {
+    throw new Error("AppKit memory requires at least one enabled backend");
+  }
+  return {
+    databricks:
+      parsed.databricks === false
+        ? false
+        : {
+            memoryStore:
+              typeof parsed.databricks === "object" && parsed.databricks.memoryStore
+                ? resourceId(parsed.databricks.memoryStore, "memory-stores")
+                : defaultMemoryStore(),
+            mode:
+              parsed.databricks === true
+                ? "required"
+                : typeof parsed.databricks === "object"
+                  ? parsed.databricks.mode
+                  : "auto",
+          },
+    graphiti:
+      config.graphiti === false
+        ? false
+        : config.graphiti === undefined || config.graphiti === true
+          ? {}
+          : graphitiOptionOverrides(config.graphiti),
+    saveTurns: parsed.saveTurns,
+  };
+}
+
+function defaultMemoryStore(): string {
+  const slug = stringUtils.toSlugWithOptions(
+    { maxLength: 56, truncateStrategy: "hash" },
+    projectUtils.name(),
+  );
+  const id = /^[a-z]/.test(slug) ? slug : `app-${slug}`;
+  return (id.length < 3 ? `${id}-store` : id).slice(0, 56);
+}
+
+function resourceId(value: string, prefix: string): string {
+  const trimmed = value.trim().replace(new RegExp(`^${prefix}/`), "");
+  if (!trimmed) throw new Error(`${prefix} id is required`);
+  return trimmed;
+}