mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-04-28 11:30:03 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
"""Dependency injection for FastAPI."""
|
|
|
|
from typing import Optional
|
|
|
|
from config.settings import Settings, get_settings as _get_settings, NVIDIA_NIM_BASE_URL
|
|
from providers.base import BaseProvider, ProviderConfig
|
|
|
|
|
|
# Global provider instance (singleton)
|
|
_provider: Optional[BaseProvider] = None
|
|
|
|
|
|
def get_settings() -> Settings:
|
|
"""Get application settings via dependency injection."""
|
|
return _get_settings()
|
|
|
|
|
|
def get_provider() -> BaseProvider:
|
|
"""Get or create the provider instance based on settings.provider_type."""
|
|
global _provider
|
|
if _provider is None:
|
|
settings = get_settings()
|
|
|
|
if settings.provider_type == "nvidia_nim":
|
|
from providers.nvidia_nim import NvidiaNimProvider
|
|
|
|
config = ProviderConfig(
|
|
api_key=settings.nvidia_nim_api_key,
|
|
base_url=NVIDIA_NIM_BASE_URL,
|
|
rate_limit=settings.nvidia_nim_rate_limit,
|
|
rate_window=settings.nvidia_nim_rate_window,
|
|
nim_settings=settings.nim,
|
|
)
|
|
_provider = NvidiaNimProvider(config)
|
|
else:
|
|
raise ValueError(
|
|
f"Unknown provider_type: '{settings.provider_type}'. "
|
|
f"Supported: 'nvidia_nim'"
|
|
)
|
|
return _provider
|
|
|
|
|
|
async def cleanup_provider():
|
|
"""Cleanup provider resources."""
|
|
global _provider
|
|
if _provider:
|
|
client = getattr(_provider, "_client", None)
|
|
if client and hasattr(client, "aclose"):
|
|
await client.aclose()
|
|
_provider = None
|