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)
