AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-04-23 14:19:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
yezhwi 2023-08-30 11:14:16 +08:00 committed by GitHub
parent 3db839a5cb
commit 8b8e510bfe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/controllers/console/datasets/datasets_document.py
View File

@ -138,6 +138,10 @@ class GetProcessRuleApi(Resource):
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)
@ -158,11 +162,9 @@ class GetProcessRuleApi(Resource):
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 if dataset_process_rule:
rules = dataset_process_rule.rules_dict mode = dataset_process_rule.mode
else: rules = dataset_process_rule.rules_dict
mode = DocumentService.DEFAULT_RULES['mode']
rules = DocumentService.DEFAULT_RULES['rules']
return { return {
'mode': mode, 'mode': mode,

3
api/core/indexing_runner.py
View File

@ -406,7 +406,8 @@ class IndexingRunner:
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()