6.1

2026-09-01 21:14:55 +08:00
parent 3b6336fde9
commit e47235dfff
4 changed files with 791 additions and 599 deletions
@@ -144,13 +144,13 @@ class DocumentService:
        return doc

    def list_by_knowledge_base(
-        self, kb_id: str, user: User, *, page: int = 1, page_size: int = 50
+        self, kb_id: str, user: User, *, category_id: str | None = None, page: int = 1, page_size: int = 50
    ):
        # 校验 KB 归属
        kb = self._kb_repo.get_by_id(kb_id)
        if kb is None or kb.user_id != user.id or kb.status == "DELETED":
            raise NotFoundError("知识库不存在。")
-        return self._doc_repo.list_by_knowledge_base(kb_id, page=page, page_size=page_size)
+        return self._doc_repo.list_by_knowledge_base(kb_id, category_id=category_id, page=page, page_size=page_size)

    def delete(self, doc_id: str, user: User) -> None:
        doc = self.get_or_404(doc_id, user)