common : improve yaml log escaping (ggerganov#4080)

* logging: improve escaping in yaml output * logging: include review feedback
Sanctum-AI · Nov 23, 2023 · 38c6ef1 · 38c6ef1
1 parent 85a81ff
commit 38c6ef1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/common/common.cpp b/common/common.cpp
@@ -1194,6 +1194,7 @@ void dump_string_yaml_multiline(FILE * stream, const char * prop_name, const cha
     if (!data_str.empty() && (std::isspace(data_str[0]) || std::isspace(data_str.back()))) {
         data_str = std::regex_replace(data_str, std::regex("\n"), "\\n");
         data_str = std::regex_replace(data_str, std::regex("\""), "\\\"");
+        data_str = std::regex_replace(data_str, std::regex(R"(\\[^n"])"), R"(\$&)");
         data_str = "\"" + data_str + "\"";
         fprintf(stream, "%s: %s\n", prop_name, data_str.c_str());
         return;