pinecone 5.2.0.dev7__tar.gz → 5.2.0.dev9__tar.gz

{pinecone-5.2.0.dev7 → pinecone-5.2.0.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pinecone
-Version: 5.2.0.dev7
+Version: 5.2.0.dev9
 Summary: Pinecone client and SDK
 Home-page: https://www.pinecone.io
 License: Apache-2.0
@@ -32,7 +32,7 @@ Requires-Dist: googleapis-common-protos (>=1.53.0) ; extra == "grpc"
 Requires-Dist: grpcio (>=1.44.0) ; (python_version >= "3.8" and python_version < "3.11") and (extra == "grpc")
 Requires-Dist: grpcio (>=1.59.0) ; (python_version >= "3.11" and python_version < "4.0") and (extra == "grpc")
 Requires-Dist: lz4 (>=3.1.3) ; extra == "grpc"
-Requires-Dist: pinecone-plugin-inference (>=1.0.3,<2.0.0)
+Requires-Dist: pinecone-plugin-inference (>=1.1.0,<2.0.0)
 Requires-Dist: pinecone-plugin-interface (>=0.0.7,<0.0.8)
 Requires-Dist: protobuf (>=4.25,<5.0) ; extra == "grpc"
 Requires-Dist: protoc-gen-openapiv2 (>=0.0.1,<0.0.2) ; extra == "grpc"

pinecone-5.2.0.dev9/pinecone/__version__

@@ -0,0 +1 @@
+5.2.0.dev9

{pinecone-5.2.0.dev7 → pinecone-5.2.0.dev9}/pinecone/core_ea/openapi/db_data/model/start_import_request.py

@@ -62,7 +62,7 @@ class StartImportRequest(ModelNormal):
     allowed_values = {}
 
     validations = {
-        ("integration",): {
+        ("integration_id",): {
             "max_length": 1000,
             "min_length": 1,
         },
@@ -102,7 +102,7 @@ class StartImportRequest(ModelNormal):
         """
         lazy_import()
         return {
-            "integration": (str,),  # noqa: E501
+            "integration_id": (str,),  # noqa: E501
             "uri": (str,),  # noqa: E501
             "error_mode": (ImportErrorMode,),  # noqa: E501
         }
@@ -112,7 +112,7 @@ class StartImportRequest(ModelNormal):
         return None
 
     attribute_map = {
-        "integration": "integration",  # noqa: E501
+        "integration_id": "integrationId",  # noqa: E501
         "uri": "uri",  # noqa: E501
         "error_mode": "errorMode",  # noqa: E501
     }
@@ -157,7 +157,7 @@ class StartImportRequest(ModelNormal):
                                 Animal class but this time we won't travel
                                 through its discriminator because we passed in
                                 _visited_composed_classes = (Animal,)
-            integration (str): The name of the storage integration that should be used to access the data.. [optional]  # noqa: E501
+            integration_id (str): The id of the storage integration that should be used to access the data.. [optional]  # noqa: E501
             uri (str): The URI prefix under which the data to import is available. All data within this prefix will be listed then imported into the target index. Currently only `s3://` URIs are supported.. [optional]  # noqa: E501
             error_mode (ImportErrorMode): [optional]  # noqa: E501
         """
@@ -246,7 +246,7 @@ class StartImportRequest(ModelNormal):
                                 Animal class but this time we won't travel
                                 through its discriminator because we passed in
                                 _visited_composed_classes = (Animal,)
-            integration (str): The name of the storage integration that should be used to access the data.. [optional]  # noqa: E501
+            integration_id (str): The id of the storage integration that should be used to access the data.. [optional]  # noqa: E501
             uri (str): The URI prefix under which the data to import is available. All data within this prefix will be listed then imported into the target index. Currently only `s3://` URIs are supported.. [optional]  # noqa: E501
             error_mode (ImportErrorMode): [optional]  # noqa: E501
         """

{pinecone-5.2.0.dev7 → pinecone-5.2.0.dev9}/pinecone/data/__init__.py

@@ -8,3 +8,5 @@ from .errors import (
     SparseValuesDictionaryExpectedError,
     MetadataDictionaryExpectedError,
 )
+
+from .features.bulk_import import ImportErrorMode

{pinecone-5.2.0.dev7 → pinecone-5.2.0.dev9}/pinecone/data/features/bulk_import.py

@@ -1,7 +1,7 @@
-import warnings
-
-from typing import Optional, Union, Literal, Iterator, List
+from enum import Enum
+from typing import Optional, Literal, Iterator, List, Type, cast
 
+from pinecone.utils.decorators import prerelease_feature
 from pinecone.config.config import ConfigBuilder
 from pinecone.core_ea.openapi.db_data import ApiClient
 from pinecone.core_ea.openapi.db_data.api.bulk_operations_api import BulkOperationsApi
@@ -20,14 +20,9 @@ from pinecone.core_ea.openapi.db_data.models import (
 for m in [StartImportResponse, ImportListResponse, ImportModel]:
     install_json_repr_override(m)
 
-
-def prerelease_feature(func):
-    def wrapper(*args, **kwargs):
-        warnmsg = f"This is a prerelease feature implemented against the {API_VERSION} version of our API."
-        warnings.warn(warnmsg)
-        return func(*args, **kwargs)
-
-    return wrapper
+ImportErrorMode: Type[Enum] = cast(
+    Type[Enum], Enum("ImportErrorMode", ImportErrorModeClass.allowed_values[("on_error",)])
+)
 
 
 class ImportFeatureMixin:
@@ -49,14 +44,14 @@ class ImportFeatureMixin:
                 api_version=API_VERSION,
             )
 
-    ImportErrorMode = Literal["CONTINUE", "ABORT"]
+    usage_warning = "The bulk import feature is in early access."
 
-    @prerelease_feature
+    @prerelease_feature(message=usage_warning, api_version=API_VERSION)
     def start_import(
         self,
         uri: str,
-        integration: Optional[str] = None,
-        error_mode: Optional[ImportErrorMode] = "CONTINUE",
+        integration_id: Optional[str] = None,
+        error_mode: Optional[Literal["CONTINUE", "ABORT"]] = "CONTINUE",
     ) -> StartImportResponse:
         """Import data from a storage provider into an index. The uri must start with the scheme of a supported
         storage provider. For buckets that are not publicly readable, you will also need to separately configure
@@ -77,16 +72,25 @@ class ImportFeatureMixin:
         Returns:
             StartImportResponse: Contains the id of the import operation.
         """
+        if isinstance(error_mode, ImportErrorMode):
+            error_mode = error_mode.value
+        elif isinstance(error_mode, str):
+            try:
+                error_mode = ImportErrorMode(error_mode.lower()).value
+            except ValueError:
+                raise ValueError(f"Invalid error_mode value: {error_mode}")
+
         args_dict = parse_non_empty_args(
             [
                 ("uri", uri),
-                ("integration", integration),
+                ("integration_id", integration_id),
                 ("error_mode", ImportErrorModeClass(on_error=error_mode)),
             ]
         )
+
         return self.__import_operations_api.start_import(StartImportRequest(**args_dict))
 
-    @prerelease_feature
+    @prerelease_feature(message=usage_warning, api_version=API_VERSION)
     def list_imports(self, **kwargs) -> Iterator[List[ImportModel]]:
         """
         Returns a generator that yields each import operation. It automatically handles pagination tokens on your behalf so you can
@@ -123,7 +127,7 @@ class ImportFeatureMixin:
             else:
                 done = True
 
-    @prerelease_feature
+    @prerelease_feature(message=usage_warning, api_version=API_VERSION)
     def list_imports_paginated(
         self,
         limit: Optional[int] = None,
@@ -168,7 +172,7 @@ class ImportFeatureMixin:
         )
         return self.__import_operations_api.list_imports(**args_dict)
 
-    @prerelease_feature
+    @prerelease_feature(message=usage_warning, api_version=API_VERSION)
     def describe_import(self, id: str) -> ImportModel:
         """
         describe_import is used to get detailed information about a specific import operation.
@@ -185,11 +189,13 @@ class ImportFeatureMixin:
 
         return self.__import_operations_api.describe_import(id=id)
 
-    @prerelease_feature
+    @prerelease_feature(message=usage_warning, api_version=API_VERSION)
     def cancel_import(self, id: str):
         """Cancel an import operation.
 
         Args:
             id (str): The id of the import operation to cancel.
         """
+        if isinstance(id, int):
+            id = str(id)
         return self.__import_operations_api.cancel_import(id=id)

{pinecone-5.2.0.dev7 → pinecone-5.2.0.dev9}/pinecone/utils/__init__.py

@@ -12,3 +12,4 @@ from .setup_openapi_client import (
 from .parse_args import parse_non_empty_args
 from .docslinks import docslinks
 from .repr_overrides import install_json_repr_override
+from .decorators import prerelease_feature

pinecone-5.2.0.dev9/pinecone/utils/decorators.py

@@ -0,0 +1,34 @@
+import warnings
+from functools import wraps
+from typing import Optional, Callable
+
+
+class PineconePrereleaseWarning(UserWarning):
+    pass
+
+
+warnings.simplefilter("once", PineconePrereleaseWarning)
+
+
+def prerelease_feature(
+    message: str = "This is a pre-release feature and may change in the future.", api_version: Optional[str] = None
+):
+    # The indirection of the function called decorator is needed for the
+    # @prerelease_feature decorator to work correctly with or without arguments.
+    def decorator(func: Callable):
+
+        # The @wraps decorator is used to preserve the metadata of the original function when calling
+        # .__name__ or .__docs__ on the decorated function.
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            if api_version is None:
+                combined_message = f"{message}"
+            else:
+                combined_message = f"{message} It is implemented against the {api_version} version of Pinecone's API."
+
+            warnings.warn(combined_message, category=PineconePrereleaseWarning, stacklevel=2)
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

{pinecone-5.2.0.dev7 → pinecone-5.2.0.dev9}/pyproject.toml

@@ -16,7 +16,7 @@ exclude = '''
 
 [tool.poetry]
 name = "pinecone"
-version = "5.2.0.dev7"
+version = "5.2.0.dev9"
 packages = [
     { include="pinecone", from="." },
 ]
@@ -68,7 +68,7 @@ lz4 = { version = ">=3.1.3", optional = true }
 protobuf = { version = "^4.25", optional = true }
 protoc-gen-openapiv2 = {version = "^0.0.1", optional = true }
 pinecone-plugin-interface = "^0.0.7"
-pinecone-plugin-inference = "^1.0.3"
+pinecone-plugin-inference = "^1.1.0"
 
 [tool.poetry.group.types]
 optional = true

pinecone-5.2.0.dev7/pinecone/__version__

@@ -1 +0,0 @@
-5.2.0.dev7