Generar texto con la función ML.GENERATE_TEXT

En este documento se explica cómo crear un modelo remoto de BigQuery ML que representa un modelo de Vertex AI y, a continuación, usar ese modelo remoto con la función ML.GENERATE_TEXT para generar texto.

Se admiten los siguientes tipos de modelos remotos:

En función del modelo de Vertex AI que elijas, puedes generar texto a partir de datos sin estructurar de tablas de objetos o de texto de tablas estándar.

Roles obligatorios

Para crear un modelo remoto y generar texto, necesitas los siguientes roles de Gestión de Identidades y Accesos (IAM):

  • Crear y usar conjuntos de datos, tablas y modelos de BigQuery: Editor de datos de BigQuery (roles/bigquery.dataEditor) en tu proyecto.
  • Crear, delegar y usar conexiones de BigQuery: Administrador de conexiones de BigQuery (roles/bigquery.connectionsAdmin) en tu proyecto.

    Si no tiene configurada una conexión predeterminada, puede crear una y definirla al ejecutar la instrucción CREATE MODEL. Para ello, debes tener el rol Administrador de BigQuery (roles/bigquery.admin) en tu proyecto. Para obtener más información, consulta Configurar la conexión predeterminada.

  • Concede permisos a la cuenta de servicio de la conexión: administrador de gestión de identidades y accesos del proyecto (roles/resourcemanager.projectIamAdmin) en el proyecto que contiene el endpoint de Vertex AI. Este es el proyecto actual de los modelos remotos que creas especificando el nombre del modelo como endpoint. Es el proyecto identificado en la URL de los modelos remotos que creas especificando una URL como endpoint.

    Si usas el modelo remoto para analizar datos no estructurados de una tabla de objetos y el segmento de Cloud Storage que usas en la tabla de objetos está en un proyecto distinto al de tu endpoint de Vertex AI, también debes tener el rol Administrador de Storage (roles/storage.admin) en el segmento de Cloud Storage que usa la tabla de objetos.

  • Crear tareas de BigQuery: Usuario de tareas de BigQuery (roles/bigquery.jobUser) en tu proyecto.

Estos roles predefinidos contienen los permisos necesarios para realizar las tareas descritas en este documento. Para ver los permisos exactos que se necesitan, despliega la sección Permisos necesarios:

Permisos obligatorios

  • Crea un conjunto de datos: bigquery.datasets.create
  • Crear, delegar y usar una conexión: bigquery.connections.*
  • Define los permisos de la cuenta de servicio: resourcemanager.projects.getIamPolicy y resourcemanager.projects.setIamPolicy
  • Crea un modelo y ejecuta la inferencia:
    • bigquery.jobs.create
    • bigquery.models.create
    • bigquery.models.getData
    • bigquery.models.updateData
    • bigquery.models.updateMetadata

También puedes obtener estos permisos con roles personalizados u otros roles predefinidos.

