diff options
Diffstat (limited to 'openbb_platform/core/openbb_core/app/static/package_builder.py')
-rw-r--r-- | openbb_platform/core/openbb_core/app/static/package_builder.py | 20 |
1 files changed, 6 insertions, 14 deletions
diff --git a/openbb_platform/core/openbb_core/app/static/package_builder.py b/openbb_platform/core/openbb_core/app/static/package_builder.py index b6e08c83696..20b2fc433a8 100644 --- a/openbb_platform/core/openbb_core/app/static/package_builder.py +++ b/openbb_platform/core/openbb_core/app/static/package_builder.py @@ -92,11 +92,11 @@ class PackageBuilder: self.directory / "assets" / "extension_map.json" ) if add: - a = ", ".join(add) + a = ", ".join(sorted(add)) print(f"Extensions to add: {a}") # noqa: T201 if remove: - r = ", ".join(remove) + r = ", ".join(sorted(remove)) print(f"Extensions to remove: {r}") # noqa: T201 if add or remove: @@ -132,18 +132,10 @@ class PackageBuilder: def _get_extension_map(self) -> Dict[str, List[str]]: """Get map of extensions available at build time.""" el = ExtensionLoader() + og = OpenBBGroups.groups() ext_map: Dict[str, List[str]] = {} - groups = [ - OpenBBGroups.core.value, - OpenBBGroups.provider.value, - ] - entry_points_ = [ - el.core_entry_points, - el.provider_entry_points, - ] - - for group, entry_point in zip(groups, entry_points_): + for group, entry_point in zip(og, el.entry_points): ext_map[group] = [ f"{e.name}@{getattr(e.dist, 'version', '')}" for e in entry_point ] @@ -252,7 +244,8 @@ class PackageBuilder: add: Set[str] = set() remove: Set[str] = set() - groups = ("openbb_core_extension", "openbb_provider_extension") + groups = OpenBBGroups.groups() + for g in groups: built = set(ext_map.get(g, {})) installed = set( @@ -556,7 +549,6 @@ class MethodDefinition: path: str, parameter_map: Dict[str, Parameter] ) -> OrderedDict[str, Parameter]: """Format the params.""" - parameter_map.pop("cc", None) # we need to add the chart parameter here bc of the docstring generation if CHARTING_INSTALLED and path.replace("/", "_")[1:] in Charting.functions(): |