refine some orm types (#22885)

2025-07-31 19:43:04 +09:00
parent a0a30bfdcc
commit 79ea94483e
12 changed files with 424 additions and 410 deletions
--- a/api/tasks/batch_create_segment_to_index_task.py
+++ b/api/tasks/batch_create_segment_to_index_task.py
@@ -134,6 +134,7 @@ def batch_create_segment_to_index_task(
            db.session.add(segment_document)
            document_segments.append(segment_document)
        # update document word count
+        assert dataset_document.word_count is not None
        dataset_document.word_count += word_count_change
        db.session.add(dataset_document)
        # add index to db