estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

refine some orm types (#22885)

Browse Source
This commit is contained in:
Asuka Minato
2025-07-31 19:43:04 +09:00
committed by GitHub
parent a0a30bfdcc
commit 79ea94483e
12 changed files with 424 additions and 410 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/services/dataset_service.py
View File

@@ -2040,6 +2040,7 @@ class SegmentService:
db.session.add(segment_document)
# update document word count
assert document.word_count is not None
document.word_count += segment_document.word_count
db.session.add(document)
db.session.commit()
@@ -2124,6 +2125,7 @@ class SegmentService:
else:
keywords_list.append(None)
# update document word count
assert document.word_count is not None
document.word_count += increment_word_count
db.session.add(document)
try:
@@ -2185,6 +2187,7 @@ class SegmentService:
db.session.commit()
# update document word count
if word_count_change != 0:
assert document.word_count is not None
document.word_count = max(0, document.word_count + word_count_change)
db.session.add(document)
# update segment index task
@@ -2260,6 +2263,7 @@ class SegmentService:
word_count_change = segment.word_count - word_count_change
# update document word count
if word_count_change != 0:
assert document.word_count is not None
document.word_count = max(0, document.word_count + word_count_change)
db.session.add(document)
db.session.add(segment)
@@ -2323,6 +2327,7 @@ class SegmentService:
delete_segment_from_index_task.delay([segment.index_node_id], dataset.id, document.id)
db.session.delete(segment)
# update document word count
assert document.word_count is not None
document.word_count -= segment.word_count
db.session.add(document)
db.session.commit()