Puede resultar práctico agregar la API KEY de Gemini al entorno de Google Colab, de esa manera ya no es necesario declararla en los parámetros de la clase ChatGoogleGenerativeAI que recibe el parámetro api_key = mi_api_key. De esta manera Langchain busca en el entorno de Colab y al encontrar la API KEY la utiliza en automático. Ejemplo:
google_api_key = userdata.get('GEMINI_API_KEY')
os.environ["GEMINI_API_KEY"] = google_api_key
llm = ChatGoogleGenerativeAI(
model="gemini-3.1-flash-lite",
temperature=0.5
)
Al crear el objeto llm ya no fue necesario declarar el parámetro api_key ya lo toma por default de las variables de entorno de Colab.