Ver código fonte

fix: page/limit param not effective (#18196)

tags/1.3.0
诗浓 6 meses atrás
pai
commit
4166f73d9d
Nenhuma conta vinculada ao e-mail do autor do commit

+ 2
- 0
api/controllers/service_api/dataset/segment.py Ver arquivo

tenant_id=current_user.current_tenant_id, tenant_id=current_user.current_tenant_id,
status_list=args["status"], status_list=args["status"],
keyword=args["keyword"], keyword=args["keyword"],
page=page,
limit=limit,
) )


response = { response = {

+ 11
- 4
api/services/dataset_service.py Ver arquivo



@classmethod @classmethod
def get_segments( def get_segments(
cls, document_id: str, tenant_id: str, status_list: list[str] | None = None, keyword: str | None = None
cls,
document_id: str,
tenant_id: str,
status_list: list[str] | None = None,
keyword: str | None = None,
page: int = 1,
limit: int = 20,
): ):
"""Get segments for a document with optional filtering.""" """Get segments for a document with optional filtering."""
query = DocumentSegment.query.filter( query = DocumentSegment.query.filter(
if keyword: if keyword:
query = query.filter(DocumentSegment.content.ilike(f"%{keyword}%")) query = query.filter(DocumentSegment.content.ilike(f"%{keyword}%"))


segments = query.order_by(DocumentSegment.position.asc()).all()
total = len(segments)
paginated_segments = query.order_by(DocumentSegment.position.asc()).paginate(
page=page, per_page=limit, max_per_page=100, error_out=False
)


return segments, total
return paginated_segments.items, paginated_segments.total


@classmethod @classmethod
def update_segment_by_id( def update_segment_by_id(

Carregando…
Cancelar
Salvar