59 lines
1.2 KiB
Python
59 lines
1.2 KiB
Python
"""
|
|
RAG 模块 Schema 导出
|
|
|
|
集中导出所有数据模型、请求和响应 DTO
|
|
"""
|
|
from .enums import ProcessType
|
|
from app.db.models.knowledge_gen import RagType, FileStatus
|
|
from .types import RagChunk
|
|
from app.db.models.knowledge_gen import KnowledgeBase, RagFile
|
|
from .request import (
|
|
KnowledgeBaseCreateReq,
|
|
KnowledgeBaseUpdateReq,
|
|
KnowledgeBaseQueryReq,
|
|
AddFilesReq,
|
|
DeleteFilesReq,
|
|
RagFileReq,
|
|
RetrieveReq,
|
|
FileInfo,
|
|
PagingQuery,
|
|
QueryRequest,
|
|
)
|
|
from .response import (
|
|
ModelConfig,
|
|
KnowledgeBaseResp,
|
|
RagFileResp,
|
|
RagChunkResp,
|
|
SearchResult,
|
|
PagedResponse,
|
|
)
|
|
|
|
__all__ = [
|
|
# Enums
|
|
"RagType",
|
|
"ProcessType",
|
|
"FileStatus",
|
|
# Entities
|
|
"KnowledgeBase",
|
|
"RagFile",
|
|
"RagChunk",
|
|
# Requests
|
|
"KnowledgeBaseCreateReq",
|
|
"KnowledgeBaseUpdateReq",
|
|
"KnowledgeBaseQueryReq",
|
|
"AddFilesReq",
|
|
"DeleteFilesReq",
|
|
"RagFileReq",
|
|
"RetrieveReq",
|
|
"FileInfo",
|
|
"PagingQuery",
|
|
"QueryRequest",
|
|
# Responses
|
|
"ModelConfig",
|
|
"KnowledgeBaseResp",
|
|
"RagFileResp",
|
|
"RagChunkResp",
|
|
"SearchResult",
|
|
"PagedResponse",
|
|
]
|