jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-08 02:11:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix docstring generator for nested models

Browse Source
This commit is contained in:
Danglewood
2026-05-03 13:55:01 -07:00
parent 30f84a3e81
commit f899588da9
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
openbb_platform/core/openbb_core/app/static/package_builder/docstring_generator.py
View File

@@ -107,7 +107,7 @@ class DocstringGenerator:
if get_origin(arg) is Literal:
continue
type_name = str(arg)
if hasattr(arg, "__name__"):
if hasattr(arg, "__name__") and not get_args(arg):
type_name = arg.__name__
type_name = (
type_name.replace("typing.", "")
@@ -116,7 +116,18 @@ class DocstringGenerator:
.replace("datetime.date", "date")
)
if "openbb_" in type_name:
type_name = type_name.rsplit(".", 1)[-1]
# Strip the dotted module path but preserve any
# generic-container prefix like ``list[`` so
# ``list[openbb_x.y.Foo]`` becomes ``list[Foo]``,
# not just ``Foo]``.
bracket_idx = type_name.find("[")
if bracket_idx != -1:
prefix = type_name[: bracket_idx + 1]
tail = type_name[bracket_idx + 1 :]
tail = tail.rsplit(".", 1)[-1]
type_name = prefix + tail
else:
type_name = type_name.rsplit(".", 1)[-1]
if type_name != "NoneType":
type_names.append(type_name)