Spaces:
Sleeping
Sleeping
| import os | |
| import google.generativeai as genai | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| api_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY") | |
| genai.configure(api_key=api_key) | |
| print("Listing available Gemini models...") | |
| try: | |
| for m in genai.list_models(): | |
| if 'generateContent' in m.supported_generation_methods: | |
| print(m.name) | |
| except Exception as e: | |
| print(f"List models failed: {e}") | |
| model_name = "gemini-1.5-flash" | |
| print(f"\nTesting model: {model_name}") | |
| try: | |
| model = genai.GenerativeModel(model_name) | |
| response = model.generate_content("Hello, can you see this?") | |
| print("Response:", response.text) | |
| except Exception as e: | |
| print(f"Test failed: {e}") | |