From 49f51f2e6a99516c98b33dbe8d5241314a70e448 Mon Sep 17 00:00:00 2001
From: Xu Wu <wuxu92@outlook.com>
Date: Mon, 17 Feb 2025 05:26:38 +0800
Subject: [PATCH] `azureaisearch`: prefer using hybrid_top_k over
similarity_top_k for hybrid search (#17612)
---
.../llama_index/vector_stores/azureaisearch/base.py | 3 ++-
.../llama-index-vector-stores-azureaisearch/pyproject.toml | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/llama_index/vector_stores/azureaisearch/base.py b/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/llama_index/vector_stores/azureaisearch/base.py
index 16ac1065cc..3e877bf9f6 100644
--- a/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/llama_index/vector_stores/azureaisearch/base.py
+++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/llama_index/vector_stores/azureaisearch/base.py
@@ -1567,7 +1567,8 @@ class AzureQueryResultSearchDefault(AzureQueryResultSearchBase):
vectorized_query = VectorizedQuery(
vector=self._query.query_embedding,
- k_nearest_neighbors=self._query.similarity_top_k,
+ k_nearest_neighbors=self._query.hybrid_top_k
+ or self._query.similarity_top_k,
fields=self._field_mapping["embedding"],
)
vector_queries = [vectorized_query]
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/pyproject.toml
index 87d6b03829..d30a06b628 100644
--- a/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/pyproject.toml
+++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/pyproject.toml
@@ -28,7 +28,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-vector-stores-azureaisearch"
readme = "README.md"
-version = "0.3.4"
+version = "0.3.5"
[tool.poetry.dependencies]
python = ">=3.9,<4.0"
--
GitLab