summaryrefslogtreecommitdiffstats
path: root/openbb_platform/core/openbb_core/app/model/hub/hub_session.py
diff options
context:
space:
mode:
Diffstat (limited to 'openbb_platform/core/openbb_core/app/model/hub/hub_session.py')
-rw-r--r--openbb_platform/core/openbb_core/app/model/hub/hub_session.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/openbb_platform/core/openbb_core/app/model/hub/hub_session.py b/openbb_platform/core/openbb_core/app/model/hub/hub_session.py
index fb0142eafc1..7a9cf615991 100644
--- a/openbb_platform/core/openbb_core/app/model/hub/hub_session.py
+++ b/openbb_platform/core/openbb_core/app/model/hub/hub_session.py
@@ -1,9 +1,13 @@
+"""Model for HubSession."""
+
from typing import Optional
from pydantic import BaseModel, SecretStr, field_serializer
class HubSession(BaseModel):
+ """Model for HubSession."""
+
username: Optional[str] = None
email: str
primary_usage: str
@@ -12,10 +16,12 @@ class HubSession(BaseModel):
access_token: SecretStr
def __repr__(self) -> str:
+ """Return string representation."""
return f"{self.__class__.__name__}\n\n" + "\n".join(
f"{k}: {v}" for k, v in self.model_dump().items()
)
@field_serializer("access_token", when_used="json-unless-none")
def _dump_secret(self, v):
+ """Dump secret."""
return v.get_secret_value()