From cf76898d1e6055ab0cf55dc7525852f5b8cddf7e Mon Sep 17 00:00:00 2001
From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com>
Date: Wed, 28 Jan 2026 08:08:57 +0000
Subject: [PATCH] [autofix.ci] apply automated fixes

---
 .../console/datasets/datasets_segments.py     |  4 +---
 api/services/summary_index_service.py         | 20 +++++--------------
 2 files changed, 6 insertions(+), 18 deletions(-)

diff --git a/api/controllers/console/datasets/datasets_segments.py b/api/controllers/console/datasets/datasets_segments.py
index 228de0914f..23a668112d 100644
--- a/api/controllers/console/datasets/datasets_segments.py
+++ b/api/controllers/console/datasets/datasets_segments.py
@@ -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 = []
diff --git a/api/services/summary_index_service.py b/api/services/summary_index_service.py
index f423fdf7ef..c0dbb1671d 100644
--- a/api/services/summary_index_service.py
+++ b/api/services/summary_index_service.py
@@ -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.