@keystrokehq/keystroke 0.1.1 → 0.1.3

package/dist/agent.cjs

@@ -1,7 +1,7 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_chunk = require("./chunk-DHdmVzO2.cjs");
 const require_dist$1 = require("./dist-B3WT5qgl.cjs");
-const require_dist$2 = require("./dist-DCMRCBXO.cjs");
+const require_dist$2 = require("./dist-DCVQ-aa6.cjs");
 require("./env-api-keys-BFptfIIs.cjs");
 const require_event_stream = require("./event-stream-CS-ls7wC.cjs");
 require("./json-parse-TT8Vr6d3.cjs");
@@ -13841,28 +13841,6 @@ function buildMemoryFromHandle(config, db) {
 		}
 	};
 }
-const MEMORY_MODULE_ENV = "KEYSTROKE_MEMORY_MODULE";
-function isMemoryFactory(value) {
-	return typeof value === "object" && value !== null && typeof value.create === "function";
-}
-function getCreateMemory(mod) {
-	if (typeof mod !== "object" || mod === null) return;
-	const record = mod;
-	const factory = record.createMemory ?? record.default;
-	return typeof factory === "function" ? factory : void 0;
-}
-async function loadMemoryFactoryFromModule(moduleSpecifier, env) {
-	const createMemory = getCreateMemory(await import(moduleSpecifier));
-	if (!createMemory) throw new Error(`Memory module "${moduleSpecifier}" must export createMemory(env): MemoryFactory`);
-	const factory = await createMemory(env);
-	if (!isMemoryFactory(factory)) throw new Error(`Memory module "${moduleSpecifier}" createMemory(env) did not return a valid MemoryFactory`);
-	return factory;
-}
-async function resolveMemoryFactoryFromEnv(env = process.env) {
-	const moduleSpecifier = env[MEMORY_MODULE_ENV]?.trim();
-	if (!moduleSpecifier) return;
-	return loadMemoryFactoryFromModule(moduleSpecifier, env);
-}
 //#endregion
 //#region ../agent/dist/index.mjs
 function createAgent(config) {
@@ -14182,11 +14160,10 @@ async function resolveAgentTools(tools, resolver) {
 		mcpConnections
 	};
 }
-async function resolveMemoryFactory(memoryOption, defMemory) {
+async function resolveMemoryFactory(memoryOption, defMemory, injectedFactory) {
 	const option = memoryOption ?? defMemory;
 	if (option === false) return;
-	const fromEnv = await resolveMemoryFactoryFromEnv();
-	if (fromEnv) return fromEnv;
+	if (injectedFactory) return injectedFactory;
 	return createDefaultMemory(typeof option === "object" ? option : {});
 }
 /** Sandbox tools + host tools → pi agent from a {@link AgentDefinition}. */
@@ -14201,7 +14178,7 @@ async function buildAgentRuntime(def, ctx, runPrompt = {}) {
 		agentId: ctx.agentId,
 		sessionId: ctx.sessionId
 	});
-	const memoryFactory = await resolveMemoryFactory(runPrompt.memory, def.memory);
+	const memoryFactory = await resolveMemoryFactory(runPrompt.memory, def.memory, runPrompt.memoryFactory);
 	const memory = memoryFactory ? await memoryFactory.create({
 		agentId: ctx.agentId,
 		sessionId: ctx.sessionId
@@ -14225,6 +14202,9 @@ async function buildAgentRuntime(def, ctx, runPrompt = {}) {
 		workspaceRoot,
 		workspacesRoot: runPrompt.workspacesRoot,
 		mode,
+		provider: runPrompt.runtimeProvider,
+		vmProvider: runPrompt.vmRuntimeProvider,
+		sessionBucketPrepare: runPrompt.sessionBucketPrepare,
 		hooks: mergeRuntimeHooks(runPrompt.runtimeHooks, def.runtimeHooks),
 		invokeTool: require_dist$2.createInvokeToolBridge(hostTools)
 	});