adding base types and module registraiton

Author: swaroopAkkineniWorkos

src/workos/types/authorization/__init__.py

@@ -1,14 +1,20 @@
+from workos.types.authorization.access_evaluation import AccessEvaluation
 from workos.types.authorization.environment_role import (
     EnvironmentRole,
     EnvironmentRoleList,
 )
+from workos.types.authorization.organization_membership import (
+    AuthorizationOrganizationMembership,
+)
 from workos.types.authorization.organization_role import (
     OrganizationRole,
     OrganizationRoleEvent,
     OrganizationRoleList,
 )
 from workos.types.authorization.permission import Permission
+from workos.types.authorization.resource import Resource
 from workos.types.authorization.role import (
     Role,
     RoleList,
 )
+from workos.types.authorization.role_assignment import RoleAssignment

src/workos/types/authorization/access_evaluation.py

@@ -0,0 +1,7 @@
+from workos.types.workos_model import WorkOSModel
+
+
+class AccessEvaluation(WorkOSModel):
+    """Representation of a WorkOS Authorization access check result."""
+
+    authorized: bool

src/workos/types/authorization/organization_membership.py

@@ -0,0 +1,23 @@
+from typing import Any, Literal, Mapping, Optional
+
+from workos.types.workos_model import WorkOSModel
+from workos.typing.literals import LiteralOrUntyped
+
+OrganizationMembershipStatus = Literal["active", "inactive", "pending"]
+
+
+class AuthorizationOrganizationMembership(WorkOSModel):
+    """Representation of an Organization Membership returned by Authorization endpoints.
+
+    This is a separate type from the user_management OrganizationMembership because
+    authorization endpoints return memberships without `role` and `organization_name` fields.
+    """
+
+    object: Literal["organization_membership"]
+    id: str
+    user_id: str
+    organization_id: str
+    status: LiteralOrUntyped[OrganizationMembershipStatus]
+    custom_attributes: Optional[Mapping[str, Any]] = None
+    created_at: str
+    updated_at: str

src/workos/types/authorization/resource.py

@@ -0,0 +1,18 @@
+from typing import Any, Literal, Mapping, Optional
+
+from workos.types.workos_model import WorkOSModel
+
+
+class Resource(WorkOSModel):
+    """Representation of a WorkOS Authorization Resource."""
+
+    object: Literal["authorization_resource"]
+    id: str
+    resource_type: str
+    resource_id: str
+    organization_id: str
+    external_id: Optional[str] = None
+    meta: Optional[Mapping[str, Any]] = None
+    environment_id: str
+    created_at: str
+    updated_at: str

src/workos/types/authorization/role_assignment.py

@@ -0,0 +1,13 @@
+from typing import Literal
+
+from workos.types.workos_model import WorkOSModel
+
+
+class RoleAssignment(WorkOSModel):
+    """Representation of a WorkOS Authorization Role Assignment."""
+
+    object: Literal["role_assignment"]
+    id: str
+    role_slug: str
+    role_name: str
+    role_id: str

src/workos/types/list_resource.py

@@ -19,7 +19,12 @@
 from typing_extensions import Required, TypedDict
 from workos.types.api_keys import ApiKey
 from workos.types.audit_logs import AuditLogAction, AuditLogSchema
+from workos.types.authorization.organization_membership import (
+    AuthorizationOrganizationMembership,
+)
 from workos.types.authorization.permission import Permission
+from workos.types.authorization.resource import Resource
+from workos.types.authorization.role_assignment import RoleAssignment
 from workos.types.directory_sync import (
     Directory,
     DirectoryGroup,
@@ -59,6 +64,9 @@
     Organization,
     OrganizationMembership,
     Permission,
+    Resource,
+    RoleAssignment,
+    AuthorizationOrganizationMembership,
     AuthorizationResource,
     AuthorizationResourceType,
     User,

src/workos/utils/_base_http_client.py

@@ -123,6 +123,7 @@ def _prepare_request(
         json: JsonType = None,
         headers: HeadersType = None,
         exclude_default_auth_headers: bool = False,
+        force_include_body: bool = False,
     ) -> PreparedRequest:
         """Executes a request against the WorkOS API.
 
@@ -134,6 +135,7 @@ def _prepare_request(
             params Optional[dict]: Query params or body payload to be added to the request
             headers Optional[dict]: Custom headers to be added to the request
             token Optional[str]: Bearer token
+            force_include_body (bool): If True, allows sending a body with DELETE requests
 
         Returns:
             dict: Response from WorkOS
@@ -149,7 +151,7 @@ def _prepare_request(
             REQUEST_METHOD_GET,
         ]
 
-        if bodyless_http_method and json is not None:
+        if bodyless_http_method and json is not None and not force_include_body:
             raise ValueError(f"Cannot send a body with a {parsed_method} request")
 
         # Remove any parameters that are None
@@ -161,7 +163,7 @@ def _prepare_request(
             json = {k: v for k, v in json.items() if v is not None}
 
         # We'll spread these return values onto the HTTP client request method
-        if bodyless_http_method:
+        if bodyless_http_method and not force_include_body:
             return {
                 "method": parsed_method,
                 "url": url,

src/workos/utils/http_client.py

@@ -14,7 +14,7 @@
     ParamsType,
     ResponseJson,
 )
-from workos.utils.request_helper import REQUEST_METHOD_GET
+from workos.utils.request_helper import REQUEST_METHOD_DELETE, REQUEST_METHOD_GET
 
 
 class SyncHttpxClientWrapper(httpx.Client):
@@ -113,6 +113,23 @@ def request(
         response = self._client.request(**prepared_request_parameters)
         return self._handle_response(response)
 
+    def delete_with_body(
+        self,
+        path: str,
+        json: JsonType = None,
+        headers: HeadersType = None,
+    ) -> ResponseJson:
+        """Executes a DELETE request with a JSON body against the WorkOS API."""
+        prepared_request_parameters = self._prepare_request(
+            path=path,
+            method=REQUEST_METHOD_DELETE,
+            json=json,
+            headers=headers,
+            force_include_body=True,
+        )
+        response = self._client.request(**prepared_request_parameters)
+        return self._handle_response(response)
+
 
 class AsyncHttpxClientWrapper(httpx.AsyncClient):
     def __del__(self) -> None:
@@ -210,5 +227,22 @@ async def request(
         response = await self._client.request(**prepared_request_parameters)
         return self._handle_response(response)
 
+    async def delete_with_body(
+        self,
+        path: str,
+        json: JsonType = None,
+        headers: HeadersType = None,
+    ) -> ResponseJson:
+        """Executes a DELETE request with a JSON body against the WorkOS API."""
+        prepared_request_parameters = self._prepare_request(
+            path=path,
+            method=REQUEST_METHOD_DELETE,
+            json=json,
+            headers=headers,
+            force_include_body=True,
+        )
+        response = await self._client.request(**prepared_request_parameters)
+        return self._handle_response(response)
+
 
 HTTPClient = Union[AsyncHTTPClient, SyncHTTPClient]