diff options
| author | Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com> | 2024-03-20 14:27:01 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-20 13:27:01 +0000 |
| commit | bb6efdffdc39f97f4c7c6c3a6f4bbd13432e329a (patch) | |
| tree | ca50dd3fbbb224b01142f58c942feea591099dca | |
| parent | 7f4007a7514410a1cbd9734f8d06cbaa0e1bc554 (diff) | |
Allow kwargs on POST endpoints (#6240)
| -rw-r--r-- | openbb_platform/core/openbb_core/app/static/package_builder.py | 6 |
1 files changed, 5 insertions, 1 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 917f9ff03df..793f82731c6 100644 --- a/openbb_platform/core/openbb_core/app/static/package_builder.py +++ b/openbb_platform/core/openbb_core/app/static/package_builder.py @@ -343,7 +343,7 @@ class ImportDefinition: code += "\nfrom pydantic import BaseModel" code += "\nfrom inspect import Parameter" code += "\nimport typing" - code += "\nfrom typing import List, Dict, Union, Optional, Literal" + code += "\nfrom typing import List, Dict, Union, Optional, Literal, Any" code += "\nfrom annotated_types import Ge, Le, Gt, Lt" code += "\nfrom warnings import warn, simplefilter" if sys.version_info < (3, 9): @@ -567,6 +567,10 @@ class MethodDefinition: for name, param in parameter_map.items(): if name == "extra_params": formatted[name] = Parameter(name="kwargs", kind=Parameter.VAR_KEYWORD) + elif name == "kwargs": + formatted["**" + name] = Parameter( + name="kwargs", kind=Parameter.VAR_KEYWORD, annotation=Any + ) elif name == "provider_choices": fields = param.annotation.__args__[0].__dataclass_fields__ field = fields["provider"] |