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

[Cummings012]

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. 

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!

[filipniziol]

Hi @Cummings012 ,

As per documentation it is not possible, as the API supports single queries.

The workaround is to process these queries in parallel to improve throughput.