Cómo usar la asistencia de vectores

Selecciona una versión de la documentación:

Usa el asistente de vectores para implementar y administrar una carga de trabajo de vectores en tus instancias de AlloyDB Omni.

Para obtener más información sobre qué es la asistencia de vectores y cómo funciona, consulta Descripción general de la asistencia de vectores.

Antes de comenzar

  • Confirma que tu instancia de AlloyDB Omni ejecute la versión 18.1.0 o una posterior. Para obtener más información sobre la actualización, consulta Administra AlloyDB Omni.

  • Habilita la extensión de asistencia de vectores con el siguiente comando en la base de datos que quieras usar:

    CREATE EXTENSION vector_assist CASCADE;

    Esto genera el esquema vector_assist, que usa el asistente de vectores.

Usa el asistente de vectores para implementar una carga de trabajo de vectores

Para usar el asistente de vectores con el objetivo de implementar y administrar una carga de trabajo de vectores, debes hacer lo siguiente:

Define la especificación de tu vector

Definir la especificación de vectores, o especificación de vectores, es el primer paso para usar la asistencia de vectores. Según el tipo de carga de trabajo de vectores, los campos que usas para definir tu especificación de vectores pueden ser diferentes.

Por ejemplo, si deseas habilitar la búsqueda semántica en una columna específica de una tabla, ejecuta la función vector_assist.define_spec para definir tu especificación de vectores:

SELECT
  recommendation_id,
  REGEXP_REPLACE(query, 'SET hnsw.ef_search TO \d+', 'SET hnsw.ef_search TO') AS query_1,
  recommendation,
  vector_spec_id,
  table_name,
  applied,
  modified
FROM vector_assist.define_spec(
  table_name => 'TABLE_NAME',
  vector_column_name => 'VECTOR_COLUMN_NAME',
  target_recall => TARGET_RECALL,
  tune_vector_index => TUNE_INDEX
);

Reemplaza los siguientes parámetros:

  • TABLE_NAME: Es el nombre de la tabla que deseas usar en tu carga de trabajo de vectores.

  • (Opcional) VECTOR_COLUMN_NAME: es la columna en la que deseas realizar una búsqueda semántica. Si solo tienes una columna con vectores en tu tabla, vector_assist puede configurarla automáticamente. Si hay más de una columna con vectores, debes especificar qué columna usar.

  • TARGET_RECALL (opcional): Es la recuperación objetivo que deseas que cumpla la asistencia de vectores. Si se especifica, vector_assist experimenta con diferentes valores de ef_search y compara la recuperación en búsquedas exactas de vecinos más cercanos. vector_assist establece ef_search en el valor que proporciona la recuperación objetivo más cercana. El valor predeterminado es 0.95.

  • TUNE_INDEX (opcional): Es un valor booleano que especifica si el asistente de vectores ajusta los índices de vectores en tu carga de trabajo. El valor predeterminado es false, lo que significa que solo se ajustan los parámetros relacionados con la búsqueda (ef_search); los parámetros de tiempo de compilación (m, ef_construction) no se ajustan.

Para ver una lista completa de todos los parámetros disponibles para la especificación de vectores, consulta Referencias de funciones de asistencia de vectores.

Después de ejecutar la consulta para crear tu especificación de vectores, el asistente de vectores genera automáticamente pasos, llamados recomendaciones, que debes ejecutar para implementar tu carga de trabajo de vectores.

Consulta las recomendaciones de Vector Assist

Para ver las recomendaciones que genera el asistente de vectores con tu especificación de vectores, ejecuta la función vector_assist.get_recommendations:

SELECT vector_assist.get_recommendations(
  spec_id => 'SPEC_ID'
);

Reemplaza SPEC_ID por el ID de la especificación de vectores para la que deseas ver las recomendaciones. Para obtener tu lista de especificaciones de vectores disponibles, consulta Enumera tus especificaciones de vectores.

vector_assist.get_recommendations devuelve una tabla llamada vector_assist.RECOMMENDATIONS que contiene todas las recomendaciones. Cada recomendación se almacena en una tabla con el spec_id asociado. Por lo general, cada recomendación contiene la siguiente información:

  • Consulta en SQL que necesitas ejecutar
  • Explicación detallada de la recomendación
  • Otra información relevante que explica la recomendación

Aplica las recomendaciones de Vector Assist

Puedes aplicar las recomendaciones generadas de forma individual o en conjunto.

Para aplicar una recomendación específica, ejecuta la función vector_assist.apply_recommendation:

SELECT vector_assist.apply_recommendation(
  recommendation_id => 'RECOMMENDATION_ID'
);

Reemplaza RECOMMENDATION_ID por el ID de la recomendación de Vector Assist que deseas aplicar de la tabla vector_assist.RECOMMENDATIONS.

Para aplicar todas las recomendaciones juntas, ejecuta la función vector_assist.apply_spec con el parámetro spec_id o table_name:

SELECT vector_assist.apply_spec(
  spec_id => 'SPEC_ID'
);

Reemplaza SPEC_ID por el ID de la especificación de vectores que quieres usar.

De manera opcional, también puedes ingresar los parámetros schema_name o column_name. Para obtener información sobre todos los parámetros de consulta disponibles, consulta Referencias de funciones de asistencia de vectores.

Una vez que apliques las recomendaciones generadas por el asistente de vectores, el índice vectorial estará listo para usarse.

Genera búsquedas

Puedes usar el asistente de vectores para crear consultas de búsqueda optimizadas para tus cargas de trabajo de vectores implementadas con la especificación de vectores de la carga de trabajo y el índice de vectores generado. Para generar una consulta de búsqueda optimizada, ejecuta la función vector_assist.generate_query:

SELECT vector_assist.generate_query(
  spec_id => 'SPEC_ID',
  search_vector => 'SEARCH_VECTOR',
  top_k => TOP_K,
  target_recall => TARGET_RECALL,
  filter_expression => 'FILTER_EXPRESSION'
);

Reemplaza los siguientes parámetros:

  • SPEC_ID: Es el ID de la especificación que deseas usar.

  • SEARCH_VECTOR: Es el vector de tu consulta de búsqueda. Por ejemplo, [1, 2, 3]

  • TOP_K (opcional): Es la cantidad de vecinos más cercanos que se mostrarán. Si no se especifica, el valor predeterminado es 10.

  • TARGET_RECALL (opcional): Es la recuperación objetivo que deseas que cumpla la asistencia de vectores. Si se especifica, vector_assist experimenta con diferentes valores de ef_search y compara la recuperación en búsquedas exactas de vecinos más cercanos. vector_assist establece ef_search en el valor que proporciona la recuperación objetivo más cercana.

  • FILTER_EXPRESSION (opcional): cualquier filtro para la búsqueda. Por ejemplo, filtrar según otras columnas.

Para ver una lista completa de todos los parámetros disponibles para generar una consulta de búsqueda, consulta las referencias de la función de asistencia de vectores.

El resultado de esta función devuelve texto que contiene una consulta en SQL. Puedes ejecutar o guardar esta consulta en SQL según sea necesario.

Enumera tus especificaciones de vectores

Si necesitas enumerar una o todas tus especificaciones de vectores existentes, ejecuta la función vector_assist.list_specs:

SELECT vector_assist.list_specs(
  spec_id => 'SPEC_ID',
  table_name => 'TABLE_NAME'
);

Reemplaza los siguientes parámetros:

  • SPEC_ID (opcional): Es el ID de la especificación de vectores que deseas usar.

  • TABLE_NAME: Es el nombre de la tabla que usaste para definir tu especificación de vectores.

¿Qué sigue?