types

Author: vivekkhimani

src/workos/types/feature_flags/__init__.py

@@ -1,3 +1,3 @@
-from workos.types.feature_flags.feature_flag import FeatureFlag
+from workos.types.feature_flags.feature_flag import FeatureFlag, FeatureFlagOwner
 
-__all__ = ["FeatureFlag"]
+__all__ = ["FeatureFlag", "FeatureFlagOwner"]

src/workos/types/feature_flags/feature_flag.py

@@ -1,15 +1,22 @@
-from typing import Any, Literal, Optional, Sequence
+from typing import Literal, Optional, Sequence
 from workos.types.workos_model import WorkOSModel
 
 
+class FeatureFlagOwner(WorkOSModel):
+    email: str
+    first_name: Optional[str]
+    last_name: Optional[str]
+
+
 class FeatureFlag(WorkOSModel):
     id: str
     object: Literal["feature_flag"]
     slug: str
     name: str
     description: Optional[str]
     tags: Sequence[str]
+    owner: Optional[FeatureFlagOwner]
     enabled: bool
-    default_value: Optional[Any]
+    default_value: bool
     created_at: str
     updated_at: str

tests/utils/fixtures/mock_feature_flag.py

@@ -13,8 +13,13 @@ def __init__(self, id, enabled=True):
             name="Test Feature",
             description="A test feature flag",
             tags=["test"],
+            owner={
+                "email": "admin@example.com",
+                "first_name": "Test",
+                "last_name": "User",
+            },
             enabled=enabled,
-            default_value=None,
+            default_value=False,
             created_at=now,
             updated_at=now,
         )