opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add expire_on_commit

This commit is contained in:
jyong 2025-12-10 00:11:35 +08:00
parent 56f8bdd724
commit 33b1aaf182
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/rag/datasource/retrieval_service.py
View File

@ -371,7 +371,7 @@ class RetrievalService:
include_segment_ids = set()
segment_child_map = {}
segment_file_map = {}
with Session(db.engine) as session:
with Session(bind=db.engine, expire_on_commit=False) as session:
# Process documents
for document in documents:
segment_id = None
@ -475,7 +475,7 @@ class RetrievalService:
session,
)
if attachment_info_dict:
attachment_info = attachment_info_dict["attchment_info"]
attachment_info = attachment_info_dict["attachment_info"]
segment_id = attachment_info_dict["segment_id"]
document_segment_stmt = select(DocumentSegment).where(
DocumentSegment.dataset_id == dataset_document.dataset_id,
@ -684,7 +684,7 @@ class RetrievalService:
.first()
)
if attachment_binding:
attchment_info = {
attachment_info = {
"id": upload_file.id,
"name": upload_file.name,
"extension": "." + upload_file.extension,
@ -692,5 +692,5 @@ class RetrievalService:
"source_url": sign_upload_file(upload_file.id, upload_file.extension),
"size": upload_file.size,
}
return {"attchment_info": attchment_info, "segment_id": attachment_binding.segment_id}
return {"attachment_info": attachment_info, "segment_id": attachment_binding.segment_id}
return None