Procházet zdrojové kódy

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

Signed-off-by: Yongtao Huang <yongtaoh2022@gmail.com>
tags/1.7.2
Yongtao Huang před 3 měsíci
rodič
revize
76d123fe19
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 7 přidání a 7 odebrání
  1. 7
    7
      api/services/dataset_service.py

+ 7
- 7
api/services/dataset_service.py Zobrazit soubor

@@ -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:

Načítá se…
Zrušit
Uložit