Some document API refined. (#53)

Add naive chunking method to RAG
2025-12-08 20:42:30 +08:00 · 2024-02-02 19:21:37 +08:00
parent 7b71fb2db6
commit 51482f3e2a
13 changed files with 447 additions and 268 deletions
--- a/api/db/services/document_service.py
+++ b/api/db/services/document_service.py
@ -36,6 +36,7 @@ class DocumentService(CommonService):
                cls.model.name.like(f"%%{keywords}%%"))
        else:
            docs = cls.model.select().where(cls.model.kb_id == kb_id)
+        count = docs.count()
        if desc:
            docs = docs.order_by(cls.model.getter_by(orderby).desc())
        else:
@ -43,7 +44,7 @@ class DocumentService(CommonService):

        docs = docs.paginate(page_number, items_per_page)

-        return list(docs.dicts())
+        return list(docs.dicts()), count

    @classmethod
    @DB.connection_context()