diff options
author | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-03-21 17:21:10 -0700 |
---|---|---|
committer | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-03-21 17:21:10 -0700 |
commit | 0239b10e91abf39dee33184ac136301d94667e68 (patch) | |
tree | 9e6ffe8e1936c54d3f62f16ae27b941081120caf | |
parent | 7a1e0b405cc8fd11e836f240871592ef19977f0d (diff) |
make that change
-rw-r--r-- | openbb_platform/core/openbb_core/app/static/package_builder.py | 9 |
1 files changed, 4 insertions, 5 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 8a03ea7c405..e9fffa956a8 100644 --- a/openbb_platform/core/openbb_core/app/static/package_builder.py +++ b/openbb_platform/core/openbb_core/app/static/package_builder.py @@ -531,7 +531,7 @@ class MethodDefinition: def reorder_params(params: Dict[str, Parameter]) -> "OrderedDict[str, Parameter]": """Reorder the params.""" formatted_keys = list(params.keys()) - for k in ["provider", "extra_params"]: + for k in ["provider", "extra_params", "**kwargs"]: if k in formatted_keys: formatted_keys.remove(k) formatted_keys.append(k) @@ -561,14 +561,14 @@ class MethodDefinition: ], default=False, ) - kwargs = None + formatted: Dict[str, Parameter] = {} for name, param in parameter_map.items(): if name == "extra_params": formatted[name] = Parameter(name="kwargs", kind=Parameter.VAR_KEYWORD) elif name == "kwargs": - kwargs = Parameter( + formatted["**" + name] = Parameter( name="kwargs", kind=Parameter.VAR_KEYWORD, annotation=Any ) elif name == "provider_choices": @@ -628,8 +628,7 @@ class MethodDefinition: annotation=updated_type, default=param.default, ) - if kwargs: - formatted["**kwargs"] = kwargs + return MethodDefinition.reorder_params(params=formatted) @staticmethod |