diff --git a/generate_sdk.py b/generate_sdk.py index 811bb8a4a2c..124fa983b16 100644 --- a/generate_sdk.py +++ b/generate_sdk.py @@ -228,6 +228,7 @@ class BuildCategoryModelClasses: if module: f.write(" def __init__(self):\r super().__init__()\r") elif sdk_root: + f.write(" __version__ = obbff.VERSION\r\r") f.write(" def __init__(self):\r SDKLogger()\r") def write_class_attributes( diff --git a/openbb_terminal/sdk.py b/openbb_terminal/sdk.py index f040c52c44b..6e5b69548a8 100644 --- a/openbb_terminal/sdk.py +++ b/openbb_terminal/sdk.py @@ -20,6 +20,7 @@ from openbb_terminal.sdk_core import ( controllers as ctrl, models as model, ) +from openbb_terminal import feature_flags as obbff from openbb_terminal.session.user import User if User.is_guest(): @@ -39,6 +40,8 @@ class OpenBBSDK: `whoami`: Display user info.\n """ + __version__ = obbff.VERSION + def __init__(self): SDKLogger() self.login = lib.sdk_session.login diff --git a/openbb_terminal/sdk_core/sdk_helpers.py b/openbb_terminal/sdk_core/sdk_helpers.py index b06251b22e5..7fef5ee4b98 100644 --- a/openbb_terminal/sdk_core/sdk_helpers.py +++ b/openbb_terminal/sdk_core/sdk_helpers.py @@ -324,6 +324,7 @@ from openbb_terminal.sdk_core import ( controllers as ctrl, models as model, ) +from openbb_terminal import feature_flags as obbff from openbb_terminal.session.user import User if User.is_guest(): diff --git a/openbb_terminal/sdk_core/trailmap.py b/openbb_terminal/sdk_core/trailmap.py index 81831e8582c..364c9daa462 100644 --- a/openbb_terminal/sdk_core/trailmap.py +++ b/openbb_terminal/sdk_core/trailmap.py @@ -112,11 +112,10 @@ class FuncAttr: definition = definition.rstrip(", ") - if location_path[0] == "root": - location_path[0] = "" - trail = ".".join([t for t in location_path if t != ""]) sdk_name = class_attr if not view else f"{class_attr}_chart" + + trail = trail.lstrip("root.") # pylint: disable=E1310 sdk_path = f"{f'openbb.{trail}' if trail else 'openbb'}.{sdk_name}" return f"{sdk_path}({definition })" diff --git a/tests/openbb_terminal/test_sdk.py b/tests/openbb_terminal/test_sdk.py index 74206a4c804..e530e760e3e 100644 --- a/tests/openbb_terminal/test_sdk.py +++ b/tests/openbb_terminal/test_sdk.py @@ -4,10 +4,12 @@ def test_openbb(mocker): """Test the openbb function""" mock = mocker.patch("openbb_terminal.base_helpers.load_dotenv_and_reload_configs") + from openbb_terminal import feature_flags as obbff from openbb_terminal.sdk import openbb mock.assert_called_once() assert "stocks" in dir(openbb) + assert openbb.__version__ == obbff.VERSION # from importlib import import_module