mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-07-04 15:42:16 +00:00
dd1443ab6f
This PR intends to add [Qdrant](https://qdrant.tech/) as a supported ingestion destination. - Implements CLI and programmatic usage. - Documentation update - Integration test script --- Clone of #2315 to run with CI secrets --------- Co-authored-by: Anush008 <anushshetty90@gmail.com> Co-authored-by: Roman Isecke <136338424+rbiseck3@users.noreply.github.com>
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from unstructured.ingest.connector.local import SimpleLocalConfig
|
|
from unstructured.ingest.connector.qdrant import (
|
|
QdrantWriteConfig,
|
|
SimpleQdrantConfig,
|
|
)
|
|
from unstructured.ingest.interfaces import (
|
|
ChunkingConfig,
|
|
EmbeddingConfig,
|
|
PartitionConfig,
|
|
ProcessorConfig,
|
|
ReadConfig,
|
|
)
|
|
from unstructured.ingest.runner import LocalRunner
|
|
from unstructured.ingest.runner.writers.base_writer import Writer
|
|
from unstructured.ingest.runner.writers.qdrant import QdrantWriter
|
|
|
|
|
|
def get_writer() -> Writer:
|
|
return QdrantWriter(
|
|
connector_config=SimpleQdrantConfig(
|
|
location="http://localhost:6333",
|
|
collection_name="test",
|
|
),
|
|
write_config=QdrantWriteConfig(batch_size=80),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
writer = get_writer()
|
|
runner = LocalRunner(
|
|
processor_config=ProcessorConfig(
|
|
verbose=True,
|
|
output_dir="local-output-to-qdrant",
|
|
num_processes=2,
|
|
),
|
|
connector_config=SimpleLocalConfig(
|
|
input_path="example-docs/book-war-and-peace-1225p.txt",
|
|
),
|
|
read_config=ReadConfig(),
|
|
partition_config=PartitionConfig(),
|
|
chunking_config=ChunkingConfig(chunk_elements=True),
|
|
embedding_config=EmbeddingConfig(provider="langchain-huggingface"),
|
|
writer=writer,
|
|
writer_kwargs={},
|
|
)
|
|
runner.run()
|