mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-06-27 02:30:08 +00:00
fd293b3e78
Closes https://github.com/Unstructured-IO/unstructured/issues/1842
Closes https://github.com/Unstructured-IO/unstructured/issues/2202
Closes https://github.com/Unstructured-IO/unstructured/issues/2203
This PR:
- Adds Elasticsearch destination connector to be able to ingest
documents from any supported source, embed them and write the embeddings
/ documents into Elasticsearch.
- Defines an example unstructured elements schema for users to be able
to setup their unstructured elasticsearch indexes easily.
- Includes parallelized upload and lazy processing for elasticsearch
destination connector.
- Rearranges elasticsearch test helpers to source, destination, and
common folders.
- Adds util functions to be able to batch iterables in a lazy way for
uploads
- Fixes a bug where removing the optional parameter `--fields` broke the
connector due to an integer processing error.
- Fixes a bug where using an [elasticsearch
config](8fa5cbf036/unstructured/ingest/connector/elasticsearch.py (L26-L35))
for a destination connector resulted in a serialization issue when
optional parameter `--fields` was not provided.
25 lines
773 B
Python
25 lines
773 B
Python
#!/usr/bin/env python3
|
|
|
|
from elasticsearch import Elasticsearch
|
|
from es_cluster_config import (
|
|
CLUSTER_URL,
|
|
INDEX_NAME,
|
|
PASSWORD,
|
|
USER,
|
|
mappings,
|
|
)
|
|
|
|
print("Connecting to the Elasticsearch cluster.")
|
|
es = Elasticsearch(CLUSTER_URL, basic_auth=(USER, PASSWORD), request_timeout=30)
|
|
print(f"{es.info()}")
|
|
|
|
print("Creating an Elasticsearch index for testing ingest elasticsearch destination connector.")
|
|
response = es.options(max_retries=5).indices.create(index=INDEX_NAME, mappings=mappings)
|
|
if response.meta.status != 200:
|
|
raise RuntimeError("failed to create index")
|
|
|
|
es.indices.refresh(index=INDEX_NAME)
|
|
response = es.cat.count(index=INDEX_NAME, format="json")
|
|
|
|
print("Succesfully created an Elasticsearch index for testing elasticsearch ingest.")
|