nits

Author: swaroopAkkineniWorkos

src/workos/types/user_management/list_filters.py

@@ -1,6 +1,6 @@
 from typing import Optional, Sequence
 from workos.types.list_resource import ListArgs
-from workos.types.user_management.organization_membership_status import (
+from workos.types.user_management.organization_membership import (
     OrganizationMembershipStatus,
 )
 

src/workos/types/user_management/organization_membership.py

@@ -3,12 +3,11 @@
 from pydantic import Field
 from typing_extensions import TypedDict
 
-from workos.types.user_management.organization_membership_status import (
-    OrganizationMembershipStatus,
-)
 from workos.types.workos_model import WorkOSModel
 from workos.typing.literals import LiteralOrUntyped
 
+OrganizationMembershipStatus = Literal["active", "inactive", "pending"]
+
 
 class BaseOrganizationMembership(WorkOSModel):
     object: Literal["organization_membership"]
@@ -26,6 +25,8 @@ class OrganizationMembershipRole(TypedDict):
 
 
 class OrganizationMembership(BaseOrganizationMembership):
+    """Representation of an WorkOS Organization Membership."""
+
     role: OrganizationMembershipRole
     roles: Optional[Sequence[OrganizationMembershipRole]] = None
-    custom_attributes: Mapping[str, Any] = Field(default_factory=dict)
+    custom_attributes: Mapping[str, Any] = {}

src/workos/types/user_management/organization_membership_status.py

@@ -137,7 +137,7 @@ def _prepare_request(
             headers Optional[dict]: Custom headers to be added to the request
             exclude_default_auth_headers (bool): If True, excludes default auth headers from the request
             force_include_body (bool): If True, allows sending a body in a bodyless request (used for DELETE requests)
-
+            exclude_none (bool): If True (default), strips keys with None values from the JSON body so only defined fields are sent.
         Returns:
             dict: Response from WorkOS
         """