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 una vista previa que te permite experimentar con el registro de un endpoint de modelo de IA y la invocación de predicciones con la gestión de endpoints de modelos. Para usar modelos de IA en entornos de producción, consulta Crea aplicaciones de IA generativa con AlloyDB AI y Trabaja con incrustaciones de vectores.

Una vez que se hayan añadido y registrado los endpoints del modelo en la gestión de endpoints del modelo, podrás hacer referencia a ellos mediante el ID del modelo para generar inserciones.

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 textembedding-gecko@002 registrado, ejecuta la siguiente instrucción:

    SELECT
      google_ml.embedding(
        model_id => 'textembedding-gecko@002',
        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');

Otros modelos de inserciones de texto

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');