fix: Add CustomAttributes field to DefaultPayloadsModel (#5870)

mujtaba1747 · web-flow · commit 6308e6daef61 · 2026-05-19T11:47:44.000-07:00
diff --git a/sagemaker-core/src/sagemaker/core/jumpstart/models.py b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
@@ -190,6 +190,7 @@ class DefaultPayloadsModel(BaseConfig):
     PromptKey: Optional[str] = None
     OutputKeys: Optional[OutputKeysModel] = None
     Body: Union[str, Dict[str, Any]]
+    CustomAttributes: Optional[str] = None
 
 
 class HostingResourceRequirementsModel(BaseConfig):
