| @@ -1,10 +1,12 @@ | |||
| import logging | |||
| from argparse import ArgumentTypeError | |||
| from datetime import datetime, timezone | |||
| from flask import request | |||
| from flask_login import current_user | |||
| from flask_restful import Resource, fields, marshal, marshal_with, reqparse | |||
| from sqlalchemy import asc, desc | |||
| from transformers.hf_argparser import string_to_bool | |||
| from werkzeug.exceptions import Forbidden, NotFound | |||
| import services | |||
| @@ -141,7 +143,11 @@ class DatasetDocumentListApi(Resource): | |||
| limit = request.args.get('limit', default=20, type=int) | |||
| search = request.args.get('keyword', default=None, type=str) | |||
| sort = request.args.get('sort', default='-created_at', type=str) | |||
| fetch = request.args.get('fetch', default=False, type=bool) | |||
| # "yes", "true", "t", "y", "1" convert to True, while others convert to False. | |||
| try: | |||
| fetch = string_to_bool(request.args.get('fetch', default='false')) | |||
| except (ArgumentTypeError, ValueError, Exception) as e: | |||
| fetch = False | |||
| dataset = DatasetService.get_dataset(dataset_id) | |||
| if not dataset: | |||
| raise NotFound('Dataset not found.') | |||