| @@ -91,18 +91,6 @@ class TestOpenSearchVector: | |||
| assert hits_by_vector[0].metadata['document_id'] == self.example_doc_id, \ | |||
| f"Expected document ID {self.example_doc_id}, got {hits_by_vector[0].metadata['document_id']}" | |||
| doc = Document(page_content="Test content to delete", metadata={"document_id": self.example_doc_id}) | |||
| embedding = [0.1] * 128 | |||
| with patch('opensearchpy.helpers.bulk') as mock_bulk: | |||
| mock_bulk.return_value = ([], []) | |||
| self.vector.add_texts([doc], [embedding]) | |||
| self.vector._client.search.return_value = {'hits': {'total': {'value': 0}, 'hits': []}} | |||
| ids = self.vector.get_ids_by_metadata_field(key='document_id', value=self.example_doc_id) | |||
| assert ids is None or len(ids) == 0 | |||
| def test_get_ids_by_metadata_field(self): | |||
| mock_response = { | |||
| 'hits': { | |||