this.queryBuilder = queryBuilder;
        if (queryBuilder instanceof BoolQueryBuilder) {
            queryType = QueryType.BOOL;
        } else if (queryBuilder instanceof DisMaxQueryBuilder) {
            queryType = QueryType.DISMAX;
        } else {
            throw new IllegalArgumentException("Unknown query builder: " + queryBuilder);
        }
    }

    /**