Antes de empezar

  1. 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 (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

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

  3. Enable the BigQuery, BigQuery Connection, and Vertex AI 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

Crear conjunto de datos

Crea un conjunto de datos de BigQuery para que contenga tus recursos:

Consola

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

    Ir a la página de BigQuery

  2. En el panel Explorador, haz clic en el nombre de tu proyecto.

  3. Haz clic en Ver acciones > Crear conjunto de datos.

  4. En la página Crear conjunto de datos, haz lo siguiente:

    • En ID del conjunto de datos, escribe un nombre para el conjunto de datos.

    • En Tipo de ubicación, seleccione una ubicación para el conjunto de datos.

    • Haz clic en Crear conjunto de datos.

bq

  1. Para crear un conjunto de datos, usa el comando bq mk con la marca --location:

    bq --location=LOCATION mk -d DATASET_ID

    Haz los cambios siguientes:

    • LOCATION: la ubicación del conjunto de datos.
    • DATASET_ID es el ID del conjunto de datos que vas a crear.
  2. Confirma que se ha creado el conjunto de datos:

    bq ls

Crear una conexión

Puedes saltarte este paso si tienes una conexión predeterminada configurada o si tienes el rol Administrador de BigQuery.

Crea una conexión de recursos de Cloud para que la use el modelo remoto y obtén la cuenta de servicio de la conexión. Crea la conexión en la misma ubicación que el conjunto de datos que has creado en el paso anterior.

Selecciona una de las opciones siguientes:

Consola

  1. Ve a la página BigQuery.

    Ir a BigQuery

  2. En el panel Explorador, haga clic en Añadir datos:

    El elemento de la interfaz de usuario para añadir datos.

    Se abrirá el cuadro de diálogo Añadir datos.

  3. En el panel Filtrar por, en la sección Tipo de fuente de datos, selecciona Aplicaciones empresariales.

    También puede introducir Vertex AI en el campo Buscar fuentes de datos.

  4. En la sección Fuentes de datos destacadas, haga clic en Vertex AI.

  5. Haz clic en la tarjeta de solución Modelos de Vertex AI: federación de BigQuery.

  6. En la lista Tipo de conexión, selecciona Modelos remotos, funciones remotas y BigLake (recurso de Cloud) de Vertex AI.

  7. En el campo ID de conexión, introduce un nombre para la conexión.

  8. Haga clic en Crear conexión.

  9. Haz clic en Ir a la conexión.

  10. En el panel Información de conexión, copia el ID de la cuenta de servicio para usarlo en un paso posterior.

bq

  1. En un entorno de línea de comandos, crea una conexión:

    bq mk --connection --location=REGION --project_id=PROJECT_ID \
        --connection_type=CLOUD_RESOURCE CONNECTION_ID

    El parámetro --project_id anula el proyecto predeterminado.

    Haz los cambios siguientes:

    • REGION: tu región de conexión
    • PROJECT_ID: tu ID de proyecto Google Cloud
    • CONNECTION_ID: un ID para tu conexión

    Cuando creas un recurso de conexión, BigQuery crea una cuenta de servicio del sistema única y la asocia a la conexión.

    Solución de problemas: si aparece el siguiente error de conexión, actualiza el SDK de Google Cloud:

    Flags parsing error: flag --connection_type=CLOUD_RESOURCE: value should be one of...
    
  2. Obtén y copia el ID de la cuenta de servicio para usarlo en un paso posterior:

    bq show --connection PROJECT_ID.REGION.CONNECTION_ID

    El resultado debería ser similar al siguiente:

    name                          properties
    1234.REGION.CONNECTION_ID     {"serviceAccountId": "connection-1234-9u56h9@gcp-sa-bigquery-condel.iam.gserviceaccount.com"}
    

Terraform

Usa el recurso google_bigquery_connection.

Para autenticarte en BigQuery, configura las credenciales predeterminadas de la aplicación. Para obtener más información, consulta Configurar la autenticación para bibliotecas de cliente.

En el siguiente ejemplo se crea una conexión de recursos de Cloud llamada my_cloud_resource_connection en la región US:


# This queries the provider for project information.
data "google_project" "default" {}

# This creates a cloud resource connection in the US region named my_cloud_resource_connection.
# Note: The cloud resource nested object has only one output field - serviceAccountId.
resource "google_bigquery_connection" "default" {
  connection_id = "my_cloud_resource_connection"
  project       = data.google_project.default.project_id
  location      = "US"
  cloud_resource {}
}

Para aplicar la configuración de Terraform en un proyecto, sigue los pasos que se indican en las siguientes secciones. Google Cloud

Preparar Cloud Shell

  1. Abre Cloud Shell.
  2. Define el Google Cloud proyecto Google Cloud predeterminado en el que quieras aplicar tus configuraciones de Terraform.

    Solo tiene que ejecutar este comando una vez por proyecto y puede hacerlo en cualquier directorio.

    export GOOGLE_CLOUD_PROJECT=PROJECT_ID

    Las variables de entorno se anulan si defines valores explícitos en el archivo de configuración de Terraform.

Preparar el directorio

Cada archivo de configuración de Terraform debe tener su propio directorio (también llamado módulo raíz).

  1. En Cloud Shell, crea un directorio y un archivo nuevo en ese directorio. El nombre del archivo debe tener la extensión .tf. Por ejemplo, main.tf. En este tutorial, nos referiremos al archivo como main.tf.
    mkdir DIRECTORY && cd DIRECTORY && touch main.tf
  2. Si estás siguiendo un tutorial, puedes copiar el código de ejemplo de cada sección o paso.

    Copia el código de ejemplo en el archivo main.tf que acabas de crear.

    También puedes copiar el código de GitHub. Se recomienda cuando el fragmento de Terraform forma parte de una solución integral.

  3. Revisa y modifica los parámetros de ejemplo para aplicarlos a tu entorno.
  4. Guarda los cambios.
  5. Inicializa Terraform. Solo tienes que hacerlo una vez por directorio.
    terraform init

    Si quieres usar la versión más reciente del proveedor de Google, incluye la opción -upgrade:

    terraform init -upgrade

Aplica los cambios

  1. Revisa la configuración y comprueba que los recursos que va a crear o actualizar Terraform se ajustan a tus expectativas:
    terraform plan

    Haga las correcciones necesarias en la configuración.

  2. Aplica la configuración de Terraform ejecutando el siguiente comando e introduciendo yes en la petición:
    terraform apply

    Espera hasta que Terraform muestre el mensaje "Apply complete!".

  3. Abre tu Google Cloud proyecto para ver los resultados. En la Google Cloud consola, ve a tus recursos en la interfaz de usuario para asegurarte de que Terraform los ha creado o actualizado.

Dar acceso a las cuentas de servicio

Debes asignar el rol Usuario de Vertex AI a la cuenta de servicio de la conexión que usa el modelo remoto. Si usas el modelo remoto para generar texto a partir de datos de una tabla de objetos, también debes asignar el rol de usuario de Vertex AI a la cuenta de servicio de la conexión que usa la tabla de objetos.

Conceder un rol a la cuenta de servicio de la conexión del modelo remoto

Asigna el rol Usuario de Vertex AI a la cuenta de servicio de la conexión.

Si tiene previsto especificar el endpoint como una URL al crear el modelo remoto (por ejemplo, endpoint = 'https://us-central1-aiplatform.googleapis.com/v1/projects/myproject/locations/us-central1/publishers/google/models/gemini-2.0-flash'), asigne este rol en el mismo proyecto que especifique en la URL.

Si tienes previsto especificar el endpoint usando el nombre del modelo al crear el modelo remoto (por ejemplo, endpoint = 'gemini-2.0-flash'), otorga este rol en el mismo proyecto en el que vayas a crear el modelo remoto.

Si se asigna el rol en otro proyecto, se produce el error bqcx-1234567890-wxyz@gcp-sa-bigquery-condel.iam.gserviceaccount.com does not have the permission to access resource.

Para conceder el rol, sigue estos pasos:

Consola

  1. Ve a la página IAM y administración.

    Ir a IAM y administración

  2. Haz clic en Añadir.

    Se abrirá el cuadro de diálogo Añadir principales.

  3. En el campo Nuevos principales, introduce el ID de la cuenta de servicio que has copiado anteriormente.

  4. En el campo Selecciona un rol, elige Vertex AI y, a continuación, Usuario de Vertex AI.

  5. Haz clic en Guardar.

gcloud

Usa el comando gcloud projects add-iam-policy-binding.

gcloud projects add-iam-policy-binding 'PROJECT_NUMBER' --member='serviceAccount:MEMBER' --role='roles/aiplatform.user' --condition=None

Haz los cambios siguientes:

  • PROJECT_NUMBER: tu número de proyecto
  • MEMBER: el ID de la cuenta de servicio que has copiado antes

Conceder un rol a la cuenta de servicio de la conexión de tabla de objetos

Si usas el modelo remoto para generar texto a partir de datos de una tabla de objetos, concede a la cuenta de servicio de la conexión de la tabla de objetos el rol de usuario de Vertex AI.

Para encontrar la cuenta de servicio de la conexión de la tabla de objetos, sigue estos pasos:

  1. Ve a la página BigQuery.

    Ir a BigQuery

  2. En el panel Explorador, despliega el conjunto de datos que contiene la tabla de objetos.

  3. Selecciona la tabla de objetos.

  4. En el panel del editor, haz clic en la pestaña Detalles.

  5. Anota el nombre de la conexión en el campo ID de conexión.

  6. En el panel Explorador, despliega la carpeta Conexiones externas.

  7. Seleccione la conexión que coincida con la del campo ID de conexión de la tabla de objetos.

  8. Copia el valor del campo ID de cuenta de servicio.

Para conceder el rol, sigue estos pasos:

Consola

  1. Ve a la página IAM y administración.

    Ir a IAM y administración

  2. Haz clic en Añadir.

    Se abrirá el cuadro de diálogo Añadir principales.

  3. En el campo Nuevos principales, introduce el ID de la cuenta de servicio que has copiado anteriormente.

  4. En el campo Selecciona un rol, elige Vertex AI y, a continuación, Usuario de Vertex AI.

  5. Haz clic en Guardar.

gcloud

Usa el comando gcloud projects add-iam-policy-binding.

gcloud projects add-iam-policy-binding 'PROJECT_NUMBER' --member='serviceAccount:MEMBER' --role='roles/aiplatform.user' --condition=None

Haz los cambios siguientes:

  • PROJECT_NUMBER: tu número de proyecto
  • MEMBER: el ID de la cuenta de servicio que has copiado antes

Habilitar un modelo de partner

Este paso solo es necesario si quieres usar los modelos de IA Claude, Llama o Mistral de Anthropic.

  1. En la Google Cloud consola, ve a la página Model Garden de Vertex AI.

    Ir a Model Garden

  2. Busca el modelo de colaborador que quieras usar.

  3. Haz clic en la tarjeta del modelo.

  4. En la página del modelo, haz clic en Habilitar.

  5. Rellena la información solicitada para habilitar la función y haz clic en Siguiente.

  6. En la sección Términos y condiciones, marca la casilla.

  7. Haz clic en Aceptar para aceptar los términos y condiciones y habilitar el modelo.

Desplegar un modelo abierto

Si quieres usar un modelo abierto compatible, primero debes desplegarlo en Vertex AI. Para obtener más información sobre cómo hacerlo, consulta Implementar modelos abiertos.

Crear un modelo remoto de BigQuery ML

Crea un modelo remoto:

Modelos abiertos

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

    Ir a BigQuery

  2. En el editor de SQL, crea un modelo remoto:

    CREATE OR REPLACE MODEL
    `PROJECT_ID.DATASET_ID.MODEL_NAME`
    REMOTE WITH CONNECTION {DEFAULT | `PROJECT_ID.REGION.CONNECTION_ID`}
    OPTIONS (ENDPOINT = 'https://ENDPOINT_REGION-aiplatform.googleapis.com/v1/projects/ENDPOINT_PROJECT_ID/locations/ENDPOINT_REGION/endpoints/ENDPOINT_ID');

    Haz los cambios siguientes:

    • PROJECT_ID: tu ID de proyecto.
    • DATASET_ID: ID del conjunto de datos que contendrá el modelo. Este conjunto de datos debe estar en la misma ubicación que la conexión que estés usando.
    • MODEL_NAME: el nombre del modelo.
    • REGION: la región que usa la conexión.
    • CONNECTION_ID: el ID de tu conexión de BigQuery.

      Para obtener este valor, consulta los detalles de la conexión en la consola Google Cloud y copia el valor de la última sección del ID de conexión completo que se muestra en ID de conexión. Por ejemplo, projects/myproject/locations/connection_location/connections/myconnection.

    • ENDPOINT_REGION: la región en la que se ha desplegado el modelo abierto.
    • ENDPOINT_PROJECT_ID: el proyecto en el que se ha desplegado el modelo abierto.
    • ENDPOINT_ID: el ID del endpoint HTTPS que usa el modelo abierto. Para obtener el ID del endpoint, busca el modelo abierto en la página Predicción online y copia el valor del campo ID.

Todos los demás modelos

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

    Ir a BigQuery

  2. En el editor de SQL, crea un modelo remoto:

    CREATE OR REPLACE MODEL
    `PROJECT_ID.DATASET_ID.MODEL_NAME`
    REMOTE WITH CONNECTION `PROJECT_ID.REGION.CONNECTION_ID`
    OPTIONS (ENDPOINT = 'ENDPOINT');

    Haz los cambios siguientes:

    • PROJECT_ID: tu ID de proyecto.
    • DATASET_ID: ID del conjunto de datos que contendrá el modelo. Este conjunto de datos debe estar en la misma ubicación que la conexión que estés usando.
    • MODEL_NAME: el nombre del modelo.
    • REGION: la región que usa la conexión.
    • CONNECTION_ID: el ID de tu conexión de BigQuery.

      Para obtener este valor, consulta los detalles de la conexión en la consola Google Cloud y copia el valor de la última sección del ID de conexión completo que se muestra en ID de conexión. Por ejemplo, projects/myproject/locations/connection_location/connections/myconnection.

    • ENDPOINT: el endpoint del modelo de Vertex AI que se va a usar.

      En el caso de los modelos preentrenados de Vertex AI, los modelos Claude y los modelos de Mistral AI, especifica el nombre del modelo. En algunos de estos modelos, puedes especificar una versión concreta del modelo como parte del nombre. En los modelos de Gemini compatibles, puedes especificar el endpoint global para mejorar la disponibilidad.

      En el caso de los modelos Llama, especifica un endpoint de la API de OpenAI con el formato openapi/<publisher_name>/<model_name>. Por ejemplo, openapi/meta/llama-3.1-405b-instruct-maas.

      Para obtener información sobre los nombres y las versiones de los modelos admitidos, consulta ENDPOINT.

      El modelo de Vertex AI que especifiques debe estar disponible en la ubicación en la que crees el modelo remoto. Para obtener más información, consulta Ubicaciones.

Generar texto a partir de datos de tablas estándar

Genera texto con la función ML.GENERATE_TEXT con datos de peticiones de una tabla estándar:

Gemini

SELECT *
FROM ML.GENERATE_TEXT(
  MODEL `PROJECT_ID.DATASET_ID.MODEL_NAME`,
  {TABLE PROJECT_ID.DATASET_ID.TABLE_NAME | (PROMPT_QUERY)},
  STRUCT(TOKENS AS max_output_tokens, TEMPERATURE AS temperature,
  TOP_P AS top_p, FLATTEN_JSON AS flatten_json_output,
  STOP_SEQUENCES AS stop_sequences,
  GROUND_WITH_GOOGLE_SEARCH AS ground_with_google_search,
  SAFETY_SETTINGS AS safety_settings,
  REQUEST_TYPE AS request_type)
);

Haz los cambios siguientes:

  • PROJECT_ID: tu ID de proyecto.
  • DATASET_ID: el ID del conjunto de datos que contiene el modelo.
  • MODEL_NAME: el nombre del modelo.
  • TABLE_NAME: el nombre de la tabla que contiene la petición. Esta tabla debe tener una columna llamada prompt. También puedes usar un alias para usar una columna con otro nombre.
  • PROMPT_QUERY: una consulta que proporciona los datos de la petición. Esta consulta debe generar una columna llamada prompt.
  • TOKENS: un valor INT64 que define el número máximo de tokens que se pueden generar en la respuesta. Este valor debe estar en el intervalo [1,8192]. Especifica un valor inferior para las respuestas más cortas y un valor superior para las respuestas más largas. El valor predeterminado es 128.
  • TEMPERATURE: un valor FLOAT64 en el intervalo [0.0,1.0] que controla el grado de aleatoriedad en la selección de tokens. El valor predeterminado es 0.

    Los valores bajos de temperature son adecuados para las peticiones que requieren una respuesta más determinista y menos abierta o creativa, mientras que los valores altos de temperature pueden dar lugar a resultados más diversos o creativos. El valor de 0 para temperature es determinista, lo que significa que siempre se selecciona la respuesta con la probabilidad más alta.

  • TOP_P: un valor FLOAT64 en el intervalo [0.0,1.0] ayuda a determinar la probabilidad de los tokens seleccionados. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. El valor predeterminado es 0.95.
  • FLATTEN_JSON: un valor BOOL que determina si se debe devolver el texto generado y los atributos de seguridad en columnas independientes. El valor predeterminado es FALSE.
  • STOP_SEQUENCES: un valor ARRAY<STRING> que elimina las cadenas especificadas si se incluyen en las respuestas del modelo. Las cadenas deben coincidir exactamente, incluidas las mayúsculas y las minúsculas. El valor predeterminado es un array vacío.
  • GROUND_WITH_GOOGLE_SEARCH: un valor BOOL que determina si el modelo de Vertex AI usa [Fundamentación con la Búsqueda de Google](/vertex-ai/generative-ai/docs/grounding/overview#ground-public) al generar respuestas. La fundamentación permite que el modelo use información adicional de Internet al generar una respuesta para que las respuestas del modelo sean más específicas y factuales. Si tanto flatten_json_output como este campo tienen el valor True, se incluye una columna ml_generate_text_grounding_result adicional en los resultados, que proporciona las fuentes que ha usado el modelo para obtener información adicional. El valor predeterminado es FALSE.
  • SAFETY_SETTINGS: valor ARRAY<STRUCT<STRING AS category, STRING AS threshold>> que configura los umbrales de seguridad del contenido para filtrar las respuestas. El primer elemento de la estructura especifica una categoría de contenido dañino y el segundo elemento especifica un umbral de bloqueo correspondiente. El modelo filtra el contenido que infringe estos ajustes. Solo puedes especificar cada categoría una vez. Por ejemplo, no puedes especificar STRUCT('HARM_CATEGORY_DANGEROUS_CONTENT' AS category, 'BLOCK_MEDIUM_AND_ABOVE' AS threshold) y STRUCT('HARM_CATEGORY_DANGEROUS_CONTENT' AS category, 'BLOCK_ONLY_HIGH' AS threshold). Si no hay ningún ajuste de seguridad para una categoría determinada, se utiliza el ajuste de seguridad BLOCK_MEDIUM_AND_ABOVE. Las categorías admitidas son las siguientes:
    • HARM_CATEGORY_HATE_SPEECH
    • HARM_CATEGORY_DANGEROUS_CONTENT
    • HARM_CATEGORY_HARASSMENT
    • HARM_CATEGORY_SEXUALLY_EXPLICIT
    Los umbrales admitidos son los siguientes:
    • BLOCK_NONE (Restringido)
    • BLOCK_LOW_AND_ABOVE
    • BLOCK_MEDIUM_AND_ABOVE (predeterminado)
    • BLOCK_ONLY_HIGH
    • HARM_BLOCK_THRESHOLD_UNSPECIFIED
    Para obtener más información, consulta la definición de categoría de seguridad y umbral de bloqueo.
  • REQUEST_TYPE: un valor STRING que especifica el tipo de solicitud de inferencia que se enviará al modelo de Gemini. El tipo de solicitud determina qué cuota usa la solicitud. Estos son los valores válidos:
    • DEDICATED: la función ML.GENERATE_TEXT solo usa la cuota de rendimiento aprovisionado. La función ML.GENERATE_TEXT devuelve el error Provisioned throughput is not purchased or is not active si no hay cuota de capacidad de procesamiento aprovisionada disponible.
    • SHARED: La función ML.GENERATE_TEXT solo usa la cuota compartida dinámica (DSQ), aunque hayas comprado cuota de rendimiento aprovisionado.
    • UNSPECIFIED: La función ML.GENERATE_TEXT usa la cuota de la siguiente manera:
      • Si no has comprado cuota de rendimiento aprovisionado, la función ML.GENERATE_TEXT usa la cuota de DSQ.
      • Si has comprado cuota de capacidad de procesamiento aprovisionada, la función ML.GENERATE_TEXT usará primero esa cuota. Si las solicitudes superan la cuota de capacidad de procesamiento aprovisionada, el tráfico de desbordamiento utiliza la cuota de DSQ.
    • El valor predeterminado es UNSPECIFIED.

      Para obtener más información, consulta Usar el rendimiento aprovisionado de Vertex AI.

    Ejemplo 1

    En el ejemplo siguiente se muestra una solicitud con estas características:

    • Pide un resumen del texto de la columna body de la tabla articles.
    • Analiza la respuesta JSON del modelo en columnas independientes.
    SELECT *
    FROM
      ML.GENERATE_TEXT(
        MODEL `mydataset.text_model`,
        (
          SELECT CONCAT('Summarize this text', body) AS prompt
          FROM mydataset.articles
        ),
        STRUCT(TRUE AS flatten_json_output));

    Ejemplo 2

    En el ejemplo siguiente se muestra una solicitud con estas características:

    • Usa una consulta para crear los datos de la petición concatenando cadenas que proporcionan prefijos de la petición con columnas de la tabla.
    • Devuelve una respuesta breve.
    • No analiza la respuesta JSON del modelo en columnas independientes.
    SELECT *
    FROM
      ML.GENERATE_TEXT(
        MODEL `mydataset.text_model`,
        (
          SELECT CONCAT(question, 'Text:', description, 'Category') AS prompt
          FROM mydataset.input_table
        ),
        STRUCT(
          100 AS max_output_tokens,
          FALSE AS flatten_json_output));

    Ejemplo 3

    En el ejemplo siguiente se muestra una solicitud con estas características:

    • Usa la columna prompt de la tabla prompts para la petición.
    • Analiza la respuesta JSON del modelo en columnas independientes.
    SELECT *
    FROM
      ML.GENERATE_TEXT(
        MODEL `mydataset.text_model`,
        TABLE mydataset.prompts,
        STRUCT(TRUE AS flatten_json_output));

    Ejemplo 4

    En el ejemplo siguiente se muestra una solicitud con estas características:

    • Usa la columna prompt de la tabla prompts para la petición.
    • Devuelve una respuesta breve.
    • Combina la respuesta JSON en columnas independientes.
    • Obtiene y devuelve datos web públicos para fundamentar las respuestas.
    • Filtra las respuestas no seguras mediante dos ajustes de seguridad.
    SELECT *
    FROM
      ML.GENERATE_TEXT(
        MODEL `mydataset.text_model`,
        TABLE mydataset.prompts,
        STRUCT(
          100 AS max_output_tokens, 0.5 AS top_p,
          TRUE AS flatten_json_output,
          TRUE AS ground_with_google_search,
          [STRUCT('HARM_CATEGORY_HATE_SPEECH' AS category,
            'BLOCK_LOW_AND_ABOVE' AS threshold),
          STRUCT('HARM_CATEGORY_DANGEROUS_CONTENT' AS category,
            'BLOCK_MEDIUM_AND_ABOVE' AS threshold)] AS safety_settings));

    Ejemplo 5

    En el ejemplo siguiente se muestra una solicitud con estas características:

    • Usa la columna prompt de la tabla prompts para la petición.
    • Devuelve una respuesta más larga.
    • Combina la respuesta JSON en columnas independientes.
    SELECT *
    FROM
      ML.GENERATE_TEXT(
        MODEL `mydataset.flash_2_model`,
        TABLE mydataset.prompts,
        STRUCT(
          0.4 AS temperature, 8192 AS max_output_tokens,
          TRUE AS flatten_json_output));

    Ejemplo 6

    En el ejemplo siguiente se muestra una solicitud con estas características:

    • Pide un resumen del texto de la columna body de la tabla articles.
    • Combina la respuesta JSON en columnas independientes.
    • Obtiene y devuelve datos web públicos para fundamentar las respuestas.
    • Filtra las respuestas no seguras mediante dos ajustes de seguridad.
    SELECT *
    FROM
      ML.GENERATE_TEXT(
        MODEL `mydataset.text_model`,
        (
          SELECT CONCAT('Summarize this text', body) AS prompt
          FROM mydataset.articles
        ),
        STRUCT(
          .1 AS TEMPERATURE,
          TRUE AS flatten_json_output, TRUE AS ground_with_google_search,
          [STRUCT('HARM_CATEGORY_HATE_SPEECH' AS category,
            'BLOCK_LOW_AND_ABOVE' AS threshold),
          STRUCT('HARM_CATEGORY_DANGEROUS_CONTENT' AS category,
            'BLOCK_MEDIUM_AND_ABOVE' AS threshold)] AS safety_settings));

Claude

SELECT *
FROM ML.GENERATE_TEXT(
  MODEL `PROJECT_ID.DATASET_ID.MODEL_NAME`,
  {TABLE PROJECT_ID.DATASET_ID.TABLE_NAME | (PROMPT_QUERY)},
  STRUCT(TOKENS AS max_output_tokens, TOP_K AS top_k,
  TOP_P AS top_p, FLATTEN_JSON AS flatten_json_output)
);

Haz los cambios siguientes:

  • PROJECT_ID: tu ID de proyecto.
  • DATASET_ID: el ID del conjunto de datos que contiene el modelo.
  • MODEL_NAME: el nombre del modelo.
  • TABLE_NAME: el nombre de la tabla que contiene la petición. Esta tabla debe tener una columna llamada prompt. También puedes usar un alias para usar una columna con otro nombre.
  • PROMPT_QUERY: una consulta que proporciona los datos de la petición. Esta consulta debe generar una columna llamada prompt.
  • TOKENS: un valor INT64 que define el número máximo de tokens que se pueden generar en la respuesta. Este valor debe estar en el intervalo [1,4096]. Especifica un valor inferior para las respuestas más cortas y un valor superior para las respuestas más largas. El valor predeterminado es 128.
  • TOP_K: un valor INT64 del intervalo [1,40] que determina el conjunto inicial de tokens que el modelo tiene en cuenta para la selección. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • TOP_P: un valor FLOAT64 en el intervalo [0.0,1.0] ayuda a determinar la probabilidad de los tokens seleccionados. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • FLATTEN_JSON: un valor BOOL que determina si se debe devolver el texto generado y los atributos de seguridad en columnas independientes. El valor predeterminado es FALSE.

Ejemplo 1

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Pide un resumen del texto de la columna body de la tabla articles.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT('Summarize this text', body) AS prompt
      FROM mydataset.articles
    ),
    STRUCT(TRUE AS flatten_json_output));

