youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-30 10:30:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/feat/knowledgebase-summaryIndex' into feat/knowledgebase-summaryIndex

Browse Source
This commit is contained in:
FFXN
2026-01-28 17:09:55 +08:00
parent a300bc5616 cf76898d1e
commit 2bcc1c540d
2 changed files with 6 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/controllers/console/datasets/datasets_segments.py
View File

@ -201,9 +201,7 @@ class DatasetDocumentSegmentListApi(Resource):
summary_records = SummaryIndexService.get_segments_summaries(segment_ids=segment_ids, dataset_id=dataset_id)
# Only include enabled summaries (already filtered by service)
summaries = {
chunk_id: summary.summary_content for chunk_id, summary in summary_records.items()
}
summaries = {chunk_id: summary.summary_content for chunk_id, summary in summary_records.items()}
# Add summary to each segment
segments_with_summary = []

20
api/services/summary_index_service.py
View File

@ -215,9 +215,7 @@ class SummaryIndexService:
with session_factory.create_session() as session:
# Refresh the summary record in the new session
summary_record_in_session = (
session.query(DocumentSegmentSummary)
.filter_by(id=summary_record_id)
.first()
session.query(DocumentSegmentSummary).filter_by(id=summary_record_id).first()
)
if summary_record_in_session:
summary_record_in_session.summary_index_node_id = summary_index_node_id
@ -278,9 +276,7 @@ class SummaryIndexService:
# Update error status in session
with session_factory.create_session() as session:
summary_record_in_session = (
session.query(DocumentSegmentSummary)
.filter_by(id=summary_record_id)
.first()
session.query(DocumentSegmentSummary).filter_by(id=summary_record_id).first()
)
if summary_record_in_session:
summary_record_in_session.status = "error"
@ -401,9 +397,7 @@ class SummaryIndexService:
try:
# Get or refresh summary record in this session
summary_record_in_session = (
session.query(DocumentSegmentSummary)
.filter_by(chunk_id=segment.id, dataset_id=dataset.id)
.first()
session.query(DocumentSegmentSummary).filter_by(chunk_id=segment.id, dataset_id=dataset.id).first()
)
if not summary_record_in_session:
@ -457,9 +451,7 @@ class SummaryIndexService:
logger.exception("Failed to generate summary for segment %s", segment.id)
# Update summary record with error status
summary_record_in_session = (
session.query(DocumentSegmentSummary)
.filter_by(chunk_id=segment.id, dataset_id=dataset.id)
.first()
session.query(DocumentSegmentSummary).filter_by(chunk_id=segment.id, dataset_id=dataset.id).first()
)
if summary_record_in_session:
summary_record_in_session.status = "error"
@ -949,9 +941,7 @@ class SummaryIndexService:
return query.all()
@staticmethod
def get_document_summary_index_status(
document_id: str, dataset_id: str, tenant_id: str
) -> str | None:
def get_document_summary_index_status(document_id: str, dataset_id: str, tenant_id: str) -> str | None:
"""
Get summary_index_status for a single document.