Add "role" to profile (#372)

* Add role to profile

* Format

Author: kendallstrautman

tests/test_sso.py

@@ -29,6 +29,7 @@ def mock_magic_link_profile(self):
             idp_id="",
             first_name=None,
             last_name=None,
+            profile=None,
             groups=None,
             raw_attributes={},
         ).dict()

tests/utils/fixtures/mock_profile.py

@@ -10,6 +10,7 @@ def __init__(self, id: str):
             email="[email protected]",
             first_name="WorkOS",
             last_name="Demo",
+            role={"slug": "admin"},
             groups=["Admins", "Developers"],
             organization_id="org_01FG53X8636WSNW2WEKB2C31ZB",
             connection_id="conn_01EMH8WAK20T42N2NBMNBCYHAG",

workos/types/sso/profile.py

@@ -2,6 +2,11 @@
 from workos.types.sso.connection import ConnectionType
 from workos.types.workos_model import WorkOSModel
 from workos.typing.literals import LiteralOrUntyped
+from typing_extensions import TypedDict
+
+
+class ProfileRole(TypedDict):
+    slug: str
 
 
 class Profile(WorkOSModel):
@@ -16,6 +21,7 @@ class Profile(WorkOSModel):
     first_name: Optional[str] = None
     last_name: Optional[str] = None
     idp_id: str
+    role: Optional[ProfileRole] = None
     groups: Optional[Sequence[str]] = None
     raw_attributes: Mapping[str, Any]