Ejemplo 2

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa una consulta para crear los datos de la petición concatenando cadenas que proporcionan prefijos de la petición con columnas de la tabla.
  • Devuelve una respuesta breve.
  • No analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT(question, 'Text:', description, 'Category') AS prompt
      FROM mydataset.input_table
    ),
    STRUCT(
      100 AS max_output_tokens,
      FALSE AS flatten_json_output));

Ejemplo 3

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa la columna prompt de la tabla prompts para la petición.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    TABLE mydataset.prompts,
    STRUCT(TRUE AS flatten_json_output));

Llama

SELECT *
FROM ML.GENERATE_TEXT(
  MODEL `PROJECT_ID.DATASET_ID.MODEL_NAME`,
  {TABLE PROJECT_ID.DATASET_ID.TABLE_NAME | (PROMPT_QUERY)},
  STRUCT(TOKENS AS max_output_tokens, TEMPERATURE AS temperature,
  TOP_P AS top_p, FLATTEN_JSON AS flatten_json_output,
  STOP_SEQUENCES AS stop_sequences)
);

Haz los cambios siguientes:

  • PROJECT_ID: tu ID de proyecto.
  • DATASET_ID: el ID del conjunto de datos que contiene el modelo.
  • MODEL_NAME: el nombre del modelo.
  • TABLE_NAME: el nombre de la tabla que contiene la petición. Esta tabla debe tener una columna llamada prompt. También puedes usar un alias para usar una columna con otro nombre.
  • PROMPT_QUERY: una consulta que proporciona los datos de la petición. Esta consulta debe generar una columna llamada prompt.
  • TOKENS: un valor INT64 que define el número máximo de tokens que se pueden generar en la respuesta. Este valor debe estar en el intervalo [1,4096]. Especifica un valor inferior para las respuestas más cortas y un valor superior para las respuestas más largas. El valor predeterminado es 128.
  • TEMPERATURE: un valor FLOAT64 en el intervalo [0.0,1.0] que controla el grado de aleatoriedad en la selección de tokens. El valor predeterminado es 0.

    Los valores bajos de temperature son adecuados para las peticiones que requieren una respuesta más determinista y menos abierta o creativa, mientras que los valores altos de temperature pueden dar lugar a resultados más diversos o creativos. El valor de 0 para temperature es determinista, lo que significa que siempre se selecciona la respuesta con la probabilidad más alta.

  • TOP_P: un valor FLOAT64 en el intervalo [0.0,1.0] ayuda a determinar la probabilidad de los tokens seleccionados. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. El valor predeterminado es 0.95.
  • FLATTEN_JSON: un valor BOOL que determina si se debe devolver el texto generado y los atributos de seguridad en columnas independientes. El valor predeterminado es FALSE.
  • STOP_SEQUENCES: un valor ARRAY<STRING> que elimina las cadenas especificadas si se incluyen en las respuestas del modelo. Las cadenas deben coincidir exactamente, incluidas las mayúsculas y las minúsculas. El valor predeterminado es un array vacío.

