reformat file document_db.py

eosphoros-ai · Dec 8, 2023 · 6dc3e99 · 6dc3e99
1 parent d05afe1
commit 6dc3e99
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pilot/server/knowledge/document_db.py b/pilot/server/knowledge/document_db.py
@@ -150,7 +150,7 @@ def get_knowledge_documents_count_bulk(self, space_names):
         counts_query = (
             session.query(
                 KnowledgeDocumentEntity.space,
-                func.count(KnowledgeDocumentEntity.id).label('document_count')
+                func.count(KnowledgeDocumentEntity.id).label("document_count"),
             )
             .filter(KnowledgeDocumentEntity.space.in_(space_names))
             .group_by(KnowledgeDocumentEntity.space)

diff --git a/pilot/server/knowledge/service.py b/pilot/server/knowledge/service.py
@@ -118,7 +118,9 @@ def get_knowledge_space(self, request: KnowledgeSpaceRequest):
         )
         spaces = knowledge_space_dao.get_knowledge_space(query)
         space_names = [space.name for space in spaces]
-        docs_count = knowledge_document_dao.get_knowledge_documents_count_bulk(space_names)
+        docs_count = knowledge_document_dao.get_knowledge_documents_count_bulk(
+            space_names
+        )
         responses = []
         for space in spaces:
             res = SpaceQueryResponse()