github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-06 10:54:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:space return doc_count

Browse Source
This commit is contained in:
aries_ckt 2023-07-04 11:39:57 +08:00
parent 723cd22507
commit 2e68d6e28d
2 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pilot/server/knowledge/request/response.py
View File

@ -21,3 +21,17 @@ class DocumentQueryResponse(BaseModel):
total: int = None
"""page: current page"""
page: int = None
class SpaceQueryResponse(BaseModel):
"""data: data"""
name: str = None
"""vector_type: vector type"""
vector_type: str = None
"""desc: description"""
desc: str = None
"""owner: owner"""
owner: str = None
"""doc_count: doc_count"""
doc_count: int = None

15
pilot/server/knowledge/service.py
View File

@ -27,7 +27,7 @@ from enum import Enum
from pilot.server.knowledge.request.response import (
ChunkQueryResponse,
DocumentQueryResponse,
DocumentQueryResponse, SpaceQueryResponse,
)
knowledge_space_dao = KnowledgeSpaceDao()
@ -86,7 +86,18 @@ class KnowledgeService:
query = KnowledgeSpaceEntity(
name=request.name, vector_type=request.vector_type, owner=request.owner
)
return knowledge_space_dao.get_knowledge_space(query)
res = SpaceQueryResponse()
space: KnowledgeSpaceEntity = knowledge_space_dao.get_knowledge_space(query)
res.name = space.name
res.vector_type = space.vector_type
res.desc = space.desc
query = KnowledgeDocumentEntity(
space=space,
)
res.doc_count = knowledge_document_dao.get_knowledge_documents_count(query)
return res
"""get knowledge get_knowledge_documents"""