from pydantic import BaseModel

from airda.framework.agent.module.planner.pipeline.pipeline_params import PipelineParams
from airda.framework.agent.module.planner.planner_params import PlannerParams


class DataAgentPlannerParams(BaseModel, PlannerParams, PipelineParams):
    question: str
    datasource_name: str
