summaryrefslogtreecommitdiffstats
path: root/openbb_platform/core/openbb_core/provider/registry_map.py
diff options
context:
space:
mode:
Diffstat (limited to 'openbb_platform/core/openbb_core/provider/registry_map.py')
-rw-r--r--openbb_platform/core/openbb_core/provider/registry_map.py16
1 files changed, 7 insertions, 9 deletions
diff --git a/openbb_platform/core/openbb_core/provider/registry_map.py b/openbb_platform/core/openbb_core/provider/registry_map.py
index a3a7fcc32d4..c3ccdb1b6f6 100644
--- a/openbb_platform/core/openbb_core/provider/registry_map.py
+++ b/openbb_platform/core/openbb_core/provider/registry_map.py
@@ -40,8 +40,8 @@ class RegistryMap:
return self._available_providers
@property
- def credentials(self) -> List[str]:
- """Get list of required credentials."""
+ def credentials(self) -> Dict[str, List[str]]:
+ """Get map of providers to credentials."""
return self._credentials
@property
@@ -59,13 +59,11 @@ class RegistryMap:
"""Get available models."""
return self._models
- def _get_credentials(self, registry: Registry) -> List[str]:
- """Get list of required credentials."""
- cred_list = []
- for provider in registry.providers.values():
- for c in provider.credentials:
- cred_list.append(c)
- return cred_list
+ def _get_credentials(self, registry: Registry) -> Dict[str, List[str]]:
+ """Get map of providers to credentials."""
+ return {
+ name: provider.credentials for name, provider in registry.providers.items()
+ }
def _get_available_providers(self, registry: Registry) -> List[str]:
"""Get list of available providers."""