docs: fix Databricks Vector Search demo notebook (#25504)

**Description:** This PR fixes an issue in the demo notebook of
Databricks Vector Search in "Work with Delta Sync Index" section.

**Issue:** N/A

**Dependencies:** N/A

---------

Co-authored-by: Chengzu Ou <chengzu.ou@databrick.com>
Co-authored-by: Erick Friis <erick@langchain.dev>
This commit is contained in:
Chengzu Ou 2024-08-16 13:24:30 -07:00 committed by GitHub
parent a2e90a5a43
commit c1bd4e05bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -122,7 +122,7 @@
"outputs": [],
"source": [
"vector_search_endpoint_name = \"vector_search_demo_endpoint\"\n",
"index_name = \"ml.llm.demo_index\"\n",
"index_name = \"vector_search_demo.vector_search.state_of_the_union_index\"\n",
"\n",
"index = vsc.create_direct_access_index(\n",
" endpoint_name=vector_search_endpoint_name,\n",
@ -206,7 +206,16 @@
"metadata": {},
"outputs": [],
"source": [
"dvs_delta_sync = DatabricksVectorSearch(\"catalog_name.schema_name.delta_sync_index\")\n",
"delta_sync_index = vsc.create_delta_sync_index(\n",
" endpoint_name=vector_search_endpoint_name,\n",
" source_table_name=\"vector_search_demo.vector_search.state_of_the_union\",\n",
" index_name=\"vector_search_demo.vector_search.state_of_the_union_index\",\n",
" pipeline_type=\"TRIGGERED\",\n",
" primary_key=\"id\",\n",
" embedding_source_column=\"text\",\n",
" embedding_model_endpoint_name=\"e5-small-v2\",\n",
")\n",
"dvs_delta_sync = DatabricksVectorSearch(delta_sync_index)\n",
"dvs_delta_sync.similarity_search(query)"
]
}