Genera embeddings de vectores con la administración de extremos de modelos

En esta página, se describe cómo generar incorporaciones con extremos de modelos registrados.

Antes de comenzar

Asegúrate de haber registrado el extremo de tu modelo con la administración de extremos de modelos. Para obtener más información, consulta Cómo registrar un extremo de modelo con la administración de extremos de modelos.

Genera embeddings

Usa la función SQL google_ml.embedding() para llamar al extremo del modelo registrado con el tipo de modelo de incorporación de texto para generar incorporaciones.

Para llamar al modelo y generar incorporaciones, usa la siguiente consulta en SQL:

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

Reemplaza lo siguiente:

  • MODEL_ID: Es el ID del modelo que definiste cuando registraste el extremo del modelo.
  • CONTENT: es el texto que se traducirá en una embedding de vector.

Ejemplos

En esta sección, se incluyen algunos ejemplos para generar incorporaciones con el extremo del modelo registrado.

Modelos de incorporación de texto con compatibilidad integrada

Para generar incorporaciones para un extremo de modelo text-embedding-005 registrado, ejecuta la siguiente sentencia:

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

Para generar incorporaciones para un extremo de modelo text-embedding-ada-002 registrado por OpenAI, ejecuta la siguiente sentencia:

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

Para generar incorporaciones para los extremos de modelos text-embedding-3-small o text-embedding-3-large registrados por OpenAI, ejecuta la siguiente sentencia:

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