test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-11-30 10:18:13 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

prohibit enable and disable function when segment is not completed (#1954)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
Co-authored-by: Joel <iamjoel007@gmail.com>
This commit is contained in:
Jyong 2024-01-05 18:18:38 +08:00 committed by GitHub
parent 28b26f67e2
commit 4d99c689f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -156,6 +156,9 @@ class DatasetDocumentSegmentApi(Resource):
if not segment: if not segment:
raise NotFound('Segment not found.') raise NotFound('Segment not found.')
if segment.status != 'completed':
raise NotFound('Segment is not completed, enable or disable function is not allowed')
document_indexing_cache_key = 'document_{}_indexing'.format(segment.document_id) document_indexing_cache_key = 'document_{}_indexing'.format(segment.document_id)
cache_result = redis_client.get(document_indexing_cache_key) cache_result = redis_client.get(document_indexing_cache_key)
if cache_result is not None: if cache_result is not None:

2
api/tasks/disable_segment_from_index_task.py
View File

@ -27,7 +27,7 @@ def disable_segment_from_index_task(segment_id: str):
raise NotFound('Segment not found') raise NotFound('Segment not found')
if segment.status != 'completed': if segment.status != 'completed':
return raise NotFound('Segment is not completed , disable action is not allowed.')
indexing_cache_key = 'segment_{}_indexing'.format(segment.id) indexing_cache_key = 'segment_{}_indexing'.format(segment.id)

2
api/tasks/enable_segment_to_index_task.py
View File

@ -29,7 +29,7 @@ def enable_segment_to_index_task(segment_id: str):
raise NotFound('Segment not found') raise NotFound('Segment not found')
if segment.status != 'completed': if segment.status != 'completed':
return raise NotFound('Segment is not completed, enable action is not allowed.')
indexing_cache_key = 'segment_{}_indexing'.format(segment.id) indexing_cache_key = 'segment_{}_indexing'.format(segment.id)

2
web/app/components/datasets/documents/detail/completed/SegmentCard.tsx
View File

@ -128,7 +128,7 @@ const SegmentCard: FC<ISegmentCardProps> = ({
> >
<Switch <Switch
size='md' size='md'
disabled={archived} disabled={archived || detail.status !== 'completed'}
defaultValue={enabled} defaultValue={enabled}
onChange={async (val) => { onChange={async (val) => {
await onChangeSwitch?.(id, val) await onChangeSwitch?.(id, val)