Fix outdated organization_membership events

Default custom_attributes to empty dict and coerce null values from API to empty dict, ensuring type safety without requiring optional fields. Add event test for null custom_attributes deserialization.

Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>

Author: mattgd

src/workos/types/user_management/organization_membership.py

@@ -1,4 +1,6 @@
 from typing import Any, Literal, Mapping, Optional, Sequence
+
+from pydantic import field_validator
 from typing_extensions import TypedDict
 
 from workos.types.workos_model import WorkOSModel
@@ -21,6 +23,13 @@ class OrganizationMembership(WorkOSModel):
     role: OrganizationMembershipRole
     roles: Optional[Sequence[OrganizationMembershipRole]] = None
     status: LiteralOrUntyped[OrganizationMembershipStatus]
-    custom_attributes: Mapping[str, Any]
+    custom_attributes: Mapping[str, Any] = {}
     created_at: str
     updated_at: str
+
+    @field_validator("custom_attributes", mode="before")
+    @classmethod
+    def _coerce_null_custom_attributes(cls, v: Any) -> Any:
+        if v is None:
+            return {}
+        return v

tests/test_events.py

@@ -4,6 +4,7 @@
 from tests.utils.fixtures.mock_event import MockEvent
 from tests.utils.syncify import syncify
 from workos.events import AsyncEvents, Events, EventsListResource
+from workos.types.events import OrganizationMembershipCreatedEvent
 
 
 @pytest.mark.sync_and_async(Events, AsyncEvents)
@@ -20,6 +21,34 @@ def mock_events(self):
             },
         }
 
+    @pytest.fixture
+    def mock_organization_membership_event_with_null_custom_attributes(self):
+        return {
+            "object": "list",
+            "data": [
+                {
+                    "object": "event",
+                    "id": "event_01234",
+                    "event": "organization_membership.created",
+                    "data": {
+                        "object": "organization_membership",
+                        "id": "om_01234",
+                        "user_id": "user_01234",
+                        "organization_id": "org_01234",
+                        "role": {"slug": "member"},
+                        "status": "active",
+                        "custom_attributes": None,
+                        "created_at": "2024-01-01T00:00:00.000Z",
+                        "updated_at": "2024-01-01T00:00:00.000Z",
+                    },
+                    "created_at": "2024-01-01T00:00:00.000Z",
+                }
+            ],
+            "list_metadata": {
+                "after": None,
+            },
+        }
+
     def test_list_events(
         self,
         module_instance: Union[Events, AsyncEvents],
@@ -40,3 +69,25 @@ def test_list_events(
         assert request_kwargs["method"] == "get"
         assert request_kwargs["params"] == {"events": ["dsync.activated"], "limit": 10}
         assert events.dict() == mock_events
+
+    def test_list_events_organization_membership_null_custom_attributes(
+        self,
+        module_instance: Union[Events, AsyncEvents],
+        mock_organization_membership_event_with_null_custom_attributes,
+        capture_and_mock_http_client_request,
+    ):
+        capture_and_mock_http_client_request(
+            http_client=module_instance._http_client,
+            status_code=200,
+            response_dict=mock_organization_membership_event_with_null_custom_attributes,
+        )
+
+        events: EventsListResource = syncify(
+            module_instance.list_events(
+                events=["organization_membership.created"]
+            )
+        )
+
+        event = events.data[0]
+        assert isinstance(event, OrganizationMembershipCreatedEvent)
+        assert event.data.custom_attributes == {}