github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-16 22:51:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add evaluation service module for RAG and Agent (#2070)

Browse Source
This commit is contained in:
Aries-ckt
2024-10-18 17:42:11 +08:00
committed by GitHub
parent 253c367ceb
commit 811ce63493
29 changed files with 1263 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
dbgpt/serve/rag/service/service.py
View File

@@ -320,6 +320,10 @@ class Service(BaseService[KnowledgeSpaceEntity, SpaceServeRequest, SpaceServeRes
entity = self._document_dao.from_response(document)
if request.doc_name:
entity.doc_name = request.doc_name
update_chunk = self._chunk_dao.get_one({"document_id": entity.id})
if update_chunk:
update_chunk.doc_name = request.doc_name
self._chunk_dao.update_chunk(update_chunk)
if len(request.questions) == 0:
request.questions = ""
questions = [
@@ -411,13 +415,20 @@ class Service(BaseService[KnowledgeSpaceEntity, SpaceServeRequest, SpaceServeRes
"""
return self._document_dao.get_list_page(request, page, page_size)
def get_chunk_list(self, request: QUERY_SPEC, page: int, page_size: int):
"""get document chunks
def get_chunk_list_page(self, request: QUERY_SPEC, page: int, page_size: int):
"""get document chunks with page
Args:
- request: QUERY_SPEC
"""
return self._chunk_dao.get_list_page(request, page, page_size)
def get_chunk_list(self, request: QUERY_SPEC):
"""get document chunks
Args:
- request: QUERY_SPEC
"""
return self._chunk_dao.get_list(request)
def update_chunk(self, request: ChunkServeRequest):
"""update knowledge document chunk"""
if not request.id: