From 88217e8c060c567f2c47a9a7eedc3e365ba58ccb Mon Sep 17 00:00:00 2001 From: sauravniraula Date: Tue, 29 Jul 2025 12:09:33 +0545 Subject: [PATCH 1/2] fix(fastapi): adds check for IMAGE_PROVIDER on not can_change_keys --- servers/fastapi/utils/model_availability.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/servers/fastapi/utils/model_availability.py b/servers/fastapi/utils/model_availability.py index c1dce981..988475ac 100644 --- a/servers/fastapi/utils/model_availability.py +++ b/servers/fastapi/utils/model_availability.py @@ -1,5 +1,6 @@ import os from constants.supported_ollama_models import SUPPORTED_OLLAMA_MODELS +from enums.image_provider import ImageProvider from enums.llm_provider import LLMProvider from utils.custom_llm_provider import list_available_custom_models from utils.get_env import ( @@ -20,6 +21,7 @@ from utils.llm_provider import ( ) from utils.ollama import pull_ollama_model from utils.image_provider import ( + get_selected_image_provider, is_pixels_selected, is_pixabay_selected, is_gemini_flash_selected, @@ -74,7 +76,13 @@ async def check_llm_and_image_provider_api_or_model_availability(): print("-" * 50) if custom_model not in models: raise Exception(f"Model {custom_model} is not available") - elif is_pixels_selected(): + + # Check for Image Provider and API keys + selected_image_provider = get_selected_image_provider() + if not selected_image_provider: + raise Exception("IMAGE_PROVIDER must be provided") + + if selected_image_provider == ImageProvider.PEXELS: pexels_api_key = get_pexels_api_key_env() if not pexels_api_key: raise Exception("PEXELS_API_KEY must be provided") From dd7ff83d212da4f12df185aa7133534155822732 Mon Sep 17 00:00:00 2001 From: sauravniraula Date: Tue, 29 Jul 2025 12:10:29 +0545 Subject: [PATCH 2/2] 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")