Is there a batch query option for similar vectors available in Mosaic AI Vector Search?

Cummings012
New Contributor II

Hi all,

I have been exploring databrick's vector search offering, and was wondering if it had the ability to run a batch query?

I would like to be able to do something like the following where both embedding_1 and embedding_2 are separate vectors that I would like 10 results for each of them. 

Cummings012_0-1729087701912.png

I've looked through the docs for both the package and api and didn't find anything that supported this kind of batch query, but wanted to make sure I wasn't missing anything.

https://docs.databricks.com/api/workspace/vectorsearchindexes/queryindex 

https://api-docs.databricks.com/python/vector-search/databricks.vector_search.html#databricks.vector... 

Thanks!