File size: 1,558 Bytes
a91be8c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import os
from pathlib import Path
from dotenv import load_dotenv
import google.generativeai as genai # 1. Import the library
# --- Robustly find and load the .env file ---
try:
# Navigate from the current script's directory up to the project root
script_dir = Path(__file__).resolve().parent
# Adjust this if your script is nested differently
project_root = script_dir.parent.parent
dotenv_path = project_root / "configs" / ".env"
if dotenv_path.exists():
load_dotenv(dotenv_path=dotenv_path)
print(f"✅ Environment variables loaded from: {dotenv_path}")
else:
print(f"⚠️ Warning: .env file not found at {dotenv_path}.")
except Exception as e:
print(f"Could not load .env file: {e}")
# --- Get API key and list models using the client ---
api_key = os.getenv("GEMINI_API_KEY")
if api_key:
try:
# 2. Create a client instance. It automatically uses the API key
# from the environment variables.
client = genai.Client()
print("\n✅ Available models for 'generateContent':")
# 3. Use the client object to list the models
for m in client.models.list():
if 'generateContent' in m.supported_generation_methods:
print(f"- {m.name}")
except Exception as e:
print(f"🔴 An error occurred while listing models: {e}")
print("💡 Tip: Make sure your API key is correct and has the right permissions.")
else:
print("🔴 Error: GEMINI_API_KEY not found. Please check your .env file.") |