diff --git a/openbb_terminal/core/models/profile_model.py b/openbb_terminal/core/models/profile_model.py index 9cbe215ddd2..56bebae2d11 100644 --- a/openbb_terminal/core/models/profile_model.py +++ b/openbb_terminal/core/models/profile_model.py @@ -14,6 +14,7 @@ class ProfileModel(BaseModel): uuid: str = "" email: str = "" username: str = "" + primary_usage: str = "" remember: bool = True def load_user_info(self, session: dict, email: str, remember: bool): @@ -28,11 +29,13 @@ class ProfileModel(BaseModel): remember : bool Remember the session. """ + self.token_type = session.get("token_type", "") self.token = session.get("access_token", "") self.uuid = session.get("uuid", "") self.email = email self.username = self.email[: self.email.find("@")] + self.primary_usage = session.get("primary_usage", "") self.remember = remember def get_uuid(self) -> str: diff --git a/openbb_terminal/loggers.py b/openbb_terminal/loggers.py index 74dcda9e213..e4a8c37596c 100644 --- a/openbb_terminal/loggers.py +++ b/openbb_terminal/loggers.py @@ -184,7 +184,11 @@ class PosthogHandler(logging.Handler): ): self.logged_in = True openbb_posthog.identify( - get_user_uuid(), {"email": get_current_user().profile.email} + get_user_uuid(), + { + "email": get_current_user().profile.email, + "primaryUsage": get_current_user().profile.primary_usage, + }, ) openbb_posthog.alias(get_user_uuid(), app_settings.identifier)