fix(fastapi): returns placeholder image is image provider is not selected
This commit is contained in:
parent
71ff6e6adf
commit
bb55e6a2d2
1 changed files with 5 additions and 2 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue