Skip to content

Commit 8aa36ec

Browse files
committed
pull in fix for pinecone
1 parent e7fd708 commit 8aa36ec

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

unstructured_ingest/v2/processes/connectors/pinecone.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,10 @@
55
from pathlib import Path
66
from 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,
@@ -18,10 +17,12 @@
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

2627
if 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

Comments
 (0)