AaronPo/the-biergarten-app
1
0
Fork 0
mirror of https://github.com/aaronpo97/the-biergarten-app.git synced 2026-10-15 01:54:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Formatted src/data_generation/prompt_formatting

Browse Source
This commit is contained in:
Aaron Po
2026-07-19 04:10:33 -04:00
parent ac98a2a22e
commit 933b728135
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tooling/pipeline/src/data_generation/prompt_formatting/gemma4_jinja_prompt_formatter.cc
View File

@@ -8,25 +8,25 @@ static constexpr std::string_view kWhitespace = " \t\n\r\f\v";
// Strips leading and trailing whitespace to ensure clean prompt injection. // Strips leading and trailing whitespace to ensure clean prompt injection.
static std::string_view Trim(std::string_view value) { static std::string_view Trim(std::string_view value) {
const size_t first_index = value.find_first_not_of(kWhitespace); const size_t first_index = value.find_first_not_of(kWhitespace);
const bool is_all_whitespace = (first_index == std::string_view::npos); const bool is_all_whitespace = (first_index == std::string_view::npos);
if (is_all_whitespace) { if (is_all_whitespace) {
return ""; return "";
} }
const size_t last_index = value.find_last_not_of(kWhitespace); const size_t last_index = value.find_last_not_of(kWhitespace);
return value.substr(first_index, last_index - first_index + 1); return value.substr(first_index, last_index - first_index + 1);
} }
std::string Gemma4JinjaPromptFormatter::Format( std::string Gemma4JinjaPromptFormatter::Format(
std::string_view system_prompt, std::string_view user_prompt) const { std::string_view system_prompt, std::string_view user_prompt) const {
std::string_view trimmed_system = Trim(system_prompt); std::string_view trimmed_system = Trim(system_prompt);
std::string_view trimmed_user = Trim(user_prompt); std::string_view trimmed_user = Trim(user_prompt);
return std::format( return std::format(
"<|turn|>system\n<|think|>\n{}\n<|turn|>\n" "<|turn|>system\n<|think|>\n{}\n<|turn|>\n"
"<|turn|>user\n{}\n<|turn|>\n" "<|turn|>user\n{}\n<|turn|>\n"
"<|turn|>model\n<|channel>thought\n", "<|turn|>model\n<|channel>thought\n",
trimmed_system, trimmed_user); trimmed_system, trimmed_user);
} }