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

  1. 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.
  2. 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 the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  3. Verify that billing is enabled for your Google Cloud project.

  4. 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 the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the APIs

  5. 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 the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  6. Verify that billing is enabled for your Google Cloud project.

  7. 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 the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the APIs

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.

Para obtener más información sobre cómo otorgar roles, consulta Administra el acceso a proyectos, carpetas y organizaciones.

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

  1. En la Google Cloud consola de, ve a la página Instancias.

    Ir a Instancias

  2. Junto al nombre de una instancia de Vertex AI Workbench, haz clic en Abrir JupyterLab.

    Tu instancia de Vertex AI Workbench abre JupyterLab.

  3. En JupyterLab, haz clic en Archivo > Nuevo lanzador.

  4. En la pestaña Lanzador, en la sección Otro, haz clic en el mosaico Gemini CLI.

  5. Si es la primera vez que abres una terminal de Gemini CLI, ingresa Y para aceptar los términos y condiciones.

    Tu instancia de Vertex AI Workbench instala la Gemini CLI.

  6. 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:

  1. Asegúrate de que Gemini CLI esté habilitada y de que el usuario o creador haya aceptado los términos y condiciones.
  2. En la primera línea de una celda nueva, ingresa %%geminicli_magic.
  3. En la misma celda, ingresa tu instrucción en la siguiente línea.
  4. 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?