Skip to content

Commit ee3caba

Browse files
committed
fix: refactor attributes building
Signed-off-by: Danju Visvanathan <danju.visvanathan@gmail.com>
1 parent 9a7468f commit ee3caba

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

openfeature_flagsmith/provider.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -134,15 +134,10 @@ def _resolve(
134134

135135
def _get_flags(self, evaluation_context: EvaluationContext = EvaluationContext()):
136136
if targeting_key := evaluation_context.targeting_key:
137-
traits = {}
138-
for key, value in evaluation_context.attributes.items():
139-
if key == "traits":
140-
continue
141-
else:
142-
traits[key] = value
143-
traits.update(evaluation_context.attributes.get("traits", {}))
137+
nested_traits = evaluation_context.attributes.pop("traits", {})
138+
flattened_traits = {**evaluation_context.attributes, **nested_traits}
144139
return self._client.get_identity_flags(
145140
identifier=targeting_key,
146-
traits=traits,
141+
traits=flattened_traits,
147142
)
148143
return self._client.get_environment_flags()

0 commit comments

Comments
 (0)