aiecs 1.5.1__py3-none-any.whl

aiecs/tools/apisource/providers/__init__.py

@@ -0,0 +1,115 @@
+"""
+API Providers Module
+
+Contains all API provider implementations for the APISource tool.
+"""
+
+from aiecs.tools.apisource.providers.base import BaseAPIProvider, RateLimiter
+from aiecs.tools.apisource.providers.fred import FREDProvider
+from aiecs.tools.apisource.providers.worldbank import WorldBankProvider
+from aiecs.tools.apisource.providers.newsapi import NewsAPIProvider
+from aiecs.tools.apisource.providers.census import CensusProvider
+
+import logging
+from typing import Dict, List, Optional, Type, Any
+
+logger = logging.getLogger(__name__)
+
+# Global provider registry
+PROVIDER_REGISTRY: Dict[str, Type[BaseAPIProvider]] = {}
+PROVIDER_INSTANCES: Dict[str, BaseAPIProvider] = {}
+
+
+def register_provider(provider_class: Type[BaseAPIProvider]):
+    """
+    Register a provider class.
+
+    Args:
+        provider_class: Provider class to register
+    """
+    # Instantiate to get name
+    temp_instance = provider_class()
+    provider_name = temp_instance.name
+
+    PROVIDER_REGISTRY[provider_name] = provider_class
+    logger.debug(f"Registered provider: {provider_name}")
+
+
+def get_provider(name: str, config: Optional[Dict] = None) -> BaseAPIProvider:
+    """
+    Get a provider instance by name.
+
+    Args:
+        name: Provider name
+        config: Optional configuration for the provider
+
+    Returns:
+        Provider instance
+
+    Raises:
+        ValueError: If provider is not registered
+    """
+    if name not in PROVIDER_REGISTRY:
+        raise ValueError(
+            f"Provider '{name}' not found. "
+            f"Available providers: {', '.join(PROVIDER_REGISTRY.keys())}"
+        )
+
+    # Return cached instance or create new one with config
+    if config is None and name in PROVIDER_INSTANCES:
+        return PROVIDER_INSTANCES[name]
+
+    provider_instance = PROVIDER_REGISTRY[name](config)
+
+    if config is None:
+        PROVIDER_INSTANCES[name] = provider_instance
+
+    return provider_instance
+
+
+def list_providers() -> List[Dict[str, Any]]:
+    """
+    List all registered providers.
+
+    Returns:
+        List of provider metadata dictionaries
+    """
+    providers = []
+    for name, provider_class in PROVIDER_REGISTRY.items():
+        try:
+            # Get or create instance to access metadata
+            provider = get_provider(name)
+            providers.append(provider.get_metadata())
+        except Exception as e:
+            logger.warning(f"Failed to get metadata for provider {name}: {e}")
+            providers.append(
+                {
+                    "name": name,
+                    "description": "Provider metadata unavailable",
+                    "operations": [],
+                    "error": str(e),
+                }
+            )
+
+    return providers
+
+
+# Auto-register all providers
+register_provider(FREDProvider)
+register_provider(WorldBankProvider)
+register_provider(NewsAPIProvider)
+register_provider(CensusProvider)
+
+
+__all__ = [
+    "BaseAPIProvider",
+    "RateLimiter",
+    "FREDProvider",
+    "WorldBankProvider",
+    "NewsAPIProvider",
+    "CensusProvider",
+    "register_provider",
+    "get_provider",
+    "list_providers",
+    "PROVIDER_REGISTRY",
+]