diff --git a/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py b/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py index 683f92cea73..ed9fa45ec86 100644 --- a/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py +++ b/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py @@ -76,3 +76,9 @@ class Fetcher( """Get the type of the provider data.""" # pylint: disable=E1101 return self.__orig_bases__[0].__args__[3] + + @property + def return_type(self): + """Get the type of the return.""" + # pylint: disable=E1101 + return self.transform_data.__annotations__["return"] diff --git a/openbb_sdk/sdk/provider/openbb_provider/registry_map.py b/openbb_sdk/sdk/provider/openbb_provider/registry_map.py index e0a7830a5a0..90504a4177e 100644 --- a/openbb_sdk/sdk/provider/openbb_provider/registry_map.py +++ b/openbb_sdk/sdk/provider/openbb_provider/registry_map.py @@ -57,17 +57,20 @@ class RegistryMap: f = fetcher() standard_query, extra_query = self.extract_info(f, "query_params") standard_data, extra_data = self.extract_info(f, "data") + return_type = f.return_type if model_name not in map_: map_[model_name] = {} map_[model_name]["openbb"] = { "QueryParams": standard_query, "Data": standard_data, + "ReturnType": return_type, } map_[model_name][p] = { "QueryParams": extra_query, "Data": extra_data, + "ReturnType": return_type, } return map_