Generar inserciones de vectores con la gestión de endpoints de modelos

Selecciona una versión de la documentación:

En esta página se describe cómo generar embeddings mediante endpoints de modelos registrados.

Antes de empezar

Asegúrate de haber registrado tu endpoint de modelo en la gestión de endpoints de modelos. Para obtener más información, consulta Registrar un endpoint de modelo con la gestión de endpoints de modelos.

Generar inserciones

Usa la función SQL google_ml.embedding() para llamar al endpoint del modelo registrado con el tipo de modelo de inserción de texto para generar inserciones.

Para llamar al modelo y generar las inserciones, usa la siguiente consulta de SQL:

SELECT
  google_ml.embedding(
    model_id => 'MODEL_ID',
    content => 'CONTENT');

Haz los cambios siguientes:

  • MODEL_ID: el ID del modelo que definiste al registrar el endpoint del modelo.
  • CONTENT: el texto que se va a traducir a una incrustación de vector.

Ejemplos

En esta sección se incluyen algunos ejemplos de cómo generar inserciones mediante un endpoint de modelo registrado.

Modelos de inserción de texto con asistencia integrada

Para generar embeddings de un endpoint de modelo text-embedding-005 registrado, ejecuta la siguiente instrucción:

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-005',
        content => 'AlloyDB is a managed, cloud-hosted SQL database service');

Para generar inserciones de un endpoint de modelo text-embedding-ada-002 registrado de OpenAI, ejecuta la siguiente instrucción:

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-ada-002',
        content => 'e-mail spam');

Para generar inserciones de los endpoints de modelos text-embedding-3-small o text-embedding-3-large registrados de OpenAI, ejecuta la siguiente instrucción:

  SELECT
    google_ml.embedding(
      model_id => 'text-embedding-3-small',
      content => 'Vector embeddings in AI');