Ejemplo 1

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Pide un resumen del texto de la columna body de la tabla articles.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT('Summarize this text', body) AS prompt
      FROM mydataset.articles
    ),
    STRUCT(TRUE AS flatten_json_output));

Ejemplo 2

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa una consulta para crear los datos de la petición concatenando cadenas que proporcionan prefijos de la petición con columnas de la tabla.
  • Devuelve una respuesta breve.
  • No analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT(question, 'Text:', description, 'Category') AS prompt
      FROM mydataset.input_table
    ),
    STRUCT(
      100 AS max_output_tokens,
      FALSE AS flatten_json_output));

Ejemplo 3

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa la columna prompt de la tabla prompts para la petición.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    TABLE mydataset.prompts,
    STRUCT(TRUE AS flatten_json_output));

Mistral AI

SELECT *
FROM ML.GENERATE_TEXT(
  MODEL `PROJECT_ID.DATASET_ID.MODEL_NAME`,
  {TABLE PROJECT_ID.DATASET_ID.TABLE_NAME | (PROMPT_QUERY)},
  STRUCT(TOKENS AS max_output_tokens, TEMPERATURE AS temperature,
  TOP_P AS top_p, FLATTEN_JSON AS flatten_json_output,
  STOP_SEQUENCES AS stop_sequences)
);

