RAG/schemas/schemas.py

from pydantic import BaseModel, Field
from typing import List, Dict, Any


class ResponseWrapper(BaseModel):
    """封装的响应模型"""
    status_code: int
    detail: str
    data: Any


class Query(BaseModel):
    text: str


class ContentParam(BaseModel):
    datasetId: int
    title: str
    text: str

class DatasetParam(BaseModel):
    name: str