youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 17:08:03 +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 14:04:04 +08:00
parent d39591018c cf3cb1eea9
commit 86e0ccd056
1 changed files with 5 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
api/services/summary_index_service.py
View File

@ -216,7 +216,7 @@ class SummaryIndexService:
db.session.add(summary_record)
db.session.flush()
# Success, exit function
return
return
except (ConnectionError, Exception) as e:
error_str = str(e).lower()
@ -333,9 +333,7 @@ class SummaryIndexService:
error: Error message
"""
summary_record = (
db.session.query(DocumentSegmentSummary)
.filter_by(chunk_id=segment.id, dataset_id=dataset.id)
.first()
db.session.query(DocumentSegmentSummary).filter_by(chunk_id=segment.id, dataset_id=dataset.id).first()
)
if summary_record:
@ -344,9 +342,7 @@ class SummaryIndexService:
db.session.add(summary_record)
db.session.flush()
else:
logger.warning(
"Summary record not found for segment %s when updating error", segment.id
)
logger.warning("Summary record not found for segment %s when updating error", segment.id)
@staticmethod
def generate_and_vectorize_summary(
@ -371,16 +367,12 @@ class SummaryIndexService:
"""
# Get existing summary record (should have been created by batch_create_summary_records)
summary_record = (
db.session.query(DocumentSegmentSummary)
.filter_by(chunk_id=segment.id, dataset_id=dataset.id)
.first()
db.session.query(DocumentSegmentSummary).filter_by(chunk_id=segment.id, dataset_id=dataset.id).first()
)
if not summary_record:
# If not found (shouldn't happen), create one
logger.warning(
"Summary record not found for segment %s, creating one", segment.id
)
logger.warning("Summary record not found for segment %s, creating one", segment.id)
summary_record = SummaryIndexService.create_summary_record(
segment, dataset, summary_content="", status="generating"
)