Haz los cambios siguientes:

  • PROJECT_ID: tu ID de proyecto.
  • DATASET_ID: el ID del conjunto de datos que contiene el modelo.
  • MODEL_NAME: el nombre del modelo.
  • TABLE_NAME: el nombre de la tabla que contiene la petición. Esta tabla debe tener una columna llamada prompt. También puedes usar un alias para usar una columna con otro nombre.
  • PROMPT_QUERY: una consulta que proporciona los datos de la petición. Esta consulta debe generar una columna llamada prompt.
  • TOKENS: un valor INT64 que define el número máximo de tokens que se pueden generar en la respuesta. Este valor debe estar en el intervalo [1,4096]. Especifica un valor inferior para las respuestas más cortas y un valor superior para las respuestas más largas. El valor predeterminado es 128.
  • TEMPERATURE: un valor FLOAT64 en el intervalo [0.0,1.0] que controla el grado de aleatoriedad en la selección de tokens. El valor predeterminado es 0.

    Los valores bajos de temperature son adecuados para las peticiones que requieren una respuesta más determinista y menos abierta o creativa, mientras que los valores altos de temperature pueden dar lugar a resultados más diversos o creativos. El valor de 0 para temperature es determinista, lo que significa que siempre se selecciona la respuesta con la probabilidad más alta.

  • TOP_P: un valor FLOAT64 en el intervalo [0.0,1.0] ayuda a determinar la probabilidad de los tokens seleccionados. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. El valor predeterminado es 0.95.
  • FLATTEN_JSON: un valor BOOL que determina si se debe devolver el texto generado y los atributos de seguridad en columnas independientes. El valor predeterminado es FALSE.
  • STOP_SEQUENCES: un valor ARRAY<STRING> que elimina las cadenas especificadas si se incluyen en las respuestas del modelo. Las cadenas deben coincidir exactamente, incluidas las mayúsculas y las minúsculas. El valor predeterminado es un array vacío.

