Skip to content

Commit b01c861

Browse files
committed
feat: reuse _extract_traits in _get_flags
1 parent 763d1c4 commit b01c861

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

openfeature_flagsmith/provider.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -200,10 +200,8 @@ def _extract_traits(
200200

201201
def _get_flags(self, evaluation_context: EvaluationContext = EvaluationContext()):
202202
if targeting_key := evaluation_context.targeting_key:
203-
nested_traits = evaluation_context.attributes.pop("traits", {})
204-
flattened_traits = {**evaluation_context.attributes, **nested_traits}
205203
return self._client.get_identity_flags(
206204
identifier=targeting_key,
207-
traits=flattened_traits,
205+
traits=self._extract_traits(evaluation_context) or {},
208206
)
209207
return self._client.get_environment_flags()

0 commit comments

Comments
 (0)