import os
from dotenv import load_dotenv
from google import genai

load_dotenv()

try:
    # Cliente de la nueva librería google-genai
    client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
    
    print("--- Buscando modelos de embedding compatibles ---")
    
    # Listamos los modelos disponibles
    for model in client.models.list():
        # Buscamos nombres que contengan 'embedding'
        if "embedding" in model.name.lower():
            print(f"Modelo encontrado: {model.name}")
            
except Exception as e:
    print(f"Error al conectar: {e}")