update trait type

khvn26 · khvn26 · commit 9ab732a20975 · 2025-08-26T10:58:16.000+01:00
diff --git a/flag_engine/context/types.py b/flag_engine/context/types.py
@@ -8,7 +8,7 @@
 
 from typing_extensions import NotRequired
 
-from flag_engine.segments.types import ConditionOperator, RuleType
+from flag_engine.segments.types import ConditionOperator, ContextValue, RuleType
 
 
 class EnvironmentContext(TypedDict):
@@ -24,7 +24,7 @@ class FeatureValue(TypedDict):
 class IdentityContext(TypedDict):
     identifier: str
     key: str
-    traits: NotRequired[Dict[str, Optional[Union[str, float, bool]]]]
+    traits: NotRequired[Dict[str, ContextValue]]
 
 
 class StrValueSegmentCondition(TypedDict):
