)`idZddlZddlmZmZmZddlmZmZm Z m Z m Z dZ GddeZ Gdd ZGd d Zejejd d ejejdd iZdS)z+SageMaker-specific configuration constants.N)AnyDictOptional) BaseModel ConfigDictFieldfield_validatormodel_validator SAGEMAKER_ceZdZUdZedZeddZee d<edd d Z e e d <ed dZ e ee d<eddZedededeeeffdZeddededefdZed dedede fdZd 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_pathreturnc|S)zCreate SagemakerConfig from environment variables. Returns: SagemakerConfig instance with values loaded from SAGEMAKER_* env vars )clss /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/sagemaker/config.pyfrom_envzSageMakerConfig.from_env/ssuu before)modedatacdtjD}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). ci|]N\}}|t|ttd|OS)N) startswithSAGEMAKER_ENV_VAR_PREFIXlenlower).0keyvals r z6SageMakerConfig.load_from_env_vars..Bsa   S~~677 ,--// 0 6 6 8 8#   r)osenvironitems isinstancedict)rr env_configs rload_from_env_varsz"SageMakerConfig.load_from_env_vars8sX  J,,..    dD ! ! *)j)D) )rvct|tr|St|tr|dvSt|S)z/Convert string values from env vars to boolean.)true1)r.boolstrr&rr2s rparse_bool_stringz!SageMakerConfig.parse_bool_stringMsK a   H a   .7799 - -Awwrct|tr|St|trt|St|S)z/Convert string values from env vars to integer.)r.intr7r8s rparse_int_stringz SageMakerConfig.parse_int_stringWsC a   H a   q66M1vv r)rr )__name__ __module__ __qualname____doc__r model_configrrr6__annotations__rr;rrr7 classmethodrr rrr1r r9r<rrrr r s&:H---L&+U#Q&&&d %u8     $)5T$$$M8C= [_(###cd38n[$#&_/h???#$[@?_*:::[;:rr ceZdZdZdZdZdS)SageMakerEnvVarsz%SageMaker environment variable names.CUSTOM_SCRIPT_FILENAMESAGEMAKER_MODEL_PATHN)r=r>r?r@rFrGrrrrErEbs"//51rrEceZdZdZdZdZdS)SageMakerDefaultszSageMaker default values.zmodel.pyz/opt/ml/model/N)r=r>r?r@SCRIPT_FILENAME SCRIPT_PATHrrrrIrIis## O"KKKrrIz2Custom script filename to load (default: model.py)rz8SageMaker model path directory (default: /opt/ml/model/))r@r+typingrrrpydanticrrrr r r$r rErIrFrJrGrKSAGEMAKER_ENV_CONFIGrrrrOs)11 &&&&&&&&&&SSSSSSSSSSSSSS'TTTTTiTTTn22222222########+$4K..)$0Q,, r