Skip to content

Commit 96524b9

Browse files
authored
feat: Bump flagsmith-flag-engine to 5.0.0 (#69)
1 parent 7721dc7 commit 96524b9

File tree

5 files changed

+264
-155
lines changed

5 files changed

+264
-155
lines changed

flagsmith/flagsmith.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
import requests
66
from flag_engine import engine
7-
from flag_engine.environments.builders import build_environment_model
87
from flag_engine.environments.models import EnvironmentModel
98
from flag_engine.identities.models import IdentityModel, TraitModel
109
from flag_engine.segments.evaluator import get_identity_segments
@@ -196,7 +195,7 @@ def update_environment(self):
196195

197196
def _get_environment_from_api(self) -> EnvironmentModel:
198197
environment_data = self._get_json_response(self.environment_url, method="GET")
199-
return build_environment_model(environment_data)
198+
return EnvironmentModel.model_validate(environment_data)
200199

201200
def _get_environment_flags_from_document(self) -> Flags:
202201
return Flags.from_feature_state_models(

flagsmith/offline_handlers.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
1-
import json
21
from abc import ABC, abstractmethod
32

4-
from flag_engine.environments.builders import build_environment_model
53
from flag_engine.environments.models import EnvironmentModel
64

75

@@ -14,8 +12,8 @@ def get_environment(self) -> EnvironmentModel:
1412
class LocalFileHandler(BaseOfflineHandler):
1513
def __init__(self, environment_document_path: str):
1614
with open(environment_document_path) as environment_document:
17-
self.environment = build_environment_model(
18-
json.loads(environment_document.read())
15+
self.environment = EnvironmentModel.model_validate_json(
16+
environment_document.read()
1917
)
2018

2119
def get_environment(self) -> EnvironmentModel:

0 commit comments

Comments
 (0)