diff --git a/dbgpt/app/knowledge/api.py b/dbgpt/app/knowledge/api.py index ee8f1f767..875dbcb18 100644 --- a/dbgpt/app/knowledge/api.py +++ b/dbgpt/app/knowledge/api.py @@ -15,11 +15,11 @@ from dbgpt.app.knowledge.request.request import ( EntityExtractRequest, KnowledgeDocumentRequest, KnowledgeQueryRequest, - KnowledgeQueryResponse, KnowledgeSpaceRequest, KnowledgeSyncRequest, SpaceArgumentRequest, ) +from dbgpt.app.knowledge.request.response import KnowledgeQueryResponse from dbgpt.app.knowledge.service import KnowledgeService from dbgpt.app.openapi.api_v1.api_v1 import no_stream_generator, stream_generator from dbgpt.app.openapi.api_view_model import Result diff --git a/dbgpt/app/knowledge/request/request.py b/dbgpt/app/knowledge/request/request.py index 0b0e2795f..fbd6a697c 100644 --- a/dbgpt/app/knowledge/request/request.py +++ b/dbgpt/app/knowledge/request/request.py @@ -110,16 +110,6 @@ class ChunkQueryRequest(BaseModel): page_size: int = 20 -class KnowledgeQueryResponse: - """source: knowledge reference source""" - - source: str - """score: knowledge vector query similarity score""" - score: float = 0.0 - """text: raw text info""" - text: str - - class SpaceArgumentRequest(BaseModel): """argument: argument""" diff --git a/dbgpt/app/knowledge/request/response.py b/dbgpt/app/knowledge/request/response.py index eb835ed86..ccdc6519f 100644 --- a/dbgpt/app/knowledge/request/response.py +++ b/dbgpt/app/knowledge/request/response.py @@ -42,3 +42,13 @@ class SpaceQueryResponse(BaseModel): gmt_modified: str = None """doc_count: doc_count""" docs: int = None + + +class KnowledgeQueryResponse(BaseModel): + """source: knowledge reference source""" + + source: str + """score: knowledge vector query similarity score""" + score: float = 0.0 + """text: raw text info""" + text: str