@keystrokehq/keystroke 0.1.36 → 0.1.37

package/dist/config.d.cts

@@ -1258,8 +1258,8 @@ type KeystrokeConfig = {
   plugins?: ServerPlugin[];
   dirs?: KeystrokeModuleDirs; /** Queue backend for queued execution. Defaults to pg-boss (Postgres) or DB polling (SQLite). */
   scheduler?: SchedulerPlugin;
-  server?: KeystrokeServerConfig; /** Platform project slug this directory deploys to. Optional; set via `keystroke project link`. */
-  project?: string; /** Platform organization slug. Optional; usually auto-filled by `keystroke project link`. */
+  server?: KeystrokeServerConfig; /** Platform project slug this directory deploys to. Optional; set via `keystroke projects link`. */
+  project?: string; /** Platform organization slug. Optional; usually auto-filled by `keystroke projects link`. */
   organization?: string;
 };
 type CreatePluginContextOptions = {

package/dist/config.d.mts

@@ -1258,8 +1258,8 @@ type KeystrokeConfig = {
   plugins?: ServerPlugin[];
   dirs?: KeystrokeModuleDirs; /** Queue backend for queued execution. Defaults to pg-boss (Postgres) or DB polling (SQLite). */
   scheduler?: SchedulerPlugin;
-  server?: KeystrokeServerConfig; /** Platform project slug this directory deploys to. Optional; set via `keystroke project link`. */
-  project?: string; /** Platform organization slug. Optional; usually auto-filled by `keystroke project link`. */
+  server?: KeystrokeServerConfig; /** Platform project slug this directory deploys to. Optional; set via `keystroke projects link`. */
+  project?: string; /** Platform organization slug. Optional; usually auto-filled by `keystroke projects link`. */
   organization?: string;
 };
 type CreatePluginContextOptions = {

package/dist/{dist-iDcp4e6O.mjs → dist-1heZMRTe.mjs}

@@ -6889,7 +6889,7 @@ var require_get_vercel_oidc_token = /* @__PURE__ */ __commonJSMin(((exports, mod
 			err = error;
 		}
 		try {
-			const [{ getTokenPayload, isExpired }, { refreshToken }] = await Promise.all([await Promise.resolve().then(() => /* @__PURE__ */ __toESM(require_token_util())), await import("./token-DUN0yjzi.mjs").then((m) => /* @__PURE__ */ __toESM(m.default))]);
+			const [{ getTokenPayload, isExpired }, { refreshToken }] = await Promise.all([await Promise.resolve().then(() => /* @__PURE__ */ __toESM(require_token_util())), await import("./token-CJ2Rrkyp.mjs").then((m) => /* @__PURE__ */ __toESM(m.default))]);
 			if (!token || isExpired(getTokenPayload(token), options?.expirationBufferMs)) {
 				await refreshToken(options);
 				token = getVercelOidcTokenSync();
@@ -12427,6 +12427,12 @@ function createRestrictedTelemetryDispatcher({ telemetry, includeRuntimeContext,
 function isStepCount(stepCount) {
 	return ({ steps }) => steps.length === stepCount;
 }
+function hasToolCall(...toolName) {
+	return ({ steps }) => {
+		var _a22, _b, _c;
+		return (_c = (_b = (_a22 = steps[steps.length - 1]) == null ? void 0 : _a22.toolCalls) == null ? void 0 : _b.some((toolCall) => toolName.includes(toolCall.toolName))) != null ? _c : false;
+	};
+}
 async function isStopConditionMet({ stopConditions, steps }) {
 	return (await Promise.all(stopConditions.map((condition) => condition({ steps })))).some((result) => result);
 }
@@ -17294,6 +17300,169 @@ createIdGenerator({
 	prefix: "aiobj",
 	size: 24
 });
+function defaultTransform(text2) {
+	return text2.replace(/^```(?:json)?\s*\n?/, "").replace(/\n?```\s*$/, "").trim();
+}
+function extractJsonMiddleware(options) {
+	var _a22;
+	const transform = (_a22 = options == null ? void 0 : options.transform) != null ? _a22 : defaultTransform;
+	const hasCustomTransform = (options == null ? void 0 : options.transform) !== void 0;
+	return {
+		specificationVersion: "v4",
+		wrapGenerate: async ({ doGenerate }) => {
+			const { content, ...rest } = await doGenerate();
+			const transformedContent = [];
+			for (const part of content) {
+				if (part.type !== "text") {
+					transformedContent.push(part);
+					continue;
+				}
+				transformedContent.push({
+					...part,
+					text: transform(part.text)
+				});
+			}
+			return {
+				content: transformedContent,
+				...rest
+			};
+		},
+		wrapStream: async ({ doStream }) => {
+			const { stream, ...rest } = await doStream();
+			const textBlocks = createIdMap();
+			const SUFFIX_BUFFER_SIZE = 12;
+			return {
+				stream: stream.pipeThrough(new TransformStream({ transform: (chunk, controller) => {
+					if (chunk.type === "text-start") {
+						textBlocks[chunk.id] = {
+							startEvent: chunk,
+							phase: hasCustomTransform ? "buffering" : "prefix",
+							buffer: "",
+							prefixStripped: false
+						};
+						return;
+					}
+					if (chunk.type === "text-delta") {
+						const block = textBlocks[chunk.id];
+						if (!block) {
+							controller.enqueue(chunk);
+							return;
+						}
+						block.buffer += chunk.delta;
+						if (block.phase === "buffering") return;
+						if (block.phase === "prefix") {
+							if (block.buffer.length > 0 && !block.buffer.startsWith("`")) {
+								block.phase = "streaming";
+								controller.enqueue(block.startEvent);
+							} else if (block.buffer.startsWith("```")) {
+								if (block.buffer.includes("\n")) {
+									const prefixMatch = block.buffer.match(/^```(?:json)?\s*\n/);
+									if (prefixMatch) {
+										block.buffer = block.buffer.slice(prefixMatch[0].length);
+										block.prefixStripped = true;
+										block.phase = "streaming";
+										controller.enqueue(block.startEvent);
+									} else {
+										block.phase = "streaming";
+										controller.enqueue(block.startEvent);
+									}
+								}
+							} else if (block.buffer.length >= 3 && !block.buffer.startsWith("```")) {
+								block.phase = "streaming";
+								controller.enqueue(block.startEvent);
+							}
+						}
+						if (block.phase === "streaming" && block.buffer.length > SUFFIX_BUFFER_SIZE) {
+							const toStream = block.buffer.slice(0, -12);
+							block.buffer = block.buffer.slice(-12);
+							controller.enqueue({
+								type: "text-delta",
+								id: chunk.id,
+								delta: toStream
+							});
+						}
+						return;
+					}
+					if (chunk.type === "text-end") {
+						const block = textBlocks[chunk.id];
+						if (block) {
+							if (block.phase === "prefix" || block.phase === "buffering") controller.enqueue(block.startEvent);
+							let remaining = block.buffer;
+							if (block.phase === "buffering") remaining = transform(remaining);
+							else if (block.prefixStripped) remaining = remaining.replace(/\n?```\s*$/, "").trimEnd();
+							else remaining = transform(remaining);
+							if (remaining.length > 0) controller.enqueue({
+								type: "text-delta",
+								id: chunk.id,
+								delta: remaining
+							});
+							controller.enqueue(chunk);
+							delete textBlocks[chunk.id];
+							return;
+						}
+					}
+					controller.enqueue(chunk);
+				} })),
+				...rest
+			};
+		}
+	};
+}
+var wrapLanguageModel = ({ model: inputModel, middleware: middlewareArg, modelId, providerId }) => {
+	const model = asLanguageModelV4(inputModel);
+	return [...asArray(middlewareArg)].reverse().reduce((wrappedModel, middleware) => {
+		return doWrap({
+			model: wrappedModel,
+			middleware,
+			modelId,
+			providerId
+		});
+	}, model);
+};
+var doWrap = ({ model, middleware: { transformParams, wrapGenerate, wrapStream, overrideProvider, overrideModelId, overrideSupportedUrls }, modelId, providerId }) => {
+	var _a22, _b, _c;
+	async function doTransform({ params, type }) {
+		return transformParams ? await transformParams({
+			params,
+			type,
+			model
+		}) : params;
+	}
+	return {
+		specificationVersion: "v4",
+		provider: (_a22 = providerId != null ? providerId : overrideProvider == null ? void 0 : overrideProvider({ model })) != null ? _a22 : model.provider,
+		modelId: (_b = modelId != null ? modelId : overrideModelId == null ? void 0 : overrideModelId({ model })) != null ? _b : model.modelId,
+		supportedUrls: (_c = overrideSupportedUrls == null ? void 0 : overrideSupportedUrls({ model })) != null ? _c : model.supportedUrls,
+		async doGenerate(params) {
+			const transformedParams = await doTransform({
+				params,
+				type: "generate"
+			});
+			const doGenerate = async () => await model.doGenerate(transformedParams);
+			const doStream = async () => await model.doStream(transformedParams);
+			return wrapGenerate ? await wrapGenerate({
+				doGenerate,
+				doStream,
+				params: transformedParams,
+				model
+			}) : await doGenerate();
+		},
+		async doStream(params) {
+			const transformedParams = await doTransform({
+				params,
+				type: "stream"
+			});
+			const doGenerate = async () => await model.doGenerate(transformedParams);
+			const doStream = async () => await model.doStream(transformedParams);
+			return wrapStream ? await wrapStream({
+				doGenerate,
+				doStream,
+				params: transformedParams,
+				model
+			}) : await doStream();
+		}
+	};
+};
 createIdGenerator({
 	prefix: "call",
 	size: 24
@@ -46974,6 +47143,6 @@ function resolveWorkflowTool(workflow, context = {}) {
 	});
 }
 //#endregion
-export { resolveRunSourceFromTraceContext as $, generateObject as A, jsonSchema as B, connectMcpDefinition as C, isMcp as D, defineMcp as E, toUIMessageStream as F, clearLiveMessage as G, MESSAGE_EVENT_TYPE as H, createGateway as I, getAgentByRoute as J, createSession as K, require_token_util as L, isFileUIPart as M, output_exports as N, ToolLoopAgent as O, readUIMessageStream as P, recordLlmUsageFromAssistantMessage as Q, require_token_error as R, withCredentialAssignments as S, connectMcpStdio as T, addAgentSessionDuration as U, tool as V, appendEvent as W, getSession as X, getProjectScopeId as Y, listMessageEvents as Z, createCredentialResolver as _, defineWorkflow as a, withSpan as at, resolveActionCredentials as b, isRunCanceledError as c, registerWorkflowToolExecutor as d, setSessionLiveMessage as et, resolveWorkflowTool as f, captureCredentialToolErrors as g, buildCredentialRunContext as h, createDirectActionRunner as i, getTraceContext as it, generateText as j, convertToModelMessages as k, isWorkflow as l, serializeWorkflowError as m, RunCanceledError as n, touchSession as nt, deserializeWorkflowError as o, runWithWorkflowContext as p, failAgentSession as q, createActionRunner as r, captureConsole as rt, executeWorkflow as s, MemoryEventLog as t, setSessionTitle as tt, promptLlm as u, enrichCredentialError as v, connectMcpServer as w, resolveMcpTools as x, isCredentialError as y, generateId as z };
+export { getSession as $, extractJsonMiddleware as A, require_token_util as B, connectMcpDefinition as C, isMcp as D, defineMcp as E, output_exports as F, MESSAGE_EVENT_TYPE as G, generateId as H, readUIMessageStream as I, clearLiveMessage as J, addAgentSessionDuration as K, toUIMessageStream as L, generateText as M, hasToolCall as N, ToolLoopAgent as O, isFileUIPart as P, getProjectScopeId as Q, wrapLanguageModel as R, withCredentialAssignments as S, connectMcpStdio as T, jsonSchema as U, require_token_error as V, tool as W, failAgentSession as X, createSession as Y, getAgentByRoute as Z, createCredentialResolver as _, defineWorkflow as a, touchSession as at, resolveActionCredentials as b, isRunCanceledError as c, withSpan as ct, registerWorkflowToolExecutor as d, listMessageEvents as et, resolveWorkflowTool as f, captureCredentialToolErrors as g, buildCredentialRunContext as h, createDirectActionRunner as i, setSessionTitle as it, generateObject as j, convertToModelMessages as k, isWorkflow as l, serializeWorkflowError as m, RunCanceledError as n, resolveRunSourceFromTraceContext as nt, deserializeWorkflowError as o, captureConsole as ot, runWithWorkflowContext as p, appendEvent as q, createActionRunner as r, setSessionLiveMessage as rt, executeWorkflow as s, getTraceContext as st, MemoryEventLog as t, recordLlmUsageFromAssistantMessage as tt, promptLlm as u, enrichCredentialError as v, connectMcpServer as w, resolveMcpTools as x, isCredentialError as y, createGateway as z };
 
-//# sourceMappingURL=dist-iDcp4e6O.mjs.map
+//# sourceMappingURL=dist-1heZMRTe.mjs.map