Ejemplo 1

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Pide un resumen del texto de la columna body de la tabla articles.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT('Summarize this text', body) AS prompt
      FROM mydataset.articles
    ),
    STRUCT(TRUE AS flatten_json_output));

Ejemplo 2

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa una consulta para crear los datos de la petición concatenando cadenas que proporcionan prefijos de la petición con columnas de la tabla.
  • Devuelve una respuesta breve.
  • No analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT(question, 'Text:', description, 'Category') AS prompt
      FROM mydataset.input_table
    ),
    STRUCT(
      100 AS max_output_tokens,
      FALSE AS flatten_json_output));

Ejemplo 3

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa la columna prompt de la tabla prompts para la petición.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    TABLE mydataset.prompts,
    STRUCT(TRUE AS flatten_json_output));

Modelos abiertos

.

SELECT *
FROM ML.GENERATE_TEXT(
  MODEL `PROJECT_ID.DATASET_ID.MODEL_NAME`,
  {TABLE PROJECT_ID.DATASET_ID.TABLE_NAME | (PROMPT_QUERY)},
  STRUCT(TOKENS AS max_output_tokens,
   TEMPERATURE AS temperature, TOP_K AS top_k,
  TOP_P AS top_p, FLATTEN_JSON AS flatten_json_output)
);

