mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-29 21:40:44 +00:00
feat: knowledge backend management
1.knowledge_type.py 2.knowledge backend api
This commit is contained in:
76
pilot/openapi/knowledge/request/knowledge_request.py
Normal file
76
pilot/openapi/knowledge/request/knowledge_request.py
Normal file
@@ -0,0 +1,76 @@
|
||||
from typing import List
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
class KnowledgeQueryRequest(BaseModel):
|
||||
"""query: knowledge query"""
|
||||
|
||||
query: str
|
||||
"""top_k: return topK documents"""
|
||||
top_k: int
|
||||
|
||||
|
||||
class KnowledgeSpaceRequest(BaseModel):
|
||||
"""name: knowledge space name"""
|
||||
|
||||
name: str = None
|
||||
"""vector_type: vector type"""
|
||||
vector_type: str = None
|
||||
"""desc: description"""
|
||||
desc: str = None
|
||||
"""owner: owner"""
|
||||
owner: str = None
|
||||
|
||||
|
||||
class KnowledgeDocumentRequest(BaseModel):
|
||||
"""doc_name: doc path"""
|
||||
|
||||
doc_name: str
|
||||
"""doc_type: doc type"""
|
||||
doc_type: str
|
||||
"""content: content"""
|
||||
content: str
|
||||
"""text_chunk_size: text_chunk_size"""
|
||||
# text_chunk_size: int
|
||||
|
||||
class DocumentQueryRequest(BaseModel):
|
||||
"""doc_name: doc path"""
|
||||
doc_name: str = None
|
||||
"""doc_type: doc type"""
|
||||
doc_type: str= None
|
||||
"""status: status"""
|
||||
status: str= None
|
||||
"""page: page"""
|
||||
page: int = 1
|
||||
"""page_size: page size"""
|
||||
page_size: int = 20
|
||||
|
||||
|
||||
class DocumentSyncRequest(BaseModel):
|
||||
"""doc_ids: doc ids"""
|
||||
doc_ids: List
|
||||
|
||||
class ChunkQueryRequest(BaseModel):
|
||||
"""id: id"""
|
||||
id: int = None
|
||||
"""document_id: doc id"""
|
||||
document_id: int = None
|
||||
"""doc_name: doc path"""
|
||||
doc_name: str = None
|
||||
"""doc_type: doc type"""
|
||||
doc_type: str = None
|
||||
"""page: page"""
|
||||
page: int = 1
|
||||
"""page_size: page size"""
|
||||
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
|
Reference in New Issue
Block a user