Browse Source

fix: child chunk API 404 due to UUID type comparison (#25234)

Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
tags/2.0.0-beta.2^2
kenwoodjw 1 month ago
parent
commit
1ba69b8abf
No account linked to committer's email address
1 changed files with 4 additions and 4 deletions
  1. 4
    4
      api/controllers/service_api/dataset/segment.py

+ 4
- 4
api/controllers/service_api/dataset/segment.py View File

@@ -440,7 +440,7 @@ class DatasetChildChunkApi(DatasetApiResource):
raise NotFound("Segment not found.")

# validate segment belongs to the specified document
if segment.document_id != document_id:
if str(segment.document_id) != str(document_id):
raise NotFound("Document not found.")

# check child chunk
@@ -451,7 +451,7 @@ class DatasetChildChunkApi(DatasetApiResource):
raise NotFound("Child chunk not found.")

# validate child chunk belongs to the specified segment
if child_chunk.segment_id != segment.id:
if str(child_chunk.segment_id) != str(segment.id):
raise NotFound("Child chunk not found.")

try:
@@ -500,7 +500,7 @@ class DatasetChildChunkApi(DatasetApiResource):
raise NotFound("Segment not found.")

# validate segment belongs to the specified document
if segment.document_id != document_id:
if str(segment.document_id) != str(document_id):
raise NotFound("Segment not found.")

# get child chunk
@@ -511,7 +511,7 @@ class DatasetChildChunkApi(DatasetApiResource):
raise NotFound("Child chunk not found.")

# validate child chunk belongs to the specified segment
if child_chunk.segment_id != segment.id:
if str(child_chunk.segment_id) != str(segment.id):
raise NotFound("Child chunk not found.")

# validate args

Loading…
Cancel
Save