@keystrokehq/keystroke 0.1.36 → 0.1.37

package/dist/{dist-Di6PqKAu.cjs → dist-DlIUXcOB.cjs}

@@ -6890,7 +6890,7 @@ var require_get_vercel_oidc_token = /* @__PURE__ */ require_chunk.__commonJSMin(
 			err = error;
 		}
 		try {
-			const [{ getTokenPayload, isExpired }, { refreshToken }] = await Promise.all([await Promise.resolve().then(() => /* @__PURE__ */ require_chunk.__toESM(require_token_util())), await Promise.resolve().then(() => /* @__PURE__ */ require_chunk.__toESM(require("./token-CQiw-m2-.cjs").default))]);
+			const [{ getTokenPayload, isExpired }, { refreshToken }] = await Promise.all([await Promise.resolve().then(() => /* @__PURE__ */ require_chunk.__toESM(require_token_util())), await Promise.resolve().then(() => /* @__PURE__ */ require_chunk.__toESM(require("./token-qixf8bl9.cjs").default))]);
 			if (!token || isExpired(getTokenPayload(token), options?.expirationBufferMs)) {
 				await refreshToken(options);
 				token = getVercelOidcTokenSync();
@@ -12428,6 +12428,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);
 }
@@ -17295,6 +17301,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
@@ -47119,6 +47288,12 @@ Object.defineProperty(exports, "executeWorkflow", {
 		return executeWorkflow;
 	}
 });
+Object.defineProperty(exports, "extractJsonMiddleware", {
+	enumerable: true,
+	get: function() {
+		return extractJsonMiddleware;
+	}
+});
 Object.defineProperty(exports, "failAgentSession", {
 	enumerable: true,
 	get: function() {
@@ -47167,6 +47342,12 @@ Object.defineProperty(exports, "getTraceContext", {
 		return getTraceContext;
 	}
 });
+Object.defineProperty(exports, "hasToolCall", {
+	enumerable: true,
+	get: function() {
+		return hasToolCall;
+	}
+});
 Object.defineProperty(exports, "isCredentialError", {
 	enumerable: true,
 	get: function() {
@@ -47329,5 +47510,11 @@ Object.defineProperty(exports, "withSpan", {
 		return withSpan;
 	}
 });
+Object.defineProperty(exports, "wrapLanguageModel", {
+	enumerable: true,
+	get: function() {
+		return wrapLanguageModel;
+	}
+});
 
-//# sourceMappingURL=dist-Di6PqKAu.cjs.map
+//# sourceMappingURL=dist-DlIUXcOB.cjs.map