11# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
22#
33# SPDX-License-Identifier: Apache-2.0
4- from typing import Any , Dict , List , Optional , Union
4+ from typing import Any , Optional , Union
55
66from haystack import component , default_from_dict , default_to_dict
77from haystack .dataclasses import Document
@@ -49,7 +49,7 @@ def __init__(
4949 self ,
5050 * ,
5151 document_store : ElasticsearchDocumentStore ,
52- filters : Optional [Dict [str , Any ]] = None ,
52+ filters : Optional [dict [str , Any ]] = None ,
5353 top_k : int = 10 ,
5454 num_candidates : Optional [int ] = None ,
5555 filter_policy : Union [str , FilterPolicy ] = FilterPolicy .REPLACE ,
@@ -78,7 +78,7 @@ def __init__(
7878 self ._num_candidates = num_candidates
7979 self ._filter_policy = FilterPolicy .from_str (filter_policy ) if isinstance (filter_policy , str ) else filter_policy
8080
81- def to_dict (self ) -> Dict [str , Any ]:
81+ def to_dict (self ) -> dict [str , Any ]:
8282 """
8383 Serializes the component to a dictionary.
8484
@@ -95,7 +95,7 @@ def to_dict(self) -> Dict[str, Any]:
9595 )
9696
9797 @classmethod
98- def from_dict (cls , data : Dict [str , Any ]) -> "ElasticsearchEmbeddingRetriever" :
98+ def from_dict (cls , data : dict [str , Any ]) -> "ElasticsearchEmbeddingRetriever" :
9999 """
100100 Deserializes the component from a dictionary.
101101
@@ -113,10 +113,10 @@ def from_dict(cls, data: Dict[str, Any]) -> "ElasticsearchEmbeddingRetriever":
113113 data ["init_parameters" ]["filter_policy" ] = FilterPolicy .from_str (filter_policy )
114114 return default_from_dict (cls , data )
115115
116- @component .output_types (documents = List [Document ])
116+ @component .output_types (documents = list [Document ])
117117 def run (
118- self , query_embedding : List [float ], filters : Optional [Dict [str , Any ]] = None , top_k : Optional [int ] = None
119- ) -> Dict [str , List [Document ]]:
118+ self , query_embedding : list [float ], filters : Optional [dict [str , Any ]] = None , top_k : Optional [int ] = None
119+ ) -> dict [str , list [Document ]]:
120120 """
121121 Retrieve documents using a vector similarity metric.
122122
@@ -138,10 +138,10 @@ def run(
138138 )
139139 return {"documents" : docs }
140140
141- @component .output_types (documents = List [Document ])
141+ @component .output_types (documents = list [Document ])
142142 async def run_async (
143- self , query_embedding : List [float ], filters : Optional [Dict [str , Any ]] = None , top_k : Optional [int ] = None
144- ) -> Dict [str , List [Document ]]:
143+ self , query_embedding : list [float ], filters : Optional [dict [str , Any ]] = None , top_k : Optional [int ] = None
144+ ) -> dict [str , list [Document ]]:
145145 """
146146 Asynchronously retrieve documents using a vector similarity metric.
147147
0 commit comments