Skip to content

Commit

Permalink
fix whitespace
Browse files Browse the repository at this point in the history
  • Loading branch information
ttmenezes committed Nov 11, 2024
1 parent 0f38ea8 commit 41672f7
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -86,10 +86,10 @@ def from_dict(cls, data: Dict[str, Any]) -> "AzureAISearchHybridRetriever":

@component.output_types(documents=List[Document])
def run(
self,
query: str,
query_embedding: List[float],
filters: Optional[Dict[str, Any]] = None,
self,
query: str,
query_embedding: List[float],
filters: Optional[Dict[str, Any]] = None,
top_k: Optional[int] = None
):
"""Retrieve documents from the AzureAISearchDocumentStore.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -488,9 +488,9 @@ def _hybrid_retrieval(

vector_query = VectorizedQuery(vector=query_embedding, k_nearest_neighbors=top_k, fields="embedding")
result = self.client.search(
search_text=query,
vector_queries=[vector_query],
select=fields, filter=filters,
search_text=query,
vector_queries=[vector_query],
select=fields, filter=filters,
top=top_k, query_type="simple"
)
azure_docs = list(result)
Expand Down

0 comments on commit 41672f7

Please sign in to comment.