vioott's picture
refactor: reorganized project by separating routes and services
5b5c1df
raw
history blame
721 Bytes
import os
import google.generativeai as genai
def configure_genai():
"""
Configura a chave de API e inicializa o modelo Gemini.
"""
api_key = os.environ.get("GOOGLE_API_KEY")
if not api_key:
raise EnvironmentError(
"❌ A variável de ambiente GOOGLE_API_KEY não está definida."
)
try:
genai.configure(api_key=api_key)
# Testa se a chave está funcionando
model = genai.GenerativeModel("gemini-1.5-flash")
model.start_chat().send_message("ping")
print("✅ Chave da API configurada com sucesso.")
return model
except Exception as e:
raise RuntimeError(f"❌ Falha ao configurar a chave da API: {e}")