def test_elasticsearch_write_read(elasticsearch_fixture): document_store = ElasticsearchDocumentStore() documents = convert_files_to_dicts(dir_path="samples/docs") document_store.write_documents(documents) sleep(2) # wait for documents to be available for query documents = document_store.get_all_documents() assert len(documents) == 2 assert documents[0].id assert documents[0].text
def test_elasticsearch_write_read(elasticsearch_fixture): document_store = ElasticsearchDocumentStore() write_documents_to_db(document_store=document_store, document_dir="samples/docs") sleep(2) # wait for documents to be available for query documents = document_store.get_all_documents() print(documents) assert len(documents) == 2 assert documents[0].id assert documents[0].text
def test_elasticsearch_custom_fields(elasticsearch_fixture): client = Elasticsearch() client.indices.delete(index='haystack_test_custom', ignore=[404]) document_store = ElasticsearchDocumentStore( index="haystack_test_custom", text_field="custom_text_field", embedding_field="custom_embedding_field") doc_to_write = { "custom_text_field": "test", "custom_embedding_field": np.random.rand(768).astype(np.float32) } document_store.write_documents([doc_to_write]) documents = document_store.get_all_documents() assert len(documents) == 1 assert documents[0].text == "test" np.testing.assert_array_equal(doc_to_write["custom_embedding_field"], documents[0].embedding)