[BUG] Fix issue with Subject not serializing properly in FGA module (#348)

* Fix issue with Subject not serializing properly

* Black

* Fix type issue

Author: atainter

tests/test_fga.py

@@ -8,9 +8,9 @@
 )
 from workos.fga import FGA
 from workos.types.fga import (
-    Subject,
     WarrantCheckInput,
     WarrantWrite,
+    SubjectInput,
 )
 
 
@@ -356,7 +356,7 @@ def test_batch_write_warrants(
                     resource_type="permission",
                     resource_id="view-balance-sheet",
                     relation="member",
-                    subject=Subject(
+                    subject=SubjectInput(
                         resource_type="role",
                         resource_id="senior-accountant",
                         relation="member",
@@ -367,7 +367,7 @@ def test_batch_write_warrants(
                     resource_type="permission",
                     resource_id="balance-sheet:edit",
                     relation="member",
-                    subject=Subject(
+                    subject=SubjectInput(
                         resource_type="user",
                         resource_id="user-b",
                     ),
@@ -392,7 +392,7 @@ def test_check(self, mock_check_warrant_response, mock_http_client_with_response
                     resource_type="schedule",
                     resource_id="schedule-A1",
                     relation="viewer",
-                    subject=Subject(resource_type="user", resource_id="user-A"),
+                    subject=SubjectInput(resource_type="user", resource_id="user-A"),
                 )
             ],
         )
@@ -451,7 +451,7 @@ def test_check_with_debug_info(
                     resource_type="report",
                     resource_id="report-a",
                     relation="editor",
-                    subject=Subject(resource_type="user", resource_id="user-b"),
+                    subject=SubjectInput(resource_type="user", resource_id="user-b"),
                     context={"tenant": "tenant-b"},
                 )
             ],
@@ -477,13 +477,13 @@ def test_check_batch(
                     resource_type="schedule",
                     resource_id="schedule-A1",
                     relation="viewer",
-                    subject=Subject(resource_type="user", resource_id="user-A"),
+                    subject=SubjectInput(resource_type="user", resource_id="user-A"),
                 ),
                 WarrantCheckInput(
                     resource_type="schedule",
                     resource_id="schedule-A1",
                     relation="editor",
-                    subject=Subject(resource_type="user", resource_id="user-B"),
+                    subject=SubjectInput(resource_type="user", resource_id="user-B"),
                 ),
             ]
         )

workos/types/fga/check.py

@@ -3,8 +3,7 @@
 from workos.types.workos_model import WorkOSModel
 from workos.typing.literals import LiteralOrUntyped
 
-from .warrant import Subject
-
+from .warrant import Subject, SubjectInput
 
 CheckOperation = Literal["any_of", "all_of", "batch"]
 
@@ -13,7 +12,7 @@ class WarrantCheckInput(TypedDict, total=False):
     resource_type: str
     resource_id: str
     relation: str
-    subject: Subject
+    subject: SubjectInput
     context: Optional[Mapping[str, Any]]
 
 

workos/types/fga/warrant.py

@@ -4,6 +4,12 @@
 from workos.types.workos_model import WorkOSModel
 
 
+class SubjectInput(TypedDict, total=False):
+    resource_type: str
+    resource_id: str
+    relation: Optional[str]
+
+
 class Subject(WorkOSModel):
     resource_type: str
     resource_id: str
@@ -30,7 +36,7 @@ class WarrantWrite(TypedDict, total=False):
     resource_type: str
     resource_id: str
     relation: str
-    subject: Subject
+    subject: SubjectInput
     policy: Optional[str]