Class CompletionArgs

src/main.cpp:85–156

};

struct CompletionArgs {
  std::string prompt;
  int num_steps;
  float temperature = 1.0;
  float top_p = 0.95;
  // Returns true if args are valid, false otherwise
  bool parse_args(const std::vector<const char*>& args) {
    std::string prompt_path = "";
    for (size_t i = 0; i < args.size();) {
      // do some basic validation
      if (args[i][0] != '-') {
        return false;
      } // must start with dash
      if (strlen(args[i]) != 2) {
        return false;
      } // must be -x (one dash, one letter)

      // read in the args
      if (args[i][1] == 'h') {
        return false;
      } else if (args[i][1] == 'i') {
        if (i + 1 >= args.size()) {
          return false;
        }
        prompt = args[i + 1];
        i += 2;
      } else if (args[i][1] == 't') {
        if (i + 1 >= args.size()) {
          return false;
        }
        temperature = std::stof(args[i + 1]);
        i += 2;
      } else if (args[i][1] == 'p') {
        if (i + 1 >= args.size()) {
          return false;
        }
        top_p = std::stof(args[i + 1]);
        i += 2;
      } else if (args[i][1] == 'f') {
        if (i + 1 >= args.size()) {
          return false;
        }
        prompt_path = args[i + 1];
        i += 2;
      } else if (args[i][1] == 'n') {
        if (i + 1 >= args.size()) {
          return false;
        }
        num_steps = std::stoi(args[i + 1]);
        i += 2;
      } else {
        return false;
      }
    }
    int has_prompt = prompt.size() > 0 ? 1 : 0;
    int has_prompt_path = prompt_path.size() > 0 ? 1 : 0;
    if ((has_prompt + has_prompt_path) != 1) {
      return false;