Haz los cambios siguientes:

  • PROJECT_ID: tu ID de proyecto.
  • DATASET_ID: el ID del conjunto de datos que contiene el modelo.
  • MODEL_NAME: el nombre del modelo.
  • TABLE_NAME: el nombre de la tabla que contiene la petición. Esta tabla debe tener una columna llamada prompt. También puedes usar un alias para usar una columna con otro nombre.
  • PROMPT_QUERY: una consulta que proporciona los datos de la petición. Esta consulta debe generar una columna llamada prompt.
  • TOKENS: un valor INT64 que define el número máximo de tokens que se pueden generar en la respuesta. Este valor debe estar en el intervalo [1,4096]. Especifica un valor inferior para las respuestas más cortas y un valor superior para las respuestas más largas. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • TEMPERATURE: un valor FLOAT64 en el intervalo [0.0,1.0] que controla el grado de aleatoriedad en la selección de tokens. Si no especifica ningún valor, el modelo determinará un valor adecuado.

    Los valores bajos de temperature son adecuados para las peticiones que requieren una respuesta más determinista y menos abierta o creativa, mientras que los valores altos de temperature pueden dar lugar a resultados más diversos o creativos. El valor de 0 para temperature es determinista, lo que significa que siempre se selecciona la respuesta con la probabilidad más alta.

  • TOP_K: un valor INT64 del intervalo [1,40] que determina el conjunto inicial de tokens que el modelo tiene en cuenta para la selección. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • TOP_P: un valor FLOAT64 en el intervalo [0.0,1.0] ayuda a determinar la probabilidad de los tokens seleccionados. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • FLATTEN_JSON: un valor BOOL que determina si se debe devolver el texto generado y los atributos de seguridad en columnas independientes. El valor predeterminado es FALSE.

