unstructured-ingest 0.5.5__tar.gz → 0.5.6__tar.gz

{unstructured_ingest-0.5.5/unstructured_ingest.egg-info → unstructured_ingest-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: unstructured-ingest
-Version: 0.5.5
+Version: 0.5.6
 Summary: A library that prepares raw documents for downstream ML tasks.
 Home-page: https://github.com/Unstructured-IO/unstructured-ingest
 Author: Unstructured Technologies
@@ -22,13 +22,13 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9.0,<3.14
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Requires-Dist: click
+Requires-Dist: python-dateutil
 Requires-Dist: pandas
+Requires-Dist: click
 Requires-Dist: tqdm
-Requires-Dist: dataclasses_json
 Requires-Dist: pydantic>=2.7
 Requires-Dist: opentelemetry-sdk
-Requires-Dist: python-dateutil
+Requires-Dist: dataclasses_json
 Provides-Extra: remote
 Requires-Dist: unstructured-client>=0.26.1; extra == "remote"
 Provides-Extra: csv
@@ -66,16 +66,16 @@ Requires-Dist: pyairtable; extra == "airtable"
 Provides-Extra: astradb
 Requires-Dist: astrapy; extra == "astradb"
 Provides-Extra: azure
-Requires-Dist: fsspec; extra == "azure"
 Requires-Dist: adlfs; extra == "azure"
+Requires-Dist: fsspec; extra == "azure"
 Provides-Extra: azure-ai-search
 Requires-Dist: azure-search-documents; extra == "azure-ai-search"
 Provides-Extra: biomed
-Requires-Dist: bs4; extra == "biomed"
 Requires-Dist: requests; extra == "biomed"
+Requires-Dist: bs4; extra == "biomed"
 Provides-Extra: box
-Requires-Dist: fsspec; extra == "box"
 Requires-Dist: boxfs; extra == "box"
+Requires-Dist: fsspec; extra == "box"
 Provides-Extra: chroma
 Requires-Dist: chromadb; extra == "chroma"
 Provides-Extra: clarifai
@@ -86,31 +86,31 @@ Requires-Dist: requests; extra == "confluence"
 Provides-Extra: couchbase
 Requires-Dist: couchbase; extra == "couchbase"
 Provides-Extra: delta-table
-Requires-Dist: boto3; extra == "delta-table"
 Requires-Dist: deltalake; extra == "delta-table"
+Requires-Dist: boto3; extra == "delta-table"
 Provides-Extra: discord
 Requires-Dist: discord.py; extra == "discord"
 Provides-Extra: dropbox
-Requires-Dist: fsspec; extra == "dropbox"
 Requires-Dist: dropboxdrivefs; extra == "dropbox"
+Requires-Dist: fsspec; extra == "dropbox"
 Provides-Extra: duckdb
 Requires-Dist: duckdb; extra == "duckdb"
 Provides-Extra: elasticsearch
 Requires-Dist: elasticsearch[async]; extra == "elasticsearch"
 Provides-Extra: gcs
+Requires-Dist: gcsfs; extra == "gcs"
 Requires-Dist: bs4; extra == "gcs"
 Requires-Dist: fsspec; extra == "gcs"
-Requires-Dist: gcsfs; extra == "gcs"
 Provides-Extra: github
-Requires-Dist: requests; extra == "github"
 Requires-Dist: pygithub>1.58.0; extra == "github"
+Requires-Dist: requests; extra == "github"
 Provides-Extra: gitlab
 Requires-Dist: python-gitlab; extra == "gitlab"
 Provides-Extra: google-drive
 Requires-Dist: google-api-python-client; extra == "google-drive"
 Provides-Extra: hubspot
-Requires-Dist: hubspot-api-client; extra == "hubspot"
 Requires-Dist: urllib3; extra == "hubspot"
+Requires-Dist: hubspot-api-client; extra == "hubspot"
 Provides-Extra: jira
 Requires-Dist: atlassian-python-api; extra == "jira"
 Provides-Extra: kafka
@@ -124,23 +124,23 @@ Requires-Dist: pymilvus; extra == "milvus"
 Provides-Extra: mongodb
 Requires-Dist: pymongo; extra == "mongodb"
 Provides-Extra: neo4j
-Requires-Dist: cymple; extra == "neo4j"
-Requires-Dist: networkx; extra == "neo4j"
 Requires-Dist: neo4j-rust-ext; extra == "neo4j"
+Requires-Dist: networkx; extra == "neo4j"
+Requires-Dist: cymple; extra == "neo4j"
 Provides-Extra: notion
+Requires-Dist: notion-client; extra == "notion"
 Requires-Dist: backoff; extra == "notion"
 Requires-Dist: htmlBuilder; extra == "notion"
-Requires-Dist: notion-client; extra == "notion"
 Requires-Dist: httpx; extra == "notion"
 Provides-Extra: onedrive
+Requires-Dist: Office365-REST-Python-Client; extra == "onedrive"
 Requires-Dist: msal; extra == "onedrive"
 Requires-Dist: bs4; extra == "onedrive"
-Requires-Dist: Office365-REST-Python-Client; extra == "onedrive"
 Provides-Extra: opensearch
 Requires-Dist: opensearch-py; extra == "opensearch"
 Provides-Extra: outlook
-Requires-Dist: msal; extra == "outlook"
 Requires-Dist: Office365-REST-Python-Client; extra == "outlook"
+Requires-Dist: msal; extra == "outlook"
 Provides-Extra: pinecone
 Requires-Dist: pinecone-client>=3.7.1; extra == "pinecone"
 Provides-Extra: postgres
@@ -152,11 +152,11 @@ Requires-Dist: praw; extra == "reddit"
 Provides-Extra: redis
 Requires-Dist: redis; extra == "redis"
 Provides-Extra: s3
-Requires-Dist: fsspec; extra == "s3"
 Requires-Dist: s3fs; extra == "s3"
+Requires-Dist: fsspec; extra == "s3"
 Provides-Extra: sharepoint
-Requires-Dist: msal; extra == "sharepoint"
 Requires-Dist: Office365-REST-Python-Client; extra == "sharepoint"
+Requires-Dist: msal; extra == "sharepoint"
 Provides-Extra: salesforce
 Requires-Dist: simple-salesforce; extra == "salesforce"
 Provides-Extra: sftp
@@ -179,8 +179,8 @@ Provides-Extra: singlestore
 Requires-Dist: singlestoredb; extra == "singlestore"
 Provides-Extra: vectara
 Requires-Dist: aiofiles; extra == "vectara"
-Requires-Dist: requests; extra == "vectara"
 Requires-Dist: httpx; extra == "vectara"
+Requires-Dist: requests; extra == "vectara"
 Provides-Extra: vastdb
 Requires-Dist: ibis; extra == "vastdb"
 Requires-Dist: pyarrow; extra == "vastdb"
@@ -188,8 +188,8 @@ Requires-Dist: vastdb; extra == "vastdb"
 Provides-Extra: embed-huggingface
 Requires-Dist: sentence-transformers; extra == "embed-huggingface"
 Provides-Extra: embed-octoai
-Requires-Dist: openai; extra == "embed-octoai"
 Requires-Dist: tiktoken; extra == "embed-octoai"
+Requires-Dist: openai; extra == "embed-octoai"
 Provides-Extra: embed-vertexai
 Requires-Dist: vertexai; extra == "embed-vertexai"
 Provides-Extra: embed-voyageai
@@ -197,11 +197,11 @@ Requires-Dist: voyageai; extra == "embed-voyageai"
 Provides-Extra: embed-mixedbreadai
 Requires-Dist: mixedbread-ai; extra == "embed-mixedbreadai"
 Provides-Extra: openai
-Requires-Dist: openai; extra == "openai"
 Requires-Dist: tiktoken; extra == "openai"
+Requires-Dist: openai; extra == "openai"
 Provides-Extra: bedrock
-Requires-Dist: boto3; extra == "bedrock"
 Requires-Dist: aioboto3; extra == "bedrock"
+Requires-Dist: boto3; extra == "bedrock"
 Provides-Extra: togetherai
 Requires-Dist: together; extra == "togetherai"
 Dynamic: author

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/test/integration/connectors/test_pinecone.py

@@ -351,3 +351,37 @@ def test_pinecone_stager(
         stager=stager,
         tmp_dir=tmp_path,
     )
+
+
+@requires_env(API_KEY)
+@pytest.mark.tags(CONNECTOR_TYPE, DESTINATION_TAG, VECTOR_DB_TAG)
+def test_pinecone_create_destination(pinecone_index):
+    uploader = PineconeUploader(
+        connection_config=PineconeConnectionConfig(
+            access_config=PineconeAccessConfig(api_key=get_api_key())
+        ),
+        upload_config=PineconeUploaderConfig(),
+    )
+
+    random_id = str(uuid4()).split("-")[0]
+
+    index_name = f"test-create-destination-{random_id}"
+
+    assert not uploader.index_exists(index_name=index_name)
+
+    try:
+        uploader.create_destination(destination_name=index_name, vector_length=1536)
+    except Exception as e:
+        error_body = getattr(e, "body", None)
+        raise pytest.fail(f"failed to create destination: {e} {error_body}")
+
+    assert uploader.index_exists(index_name=index_name), "destination was not created successfully"
+
+    try:
+        pc = uploader.connection_config.get_client()
+        logger.info(f"deleting index for test create destination: {index_name}")
+        pc.delete_index(name=index_name)
+    except Exception as e:
+        raise pytest.fail(f"failed to cleanup / delete the destination: {e}")
+
+    assert not uploader.index_exists(index_name=index_name), "cleanup failed"

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/test/integration/connectors/weaviate/test_local.py

@@ -78,7 +78,6 @@ def run_uploader_and_validate(
     validate_count(expected_count=expected_count)
 
 
-@pytest.mark.asyncio
 @pytest.mark.tags(CONNECTOR_TYPE, DESTINATION_TAG, VECTOR_DB_TAG)
 def test_weaviate_local_destination(upload_file: Path, collection: str, tmp_path: Path):
     file_data = FileData(
@@ -142,11 +141,12 @@ def test_weaviate_local_create_destination(weaviate_instance):
         upload_config=LocalWeaviateUploaderConfig(),
         connection_config=LocalWeaviateConnectionConfig(),
     )
-    collection_name = "system_created"
+    collection_name = "system_created-123"
+    formatted_collection_name = "System_created_123"
     created = uploader.create_destination(destination_name=collection_name)
     assert created
     with uploader.connection_config.get_client() as weaviate_client:
-        assert weaviate_client.collections.exists(name=collection_name)
+        assert weaviate_client.collections.exists(name=formatted_collection_name)
 
     created = uploader.create_destination(destination_name=collection_name)
     assert not created

unstructured_ingest-0.5.6/unstructured_ingest/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.5.6"  # pragma: no cover

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/unstructured_ingest/v2/interfaces/process.py

@@ -8,7 +8,7 @@ class BaseProcess(ABC):
     def is_async(self) -> bool:
         return False
 
-    def init(self, *kwargs: Any) -> None:
+    def init(self, **kwargs: Any) -> None:
         pass
 
     def precheck(self) -> None:

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/unstructured_ingest/v2/interfaces/uploader.py

@@ -1,7 +1,7 @@
 from abc import ABC
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Any, Optional, TypeVar
+from typing import Any, TypeVar
 
 from pydantic import BaseModel
 
@@ -61,6 +61,6 @@ class Uploader(BaseProcess, BaseConnector, ABC):
 @dataclass
 class VectorDBUploader(Uploader, ABC):
     def create_destination(
-        self, destination_name: str = "elements", vector_length: Optional[int] = None, **kwargs: Any
+        self, vector_length: int, destination_name: str = "elements", **kwargs: Any
     ) -> bool:
         return False

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/unstructured_ingest/v2/pipeline/pipeline.py

@@ -126,14 +126,32 @@ class Pipeline:
                 for kk, vv in v.items():
                     logger.error(f"{k}: [{kk}] {vv}")
 
+    def _run_initialization(self):
+        failures = {}
+        init_kwargs = {}
+        for step in self._get_ordered_steps():
+            try:
+                step.process.init(**init_kwargs)
+                step.process.precheck()
+                # Make sure embedder dimensions available for downstream steps
+                if isinstance(step.process, Embedder):
+                    embed_dimensions = step.process.config.get_embedder().dimension
+                    init_kwargs["vector_length"] = embed_dimensions
+
+            except Exception as e:
+                failures[step.process.__class__.__name__] = f"[{type(e).__name__}] {e}"
+        if failures:
+            for k, v in failures.items():
+                logger.error(f"Step initialization failure: {k}: {v}")
+            raise PipelineError("Initialization failed")
+
     def run(self):
         otel_handler = OtelHandler(otel_endpoint=self.context.otel_endpoint, log_out=logger.info)
         try:
             with otel_handler.get_tracer().start_as_current_span(
                 "ingest process", record_exception=True
             ):
-                self._run_inits()
-                self._run_prechecks()
+                self._run_initialization()
                 self._run()
         finally:
             self.log_statuses()
@@ -154,43 +172,20 @@ class Pipeline:
         final = [f for f in flat if f]
         return final or None
 
-    def _get_all_steps(self) -> list[PipelineStep]:
-        steps = [self.indexer_step, self.downloader_step, self.partitioner_step, self.uploader_step]
+    def _get_ordered_steps(self) -> list[PipelineStep]:
+        steps = [self.indexer_step, self.downloader_step]
+        if self.uncompress_step:
+            steps.append(self.uncompress_step)
+        steps.append(self.partitioner_step)
         if self.chunker_step:
             steps.append(self.chunker_step)
         if self.embedder_step:
             steps.append(self.embedder_step)
-        if self.uncompress_step:
-            steps.append(self.uncompress_step)
         if self.stager_step:
             steps.append(self.stager_step)
+        steps.append(self.uploader_step)
         return steps
 
-    def _run_inits(self):
-        failures = {}
-
-        for step in self._get_all_steps():
-            try:
-                step.process.init()
-            except Exception as e:
-                failures[step.process.__class__.__name__] = f"[{type(e).__name__}] {e}"
-        if failures:
-            for k, v in failures.items():
-                logger.error(f"Step init failure: {k}: {v}")
-            raise PipelineError("Init failed")
-
-    def _run_prechecks(self):
-        failures = {}
-        for step in self._get_all_steps():
-            try:
-                step.process.precheck()
-            except Exception as e:
-                failures[step.process.__class__.__name__] = f"[{type(e).__name__}] {e}"
-        if failures:
-            for k, v in failures.items():
-                logger.error(f"Step precheck failure: {k}: {v}")
-            raise PipelineError("Precheck failed")
-
     def apply_filter(self, records: list[dict]) -> list[dict]:
         if not self.filter_step:
             return records

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/unstructured_ingest/v2/processes/connectors/pinecone.py

@@ -1,6 +1,7 @@
 import json
+import re
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any, Literal, Optional
 
 from pydantic import Field, Secret
 
@@ -13,10 +14,10 @@ from unstructured_ingest.v2.interfaces import (
     AccessConfig,
     ConnectionConfig,
     FileData,
-    Uploader,
     UploaderConfig,
     UploadStager,
     UploadStagerConfig,
+    VectorDBUploader,
 )
 from unstructured_ingest.v2.logger import logger
 from unstructured_ingest.v2.processes.connector_registry import DestinationRegistryEntry
@@ -41,7 +42,7 @@ class PineconeAccessConfig(AccessConfig):
 
 
 class PineconeConnectionConfig(ConnectionConfig):
-    index_name: str = Field(description="Name of the index to connect to.")
+    index_name: Optional[str] = Field(description="Name of the index to connect to.", default=None)
     access_config: Secret[PineconeAccessConfig] = Field(
         default=PineconeAccessConfig(), validate_default=True
     )
@@ -160,18 +161,101 @@ class PineconeUploadStager(UploadStager):
 
 
 @dataclass
-class PineconeUploader(Uploader):
+class PineconeUploader(VectorDBUploader):
     upload_config: PineconeUploaderConfig
     connection_config: PineconeConnectionConfig
     connector_type: str = CONNECTOR_TYPE
 
+    def init(self, **kwargs: Any) -> None:
+        self.create_destination(**kwargs)
+
+    def index_exists(self, index_name: Optional[str]) -> bool:
+        from pinecone.exceptions import NotFoundException
+
+        index_name = index_name or self.connection_config.index_name
+        pc = self.connection_config.get_client()
+        try:
+            pc.describe_index(index_name)
+            return True
+        except NotFoundException:
+            return False
+        except Exception as e:
+            logger.error(f"failed to check if pinecone index exists : {e}")
+            raise DestinationConnectionError(f"failed to check if pinecone index exists : {e}")
+
     def precheck(self):
         try:
-            self.connection_config.get_index()
+            # just a connection check here. not an actual index_exists check
+            self.index_exists("just-checking-our-connection")
+
+            if self.connection_config.index_name and not self.index_exists(
+                self.connection_config.index_name
+            ):
+                raise DestinationConnectionError(
+                    f"index {self.connection_config.index_name} does not exist"
+                )
         except Exception as e:
             logger.error(f"failed to validate connection: {e}", exc_info=True)
             raise DestinationConnectionError(f"failed to validate connection: {e}")
 
+    def format_destination_name(self, destination_name: str) -> str:
+        # Pinecone naming requirements:
+        # can only contain lowercase letters, numbers, and hyphens
+        # must be 45 characters or less
+        formatted = re.sub(r"[^a-z0-9]", "-", destination_name.lower())
+        return formatted
+
+    def create_destination(
+        self,
+        vector_length: int,
+        destination_name: str = "elements",
+        destination_type: Literal["pod", "serverless"] = "serverless",
+        serverless_cloud: str = "aws",
+        serverless_region: str = "us-west-2",
+        pod_environment: str = "us-east1-gcp",
+        pod_type: str = "p1.x1",
+        pod_count: int = 1,
+        **kwargs: Any,
+    ) -> bool:
+        from pinecone import PodSpec, ServerlessSpec
+
+        index_name = destination_name or self.connection_config.index_name
+        index_name = self.format_destination_name(index_name)
+        self.connection_config.index_name = index_name
+
+        if not self.index_exists(index_name):
+
+            logger.info(f"creating pinecone index {index_name}")
+
+            pc = self.connection_config.get_client()
+
+            if destination_type == "serverless":
+                pc.create_index(
+                    name=destination_name,
+                    dimension=vector_length,
+                    spec=ServerlessSpec(cloud=serverless_cloud, region=serverless_region),
+                    **kwargs,
+                )
+
+                return True
+
+            elif destination_type == "pod":
+                pc.create_index(
+                    name=destination_name,
+                    dimension=vector_length,
+                    spec=PodSpec(environment=pod_environment, pod_type=pod_type, pods=pod_count),
+                    **kwargs,
+                )
+
+                return True
+
+            else:
+                raise ValueError(f"unexpected destination type: {destination_type}")
+
+        else:
+            logger.debug(f"index {index_name} already exists, skipping creation")
+            return False
+
     def pod_delete_by_record_id(self, file_data: FileData) -> None:
         logger.debug(
             f"deleting any content with metadata "
@@ -266,6 +350,10 @@ class PineconeUploader(Uploader):
         )
         # Determine if serverless or pod based index
         pinecone_client = self.connection_config.get_client()
+
+        if not self.connection_config.index_name:
+            raise ValueError("No index name specified")
+
         index_description = pinecone_client.describe_index(name=self.connection_config.index_name)
         if "serverless" in index_description.get("spec"):
             self.serverless_delete_by_record_id(file_data=file_data)

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/unstructured_ingest/v2/processes/connectors/weaviate/weaviate.py

@@ -1,4 +1,5 @@
 import json
+import re
 from abc import ABC, abstractmethod
 from contextlib import contextmanager
 from dataclasses import dataclass, field
@@ -229,19 +230,29 @@ class WeaviateUploader(VectorDBUploader, ABC):
             logger.error(f"Failed to validate connection {e}", exc_info=True)
             raise DestinationConnectionError(f"failed to validate connection: {e}")
 
-    def init(self, *kwargs: Any) -> None:
-        self.create_destination()
+    def init(self, **kwargs: Any) -> None:
+        self.create_destination(**kwargs)
+
+    def format_destination_name(self, destination_name: str) -> str:
+        # Weaviate naming requirements:
+        # must be alphanumeric and underscores only
+        formatted = re.sub(r"[^a-zA-Z0-9]", "_", destination_name)
+        # must begin with capital letter
+        return formatted.capitalize()
 
     def create_destination(
         self, destination_name: str = "elements", vector_length: Optional[int] = None, **kwargs: Any
     ) -> bool:
+        destination_name = self.format_destination_name(destination_name)
         collection_name = self.upload_config.collection or destination_name
         self.upload_config.collection = collection_name
+
         connectors_dir = Path(__file__).parents[1]
         collection_config_file = connectors_dir / "assets" / "weaviate_collection_config.json"
         with collection_config_file.open() as f:
             collection_config = json.load(f)
         collection_config["class"] = collection_name
+
         if not self._collection_exists():
             logger.info(
                 f"creating default weaviate collection '{collection_name}' with default configs"

{unstructured_ingest-0.5.5 → unstructured_ingest-0.5.6}/unstructured_ingest/v2/processes/embedder.py

@@ -186,7 +186,7 @@ class EmbedderConfig(BaseModel):
 class Embedder(BaseProcess, ABC):
     config: EmbedderConfig
 
-    def init(self, *kwargs: Any) -> None:
+    def init(self, **kwargs: Any) -> None:
         self.config.get_embedder().initialize()
 
     def run(self, elements_filepath: Path, **kwargs: Any) -> list[dict]: