Spaces:

Mike0021
/

chat-ui-agent-demo

Paused

chat-ui-agent-demo / src /lib /server /textGeneration /utils /routing.ts

Deploy verified chat and coding agent UX (part 2)

fae625a verified 3 months ago

693 Bytes

import type { EndpointMessage } from "../../endpoints/endpoints";

const ROUTER_REASONING_REGEX = /<think>[\s\S]*?(?:<\/think>|$)/g;

export function stripReasoningBlocks(text: string): string {
	const stripped = text.replace(ROUTER_REASONING_REGEX, "");
	return stripped === text ? text : stripped.trim();
}

export function stripReasoningFromMessageForRouting(message: EndpointMessage): EndpointMessage {
	const clone = { ...message } as EndpointMessage & { reasoning?: string };
	if ("reasoning" in clone) {
		delete clone.reasoning;
	}
	const content =
		typeof message.content === "string" ? stripReasoningBlocks(message.content) : message.content;
	return {
		...clone,
		content,
	};
}

	import type { EndpointMessage } from "../../endpoints/endpoints";

	const ROUTER_REASONING_REGEX = /<think>[\s\S]*?(?:<\/think>\|$)/g;

	export function stripReasoningBlocks(text: string): string {
	const stripped = text.replace(ROUTER_REASONING_REGEX, "");
	return stripped === text ? text : stripped.trim();
	}

	export function stripReasoningFromMessageForRouting(message: EndpointMessage): EndpointMessage {
	const clone = { ...message } as EndpointMessage & { reasoning?: string };
	if ("reasoning" in clone) {
	delete clone.reasoning;
	}
	const content =
	typeof message.content === "string" ? stripReasoningBlocks(message.content) : message.content;
	return {
	...clone,
	content,
	};
	}