[ggma] Add prompt option to ggma_run tool

runtime/tests/tools/ggma_run/README.md

@@ -4,13 +4,19 @@
 
 It takes GGMA package as input. It uses **GGMA API** internally.
 
+## Options
+
+- `-p, --prompt`: Specify the input prompt text (default: "Lily picked up a flower.")
+
 ## Usage
 
 ```
 $ ./ggma_run path_to_ggma_package
+$ ./ggma_run path_to_ggma_package -p "Your custom prompt here"
+$ ./ggma_run path_to_ggma_package --prompt "What is the weather today?"
 ```
 
-It will run a GGML package to generate the output using the default prompt.
+It will run a GGMA package to generate the output.
 
 ## Example
 

runtime/tests/tools/ggma_run/src/args.cc

@@ -50,6 +50,10 @@ Args::Args(const int argc, char **argv)
 void Args::initialize(void)
 {
   _arser.add_argument("path").type(arser::DataType::STR).help("nnpackage path");
+  _arser.add_argument("-p", "--prompt")
+    .type(arser::DataType::STR)
+    .default_value("Lily picked up a flower.")
+    .help("input prompt text");
   arser::Helper::add_version(_arser, print_version);
 }
 
@@ -80,6 +84,11 @@ void Args::parse(const int argc, char **argv)
         exit(1);
       }
     }
+
+    if (_arser["--prompt"])
+    {
+      _prompt = _arser.get<std::string>("--prompt");
+    }
   }
   catch (const std::bad_cast &e)
   {

runtime/tests/tools/ggma_run/src/args.h

@@ -33,6 +33,7 @@ class Args
   void print(void);
 
   const std::string &packagePath() const { return _package_path; }
+  const std::string &prompt() const { return _prompt; }
   bool printVersion() const { return _print_version; }
 
 private:
@@ -43,6 +44,7 @@ class Args
   arser::Arser _arser;
 
   std::string _package_path;
+  std::string _prompt;
   bool _print_version = false;
 };
 

runtime/tests/tools/ggma_run/src/ggma_run.cc

@@ -38,7 +38,7 @@ int main(const int argc, char **argv)
   {
     Args args(argc, argv);
 
-    std::string prompt = "Lily picked up a flower.";
+    std::string prompt = args.prompt();
     constexpr size_t n_tokens_max = 32;
     ggma_token tokens[n_tokens_max];
     size_t n_tokens;