Skip to content

Commit 767de07

Browse files
authored
Allow querying AzureAISearch without non-null metadata field (run-llama#13531)
1 parent 8ed61b7 commit 767de07

File tree

2 files changed

+9
-5
lines changed
  • llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch

2 files changed

+9
-5
lines changed

llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/llama_index/vector_stores/azureaisearch/base.py

+8-4
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,8 @@ class IndexManagement(int, enum.Enum):
5151

5252

5353
class AzureAISearchVectorStore(BasePydanticVectorStore):
54-
"""Azure AI Search vector store.
54+
"""
55+
Azure AI Search vector store.
5556
5657
Examples:
5758
`pip install llama-index-vector-stores-azureaisearch`
@@ -495,7 +496,8 @@ def add(
495496
nodes: List[BaseNode],
496497
**add_kwargs: Any,
497498
) -> List[str]:
498-
"""Add nodes to index associated with the configured search client.
499+
"""
500+
Add nodes to index associated with the configured search client.
499501
500502
Args:
501503
nodes: List[BaseNode]: nodes with embeddings
@@ -678,7 +680,8 @@ def _create_query_result(
678680
score_result = []
679681
for result in results:
680682
node_id = result[self._field_mapping["id"]]
681-
metadata = json.loads(result[self._field_mapping["metadata"]])
683+
metadata_str = result[self._field_mapping["metadata"]]
684+
metadata = json.loads(metadata_str) if metadata_str else {}
682685
score = result["@search.score"]
683686
chunk = result[self._field_mapping["chunk"]]
684687

@@ -795,7 +798,8 @@ def _create_query_result(
795798
score_result = []
796799
for result in results:
797800
node_id = result[self._field_mapping["id"]]
798-
metadata = json.loads(result[self._field_mapping["metadata"]])
801+
metadata_str = result[self._field_mapping["metadata"]]
802+
metadata = json.loads(metadata_str) if metadata_str else {}
799803
# use reranker_score instead of score
800804
score = result["@search.reranker_score"]
801805
chunk = result[self._field_mapping["chunk"]]

llama-index-integrations/vector_stores/llama-index-vector-stores-azureaisearch/pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ exclude = ["**/BUILD"]
2828
license = "MIT"
2929
name = "llama-index-vector-stores-azureaisearch"
3030
readme = "README.md"
31-
version = "0.1.6"
31+
version = "0.1.7"
3232

3333
[tool.poetry.dependencies]
3434
python = ">=3.8.1,<4.0"

0 commit comments

Comments
 (0)