aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--examples/common.cpp29
1 files changed, 28 insertions, 1 deletions
diff --git a/examples/common.cpp b/examples/common.cpp
index 222b4fa..1a2f474 100644
--- a/examples/common.cpp
+++ b/examples/common.cpp
@@ -66,6 +66,33 @@ int32_t get_num_physical_cores() {
return n_threads > 0 ? (n_threads <= 4 ? n_threads : n_threads / 2) : 4;
}
+std::string process_escapes(const char* input) {
+ std::string output;
+
+ if (input != nullptr) {
+ std::size_t input_len = std::strlen(input);
+ output.reserve(input_len);
+
+ for (std::size_t i = 0; i < input_len; ++i) {
+ if (input[i] == '\\' && i + 1 < input_len) {
+ switch (input[++i]) {
+ case 'n': output.push_back('\n'); break;
+ case 't': output.push_back('\t'); break;
+ case '\'': output.push_back('\''); break;
+ case '\"': output.push_back('\"'); break;
+ case '\\': output.push_back('\\'); break;
+ default: output.push_back('\\');
+ output.push_back(input[i]); break;
+ }
+ } else {
+ output.push_back(input[i]);
+ }
+ }
+ }
+
+ return output;
+}
+
bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
bool invalid_param = false;
std::string arg;
@@ -91,7 +118,7 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
invalid_param = true;
break;
}
- params.prompt = argv[i];
+ params.prompt = process_escapes(argv[i]);
} else if (arg == "--session") {
if (++i >= argc) {
invalid_param = true;