TrainingWeightAuth

class baseten.client.managementapi.TrainingWeightAuth(*, auth_method, auth_secret_name=None, aws_oidc_role_arn=None, aws_oidc_region=None, gcp_oidc_service_account=None, gcp_oidc_workload_id_provider=None, aws_assume_role_arn=None, aws_assume_role_region=None)

Bases: BaseModel

Parameters:

• auth_method (Annotated[Literal['CUSTOM_SECRET', 'AWS_OIDC', 'GCP_OIDC', 'HUGGING_FACE_OIDC', 'AWS_ASSUME_ROLE'] | str, FieldInfo(annotation=NoneType, required=True, title='AuthMethod', description='Method used to authenticate the weight source.')])

• auth_secret_name (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Auth Secret Name', description='Name of the workspace secret used for custom-secret authentication.')])

• aws_oidc_role_arn (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Aws Oidc Role Arn', description='AWS IAM role ARN used for OIDC authentication.')])

• aws_oidc_region (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Aws Oidc Region', description='AWS region used for OIDC authentication.')])

• gcp_oidc_service_account (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Gcp Oidc Service Account', description='GCP service account used for OIDC authentication.')])

• gcp_oidc_workload_id_provider (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Gcp Oidc Workload Id Provider', description='GCP workload identity provider used for OIDC authentication.')])

• aws_assume_role_arn (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Aws Assume Role Arn', description='AWS IAM role ARN that Baseten assumes to access the weight source.')])

• aws_assume_role_region (Annotated[str | None, FieldInfo(annotation=NoneType, required=True, title='Aws Assume Role Region', description='AWS region used for assume-role authentication.')])

model_config = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].