@huggingface/tasks
Version:
List of ML tasks for huggingface.co/tasks
20 lines (19 loc) • 728 B
JavaScript
export function stringifyMessages(messages, opts) {
let messagesStr = JSON.stringify(messages, null, "\t");
if (opts?.indent) {
messagesStr = messagesStr.replaceAll("\n", `\n${opts.indent}`);
}
if (!opts?.attributeKeyQuotes) {
messagesStr = messagesStr.replace(/"([^"]+)":/g, "$1:");
}
if (opts?.customContentEscaper) {
messagesStr = opts.customContentEscaper(messagesStr);
}
return messagesStr;
}
export function stringifyGenerationConfig(config, opts) {
const quote = opts.attributeKeyQuotes ? `"` : "";
return Object.entries(config)
.map(([key, val]) => `${quote}${key}${quote}${opts.attributeValueConnector}${val},`)
.join(`${opts.indent}`);
}