run: allow to customize prompt by env var LLAMA_PROMPT_PREFIX

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

Author: benoitf

examples/run/run.cpp

@@ -976,8 +976,11 @@ static int generate(LlamaData & llama_data, const std::string & prompt, std::str
     return 0;
 }
 
+static const char * prompt_prefix_env = std::getenv("LLAMA_PROMPT_PREFIX");
+static const char * prompt_prefix     = prompt_prefix_env ? prompt_prefix_env : "> ";
+
 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);