Merge pull request #156 from presenton/fix/image-generation

fix(fastapi): returns placeholder image is image provider is not selected
This commit is contained in:
Saurav Niraula 2025-07-29 12:01:01 +05:45 committed by GitHub
commit 88437178b7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -24,13 +24,16 @@ def is_dalle3_selected() -> bool:
return ImageProvider.DALLE3 == get_selected_image_provider()
def get_selected_image_provider() -> ImageProvider:
def get_selected_image_provider() -> ImageProvider | None:
"""
Get the selected image provider from environment variables.
Returns:
ImageProvider: The selected image provider.
"""
return ImageProvider(get_image_provider_env())
image_provider_env = get_image_provider_env()
if image_provider_env:
return ImageProvider(image_provider_env)
return None
def get_image_provider_api_key() -> str: