yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-03 21:33:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix filters in query_embedding for ElasticsearchDocumentStore (#464)

Browse Source 
Co-authored-by: Pierre Pereira <pierre.pereira@lexistems.com>
This commit is contained in:
Futurne 2020-10-05 11:25:07 +02:00 committed by GitHub
parent 669c72d538
commit 072e32b38a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
haystack/document_store/elasticsearch.py
View File

@ -437,14 +437,13 @@ class ElasticsearchDocumentStore(BaseDocumentStore):
} # type: Dict[str,Any] } # type: Dict[str,Any]
if filters: if filters:
filter_clause = []
for key, values in filters.items(): for key, values in filters.items():
filter_clause.append( if type(values) != list:
{ raise ValueError(f'Wrong filter format for key "{key}": Please provide a list of allowed values for each key. '
"terms": {key: values} 'Example: {"name": ["some", "more"], "category": ["only_one"]} ')
} body["query"]["script_score"]["query"] = {
) "terms": filters
body["query"]["bool"]["filter"] = filter_clause }
if self.excluded_meta_data: if self.excluded_meta_data:
body["_source"] = {"excludes": self.excluded_meta_data} body["_source"] = {"excludes": self.excluded_meta_data}