Pārlūkot izejas kodu

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

Co-authored-by: jyong <jyong@dify.ai>
Co-authored-by: Joel <iamjoel007@gmail.com>
tags/0.4.4
Jyong pirms 1 gada
vecāks
revīzija
4d99c689f0
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam

+ 3
- 0
api/controllers/console/datasets/datasets_segments.py Parādīt failu

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:

+ 1
- 1
api/tasks/disable_segment_from_index_task.py Parādīt failu

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)



+ 1
- 1
api/tasks/enable_segment_to_index_task.py Parādīt failu

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)



+ 1
- 1
web/app/components/datasets/documents/detail/completed/SegmentCard.tsx Parādīt failu

> >
<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)

Notiek ielāde…
Atcelt
Saglabāt