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 1f0638b1e86cbc38221531519b65daf387f44844..0809d8aed4cac5c2614b6e299dbfd6d356ee144c 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 @@ -51,7 +51,8 @@ class IndexManagement(int, enum.Enum): class AzureAISearchVectorStore(BasePydanticVectorStore): - """Azure AI Search vector store. + """ + Azure AI Search vector store. Examples: `pip install llama-index-vector-stores-azureaisearch` @@ -495,7 +496,8 @@ class AzureAISearchVectorStore(BasePydanticVectorStore): nodes: List[BaseNode], **add_kwargs: Any, ) -> List[str]: - """Add nodes to index associated with the configured search client. + """ + Add nodes to index associated with the configured search client. Args: nodes: List[BaseNode]: nodes with embeddings @@ -678,7 +680,8 @@ class AzureQueryResultSearchBase: score_result = [] for result in results: node_id = result[self._field_mapping["id"]] - metadata = json.loads(result[self._field_mapping["metadata"]]) + metadata_str = result[self._field_mapping["metadata"]] + metadata = json.loads(metadata_str) if metadata_str else {} score = result["@search.score"] chunk = result[self._field_mapping["chunk"]] @@ -795,7 +798,8 @@ class AzureQueryResultSearchSemanticHybrid(AzureQueryResultSearchHybrid): score_result = [] for result in results: node_id = result[self._field_mapping["id"]] - metadata = json.loads(result[self._field_mapping["metadata"]]) + metadata_str = result[self._field_mapping["metadata"]] + metadata = json.loads(metadata_str) if metadata_str else {} # use reranker_score instead of score score = result["@search.reranker_score"] chunk = result[self._field_mapping["chunk"]] 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 e277a1b5436973e1d1c97a252a29ae026998f3f7..e0bc6ecd3ec16767ffbd39b662177f0e5ba0efec 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.1.6" +version = "0.1.7" [tool.poetry.dependencies] python = ">=3.8.1,<4.0"