55from pathlib import Path
66from typing import TYPE_CHECKING , Any , Optional
77
8- from unstructured_ingest .enhanced_dataclass import enhanced_field
9- from unstructured_ingest .error import DestinationConnectionError
10- from unstructured_ingest .utils .data_prep import batch_generator , flatten_dict
11- from unstructured_ingest .utils .dep_check import requires_dependencies
12- from unstructured_ingest .v2 .interfaces import (
8+ from unstructured .ingest .enhanced_dataclass import enhanced_field
9+ from unstructured .ingest .error import DestinationConnectionError
10+ from unstructured .ingest .utils .data_prep import batch_generator
11+ from unstructured .ingest .v2 .interfaces import (
1312 AccessConfig ,
1413 ConnectionConfig ,
1514 UploadContent ,
1817 UploadStager ,
1918 UploadStagerConfig ,
2019)
21- from unstructured_ingest .v2 .logger import logger
22- from unstructured_ingest .v2 .processes .connector_registry import (
20+ from unstructured . ingest .v2 .logger import logger
21+ from unstructured . ingest .v2 .processes .connector_registry import (
2322 DestinationRegistryEntry ,
2423)
24+ from unstructured .staging .base import flatten_dict
25+ from unstructured .utils import requires_dependencies
2526
2627if TYPE_CHECKING :
2728 from pinecone import Index as PineconeIndex
@@ -44,6 +45,7 @@ class PineconeConnectionConfig(ConnectionConfig):
4445 @requires_dependencies (["pinecone" ], extras = "pinecone" )
4546 def get_index (self ) -> "PineconeIndex" :
4647 from pinecone import Pinecone
48+
4749 from unstructured import __version__ as unstructured_version
4850
4951 pc = Pinecone (
@@ -127,7 +129,7 @@ def check_connection(self):
127129
128130 @requires_dependencies (["pinecone" ], extras = "pinecone" )
129131 def upsert_batch (self , batch ):
130- from pinecone .core . client . exceptions import PineconeApiException
132+ from pinecone .exceptions import PineconeApiException
131133
132134 try :
133135 index = self .connection_config .get_index ()
@@ -173,4 +175,4 @@ def run(self, contents: list[UploadContent], **kwargs: Any) -> None:
173175 uploader_config = PineconeUploaderConfig ,
174176 upload_stager = PineconeUploadStager ,
175177 upload_stager_config = PineconeUploadStagerConfig ,
176- )
178+ )
0 commit comments