from pydantic import BaseModel

# 定义请求体
class RecommendRequest(BaseModel):
    city_uuid: str              # 城市id
    product_code: str           # 卷烟编码
    recall_cust_count: int      # 推荐的商户数量
    delivery_count: int         # 投放的品规数量
    cultivacation_id: str       # 投放策略id
    limit_cycle_name: str       # 投放周期名称
    
class ReportRequest(BaseModel):
    cultivacation_id: str       # 投放策略id
    
class EvalReportRequest(BaseModel):
    city_uuid: str              # 城市id
    product_code: str           # 卷烟编码
    cultivacation_id: str       # 投放策略id
    limit_cycle_name: str       # 投放周期名称