Oliver/presenton
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source 
fix/image generation
This commit is contained in:
Saurav Niraula 2025-07-29 12:11:05 +05:45 • committed by GitHub
commit 18d2935bee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
servers/fastapi/utils/model_availability.py
View file

@ -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,22 +76,28 @@ 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")
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")