Refactored types and updated where session is exposed based on review feedback

Author: awolfden

workos/__init__.py

@@ -1,5 +1,4 @@
 from workos.client import SyncClient as WorkOSClient
 from workos.async_client import AsyncClient as AsyncWorkOSClient
-from workos.types.user_management.session import Session as Session
 
-__all__ = ["WorkOSClient", "AsyncWorkOSClient", "Session"]
+__all__ = ["WorkOSClient", "AsyncWorkOSClient"]

workos/types/user_management/session.py

@@ -48,6 +48,19 @@ class SessionConfig(TypedDict, total=False):
     cookie_password: str
 
 
+AuthMethodType = Literal[
+    "external_auth",
+    "impersonation",
+    "magic_code",
+    "migrated_session",
+    "oauth",
+    "passkey",
+    "password",
+    "sso",
+    "unknown",
+]
+
+
 class Session(WorkOSModel):
     """Representation of a WorkOS User Management Session."""
 
@@ -56,10 +69,11 @@ class Session(WorkOSModel):
     user_id: str
     organization_id: Optional[str] = None
     status: str
-    auth_method: Optional[str] = None
+    auth_method: AuthMethodType
+    impersonator: Optional[Impersonator] = None
     ip_address: Optional[str] = None
     user_agent: Optional[str] = None
-    expires_at: Optional[str] = None
+    expires_at: str
     ended_at: Optional[str] = None
     created_at: str
     updated_at: str

workos/user_management.py

@@ -1404,13 +1404,13 @@ def list_sessions(
         self,
         *,
         user_id: str,
-        limit: Optional[int] = None,
+        limit: Optional[int] = DEFAULT_LIST_RESPONSE_LIMIT,
         before: Optional[str] = None,
         after: Optional[str] = None,
         order: Optional[PaginationOrder] = None,
     ) -> "SessionsListResource":
         params: ListArgs = {
-            "limit": limit or DEFAULT_LIST_RESPONSE_LIMIT,
+            "limit": limit,
             "before": before,
             "after": after,
             "order": order,
@@ -1423,7 +1423,7 @@ def list_sessions(
         )
 
         list_args: SessionsListFilters = {
-            "limit": limit or DEFAULT_LIST_RESPONSE_LIMIT,
+            "limit": limit,
             "before": before,
             "after": after,
             "order": order,
@@ -2105,13 +2105,13 @@ async def list_sessions(
         self,
         *,
         user_id: str,
-        limit: Optional[int] = None,
+        limit: Optional[int] = DEFAULT_LIST_RESPONSE_LIMIT,
         before: Optional[str] = None,
         after: Optional[str] = None,
         order: Optional[PaginationOrder] = None,
     ) -> "SessionsListResource":
         params: ListArgs = {
-            "limit": limit or DEFAULT_LIST_RESPONSE_LIMIT,
+            "limit": limit,
             "before": before,
             "after": after,
             "order": order,
@@ -2124,7 +2124,7 @@ async def list_sessions(
         )
 
         list_args: SessionsListFilters = {
-            "limit": limit or DEFAULT_LIST_RESPONSE_LIMIT,
+            "limit": limit,
             "before": before,
             "after": after,
             "order": order,