types-boto3-full 1.35.72__py3-none-any.whl → 1.35.75__py3-none-any.whl

types_boto3_bedrock/type_defs.py

@@ -33,6 +33,7 @@ from .literals import (
     GuardrailContentFilterTypeType,
     GuardrailContextualGroundingFilterTypeType,
     GuardrailFilterStrengthType,
+    GuardrailModalityType,
     GuardrailPiiEntityTypeType,
     GuardrailSensitiveInformationActionType,
     GuardrailStatusType,
@@ -44,10 +45,12 @@ from .literals import (
     ModelImportJobStatusType,
     ModelInvocationJobStatusType,
     ModelModalityType,
+    PromptRouterTypeType,
     ProvisionedModelStatusType,
     RetrieveAndGenerateTypeType,
     SearchTypeType,
     SortOrderType,
+    StatusType,
 )
 
 if sys.version_info >= (3, 12):
@@ -78,6 +81,8 @@ __all__ = (
     "CreateGuardrailVersionResponseTypeDef",
     "CreateInferenceProfileRequestRequestTypeDef",
     "CreateInferenceProfileResponseTypeDef",
+    "CreateMarketplaceModelEndpointRequestRequestTypeDef",
+    "CreateMarketplaceModelEndpointResponseTypeDef",
     "CreateModelCopyJobRequestRequestTypeDef",
     "CreateModelCopyJobResponseTypeDef",
     "CreateModelCustomizationJobRequestRequestTypeDef",
@@ -89,11 +94,17 @@ __all__ = (
     "CreateProvisionedModelThroughputRequestRequestTypeDef",
     "CreateProvisionedModelThroughputResponseTypeDef",
     "CustomModelSummaryTypeDef",
+    "CustomizationConfigTypeDef",
     "DeleteCustomModelRequestRequestTypeDef",
     "DeleteGuardrailRequestRequestTypeDef",
     "DeleteImportedModelRequestRequestTypeDef",
     "DeleteInferenceProfileRequestRequestTypeDef",
+    "DeleteMarketplaceModelEndpointRequestRequestTypeDef",
     "DeleteProvisionedModelThroughputRequestRequestTypeDef",
+    "DeregisterMarketplaceModelEndpointRequestRequestTypeDef",
+    "DistillationConfigTypeDef",
+    "EndpointConfigOutputTypeDef",
+    "EndpointConfigTypeDef",
     "EvaluationBedrockModelTypeDef",
     "EvaluationConfigOutputTypeDef",
     "EvaluationConfigTypeDef",
@@ -140,6 +151,8 @@ __all__ = (
     "GetImportedModelResponseTypeDef",
     "GetInferenceProfileRequestRequestTypeDef",
     "GetInferenceProfileResponseTypeDef",
+    "GetMarketplaceModelEndpointRequestRequestTypeDef",
+    "GetMarketplaceModelEndpointResponseTypeDef",
     "GetModelCopyJobRequestRequestTypeDef",
     "GetModelCopyJobResponseTypeDef",
     "GetModelCustomizationJobRequestRequestTypeDef",
@@ -149,6 +162,8 @@ __all__ = (
     "GetModelInvocationJobRequestRequestTypeDef",
     "GetModelInvocationJobResponseTypeDef",
     "GetModelInvocationLoggingConfigurationResponseTypeDef",
+    "GetPromptRouterRequestRequestTypeDef",
+    "GetPromptRouterResponseTypeDef",
     "GetProvisionedModelThroughputRequestRequestTypeDef",
     "GetProvisionedModelThroughputResponseTypeDef",
     "GuardrailConfigurationTypeDef",
@@ -186,6 +201,10 @@ __all__ = (
     "InferenceProfileModelSourceTypeDef",
     "InferenceProfileModelTypeDef",
     "InferenceProfileSummaryTypeDef",
+    "InvocationLogSourceTypeDef",
+    "InvocationLogsConfigOutputTypeDef",
+    "InvocationLogsConfigTypeDef",
+    "InvocationLogsConfigUnionTypeDef",
     "KbInferenceConfigOutputTypeDef",
     "KbInferenceConfigTypeDef",
     "KbInferenceConfigUnionTypeDef",
@@ -218,6 +237,9 @@ __all__ = (
     "ListInferenceProfilesRequestListInferenceProfilesPaginateTypeDef",
     "ListInferenceProfilesRequestRequestTypeDef",
     "ListInferenceProfilesResponseTypeDef",
+    "ListMarketplaceModelEndpointsRequestListMarketplaceModelEndpointsPaginateTypeDef",
+    "ListMarketplaceModelEndpointsRequestRequestTypeDef",
+    "ListMarketplaceModelEndpointsResponseTypeDef",
     "ListModelCopyJobsRequestListModelCopyJobsPaginateTypeDef",
     "ListModelCopyJobsRequestRequestTypeDef",
     "ListModelCopyJobsResponseTypeDef",
@@ -230,12 +252,17 @@ __all__ = (
     "ListModelInvocationJobsRequestListModelInvocationJobsPaginateTypeDef",
     "ListModelInvocationJobsRequestRequestTypeDef",
     "ListModelInvocationJobsResponseTypeDef",
+    "ListPromptRoutersRequestListPromptRoutersPaginateTypeDef",
+    "ListPromptRoutersRequestRequestTypeDef",
+    "ListPromptRoutersResponseTypeDef",
     "ListProvisionedModelThroughputsRequestListProvisionedModelThroughputsPaginateTypeDef",
     "ListProvisionedModelThroughputsRequestRequestTypeDef",
     "ListProvisionedModelThroughputsResponseTypeDef",
     "ListTagsForResourceRequestRequestTypeDef",
     "ListTagsForResourceResponseTypeDef",
     "LoggingConfigTypeDef",
+    "MarketplaceModelEndpointSummaryTypeDef",
+    "MarketplaceModelEndpointTypeDef",
     "ModelCopyJobSummaryTypeDef",
     "ModelCustomizationJobSummaryTypeDef",
     "ModelDataSourceTypeDef",
@@ -248,6 +275,8 @@ __all__ = (
     "OrchestrationConfigurationTypeDef",
     "OutputDataConfigTypeDef",
     "PaginatorConfigTypeDef",
+    "PromptRouterSummaryTypeDef",
+    "PromptRouterTargetModelTypeDef",
     "PromptTemplateTypeDef",
     "ProvisionedModelSummaryTypeDef",
     "PutModelInvocationLoggingConfigurationRequestRequestTypeDef",
@@ -255,6 +284,14 @@ __all__ = (
     "RAGConfigOutputTypeDef",
     "RAGConfigTypeDef",
     "RAGConfigUnionTypeDef",
+    "RegisterMarketplaceModelEndpointRequestRequestTypeDef",
+    "RegisterMarketplaceModelEndpointResponseTypeDef",
+    "RequestMetadataBaseFiltersOutputTypeDef",
+    "RequestMetadataBaseFiltersTypeDef",
+    "RequestMetadataBaseFiltersUnionTypeDef",
+    "RequestMetadataFiltersOutputTypeDef",
+    "RequestMetadataFiltersTypeDef",
+    "RequestMetadataFiltersUnionTypeDef",
     "ResponseMetadataTypeDef",
     "RetrievalFilterOutputTypeDef",
     "RetrievalFilterTypeDef",
@@ -265,23 +302,31 @@ __all__ = (
     "RetrieveConfigOutputTypeDef",
     "RetrieveConfigTypeDef",
     "RetrieveConfigUnionTypeDef",
+    "RoutingCriteriaTypeDef",
     "S3ConfigTypeDef",
     "S3DataSourceTypeDef",
     "S3ObjectDocTypeDef",
+    "SageMakerEndpointOutputTypeDef",
+    "SageMakerEndpointTypeDef",
+    "SageMakerEndpointUnionTypeDef",
     "StopEvaluationJobRequestRequestTypeDef",
     "StopModelCustomizationJobRequestRequestTypeDef",
     "StopModelInvocationJobRequestRequestTypeDef",
     "TagResourceRequestRequestTypeDef",
     "TagTypeDef",
+    "TeacherModelConfigTypeDef",
     "TextInferenceConfigOutputTypeDef",
     "TextInferenceConfigTypeDef",
     "TextInferenceConfigUnionTypeDef",
     "TimestampTypeDef",
+    "TrainingDataConfigOutputTypeDef",
     "TrainingDataConfigTypeDef",
     "TrainingMetricsTypeDef",
     "UntagResourceRequestRequestTypeDef",
     "UpdateGuardrailRequestRequestTypeDef",
     "UpdateGuardrailResponseTypeDef",
+    "UpdateMarketplaceModelEndpointRequestRequestTypeDef",
+    "UpdateMarketplaceModelEndpointResponseTypeDef",
     "UpdateProvisionedModelThroughputRequestRequestTypeDef",
     "ValidationDataConfigOutputTypeDef",
     "ValidationDataConfigTypeDef",
@@ -289,6 +334,7 @@ __all__ = (
     "ValidatorTypeDef",
     "VpcConfigOutputTypeDef",
     "VpcConfigTypeDef",
+    "VpcConfigUnionTypeDef",
 )
 
 
@@ -356,10 +402,6 @@ class OutputDataConfigTypeDef(TypedDict):
     s3Uri: str
 
 
-class TrainingDataConfigTypeDef(TypedDict):
-    s3Uri: str
-
-
 class VpcConfigTypeDef(TypedDict):
     subnetIds: Sequence[str]
     securityGroupIds: Sequence[str]
@@ -392,10 +434,23 @@ class DeleteInferenceProfileRequestRequestTypeDef(TypedDict):
     inferenceProfileIdentifier: str
 
 
+class DeleteMarketplaceModelEndpointRequestRequestTypeDef(TypedDict):
+    endpointArn: str
+
+
 class DeleteProvisionedModelThroughputRequestRequestTypeDef(TypedDict):
     provisionedModelId: str
 
 
+class DeregisterMarketplaceModelEndpointRequestRequestTypeDef(TypedDict):
+    endpointArn: str
+
+
+class TeacherModelConfigTypeDef(TypedDict):
+    teacherModelIdentifier: str
+    maxResponseLengthForInference: NotRequired[int]
+
+
 class EvaluationBedrockModelTypeDef(TypedDict):
     modelIdentifier: str
     inferenceParams: NotRequired[str]
@@ -482,6 +537,10 @@ class InferenceProfileModelTypeDef(TypedDict):
     modelArn: NotRequired[str]
 
 
+class GetMarketplaceModelEndpointRequestRequestTypeDef(TypedDict):
+    endpointArn: str
+
+
 class GetModelCopyJobRequestRequestTypeDef(TypedDict):
     jobArn: str
 
@@ -503,6 +562,18 @@ class GetModelInvocationJobRequestRequestTypeDef(TypedDict):
     jobIdentifier: str
 
 
+class GetPromptRouterRequestRequestTypeDef(TypedDict):
+    promptRouterArn: str
+
+
+class PromptRouterTargetModelTypeDef(TypedDict):
+    modelArn: NotRequired[str]
+
+
+class RoutingCriteriaTypeDef(TypedDict):
+    responseQualityDifference: float
+
+
 class GetProvisionedModelThroughputRequestRequestTypeDef(TypedDict):
     provisionedModelId: str
 
@@ -513,6 +584,8 @@ GuardrailContentFilterConfigTypeDef = TypedDict(
         "type": GuardrailContentFilterTypeType,
         "inputStrength": GuardrailFilterStrengthType,
         "outputStrength": GuardrailFilterStrengthType,
+        "inputModalities": NotRequired[Sequence[GuardrailModalityType]],
+        "outputModalities": NotRequired[Sequence[GuardrailModalityType]],
     },
 )
 GuardrailContentFilterTypeDef = TypedDict(
@@ -521,6 +594,8 @@ GuardrailContentFilterTypeDef = TypedDict(
         "type": GuardrailContentFilterTypeType,
         "inputStrength": GuardrailFilterStrengthType,
         "outputStrength": GuardrailFilterStrengthType,
+        "inputModalities": NotRequired[List[GuardrailModalityType]],
+        "outputModalities": NotRequired[List[GuardrailModalityType]],
     },
 )
 GuardrailContextualGroundingFilterConfigTypeDef = TypedDict(
@@ -639,6 +714,10 @@ class ImportedModelSummaryTypeDef(TypedDict):
     modelArchitecture: NotRequired[str]
 
 
+class InvocationLogSourceTypeDef(TypedDict):
+    s3Uri: NotRequired[str]
+
+
 class TextInferenceConfigOutputTypeDef(TypedDict):
     temperature: NotRequired[float]
     topP: NotRequired[float]
@@ -674,6 +753,21 @@ class ListInferenceProfilesRequestRequestTypeDef(TypedDict):
     typeEquals: NotRequired[InferenceProfileTypeType]
 
 
+class ListMarketplaceModelEndpointsRequestRequestTypeDef(TypedDict):
+    maxResults: NotRequired[int]
+    nextToken: NotRequired[str]
+    modelSourceEquals: NotRequired[str]
+
+
+class MarketplaceModelEndpointSummaryTypeDef(TypedDict):
+    endpointArn: str
+    modelSourceIdentifier: str
+    createdAt: datetime
+    updatedAt: datetime
+    status: NotRequired[StatusType]
+    statusMessage: NotRequired[str]
+
+
 class ModelCustomizationJobSummaryTypeDef(TypedDict):
     jobArn: str
     baseModelArn: str
@@ -698,6 +792,11 @@ class ModelImportJobSummaryTypeDef(TypedDict):
     importedModelName: NotRequired[str]
 
 
+class ListPromptRoutersRequestRequestTypeDef(TypedDict):
+    maxResults: NotRequired[int]
+    nextToken: NotRequired[str]
+
+
 class ProvisionedModelSummaryTypeDef(TypedDict):
     provisionedModelName: str
     provisionedModelArn: str
@@ -741,6 +840,21 @@ QueryTransformationConfigurationTypeDef = TypedDict(
 )
 
 
+class RegisterMarketplaceModelEndpointRequestRequestTypeDef(TypedDict):
+    endpointIdentifier: str
+    modelSourceIdentifier: str
+
+
+class RequestMetadataBaseFiltersOutputTypeDef(TypedDict):
+    equals: NotRequired[Dict[str, str]]
+    notEquals: NotRequired[Dict[str, str]]
+
+
+class RequestMetadataBaseFiltersTypeDef(TypedDict):
+    equals: NotRequired[Mapping[str, str]]
+    notEquals: NotRequired[Mapping[str, str]]
+
+
 class StopEvaluationJobRequestRequestTypeDef(TypedDict):
     jobIdentifier: str
 
@@ -946,6 +1060,10 @@ class ListCustomModelsResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class DistillationConfigTypeDef(TypedDict):
+    teacherModelConfig: TeacherModelConfigTypeDef
+
+
 class EvaluationModelConfigTypeDef(TypedDict):
     bedrockModel: NotRequired[EvaluationBedrockModelTypeDef]
 
@@ -1045,6 +1163,48 @@ InferenceProfileSummaryTypeDef = TypedDict(
 )
 
 
+class SageMakerEndpointOutputTypeDef(TypedDict):
+    initialInstanceCount: int
+    instanceType: str
+    executionRole: str
+    kmsEncryptionKey: NotRequired[str]
+    vpc: NotRequired[VpcConfigOutputTypeDef]
+
+
+VpcConfigUnionTypeDef = Union[VpcConfigTypeDef, VpcConfigOutputTypeDef]
+GetPromptRouterResponseTypeDef = TypedDict(
+    "GetPromptRouterResponseTypeDef",
+    {
+        "promptRouterName": str,
+        "routingCriteria": RoutingCriteriaTypeDef,
+        "description": str,
+        "createdAt": datetime,
+        "updatedAt": datetime,
+        "promptRouterArn": str,
+        "models": List[PromptRouterTargetModelTypeDef],
+        "fallbackModel": PromptRouterTargetModelTypeDef,
+        "status": Literal["AVAILABLE"],
+        "type": PromptRouterTypeType,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
+PromptRouterSummaryTypeDef = TypedDict(
+    "PromptRouterSummaryTypeDef",
+    {
+        "promptRouterName": str,
+        "routingCriteria": RoutingCriteriaTypeDef,
+        "promptRouterArn": str,
+        "models": List[PromptRouterTargetModelTypeDef],
+        "fallbackModel": PromptRouterTargetModelTypeDef,
+        "status": Literal["AVAILABLE"],
+        "type": PromptRouterTypeType,
+        "description": NotRequired[str],
+        "createdAt": NotRequired[datetime],
+        "updatedAt": NotRequired[datetime],
+    },
+)
+
+
 class GuardrailContentPolicyConfigTypeDef(TypedDict):
     filtersConfig: Sequence[GuardrailContentFilterConfigTypeDef]
 
@@ -1115,6 +1275,15 @@ class ListInferenceProfilesRequestListInferenceProfilesPaginateTypeDef(TypedDict
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
+class ListMarketplaceModelEndpointsRequestListMarketplaceModelEndpointsPaginateTypeDef(TypedDict):
+    modelSourceEquals: NotRequired[str]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
+class ListPromptRoutersRequestListPromptRoutersPaginateTypeDef(TypedDict):
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
 class ListCustomModelsRequestListCustomModelsPaginateTypeDef(TypedDict):
     creationTimeBefore: NotRequired[TimestampTypeDef]
     creationTimeAfter: NotRequired[TimestampTypeDef]
@@ -1295,6 +1464,12 @@ class ListProvisionedModelThroughputsRequestRequestTypeDef(TypedDict):
     sortOrder: NotRequired[SortOrderType]
 
 
+class ListMarketplaceModelEndpointsResponseTypeDef(TypedDict):
+    marketplaceModelEndpoints: List[MarketplaceModelEndpointSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    nextToken: NotRequired[str]
+
+
 class ListModelCustomizationJobsResponseTypeDef(TypedDict):
     modelCustomizationJobSummaries: List[ModelCustomizationJobSummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -1329,6 +1504,16 @@ class OrchestrationConfigurationTypeDef(TypedDict):
     queryTransformationConfiguration: QueryTransformationConfigurationTypeDef
 
 
+class RequestMetadataFiltersOutputTypeDef(TypedDict):
+    equals: NotRequired[Dict[str, str]]
+    notEquals: NotRequired[Dict[str, str]]
+    andAll: NotRequired[List[RequestMetadataBaseFiltersOutputTypeDef]]
+    orAll: NotRequired[List[RequestMetadataBaseFiltersOutputTypeDef]]
+
+
+RequestMetadataBaseFiltersUnionTypeDef = Union[
+    RequestMetadataBaseFiltersTypeDef, RequestMetadataBaseFiltersOutputTypeDef
+]
 TextInferenceConfigUnionTypeDef = Union[
     TextInferenceConfigTypeDef, TextInferenceConfigOutputTypeDef
 ]
@@ -1354,6 +1539,7 @@ class LoggingConfigTypeDef(TypedDict):
     textDataDeliveryEnabled: NotRequired[bool]
     imageDataDeliveryEnabled: NotRequired[bool]
     embeddingDataDeliveryEnabled: NotRequired[bool]
+    videoDataDeliveryEnabled: NotRequired[bool]
 
 
 class ListModelCopyJobsResponseTypeDef(TypedDict):
@@ -1362,6 +1548,10 @@ class ListModelCopyJobsResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class CustomizationConfigTypeDef(TypedDict):
+    distillationConfig: NotRequired[DistillationConfigTypeDef]
+
+
 class EvaluationDatasetMetricConfigOutputTypeDef(TypedDict):
     taskType: EvaluationTaskTypeType
     dataset: EvaluationDatasetTypeDef
@@ -1418,6 +1608,24 @@ class ListInferenceProfilesResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class EndpointConfigOutputTypeDef(TypedDict):
+    sageMaker: NotRequired[SageMakerEndpointOutputTypeDef]
+
+
+class SageMakerEndpointTypeDef(TypedDict):
+    initialInstanceCount: int
+    instanceType: str
+    executionRole: str
+    kmsEncryptionKey: NotRequired[str]
+    vpc: NotRequired[VpcConfigUnionTypeDef]
+
+
+class ListPromptRoutersResponseTypeDef(TypedDict):
+    promptRouterSummaries: List[PromptRouterSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    nextToken: NotRequired[str]
+
+
 class CreateGuardrailRequestRequestTypeDef(TypedDict):
     name: str
     blockedInputMessaging: str
@@ -1574,68 +1782,21 @@ class ModelInvocationJobSummaryTypeDef(TypedDict):
     jobExpirationTime: NotRequired[datetime]
 
 
-class KbInferenceConfigTypeDef(TypedDict):
-    textInferenceConfig: NotRequired[TextInferenceConfigUnionTypeDef]
-
-
-class GetCustomModelResponseTypeDef(TypedDict):
-    modelArn: str
-    modelName: str
-    jobName: str
-    jobArn: str
-    baseModelArn: str
-    customizationType: CustomizationTypeType
-    modelKmsKeyArn: str
-    hyperParameters: Dict[str, str]
-    trainingDataConfig: TrainingDataConfigTypeDef
-    validationDataConfig: ValidationDataConfigOutputTypeDef
-    outputDataConfig: OutputDataConfigTypeDef
-    trainingMetrics: TrainingMetricsTypeDef
-    validationMetrics: List[ValidatorMetricTypeDef]
-    creationTime: datetime
-    ResponseMetadata: ResponseMetadataTypeDef
+class InvocationLogsConfigOutputTypeDef(TypedDict):
+    invocationLogSource: InvocationLogSourceTypeDef
+    usePromptResponse: NotRequired[bool]
+    requestMetadataFilters: NotRequired[RequestMetadataFiltersOutputTypeDef]
 
 
-class GetModelCustomizationJobResponseTypeDef(TypedDict):
-    jobArn: str
-    jobName: str
-    outputModelName: str
-    outputModelArn: str
-    clientRequestToken: str
-    roleArn: str
-    status: ModelCustomizationJobStatusType
-    failureMessage: str
-    creationTime: datetime
-    lastModifiedTime: datetime
-    endTime: datetime
-    baseModelArn: str
-    hyperParameters: Dict[str, str]
-    trainingDataConfig: TrainingDataConfigTypeDef
-    validationDataConfig: ValidationDataConfigOutputTypeDef
-    outputDataConfig: OutputDataConfigTypeDef
-    customizationType: CustomizationTypeType
-    outputModelKmsKeyArn: str
-    trainingMetrics: TrainingMetricsTypeDef
-    validationMetrics: List[ValidatorMetricTypeDef]
-    vpcConfig: VpcConfigOutputTypeDef
-    ResponseMetadata: ResponseMetadataTypeDef
+class RequestMetadataFiltersTypeDef(TypedDict):
+    equals: NotRequired[Mapping[str, str]]
+    notEquals: NotRequired[Mapping[str, str]]
+    andAll: NotRequired[Sequence[RequestMetadataBaseFiltersUnionTypeDef]]
+    orAll: NotRequired[Sequence[RequestMetadataBaseFiltersTypeDef]]
 
 
-class CreateModelCustomizationJobRequestRequestTypeDef(TypedDict):
-    jobName: str
-    customModelName: str
-    roleArn: str
-    baseModelIdentifier: str
-    trainingDataConfig: TrainingDataConfigTypeDef
-    outputDataConfig: OutputDataConfigTypeDef
-    hyperParameters: Mapping[str, str]
-    clientRequestToken: NotRequired[str]
-    customizationType: NotRequired[CustomizationTypeType]
-    customModelKmsKeyId: NotRequired[str]
-    jobTags: NotRequired[Sequence[TagTypeDef]]
-    customModelTags: NotRequired[Sequence[TagTypeDef]]
-    validationDataConfig: NotRequired[ValidationDataConfigTypeDef]
-    vpcConfig: NotRequired[VpcConfigTypeDef]
+class KbInferenceConfigTypeDef(TypedDict):
+    textInferenceConfig: NotRequired[TextInferenceConfigUnionTypeDef]
 
 
 class ExternalSourceTypeDef(TypedDict):
@@ -1682,6 +1843,21 @@ class KnowledgeBaseRetrievalConfigurationOutputTypeDef(TypedDict):
 RetrievalFilterUnionTypeDef = Union[RetrievalFilterTypeDef, RetrievalFilterOutputTypeDef]
 
 
+class MarketplaceModelEndpointTypeDef(TypedDict):
+    endpointArn: str
+    modelSourceIdentifier: str
+    createdAt: datetime
+    updatedAt: datetime
+    endpointConfig: EndpointConfigOutputTypeDef
+    endpointStatus: str
+    status: NotRequired[StatusType]
+    statusMessage: NotRequired[str]
+    endpointStatusMessage: NotRequired[str]
+
+
+SageMakerEndpointUnionTypeDef = Union[SageMakerEndpointTypeDef, SageMakerEndpointOutputTypeDef]
+
+
 class ExternalSourcesRetrieveAndGenerateConfigurationOutputTypeDef(TypedDict):
     modelArn: str
     sources: List[ExternalSourceOutputTypeDef]
@@ -1694,6 +1870,14 @@ class ListModelInvocationJobsResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class TrainingDataConfigOutputTypeDef(TypedDict):
+    s3Uri: NotRequired[str]
+    invocationLogsConfig: NotRequired[InvocationLogsConfigOutputTypeDef]
+
+
+RequestMetadataFiltersUnionTypeDef = Union[
+    RequestMetadataFiltersTypeDef, RequestMetadataFiltersOutputTypeDef
+]
 KbInferenceConfigUnionTypeDef = Union[KbInferenceConfigTypeDef, KbInferenceConfigOutputTypeDef]
 ExternalSourceUnionTypeDef = Union[ExternalSourceTypeDef, ExternalSourceOutputTypeDef]
 
@@ -1736,6 +1920,81 @@ KnowledgeBaseVectorSearchConfigurationTypeDef = TypedDict(
 )
 
 
+class CreateMarketplaceModelEndpointResponseTypeDef(TypedDict):
+    marketplaceModelEndpoint: MarketplaceModelEndpointTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class GetMarketplaceModelEndpointResponseTypeDef(TypedDict):
+    marketplaceModelEndpoint: MarketplaceModelEndpointTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class RegisterMarketplaceModelEndpointResponseTypeDef(TypedDict):
+    marketplaceModelEndpoint: MarketplaceModelEndpointTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class UpdateMarketplaceModelEndpointResponseTypeDef(TypedDict):
+    marketplaceModelEndpoint: MarketplaceModelEndpointTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class EndpointConfigTypeDef(TypedDict):
+    sageMaker: NotRequired[SageMakerEndpointUnionTypeDef]
+
+
+class GetCustomModelResponseTypeDef(TypedDict):
+    modelArn: str
+    modelName: str
+    jobName: str
+    jobArn: str
+    baseModelArn: str
+    customizationType: CustomizationTypeType
+    modelKmsKeyArn: str
+    hyperParameters: Dict[str, str]
+    trainingDataConfig: TrainingDataConfigOutputTypeDef
+    validationDataConfig: ValidationDataConfigOutputTypeDef
+    outputDataConfig: OutputDataConfigTypeDef
+    trainingMetrics: TrainingMetricsTypeDef
+    validationMetrics: List[ValidatorMetricTypeDef]
+    creationTime: datetime
+    customizationConfig: CustomizationConfigTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class GetModelCustomizationJobResponseTypeDef(TypedDict):
+    jobArn: str
+    jobName: str
+    outputModelName: str
+    outputModelArn: str
+    clientRequestToken: str
+    roleArn: str
+    status: ModelCustomizationJobStatusType
+    failureMessage: str
+    creationTime: datetime
+    lastModifiedTime: datetime
+    endTime: datetime
+    baseModelArn: str
+    hyperParameters: Dict[str, str]
+    trainingDataConfig: TrainingDataConfigOutputTypeDef
+    validationDataConfig: ValidationDataConfigOutputTypeDef
+    outputDataConfig: OutputDataConfigTypeDef
+    customizationType: CustomizationTypeType
+    outputModelKmsKeyArn: str
+    trainingMetrics: TrainingMetricsTypeDef
+    validationMetrics: List[ValidatorMetricTypeDef]
+    vpcConfig: VpcConfigOutputTypeDef
+    customizationConfig: CustomizationConfigTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class InvocationLogsConfigTypeDef(TypedDict):
+    invocationLogSource: InvocationLogSourceTypeDef
+    usePromptResponse: NotRequired[bool]
+    requestMetadataFilters: NotRequired[RequestMetadataFiltersUnionTypeDef]
+
+
 class ExternalSourcesGenerationConfigurationTypeDef(TypedDict):
     promptTemplate: NotRequired[PromptTemplateTypeDef]
     guardrailConfiguration: NotRequired[GuardrailConfigurationTypeDef]
@@ -1769,6 +2028,26 @@ KnowledgeBaseVectorSearchConfigurationUnionTypeDef = Union[
     KnowledgeBaseVectorSearchConfigurationTypeDef,
     KnowledgeBaseVectorSearchConfigurationOutputTypeDef,
 ]
+
+
+class CreateMarketplaceModelEndpointRequestRequestTypeDef(TypedDict):
+    modelSourceIdentifier: str
+    endpointConfig: EndpointConfigTypeDef
+    endpointName: str
+    acceptEula: NotRequired[bool]
+    clientRequestToken: NotRequired[str]
+    tags: NotRequired[Sequence[TagTypeDef]]
+
+
+class UpdateMarketplaceModelEndpointRequestRequestTypeDef(TypedDict):
+    endpointArn: str
+    endpointConfig: EndpointConfigTypeDef
+    clientRequestToken: NotRequired[str]
+
+
+InvocationLogsConfigUnionTypeDef = Union[
+    InvocationLogsConfigTypeDef, InvocationLogsConfigOutputTypeDef
+]
 ExternalSourcesGenerationConfigurationUnionTypeDef = Union[
     ExternalSourcesGenerationConfigurationTypeDef,
     ExternalSourcesGenerationConfigurationOutputTypeDef,
@@ -1792,6 +2071,11 @@ class KnowledgeBaseRetrievalConfigurationTypeDef(TypedDict):
     vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationUnionTypeDef
 
 
+class TrainingDataConfigTypeDef(TypedDict):
+    s3Uri: NotRequired[str]
+    invocationLogsConfig: NotRequired[InvocationLogsConfigUnionTypeDef]
+
+
 class ExternalSourcesRetrieveAndGenerateConfigurationTypeDef(TypedDict):
     modelArn: str
     sources: Sequence[ExternalSourceUnionTypeDef]
@@ -1805,6 +2089,26 @@ class RAGConfigOutputTypeDef(TypedDict):
 KnowledgeBaseRetrievalConfigurationUnionTypeDef = Union[
     KnowledgeBaseRetrievalConfigurationTypeDef, KnowledgeBaseRetrievalConfigurationOutputTypeDef
 ]
+
+
+class CreateModelCustomizationJobRequestRequestTypeDef(TypedDict):
+    jobName: str
+    customModelName: str
+    roleArn: str
+    baseModelIdentifier: str
+    trainingDataConfig: TrainingDataConfigTypeDef
+    outputDataConfig: OutputDataConfigTypeDef
+    clientRequestToken: NotRequired[str]
+    customizationType: NotRequired[CustomizationTypeType]
+    customModelKmsKeyId: NotRequired[str]
+    jobTags: NotRequired[Sequence[TagTypeDef]]
+    customModelTags: NotRequired[Sequence[TagTypeDef]]
+    validationDataConfig: NotRequired[ValidationDataConfigTypeDef]
+    hyperParameters: NotRequired[Mapping[str, str]]
+    vpcConfig: NotRequired[VpcConfigTypeDef]
+    customizationConfig: NotRequired[CustomizationConfigTypeDef]
+
+
 ExternalSourcesRetrieveAndGenerateConfigurationUnionTypeDef = Union[
     ExternalSourcesRetrieveAndGenerateConfigurationTypeDef,
     ExternalSourcesRetrieveAndGenerateConfigurationOutputTypeDef,