change KnowledgeQueryResponse

dbgpt/app/knowledge/api.py

@@ -15,11 +15,11 @@
     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

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"""
 

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