jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-27 12:40:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix] json_schema_extra getting overridden (#6300)

Browse Source 
* fix json_schema_extra getting overriden

* keeping it as it was

---------

Co-authored-by: hjoaquim <h.joaquim@campus.fct.unl.pt>
Co-authored-by: Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com>
This commit is contained in:
Henrique Joaquim
2024-04-10 14:50:09 +01:00
committed by GitHub
parent 951749156d
commit 1c26d24e95
1 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
openbb_platform/core/openbb_core/app/provider_interface.py
View File

@@ -164,9 +164,11 @@ class ProviderInterface(metaclass=SingletonMeta):
curr_name = current.name
curr_type: Optional[Type] = current.annotation
curr_desc = getattr(current.default, "description", "")
curr_json_schema_extra = getattr(current.default, "json_schema_extra", {})
inc_type: Optional[Type] = incoming.annotation
inc_desc = getattr(incoming.default, "description", "")
inc_json_schema_extra = getattr(incoming.default, "json_schema_extra", {})
def split_desc(desc: str) -> str:
"""Split field description."""
@@ -174,6 +176,24 @@ class ProviderInterface(metaclass=SingletonMeta):
detail = item[0] if item else ""
return detail
def merge_json_schema_extra(curr: dict, inc: dict) -> dict:
"""Merge json schema extra."""
for key in curr.keys() & inc.keys():
# Merge keys that are in both dictionaries if both are lists
curr_value = curr[key]
inc_value = inc[key]
if isinstance(curr_value, list) and isinstance(inc_value, list):
curr[key] = list(set(curr.get(key, []) + inc.get(key, [])))
inc.pop(key)
# Add any remaining keys from inc to curr
curr.update(inc)
return curr
json_schema_extra: dict = merge_json_schema_extra(
curr=curr_json_schema_extra or {}, inc=inc_json_schema_extra or {}
)
curr_detail = split_desc(curr_desc)
inc_detail = split_desc(inc_desc)
@@ -192,6 +212,7 @@ class ProviderInterface(metaclass=SingletonMeta):
default=getattr(current.default, "default", None),
title=providers,
description=new_desc,
json_schema_extra=json_schema_extra,
)
merged_type: Optional[Type] = (