feat: refactor: save_document_with_dataset_id, fix error shadowing

pull/21528/head
neatguycoding 11 months ago
parent f668d89621
commit 4e1b17332c

@ -1075,7 +1075,7 @@ class DocumentService:
raise ValueError("No notion info list found.") raise ValueError("No notion info list found.")
exist_page_ids = [] exist_page_ids = []
exist_document = {} exist_document = {}
documents = ( documents_from_db = (
db.session.query(Document) db.session.query(Document)
.filter_by( .filter_by(
dataset_id=dataset.id, dataset_id=dataset.id,
@ -1085,8 +1085,8 @@ class DocumentService:
) )
.all() .all()
) )
if documents: if documents_from_db:
for document in documents: for document in documents_from_db:
data_source_info = json.loads(document.data_source_info) data_source_info = json.loads(document.data_source_info)
exist_page_ids.append(data_source_info["notion_page_id"]) exist_page_ids.append(data_source_info["notion_page_id"])
exist_document[data_source_info["notion_page_id"]] = document.id exist_document[data_source_info["notion_page_id"]] = document.id

Loading…
Cancel
Save