瀏覽代碼

Fixed a bug where any content in the 'fetch' was converted to True (#4400)

tags/0.6.10
Moonlit 1 年之前
父節點
當前提交
86e7c7321f
No account linked to committer's email address
共有 1 個文件被更改,包括 7 次插入1 次删除
  1. 7
    1
      api/controllers/console/datasets/datasets_document.py

+ 7
- 1
api/controllers/console/datasets/datasets_document.py 查看文件

@@ -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.')

Loading…
取消
儲存