Method ParseCreateModel

src/custom_parser/query/model_parser.cpp:138–223

}

void ModelParser::ParseCreateModel(Tokenizer& tokenizer, std::unique_ptr<QueryStatement>& statement) {
    auto token = tokenizer.NextToken();
    auto value = duckdb::StringUtil::Upper(token.value);

    std::string catalog;
    if (token.type == TokenType::KEYWORD && (value == "GLOBAL" || value == "LOCAL")) {
        if (value == "GLOBAL") {
            catalog = "flock_storage.";
        }
        token = tokenizer.NextToken();
        value = duckdb::StringUtil::Upper(token.value);
    }

    if (token.type != TokenType::KEYWORD || value != "MODEL") {
        throw std::runtime_error("Expected 'MODEL' after 'CREATE'.");
    }

    token = tokenizer.NextToken();
    if (token.type != TokenType::PARENTHESIS || token.value != "(") {
        throw std::runtime_error("Expected opening parenthesis '(' after 'MODEL'.");
    }

    token = tokenizer.NextToken();
    if (token.type != TokenType::STRING_LITERAL || token.value.empty()) {
        throw std::runtime_error("Expected non-empty string literal for model name.");
    }
    auto model_name = token.value;

    token = tokenizer.NextToken();
    if (token.type != TokenType::SYMBOL || token.value != ",") {
        throw std::runtime_error("Expected comma ',' after model name.");
    }

    token = tokenizer.NextToken();
    if (token.type != TokenType::STRING_LITERAL || token.value.empty()) {
        throw std::runtime_error("Expected non-empty string literal for model.");
    }
    auto model = token.value;

    token = tokenizer.NextToken();
    if (token.type != TokenType::SYMBOL || token.value != ",") {
        throw std::runtime_error("Expected comma ',' after model.");
    }

    token = tokenizer.NextToken();
    if (token.type != TokenType::STRING_LITERAL || token.value.empty()) {
        throw std::runtime_error("Expected non-empty string literal for provider_name.");
    }
    std::string provider_name = token.value;

    token = tokenizer.NextToken();
    nlohmann::json model_args = nlohmann::json::object();
    // The JSON argument is optional. If present, extract supported model arguments.
    if (token.type == TokenType::SYMBOL || token.value == ",") {
        token = tokenizer.NextToken();
        try {
            nlohmann::json input_args = nlohmann::json::parse(token.value);
            for (auto it = input_args.begin(); it != input_args.end(); ++it) {