dify/api/core/callback_handler/entity/dataset_query.py

from pydantic import BaseModel


class DatasetQueryObj(BaseModel):
    dataset_id: str = None
    query: str = None