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

8
dbgpt/app/initialization/serve_initialization.py
View File

@@ -122,3 +122,11 @@ def register_serve_apps(system_app: SystemApp, cfg: Config, webserver_port: int)
system_app.register(FileServe)
# ################################ File Serve Register End ########################################
# ################################ Evaluate Serve Register Begin #######################################
from dbgpt.serve.evaluate.serve import Serve as EvaluateServe
# Register serve Evaluate
system_app.register(EvaluateServe)
# ################################ Evaluate Serve Register End ########################################

2
dbgpt/app/knowledge/api.py
View File

@@ -447,7 +447,7 @@ def chunk_list(
"doc_type": query_request.doc_type,
"content": query_request.content,
}
chunk_res = service.get_chunk_list(
chunk_res = service.get_chunk_list_page(
query, query_request.page, query_request.page_size
)
res = ChunkQueryResponse(