mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-07 03:14:42 +00:00
change KnowledgeQueryResponse (#1202)
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
parent
7de98d5363
commit
08fcb4f744
@ -15,11 +15,11 @@ from dbgpt.app.knowledge.request.request import (
|
|||||||
EntityExtractRequest,
|
EntityExtractRequest,
|
||||||
KnowledgeDocumentRequest,
|
KnowledgeDocumentRequest,
|
||||||
KnowledgeQueryRequest,
|
KnowledgeQueryRequest,
|
||||||
KnowledgeQueryResponse,
|
|
||||||
KnowledgeSpaceRequest,
|
KnowledgeSpaceRequest,
|
||||||
KnowledgeSyncRequest,
|
KnowledgeSyncRequest,
|
||||||
SpaceArgumentRequest,
|
SpaceArgumentRequest,
|
||||||
)
|
)
|
||||||
|
from dbgpt.app.knowledge.request.response import KnowledgeQueryResponse
|
||||||
from dbgpt.app.knowledge.service import KnowledgeService
|
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_v1.api_v1 import no_stream_generator, stream_generator
|
||||||
from dbgpt.app.openapi.api_view_model import Result
|
from dbgpt.app.openapi.api_view_model import Result
|
||||||
|
@ -110,16 +110,6 @@ class ChunkQueryRequest(BaseModel):
|
|||||||
page_size: int = 20
|
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):
|
class SpaceArgumentRequest(BaseModel):
|
||||||
"""argument: argument"""
|
"""argument: argument"""
|
||||||
|
|
||||||
|
@ -42,3 +42,13 @@ class SpaceQueryResponse(BaseModel):
|
|||||||
gmt_modified: str = None
|
gmt_modified: str = None
|
||||||
"""doc_count: doc_count"""
|
"""doc_count: doc_count"""
|
||||||
docs: int = None
|
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
|
||||||
|
Loading…
Reference in New Issue
Block a user