Add custom_attributes field to SSO Profile. (#434)

mattgd · web-flow · commit aa98f87ba356 · 2025-04-25T07:40:04.000-04:00
diff --git a/tests/test_sso.py b/tests/test_sso.py
@@ -29,7 +29,7 @@ def mock_magic_link_profile(self):
             idp_id="",
             first_name=None,
             last_name=None,
-            profile=None,
+            role=None,
             groups=None,
             raw_attributes={},
         ).dict()
diff --git a/tests/utils/fixtures/mock_profile.py b/tests/utils/fixtures/mock_profile.py
@@ -6,7 +6,7 @@ class MockProfile(Profile):
     def __init__(self, id: str):
         super().__init__(
             object="profile",
-            id="prof_01DWAS7ZQWM70PV93BFV1V78QV",
+            id=id or "prof_01DWAS7ZQWM70PV93BFV1V78QV",
             email="demo@workos-okta.com",
             first_name="WorkOS",
             last_name="Demo",
@@ -16,10 +16,14 @@ def __init__(self, id: str):
             connection_id="conn_01EMH8WAK20T42N2NBMNBCYHAG",
             connection_type="OktaSAML",
             idp_id="00u1klkowm8EGah2H357",
+            custom_attributes={
+                "license": "professional",
+            },
             raw_attributes={
                 "email": "demo@workos-okta.com",
                 "first_name": "WorkOS",
                 "last_name": "Demo",
                 "groups": ["Admins", "Developers"],
+                "license": "professional",
             },
         )
diff --git a/workos/types/sso/profile.py b/workos/types/sso/profile.py
@@ -23,7 +23,8 @@ class Profile(WorkOSModel):
     idp_id: str
     role: Optional[ProfileRole] = None
     groups: Optional[Sequence[str]] = None
-    raw_attributes: Mapping[str, Any]
+    custom_attributes: Optional[Mapping[str, Any]] = None
+    raw_attributes: Optional[Mapping[str, Any]] = None
 
 
 class ProfileAndToken(WorkOSModel):
