# ！/usr/bin/ python
# -*- coding: utf-8 -*-
'''
@Project    : lq-agent-api
@File       :cattle_farm.py
@IDE        :PyCharm
@Author     :
@Date       :2025/7/11 12:41
'''
from typing import Optional

from pydantic import BaseModel, constr, Field
from enums.common_enums import BusinessSceneEnum, ErrorCodeEnum, UserRoleEnum


class FarmConfig(BaseModel):
    sessionId: constr(max_length=128) =Field(description="会话id")
    userRole: constr(max_length=15) =Field(default=UserRoleEnum.COMMON.code,description="用户角色")



class CattleFarm(BaseModel):
    config: FarmConfig
    input: Optional[str] = Field(description="用户输入")
    businessScene: str = Field(default=None, description="业务场景")
    context: Optional[str]  = Field(default=None, description="参考上下文")
    supplementInfo: Optional[str] = Field(default=None, description="补充信息")