6.1

2026-09-01 21:14:55 +08:00
parent 3b6336fde9
commit e47235dfff
4 changed files with 791 additions and 599 deletions
@@ -74,13 +74,14 @@ def create_text_document(
@router.get("", response_model=DocumentListResponse)
 def list_documents(
    kb_id: str = Query(...),
+    category_id: str = Query(None),
    page: int = Query(1, ge=1),
    page_size: int = Query(50, ge=1, le=200),
    user: User = Depends(get_current_user),
    db: Session = Depends(get_db),
 ) -> DocumentListResponse:
    svc = DocumentService(db)
-    items, total = svc.list_by_knowledge_base(kb_id, user, page=page, page_size=page_size)
+    items, total = svc.list_by_knowledge_base(kb_id, user, category_id=category_id, page=page, page_size=page_size)
    return DocumentListResponse(
        items=[_to_response(doc) for doc in items],
        total=total,