| synchronize_session=False, | synchronize_session=False, | ||||
| ) | ) | ||||
| ) | ) | ||||
| db.session.commit() | |||||
| else: | else: | ||||
| query = db.session.query(DocumentSegment).where( | query = db.session.query(DocumentSegment).where( | ||||
| DocumentSegment.index_node_id == document.metadata["doc_id"] | DocumentSegment.index_node_id == document.metadata["doc_id"] | ||||
| with flask_app.app_context(): | with flask_app.app_context(): | ||||
| with Session(db.engine) as session: | with Session(db.engine) as session: | ||||
| dataset_stmt = select(Dataset).where(Dataset.id == dataset_id) | dataset_stmt = select(Dataset).where(Dataset.id == dataset_id) | ||||
| dataset = db.session.scalar(dataset_stmt) | |||||
| dataset = session.scalar(dataset_stmt) | |||||
| if not dataset: | if not dataset: | ||||
| return [] | return [] |