outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: enhance get_segments method signature for improved type hinting and clarity

Browse Source
This commit is contained in:
ZeroZ_JQ 2025-03-18 16:50:26 +08:00
parent f9bdc9917a
commit eb22554448
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/services/dataset_service.py
View File

@ -2156,7 +2156,13 @@ class SegmentService:
return query.paginate(page=page, per_page=limit, max_per_page=100, error_out=False) return query.paginate(page=page, per_page=limit, max_per_page=100, error_out=False)
@classmethod @classmethod
def get_segments(cls, document_id: str, tenant_id: str, status_list: list = None, keyword: str = None): def get_segments(
cls,
document_id: str,
tenant_id: str,
status_list: list[str] | None = None,
keyword: str | None = None
):
"""Get segments for a document with optional filtering.""" """Get segments for a document with optional filtering."""
query = DocumentSegment.query.filter( query = DocumentSegment.query.filter(
DocumentSegment.document_id == document_id, DocumentSegment.document_id == document_id,