run: allow to customize prompt by env var LLAMA_PROMPT_PREFIX

Signed-off-by: Florent Benoit <[email protected]>

Author: benoitf

examples/run/run.cpp

@@ -975,9 +975,16 @@ static int generate(LlamaData & llama_data, const std::string & prompt, std::str
     printf(LOG_COL_DEFAULT);
     return 0;
 }
+static const char *default_prompt_prefix = "> ";
+static const char *prompt_prefix = nullptr;
+
+static void initialize_prompt_prefix() {
+    const char *llama_prompt_prefix = std::getenv("LLAMA_PROMPT_PREFIX");
+    prompt_prefix = (llama_prompt_prefix != nullptr) ? llama_prompt_prefix : default_prompt_prefix;
+}
+
 
 static int read_user_input(std::string & user_input) {
-    static const char * prompt_prefix = "> ";
 #ifdef WIN32
     printf("\r" LOG_CLR_TO_EOL LOG_COL_DEFAULT "%s", prompt_prefix);
 
@@ -1185,6 +1192,7 @@ static void ctrl_c_handling() {
 
 int main(int argc, const char ** argv) {
     ctrl_c_handling();
+    initialize_prompt_prefix();
     Opt       opt;
     const int ret = opt.init(argc, argv);
     if (ret == 2) {