Sfoglia il codice sorgente

Fix segment query tenant bug and variable naming typo (#23321)

Signed-off-by: Yongtao Huang <yongtaoh2022@gmail.com>
tags/1.7.2
Yongtao Huang 3 mesi fa
parent
commit
76d123fe19
Nessun account collegato all'indirizzo email del committer
1 ha cambiato i file con 7 aggiunte e 7 eliminazioni
  1. 7
    7
      api/services/dataset_service.py

+ 7
- 7
api/services/dataset_service.py Vedi File

@@ -2372,7 +2372,7 @@ class SegmentService:
)
if not segments:
return
real_deal_segmment_ids = []
real_deal_segment_ids = []
for segment in segments:
indexing_cache_key = f"segment_{segment.id}_indexing"
cache_result = redis_client.get(indexing_cache_key)
@@ -2382,10 +2382,10 @@ class SegmentService:
segment.disabled_at = None
segment.disabled_by = None
db.session.add(segment)
real_deal_segmment_ids.append(segment.id)
real_deal_segment_ids.append(segment.id)
db.session.commit()

enable_segments_to_index_task.delay(real_deal_segmment_ids, dataset.id, document.id)
enable_segments_to_index_task.delay(real_deal_segment_ids, dataset.id, document.id)
elif action == "disable":
segments = (
db.session.query(DocumentSegment)
@@ -2399,7 +2399,7 @@ class SegmentService:
)
if not segments:
return
real_deal_segmment_ids = []
real_deal_segment_ids = []
for segment in segments:
indexing_cache_key = f"segment_{segment.id}_indexing"
cache_result = redis_client.get(indexing_cache_key)
@@ -2409,10 +2409,10 @@ class SegmentService:
segment.disabled_at = datetime.datetime.now(datetime.UTC).replace(tzinfo=None)
segment.disabled_by = current_user.id
db.session.add(segment)
real_deal_segmment_ids.append(segment.id)
real_deal_segment_ids.append(segment.id)
db.session.commit()

disable_segments_from_index_task.delay(real_deal_segmment_ids, dataset.id, document.id)
disable_segments_from_index_task.delay(real_deal_segment_ids, dataset.id, document.id)
else:
raise InvalidActionError()

@@ -2670,7 +2670,7 @@ class SegmentService:
# check segment
segment = (
db.session.query(DocumentSegment)
.where(DocumentSegment.id == segment_id, DocumentSegment.tenant_id == user_id)
.where(DocumentSegment.id == segment_id, DocumentSegment.tenant_id == tenant_id)
.first()
)
if not segment:

Loading…
Annulla
Salva