File size: 693 Bytes
e67ab0e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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,
};
}
|