index-client/README.md

ElasticSearch Index

Description

The index operator index the given documents in ElasticSearch to get ready for retrieval. It accepts a single document in dictionary or a list of documents (dictionaries) as input. For each document, the index automatically generates a unique id. To use this operator, you need to set up ElasticSearch in advance.

Code Example

Insert an example document into ElasticSearch with address of localhost:9200 and index of 'test_index'.

from datetime import datetime
from towhee.dc2 import pipe, ops, DataCollection


example_doc = {
    'title': 'Test Title',
    'author': 'Towhee',
    'content': 'This is an example.',
    'timestamp': datetime.now()
    }

es_insert = (
    pipe.input('doc')
        .map('doc', 'res', ops.elasticsearch.index_client(
            host='localhost', port=9200, index_name='test_index'
        ))
        .output('doc', 'res')
)

res = es_insert(example_doc) # OR: es_insert([example_doc])
DataCollection(res).show() # Optional: display output data