Ejemplo 1

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Pide un resumen del texto de la columna body de la tabla articles.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT('Summarize this text', body) AS prompt
      FROM mydataset.articles
    ),
    STRUCT(TRUE AS flatten_json_output));

Ejemplo 2

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa una consulta para crear los datos de la petición concatenando cadenas que proporcionan prefijos de la petición con columnas de la tabla.
  • Devuelve una respuesta breve.
  • No analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    (
      SELECT CONCAT(question, 'Text:', description, 'Category') AS prompt
      FROM mydataset.input_table
    ),
    STRUCT(
      100 AS max_output_tokens,
      FALSE AS flatten_json_output));

Ejemplo 3

En el ejemplo siguiente se muestra una solicitud con estas características:

  • Usa la columna prompt de la tabla prompts para la petición.
  • Analiza la respuesta JSON del modelo en columnas independientes.
SELECT *
FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.text_model`,
    TABLE mydataset.prompts,
    STRUCT(TRUE AS flatten_json_output));

Generar texto a partir de datos de una tabla de objetos

Genera texto usando la función ML.GENERATE_TEXT con un modelo de Gemini para analizar datos sin estructurar de una tabla de objetos. Los datos de la petición se proporcionan en el parámetro prompt.

SELECT *
FROM ML.GENERATE_TEXT(
  MODEL `PROJECT_ID.DATASET_ID.MODEL_NAME`,
  TABLE PROJECT_ID.DATASET_ID.TABLE_NAME,
  STRUCT(PROMPT AS prompt, TOKENS AS max_output_tokens,
  TEMPERATURE AS temperature, TOP_P AS top_p,
  FLATTEN_JSON AS flatten_json_output,
  STOP_SEQUENCES AS stop_sequences,
  SAFETY_SETTINGS AS safety_settings)
);

Haz los cambios siguientes:

  • PROJECT_ID: tu ID de proyecto.
  • DATASET_ID: el ID del conjunto de datos que contiene el modelo.
  • MODEL_NAME: el nombre del modelo. Debe ser un modelo de Gemini.
  • TABLE_NAME: nombre de la tabla de objetos que contiene el contenido que se va a analizar. Para obtener más información sobre los tipos de contenido que puedes analizar, consulta Entrada.

    El segmento de Cloud Storage que usa la tabla de objetos debe estar en el mismo proyecto en el que has creado el modelo y en el que llamas a la función ML.GENERATE_TEXT. Si quieres llamar a la función ML.GENERATE_TEXT en un proyecto distinto del que contiene el segmento de Cloud Storage que usa la tabla de objetos, debes asignar el rol Administrador de almacenamiento a nivel de segmento a la cuenta de servicio service-A@gcp-sa-aiplatform.iam.gserviceaccount.com.

  • PROMPT: la petición que se va a usar para analizar el contenido.
  • TOKENS: un valor INT64 que define el número máximo de tokens que se pueden generar en la respuesta. Este valor debe estar en el intervalo [1,4096]. Especifica un valor inferior para las respuestas más cortas y un valor superior para las respuestas más largas. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • TEMPERATURE: un valor FLOAT64 en el intervalo [0.0,1.0] que controla el grado de aleatoriedad en la selección de tokens. Si no especifica ningún valor, el modelo determinará un valor adecuado.

    Los valores bajos de temperature son adecuados para las peticiones que requieren una respuesta más determinista y menos abierta o creativa, mientras que los valores altos de temperature pueden dar lugar a resultados más diversos o creativos. El valor de 0 para temperature es determinista, lo que significa que siempre se selecciona la respuesta con la probabilidad más alta.

  • TOP_K: un valor INT64 del intervalo [1,40] que determina el conjunto inicial de tokens que el modelo tiene en cuenta para la selección. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • TOP_P: un valor FLOAT64 en el intervalo [0.0,1.0] ayuda a determinar la probabilidad de los tokens seleccionados. Especifica un valor más bajo para obtener respuestas menos aleatorias y un valor más alto para obtener respuestas más aleatorias. Si no especifica ningún valor, el modelo determinará un valor adecuado.
  • FLATTEN_JSON: un valor BOOL que determina si se debe devolver el texto generado y los atributos de seguridad en columnas independientes. El valor predeterminado es FALSE.

Ejemplos

En este ejemplo, se traduce y transcribe contenido de audio de una tabla de objetos llamada feedback:

SELECT * FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.audio_model`,
    TABLE `mydataset.feedback`,
      STRUCT('What is the content of this audio clip, translated into Spanish?' AS PROMPT,
      TRUE AS FLATTEN_JSON_OUTPUT));

En este ejemplo se clasifica el contenido PDF de una tabla de objetos llamada invoices:

SELECT * FROM
  ML.GENERATE_TEXT(
    MODEL `mydataset.classify_model`,
    TABLE `mydataset.invoices`,
      STRUCT('Classify this document based on the invoice total, using the following categories: 0 to 100, 101 to 200, greater than 200' AS PROMPT,
      TRUE AS FLATTEN_JSON_OUTPUT));