From dd7ff83d212da4f12df185aa7133534155822732 Mon Sep 17 00:00:00 2001 From: sauravniraula Date: Tue, 29 Jul 2025 12:10:29 +0545 Subject: [PATCH] chore(fastapi) --- servers/fastapi/utils/model_availability.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/servers/fastapi/utils/model_availability.py b/servers/fastapi/utils/model_availability.py index 988475ac..210fcdd2 100644 --- a/servers/fastapi/utils/model_availability.py +++ b/servers/fastapi/utils/model_availability.py @@ -87,17 +87,17 @@ async def check_llm_and_image_provider_api_or_model_availability(): if not pexels_api_key: raise Exception("PEXELS_API_KEY must be provided") - elif is_pixabay_selected(): + elif selected_image_provider == ImageProvider.PIXABAY: pixabay_api_key = get_pixabay_api_key_env() if not pixabay_api_key: raise Exception("PIXABAY_API_KEY must be provided") - elif is_gemini_flash_selected(): + elif selected_image_provider == ImageProvider.GEMINI_FLASH: google_api_key = get_google_api_key_env() if not google_api_key: raise Exception("GOOGLE_API_KEY must be provided") - elif is_dalle3_selected(): + elif selected_image_provider == ImageProvider.DALLE3: openai_api_key = get_openai_api_key_env() if not openai_api_key: raise Exception("OPENAI_API_KEY must be provided")