TanWenyan/leaudit-platform-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: restore rag dataset management and linkage

Browse Source
This commit is contained in:
wren
2026-05-11 17:21:33 +08:00
parent da2bb8310d
commit dcc0f3c30d
6 changed files with 2208 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastapi_modules/fastapi_leaudit/domian/Dto/ragDatasetDto.py
+6
View File
@@ -0,0 +1,6 @@
from pydantic import BaseModel, Field
class RagDatasetUpdateDTO(BaseModel):
name: str | None = Field(None, min_length=1, max_length=255)
retrieval_model: dict | None = Field(None)
fastapi_modules/fastapi_leaudit/domian/vo/ragDatasetVo.py
+106
View File
@@ -11,8 +11,114 @@ class RagDatasetItemVO(BaseModel):
documentCount: int = Field(0)
totalChunks: int = Field(0)
status: int = Field(1)
sortOrder: int = Field(0)
createdAt: int = Field(0)
updatedAt: int = Field(0)
appId: int | None = Field(default=None)
appName: str = Field("")
appIsDefault: bool = Field(False)
class RagDatasetPageVO(BaseModel):
data: list[RagDatasetItemVO] = Field(default_factory=list)
total: int = Field(0)
class RagDatasetDetailVO(BaseModel):
id: int = Field(...)
name: str = Field(...)
description: str = Field("")
area: str = Field("")
isPublic: bool = Field(False)
isDefault: bool = Field(False)
status: int = Field(1)
documentCount: int = Field(0)
totalChunks: int = Field(0)
chunkMaxSize: int = Field(800)
chunkMinSize: int = Field(20)
sortOrder: int = Field(0)
retrievalModel: dict = Field(default_factory=dict)
createdAt: int = Field(0)
updatedAt: int = Field(0)
appId: int | None = Field(default=None)
appName: str = Field("")
appIsDefault: bool = Field(False)
class RagDatasetDocumentItemVO(BaseModel):
id: int = Field(...)
datasetId: int = Field(...)
name: str = Field(...)
fileType: str = Field("")
fileSize: int = Field(0)
chunkCount: int = Field(0)
indexingStatus: str = Field("waiting")
error: str = Field("")
enabled: bool = Field(True)
hitCount: int = Field(0)
createdBy: int | None = Field(None)
createdAt: int = Field(0)
updatedAt: int = Field(0)
class RagDatasetDocumentPageVO(BaseModel):
data: list[RagDatasetDocumentItemVO] = Field(default_factory=list)
total: int = Field(0)
page: int = Field(1)
limit: int = Field(20)
hasMore: bool = Field(False)
class RagDatasetUploadDocumentVO(BaseModel):
document: dict = Field(default_factory=dict)
batch: str = Field("")
class RagDatasetSegmentItemVO(BaseModel):
id: str = Field(...)
position: int = Field(0)
documentId: str = Field("")
content: str = Field("")
wordCount: int = Field(0)
hitCount: int = Field(0)
enabled: bool = Field(True)
status: str = Field("completed")
createdAt: int = Field(0)
class RagDatasetSegmentPageVO(BaseModel):
data: list[RagDatasetSegmentItemVO] = Field(default_factory=list)
total: int = Field(0)
limit: int = Field(20)
hasMore: bool = Field(False)
class RagDatasetRetrieveDocumentVO(BaseModel):
id: str = Field("")
dataSourceType: str = Field("upload_file")
name: str = Field("")
docType: str | None = Field(default=None)
class RagDatasetRetrieveSegmentVO(BaseModel):
id: str = Field(...)
position: int = Field(0)
documentId: str = Field("")
content: str = Field("")
answer: str = Field("")
wordCount: int = Field(0)
hitCount: int = Field(0)
enabled: bool = Field(True)
status: str = Field("completed")
createdAt: int = Field(0)
document: RagDatasetRetrieveDocumentVO | None = Field(default=None)
class RagDatasetRetrieveRecordVO(BaseModel):
segment: RagDatasetRetrieveSegmentVO = Field(...)
score: float = Field(0.0)
class RagDatasetRetrieveResponseVO(BaseModel):
query: dict = Field(default_factory=dict)
records: list[RagDatasetRetrieveRecordVO] = Field(default_factory=list)