Spaces:
Sleeping
Sleeping
| import os | |
| import shutil | |
| import logging | |
| from typing import List, Dict | |
| logger = logging.getLogger(__name__) | |
| def list_available_models(api_key: str) -> List[str]: | |
| """ | |
| Lists available Gemini models using the provided API key. | |
| Filters for strictly 'gemini-pro' and 'gemini-1.5' variants suitable for chat. | |
| """ | |
| import google.generativeai as genai | |
| try: | |
| genai.configure(api_key=api_key) | |
| models = [] | |
| for m in genai.list_models(): | |
| if 'generateContent' in m.supported_generation_methods: | |
| if 'gemini' in m.name: | |
| # Clean up the name (models/gemini-pro -> gemini-pro) | |
| clean_name = m.name.replace('models/', '') | |
| models.append(clean_name) | |
| return models | |
| except Exception as e: | |
| logger.error(f"Failed to list models: {e}") | |
| # Fallback list if API call fails | |
| return ["gemini-pro", "gemini-1.5-flash", "gemini-1.5-pro-latest"] | |