github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 15:21:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

change KnowledgeQueryResponse (#1202)

Browse Source 
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
比奇堡 扛把子 2024-02-27 11:23:25 +08:00 committed by GitHub
parent 7de98d5363
commit 08fcb4f744
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

10
dbgpt/app/knowledge/request/request.py
View File

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

10
dbgpt/app/knowledge/request/response.py
View File

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