o )i@sdZddlZddlmZmZmZddlmZmZm Z m Z m Z dZ GdddeZ Gdd d ZGd d d Zejejd d ejejdd iZdS)z+SageMaker-specific configuration constants.N)AnyDictOptional) BaseModel ConfigDictFieldfield_validatormodel_validator SAGEMAKER_c@seZdZUdZeddZedddZee d<edd d d Z e e d <ed ddZ e ee d<edddZeddededeeeffddZedddededefddZed ddedede fddZd S)SageMakerConfigaPydantic model for SageMaker configuration. Automatically loads configuration from environment variables with SAGEMAKER_ prefix. Example: SAGEMAKER_ENABLE_STATEFUL_SESSIONS=true -> enable_stateful_sessions=True Only fields defined in this model are loaded. Other SAGEMAKER_* env vars (like SAGEMAKER_MODEL_PATH) are ignored. Usage: # Create from environment variables config = SagemakerConfig.from_env() # Or just instantiate (automatically loads from env) config = SagemakerConfig() # Override specific values config = SagemakerConfig(enable_stateful_sessions=True) ignore)extraFz,Enable stateful sessions for the applicationdefault descriptionenable_stateful_sessionsiz"Session expiration time in secondsr)rrgtsessions_expirationNz>Custom path for session storage (defaults to /dev/shm or temp) sessions_pathreturncCs|S)zCreate SagemakerConfig from environment variables. Returns: SagemakerConfig instance with values loaded from SAGEMAKER_* env vars )clsrro/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/model_hosting_container_standards/sagemaker/config.pyfrom_env/szSageMakerConfig.from_envbefore)modedatacCs.ddtjD}t|tri||S|S)a%Load configuration from environment variables. Extracts SAGEMAKER_* environment variables and merges with any provided data. Provided data takes precedence over environment variables. Unknown SAGEMAKER_* variables are ignored (only defined fields are loaded). cSs0i|]\}}|tr|ttd|qS)N) startswithSAGEMAKER_ENV_VAR_PREFIXlenlower).0keyvalrrr Bs z6SageMakerConfig.load_from_env_vars..)osenvironitems isinstancedict)rr env_configrrrload_from_env_vars8s    z"SageMakerConfig.load_from_env_varsvcCs,t|tr|St|tr|dvSt|S)z/Convert string values from env vars to boolean.)true1)r(boolstrr rr,rrrparse_bool_stringMs   z!SageMakerConfig.parse_bool_stringcCs(t|tr|St|trt|St|S)z/Convert string values from env vars to integer.)r(intr0r1rrrparse_int_stringWs  z SageMakerConfig.parse_int_string)rr )__name__ __module__ __qualname____doc__r model_configrrr/__annotations__rr3rrr0 classmethodrr rrr+rr2r4rrrrr s4     r c@eZdZdZdZdZdS)SageMakerEnvVarsz%SageMaker environment variable names.CUSTOM_SCRIPT_FILENAMESAGEMAKER_MODEL_PATHN)r5r6r7r8r>r?rrrrr=br=c@r<)SageMakerDefaultszSageMaker default values.zmodel.pyz/opt/ml/model/N)r5r6r7r8SCRIPT_FILENAME SCRIPT_PATHrrrrrAir@rAz2Custom script filename to load (default: model.py)rz8SageMaker model path directory (default: /opt/ml/model/))r8r%typingrrrpydanticrrrrr rr r=rAr>rBr?rCSAGEMAKER_ENV_CONFIGrrrrs"W