feat: add pydantic type

gotrue/_async/gotrue_client.py

@@ -48,6 +48,7 @@
     AuthResponse,
     CodeExchangeParams,
     DecodedJWTDict,
+    IdentitiesResponse,
     MFAChallengeAndVerifyParams,
     MFAChallengeParams,
     MFAEnrollParams,
@@ -291,14 +292,16 @@ async def link_identity(self, credentials):
 
     async def get_user_identities(self):
         response = self.get_user()
-        user = response.get("user")
-        identities = user.get("identities", []) if user else []
-        return {"data": {"identities": identities}}
+        return (
+            IdentitiesResponse(identities=response.user.identities)
+            if response.user
+            else AuthSessionMissingError()
+        )
 
     async def unlink_identity(self, identity):
         return await self._request(
             "POST",
-            f"/user/identities/{identity.identity_id}",
+            f"/user/identities/{identity.id}",
         )
 
     async def sign_in_with_otp(

gotrue/_sync/gotrue_client.py

@@ -48,6 +48,7 @@
     AuthResponse,
     CodeExchangeParams,
     DecodedJWTDict,
+    IdentitiesResponse,
     MFAChallengeAndVerifyParams,
     MFAChallengeParams,
     MFAEnrollParams,
@@ -291,14 +292,16 @@ def link_identity(self, credentials):
 
     def get_user_identities(self):
         response = self.get_user()
-        user = response.get("user")
-        identities = user.get("identities", []) if user else []
-        return {"data": {"identities": identities}}
+        return (
+            IdentitiesResponse(identities=response.user.identities)
+            if response.user
+            else AuthSessionMissingError()
+        )
 
     def unlink_identity(self, identity):
         return self._request(
             "POST",
-            f"/user/identities/{identity.identity_id}",
+            f"/user/identities/{identity.id}",
         )
 
     def sign_in_with_otp(

gotrue/types.py

@@ -94,6 +94,14 @@ class OAuthResponse(BaseModel):
     url: str
 
 
+class SSOResponse(BaseModel):
+    url: str
+
+
+class IdentitiesResponse(BaseModel):
+    identities: List[UserIdentity]
+
+
 class UserResponse(BaseModel):
     user: User
 
@@ -323,6 +331,17 @@ class SignInWithOAuthCredentials(TypedDict):
     options: NotRequired[SignInWithOAuthCredentialsOptions]
 
 
+class SignInWithSSOCredentials(TypedDict):
+    provider_id: NotRequired[str]
+    domain: NotRequired[str]
+    options: NotRequired[SignInWithSSOOptions]
+
+
+class SignInWithSSOOptions(TypedDict):
+    redirect_to: NotRequired[str]
+    skip_http_redirect: NotRequired[bool]
+
+
 class VerifyOtpParamsOptions(TypedDict):
     redirect_to: NotRequired[str]
     captcha_token: NotRequired[str]