Method parse_args

src/main.cpp:91–155

Returns true if args are valid, false otherwise

  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;
    } else if (has_prompt_path) {
      std::ifstream file(prompt_path);
      if (!file.is_open()) {
        std::cerr << "Error: could not open file " << prompt_path << std::endl;
        return false;
      }