Gerar embeddings de vetor com o gerenciamento de endpoints de modelo

Nesta página, descrevemos como registrar um endpoint de modelo de IA e gerar incorporações de vetores com o gerenciamento de endpoints de modelo. Para usar modelos de IA em ambientes de produção, consulte Gerar e gerenciar embeddings de vetor.

Depois que os endpoints de modelo são adicionados e registrados no gerenciamento de endpoints de modelo, é possível referenciá-los usando o ID do modelo para gerar incorporações.

Antes de começar

Faça o seguinte:

Gerar embeddings

Use a função SQL mysql.ml_embedding() para chamar o endpoint do modelo registrado com o tipo de modelo de embedding de texto e gerar embeddings.

Para chamar o modelo e gerar embeddings, use a seguinte consulta SQL:

SELECT
  mysql.ml_embedding(
    'MODEL_ID',
    'CONTENT');

Substitua:

  • MODEL_ID: o ID do modelo definido ao registrar o endpoint do modelo
  • CONTENT: o texto a ser traduzido em um embedding vetorial.

Exemplos

Alguns exemplos de geração de embeddings usando endpoints de modelos registrados estão listados nesta seção.

Modelos de embedding de texto com suporte integrado

Os modelos de embedding da Vertex AI são compatíveis por padrão. Não é necessário registrar esses dispositivos. Para gerar embeddings para um endpoint de modelo gemini-embedding-001 registrado, execute a seguinte instrução:

    SELECT
      mysql.ml_embedding(
        'gemini-embedding-001',
        'Cloud SQL is a managed, cloud-hosted SQL database service');