瀏覽代碼

fix: handle AttributeError for datasets and index (#1052)

tags/0.3.19
yezhwi 2 年之前
父節點
當前提交
8b8e510bfe
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 9 行新增6 行删除
  1. 7
    5
      api/controllers/console/datasets/datasets_document.py
  2. 2
    1
      api/core/indexing_runner.py

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

req_data = request.args req_data = request.args


document_id = req_data.get('document_id') document_id = req_data.get('document_id')
# get default rules
mode = DocumentService.DEFAULT_RULES['mode']
rules = DocumentService.DEFAULT_RULES['rules']
if document_id: if document_id:
# get the latest process rule # get the latest process rule
document = Document.query.get_or_404(document_id) document = Document.query.get_or_404(document_id)
order_by(DatasetProcessRule.created_at.desc()). \ order_by(DatasetProcessRule.created_at.desc()). \
limit(1). \ limit(1). \
one_or_none() one_or_none()
mode = dataset_process_rule.mode
rules = dataset_process_rule.rules_dict
else:
mode = DocumentService.DEFAULT_RULES['mode']
rules = DocumentService.DEFAULT_RULES['rules']
if dataset_process_rule:
mode = dataset_process_rule.mode
rules = dataset_process_rule.rules_dict


return { return {
'mode': mode, 'mode': mode,

+ 2
- 1
api/core/indexing_runner.py 查看文件

filter(UploadFile.id == data_source_info['upload_file_id']). \ filter(UploadFile.id == data_source_info['upload_file_id']). \
one_or_none() one_or_none()


text_docs = FileExtractor.load(file_detail)
if file_detail:
text_docs = FileExtractor.load(file_detail)
elif dataset_document.data_source_type == 'notion_import': elif dataset_document.data_source_type == 'notion_import':
loader = NotionLoader.from_document(dataset_document) loader = NotionLoader.from_document(dataset_document)
text_docs = loader.load() text_docs = loader.load()

Loading…
取消
儲存