| @@ -189,7 +189,10 @@ class WeaviateVector(BaseVector): | |||
| vector = {"vector": query_vector} | |||
| document_ids_filter = kwargs.get("document_ids_filter") | |||
| if document_ids_filter: | |||
| where_filter = {"operator": "ContainsAny", "path": ["document_id"], "valueTextArray": document_ids_filter} | |||
| operands = [] | |||
| for document_id_filter in document_ids_filter: | |||
| operands.append({"path": ["document_id"], "operator": "Equal", "valueText": document_id_filter}) | |||
| where_filter = {"operator": "Or", "operands": operands} | |||
| query_obj = query_obj.with_where(where_filter) | |||
| result = ( | |||
| query_obj.with_near_vector(vector) | |||
| @@ -237,7 +240,10 @@ class WeaviateVector(BaseVector): | |||
| query_obj = self._client.query.get(collection_name, properties) | |||
| document_ids_filter = kwargs.get("document_ids_filter") | |||
| if document_ids_filter: | |||
| where_filter = {"operator": "ContainsAny", "path": ["document_id"], "valueTextArray": document_ids_filter} | |||
| operands = [] | |||
| for document_id_filter in document_ids_filter: | |||
| operands.append({"path": ["document_id"], "operator": "Equal", "valueText": document_id_filter}) | |||
| where_filter = {"operator": "Or", "operands": operands} | |||
| query_obj = query_obj.with_where(where_filter) | |||
| query_obj = query_obj.with_additional(["vector"]) | |||
| properties = ["text"] | |||