Usa Gemini CLI
En esta página, se describe cómo usar la interfaz de línea de comandos (CLI) de Gemini con una instancia de Vertex AI Workbench.
Este documento está dirigido a analistas de datos, científicos de datos y desarrolladores de datos que trabajan con Vertex AI Workbench. En este documento, se supone que tienes conocimientos sobre cómo escribir código en un entorno de notebook.
Descripción general
Gemini CLI es un agente de IA de código abierto que proporciona acceso a Gemini directamente en una terminal. Para obtener más información, consulta geminicli.com.
Gemini CLI está disponible en instancias de Vertex AI Workbench. Puedes usar Gemini CLI para hacer lo siguiente:
- Crea un cuaderno nuevo.
- Ejecuta celdas de notebook.
- Escribe y edita las celdas de código y texto de un notebook.
- Explica código y conceptos técnicos.
- Interactúa con el sistema de archivos local de una instancia de Vertex AI Workbench, lo que incluye realizar operaciones de archivos complejas que abarcan varios archivos en función de una sola instrucción de alto nivel.
- Ejecuta comandos básicos de shell.
- Ejecuta comandos para interactuar con otros Google Cloud servicios, como Gemini Enterprise Agent Platform y BigQuery.
Limitaciones
Ten en cuenta las siguientes limitaciones cuando uses Gemini CLI con Vertex AI Workbench:
Gemini CLI es solo una CLI. No se incluyen una interfaz gráfica de chat ni herramientas avanzadas en el editor.
Cuando le pides a Gemini CLI que modifique un notebook, Gemini CLI cambia el archivo de notebook directamente en el disco de la instancia. Por este motivo, no puedes deshacer las ediciones realizadas por Gemini CLI con el botón Deshacer del editor de notebooks ni con Control+Z (Command+Z en macOS). Sin embargo, puedes pedirle a Gemini CLI que deshaga un cambio con un comando de lenguaje natural, como
Undo your last change.
Antes de comenzar
- Accede a tu Google Cloud cuenta de. Si eres nuevo en Google Cloud, crea una cuenta para evaluar el rendimiento de nuestros productos en situaciones reales. Los clientes nuevos también obtienen $300 en créditos gratuitos para ejecutar, probar y, además, implementar cargas de trabajo.
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Roles required to select or create a project
- Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
-
Create a project: To create a project, you need the Project Creator role
(
roles/resourcemanager.projectCreator), which contains theresourcemanager.projects.createpermission. Learn how to grant roles.
-
Verify that billing is enabled for your Google Cloud project.
Enable the Compute Engine, Notebooks, and Agent Platform APIs.
Roles required to enable APIs
To enable APIs, you need the Service Usage Admin IAM role (
roles/serviceusage.serviceUsageAdmin), which contains theserviceusage.services.enablepermission. Learn how to grant roles.-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Roles required to select or create a project
- Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
-
Create a project: To create a project, you need the Project Creator role
(
roles/resourcemanager.projectCreator), which contains theresourcemanager.projects.createpermission. Learn how to grant roles.
-
Verify that billing is enabled for your Google Cloud project.
Enable the Compute Engine, Notebooks, and Agent Platform APIs.
Roles required to enable APIs
To enable APIs, you need the Service Usage Admin IAM role (
roles/serviceusage.serviceUsageAdmin), which contains theserviceusage.services.enablepermission. Learn how to grant roles.
Roles obligatorios
Para usar Gemini CLI en Vertex AI Workbench, debes otorgar permisos al usuario de la instancia de Vertex AI Workbench y a la cuenta de servicio de la instancia.
Otorga permisos al usuario de la instancia
Para obtener los permisos que
necesitas para usar Gemini CLI en una instancia de Vertex AI Workbench,
pídele a tu administrador que te otorgue el
rol de IAM Usuario de Agent Platform (roles/aiplatform.user)
en el proyecto.
Para obtener más información sobre cómo otorgar roles, consulta Administra el acceso a proyectos, carpetas y organizaciones.
También puedes obtener los permisos necesarios a través de roles personalizados o cualquier otro rol predefinido.
Otorga un permiso a la cuenta de servicio de tu instancia
Para asegurarte de que la cuenta de servicio de tu instancia de Vertex AI Workbench tenga el permiso necesario para permitir que Gemini CLI se ejecute en una instancia de Vertex AI Workbench, pídele a tu administrador que le otorgue el rol de IAM Usuario de Agent Platform (roles/aiplatform.user)) a la cuenta de servicio de tu instancia de Vertex AI Workbench en el proyecto.
Este rol predefinido contiene el
aiplatform.endpoints.predict
permiso,
que es necesario para
permitir que Gemini CLI se ejecute en una instancia de Vertex AI Workbench.
Es posible que tu administrador también pueda otorgar este permiso a la cuenta de servicio de tu instancia de Vertex AI Workbench con este permiso con roles personalizados o con otros roles predefinidos.
Usa Gemini CLI
-
En la Google Cloud consola de, ve a la página Instancias.
-
Junto al nombre de una instancia de Vertex AI Workbench, haz clic en Abrir JupyterLab.
Tu instancia de Vertex AI Workbench abre JupyterLab.
-
En JupyterLab, haz clic en Archivo > Nuevo lanzador.
-
En la pestaña Lanzador, en la sección Otro, haz clic en el mosaico Gemini CLI.
-
Si es la primera vez que abres una terminal de Gemini CLI, ingresa
Ypara aceptar los términos y condiciones.Tu instancia de Vertex AI Workbench instala la Gemini CLI.
-
En la terminal de Gemini CLI, ingresa un mensaje.
Por ejemplo, puedes ingresar
Create a new notebook named 'test-notebook'. Para ver ejemplos de mensajes que podrían ser útiles, consulta Ejemplos de instrucciones.
Ejemplos de instrucciones
Para ayudarte a obtener ideas sobre cómo usar Gemini CLI, consulta los siguientes ejemplos de instrucciones:
"Create a new notebook that trains a model to predict 'income bracket' from bigquery-public-data.ml_datasets.census_adult_income, using BigQuery and Python."
"Summarize the notebook named 'test-file', and propose next steps for the project."
"I want to get a quick overview of the notebooks in this directory. For every .ipynb file, show me the first 5 lines of the file."
"Create a script using the contents of the 'test-file' notebook."
"Show me how to access data from BigQuery tables from within Vertex AI Workbench."
"Query the bigquery-public-data.ml_datasets.census_adult_income table to find the number of people with an income bracket of > 50K."
"Set my default Google Cloud project to my-project."
"Create a bucket de Cloud Storage, and upload all the CSV files from my current directory to it."
"Create a Compute Engine instance with a Debian 11 image and an n1-standard-4 tipo de máquina."
"Create a notebook file that runs through the code in the 'test-script'. Add text cells that explain the code."
Controla el acceso a Gemini CLI
Puedes controlar el acceso a Gemini CLI en Vertex AI Workbench con los siguientes métodos:
Un administrador puede configurar una política de la organización para restringir el uso de modelos específicos de Gemini a nivel de la organización, la carpeta o el proyecto. Consulta Controla el acceso a los modelos de Model Garden models. Gemini CLI sigue apareciendo en JupyterLab, pero la CLI no responde a las instrucciones.
Si no otorga el permiso
aiplatform.endpoints.predict, un administrador puede bloquear algunas identidades para que no puedan usar los extremos de Gemini para la inferencia.
Usa el comando mágico de Gemini CLI
Para usar Gemini CLI directamente dentro de una celda en tu archivo de notebook, haz lo siguiente:
- Asegúrate de que Gemini CLI esté habilitada y de que el usuario o creador haya aceptado los términos y condiciones.
- En la primera línea de una celda nueva, ingresa
%%geminicli_magic. - En la misma celda, ingresa tu instrucción en la siguiente línea.
- Ejecuta la celda.
Gemini CLI agrega una celda nueva a continuación con su respuesta.
Solucionar problemas
Si tienes algún problema para usar Gemini CLI con instancias de Vertex AI Workbench, consulta Solución de problemas de Vertex AI Workbench para obtener ayuda con problemas habituales.
¿Qué sigue?
Obtén más información sobre Gemini.
Para obtener información sobre los métodos para consultar datos de BigQuery en notebooks de Vertex AI Workbench, consulta Consulta datos en BigQuery desde JupyterLab.