Compartir a través de

Adición de un recurso de índice de búsqueda vectorial a una aplicación de Databricks

Agregue índices de búsqueda vectorial como recursos de Databricks Apps para habilitar la búsqueda semántica y la recuperación basada en similitud en las aplicaciones. Los índices de búsqueda vectorial almacenan y consultan incrustaciones de vectores de alta dimensión, lo que potencia los casos de uso, como la generación aumentada por recuperación (RAG), la búsqueda semántica y los sistemas de recomendaciones.

Requisitos de privilegios

Para acceder a un índice de búsqueda vectorial, la entidad de servicio de la aplicación debe tener el USE CATALOG privilegio en el catálogo primario, el USE SCHEMA privilegio en el esquema primario y el SELECT privilegio en el índice. Al agregar el recurso de índice, Azure Databricks concede automáticamente estos privilegios a la entidad de servicio de la aplicación.

Para que esta concesión automática se realice correctamente, uno de los siguientes valores debe ser verdadero para cada privilegio.

  • Para USE CATALOG: todos los usuarios de la cuenta tienen el USE CATALOG privilegio en el catálogo, o todos tienen el MANAGE privilegio en el catálogo.
  • Para USE SCHEMA: o todos los usuarios de la cuenta tienen el USE SCHEMA privilegio en el esquema, o usted tiene el MANAGE privilegio en el esquema.
  • Para SELECT: Todos los usuarios de la cuenta tienen el SELECT privilegio en el índice o usted tiene el MANAGE privilegio en el índice.

Para obtener más información sobre cómo consultar los índices de búsqueda vectorial con estos permisos, vea Cómo consultar un índice de búsqueda vectorial.

Consulte Privilegios de Unity Catalog y objetos protegibles.

Adición de un recurso de índice de búsqueda vectorial

Antes de agregar un índice de búsqueda vectorial como recurso, revise los requisitos previos del recurso de la aplicación.

  1. Al crear o editar una aplicación, vaya al paso Configurar .
  2. En la sección Recursos de la aplicación, haga clic en + Agregar recurso.
  3. Seleccione Índice de búsqueda vectorial como tipo de recurso.
  4. Elija un índice de búsqueda vectorial de los índices disponibles en el área de trabajo. El índice ya debe existir en el catálogo de Unity.
  5. Seleccione el nivel de permiso de la aplicación:
    • Puede seleccionar: Concede permiso a la aplicación para consultar el índice de búsqueda vectorial para búsquedas de similitud. Corresponde al SELECT privilegio.
  6. (Opcional) Especifique una clave de recurso personalizada, que es la forma en que hace referencia al índice en la configuración de la aplicación. La clave predeterminada es vector-search-index.

Nota:

Los índices de búsqueda vectorial son tablas de catálogo de Unity con tipo TABLE_ONLINE_VECTOR_INDEX_REPLICA o TABLE_ONLINE_VECTOR_INDEX_DIRECT. Al seleccionar un índice de búsqueda vectorial, se selecciona una tabla especialmente configurada que admite operaciones de búsqueda semántica.

Variables de entorno

Al implementar una aplicación con un recurso de índice de búsqueda vectorial, Azure Databricks expone el nombre completo de tres niveles a través de variables de entorno a las que puede hacer referencia mediante el valueFrom campo .

Configuración de ejemplo:

env:
  - name: VECTOR_SEARCH_INDEX
    valueFrom: vector-search-index # Use your custom resource key if different

Utilizando el índice en su aplicación:

import os
from databricks.sdk import WorkspaceClient

# Access the vector search index name
index_name = os.getenv("VECTOR_SEARCH_INDEX")

# Initialize workspace client
w = WorkspaceClient()

# Query the vector search index
results = w.vector_search_indexes.query_index(
    index_name=index_name,
    query_text="What is machine learning?",
    num_results=10
)

# Process results
for result in results.manifest.columns:
    print(f"Result: {result}")

Para obtener más información, consulte Uso de variables de entorno para acceder a los recursos.

Eliminación de un recurso de índice de búsqueda vectorial

Al quitar un recurso de índice de búsqueda vectorial de una aplicación, la entidad de servicio de la aplicación pierde el acceso al índice. El propio índice permanece sin cambios y sigue estando disponible para otros usuarios y aplicaciones que tienen los permisos adecuados.

procedimientos recomendados

Tenga en cuenta lo siguiente al trabajar con recursos de índice de búsqueda vectorial:

  • Asegúrese de que la entidad de servicio de la aplicación tiene acceso a orígenes de datos subyacentes si el índice hace referencia a otras tablas.
  • Supervise el rendimiento de las consultas y ajuste la configuración del índice o la inserción de modelos si los tiempos de respuesta se degradan.
  • Considere las programaciones de actualización de índices para mantener las incrustaciones sincronizadas con los datos de origen.
  • Utilice las métricas de similitud adecuadas (coseno, euclidiana, producto punto) en función del modelo de incrustación.
  • Last updated on