Skip to content

Commit 1346644

Browse files
authored
Expose document_debug_info and fix samples (Azure#39019)
* Expose document_debug_info and fix samples * update
1 parent 174f2fd commit 1346644

File tree

13 files changed

+28
-9
lines changed

13 files changed

+28
-9
lines changed

sdk/search/azure-search-documents/CHANGELOG.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@
88

99
### Bugs Fixed
1010

11+
- Exposed `@search.document_debug_info` in the search results.
12+
1113
### Other Changes
1214

1315
## 11.6.0b8 (2024-11-21)

sdk/search/azure-search-documents/azure/search/documents/_paging.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ def convert_search_result(result):
2020
ret["@search.reranker_score"] = result.reranker_score
2121
ret["@search.highlights"] = result.highlights
2222
ret["@search.captions"] = result.captions
23+
ret["@search.document_debug_info"] = result.document_debug_info
2324
return ret
2425

2526

@@ -44,6 +45,7 @@ def unpack_continuation_token(token):
4445

4546
class SearchItemPaged(ItemPaged[ReturnType]):
4647
"""A pageable list of search results."""
48+
4749
def __init__(self, *args, **kwargs) -> None:
4850
super(SearchItemPaged, self).__init__(*args, **kwargs)
4951
self._first_page_iterator_instance: Optional[SearchPageIterator] = None
@@ -118,6 +120,7 @@ def wrapper(self, *args, **kw):
118120

119121
class SearchPageIterator(PageIterator):
120122
"""An iterator over search results."""
123+
121124
def __init__(self, client, initial_query, kwargs, continuation_token=None) -> None:
122125
super(SearchPageIterator, self).__init__(
123126
get_next=self._get_next_cb,

sdk/search/azure-search-documents/azure/search/documents/_search_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def __repr__(self) -> str:
101101

102102
def close(self) -> None:
103103
"""Close the session.
104-
104+
105105
:return: None
106106
:rtype: None
107107
"""

sdk/search/azure-search-documents/azure/search/documents/_search_indexing_buffered_sender.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def actions(self) -> List[IndexAction]:
111111
@distributed_trace
112112
def close(self, **kwargs) -> None: # pylint: disable=unused-argument
113113
"""Close the session.
114-
114+
115115
:return: None
116116
:rtype: None
117117
"""

sdk/search/azure-search-documents/azure/search/documents/aio/_paging.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
class AsyncSearchItemPaged(AsyncItemPaged[ReturnType]):
2020
"""A pageable list of search results."""
21+
2122
def __init__(self, *args, **kwargs) -> None:
2223
super(AsyncSearchItemPaged, self).__init__(*args, **kwargs)
2324
self._first_page_iterator_instance: Optional[AsyncSearchPageIterator] = None
@@ -86,6 +87,7 @@ async def get_debug_info(self) -> DebugInfo:
8687
"""
8788
return cast(DebugInfo, await self._first_iterator_instance().get_debug_info())
8889

90+
8991
# The pylint error silenced below seems spurious, as the inner wrapper does, in
9092
# fact, become a method of the class when it is applied.
9193
def _ensure_response(f):
@@ -101,6 +103,7 @@ async def wrapper(self, *args, **kw):
101103

102104
class AsyncSearchPageIterator(AsyncPageIterator[ReturnType]):
103105
"""An iterator of search results."""
106+
104107
def __init__(self, client, initial_query, kwargs, continuation_token=None) -> None:
105108
super(AsyncSearchPageIterator, self).__init__(
106109
get_next=self._get_next_cb,

sdk/search/azure-search-documents/azure/search/documents/aio/_search_client_async.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def __repr__(self) -> str:
103103

104104
async def close(self) -> None:
105105
"""Close the session.
106-
106+
107107
:return: None
108108
:rtype: None
109109
"""

sdk/search/azure-search-documents/azure/search/documents/indexes/_search_index_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def list_index_names(self, **kwargs: Any) -> ItemPaged[str]:
136136

137137
@distributed_trace
138138
def get_index(self, name: str, **kwargs: Any) -> SearchIndex:
139-
""" Retrieve a named index in an Azure Search service
139+
"""Retrieve a named index in an Azure Search service
140140
141141
:param name: The name of the index to retrieve.
142142
:type name: str

sdk/search/azure-search-documents/azure/search/documents/indexes/aio/_search_index_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def list_index_names(self, **kwargs: Any) -> AsyncItemPaged[str]:
134134

135135
@distributed_trace_async
136136
async def get_index(self, name: str, **kwargs: Any) -> SearchIndex:
137-
""" Retrieve a named index in an Azure Search service
137+
"""Retrieve a named index in an Azure Search service
138138
139139
:param name: The name of the index to retrieve.
140140
:type name: str
@@ -513,7 +513,7 @@ def list_alias_names(self, **kwargs) -> AsyncItemPaged[str]:
513513

514514
@distributed_trace_async
515515
async def get_alias(self, name: str, **kwargs) -> SearchAlias:
516-
""" Retrieve a named alias in an Azure Search service
516+
"""Retrieve a named alias in an Azure Search service
517517
518518
:param name: The name of the alias to retrieve.
519519
:type name: str

sdk/search/azure-search-documents/azure/search/documents/indexes/models/_models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -442,6 +442,7 @@ def _from_generated(cls, skill):
442442

443443
class AnalyzeTextOptions(_serialization.Model):
444444
"""Specifies some text and analysis components used to break that text into tokens."""
445+
445446
def __init__(
446447
self,
447448
*,
@@ -1414,7 +1415,7 @@ def _from_generated(cls, search_indexer) -> Optional[Self]:
14141415

14151416
def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> MutableMapping[str, Any]:
14161417
"""Return the JSON that would be sent to server from this model.
1417-
1418+
14181419
:param bool keep_readonly: If you want to serialize the readonly attributes
14191420
:returns: A dict JSON compatible object
14201421
:rtype: dict

sdk/search/azure-search-documents/samples/async_samples/sample_semantic_search_async.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,14 @@ async def semantic_ranking():
5252

5353
credential = AzureKeyCredential(key)
5454
client = SearchClient(endpoint=service_endpoint, index_name=index_name, credential=credential)
55-
results = list(client.search(search_text="luxury", query_type="semantic", query_language="en-us"))
55+
results = list(
56+
client.search(
57+
search_text="luxury",
58+
query_type="semantic",
59+
semantic_configuration_name="semantic_config_name",
60+
query_language="en-us",
61+
)
62+
)
5663

5764
for result in results:
5865
print("{}\n{}\n)".format(result["hotelId"], result["hotelName"]))

0 commit comments

Comments
 (0)