feat: Generic segment metadata

Author: khvn26

flag_engine/context/types.py

@@ -4,11 +4,16 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, List, Literal, Optional, TypedDict, Union
+from typing import Any, Dict, Generic, List, Literal, Optional, Union
 
-from typing_extensions import NotRequired
+from typing_extensions import NotRequired, TypedDict
 
-from flag_engine.segments.types import ConditionOperator, ContextValue, RuleType
+from flag_engine.segments.types import (
+    ConditionOperator,
+    ContextValue,
+    MetadataT,
+    RuleType,
+)
 
 
 class EnvironmentContext(TypedDict):
@@ -58,16 +63,16 @@ class FeatureContext(TypedDict):
     priority: NotRequired[float]
 
 
-class SegmentContext(TypedDict):
+class SegmentContext(TypedDict, Generic[MetadataT]):
     key: str
     name: str
     rules: List[SegmentRule]
     overrides: NotRequired[List[FeatureContext]]
-    metadata: NotRequired[Dict[str, Any]]
+    metadata: NotRequired[MetadataT]
 
 
-class EvaluationContext(TypedDict):
+class EvaluationContext(TypedDict, Generic[MetadataT]):
     environment: EnvironmentContext
     identity: NotRequired[Optional[IdentityContext]]
-    segments: NotRequired[Dict[str, SegmentContext]]
+    segments: NotRequired[Dict[str, SegmentContext[MetadataT]]]
     features: NotRequired[Dict[str, FeatureContext]]

flag_engine/result/types.py

@@ -4,9 +4,11 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, List, TypedDict
+from typing import Any, Dict, Generic, List
 
-from typing_extensions import NotRequired
+from typing_extensions import NotRequired, TypedDict
+
+from flag_engine.segments.types import MetadataT
 
 
 class FlagResult(TypedDict):
@@ -17,12 +19,12 @@ class FlagResult(TypedDict):
     reason: str
 
 
-class SegmentResult(TypedDict):
+class SegmentResult(TypedDict, Generic[MetadataT]):
     key: str
     name: str
-    metadata: NotRequired[Dict[str, Any]]
+    metadata: NotRequired[MetadataT]
 
 
-class EvaluationResult(TypedDict):
+class EvaluationResult(TypedDict, Generic[MetadataT]):
     flags: Dict[str, FlagResult]
-    segments: List[SegmentResult]
+    segments: List[SegmentResult[MetadataT]]

flag_engine/segments/types.py

@@ -1,6 +1,10 @@
-from typing import Any, Literal, Union, get_args
+from __future__ import annotations
 
-from typing_extensions import TypeGuard
+from typing import Any, Dict, Literal, Union, get_args
+
+from typing_extensions import TypeGuard, TypeVar
+
+MetadataT = TypeVar("MetadataT", default=Dict[str, Any])
 
 ConditionOperator = Literal[
     "EQUAL",