Spaces:
Paused
Paused
| import type { Message } from "$lib/types/Message"; | |
| import Handlebars from "handlebars"; | |
| Handlebars.registerHelper("ifUser", function (this: Pick<Message, "from" | "content">, options) { | |
| if (this.from == "user") return options.fn(this); | |
| }); | |
| Handlebars.registerHelper( | |
| "ifAssistant", | |
| function (this: Pick<Message, "from" | "content">, options) { | |
| if (this.from == "assistant") return options.fn(this); | |
| } | |
| ); | |
| export function compileTemplate<T>(input: string, model: { preprompt: string }) { | |
| const template = Handlebars.compile<T>(input, { | |
| knownHelpers: { ifUser: true, ifAssistant: true }, | |
| knownHelpersOnly: true, | |
| noEscape: true, | |
| strict: true, | |
| preventIndent: true, | |
| }); | |
| return function render(inputs: T, options?: RuntimeOptions) { | |
| return template({ ...model, ...inputs }, options); | |
| }; | |
| } | |