[AI-FSSDK] [FSSDK-12337] Type-restrict Experiment.type to ExperimentType

- Add ExperimentType Literal type in helpers/types.py: 'a/b', 'mab', 'cmab', 'feature_rollout'
- Change Experiment.type from Optional[str] to Optional[ExperimentType]

Author: FarhanAnjum-opti

optimizely/entities.py

@@ -22,7 +22,7 @@
 
 if TYPE_CHECKING:
     # prevent circular dependenacy by skipping import at runtime
-    from .helpers.types import ExperimentDict, TrafficAllocation, VariableDict, VariationDict, CmabDict
+    from .helpers.types import ExperimentDict, ExperimentType, TrafficAllocation, VariableDict, VariationDict, CmabDict
 
 
 class BaseEntity:
@@ -87,7 +87,7 @@ def __init__(
         groupId: Optional[str] = None,
         groupPolicy: Optional[str] = None,
         cmab: Optional[CmabDict] = None,
-        type: Optional[str] = None,
+        type: Optional[ExperimentType] = None,
         **kwargs: Any
     ):
         self.id = id

optimizely/helpers/types.py

@@ -117,6 +117,8 @@ class CmabDict(BaseEntity):
     trafficAllocation: int
 
 
+ExperimentType = Literal['a/b', 'mab', 'cmab', 'feature_rollout']
+
 HoldoutStatus = Literal['Draft', 'Running', 'Concluded', 'Archived']