pinecone 5.2.0.dev0__tar.gz → 5.2.0.dev2__tar.gz

{pinecone-5.2.0.dev0 → pinecone-5.2.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pinecone
-Version: 5.2.0.dev0
+Version: 5.2.0.dev2
 Summary: Pinecone client and SDK
 Home-page: https://www.pinecone.io
 License: Apache-2.0

pinecone-5.2.0.dev2/pinecone/__version__

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

{pinecone-5.2.0.dev0 → pinecone-5.2.0.dev2}/pinecone/config/openapi.py

@@ -24,6 +24,8 @@ class OpenApiConfigFactory:
         openapi_config.host = host
         openapi_config.ssl_ca_cert = certifi.where()
         openapi_config.socket_options = cls._get_socket_options()
+        openapi_config.discard_unknown_keys = True
+
         return openapi_config
 
     @classmethod

{pinecone-5.2.0.dev0 → pinecone-5.2.0.dev2}/pinecone/control/repr_overrides.py

@@ -1,8 +1,7 @@
+from pinecone.utils import install_json_repr_override
 from pinecone.models.index_model import IndexModel
 from pinecone.core.openapi.control.models import CollectionModel
 
-import json
-
 
 def install_repr_overrides():
     """
@@ -14,4 +13,4 @@ def install_repr_overrides():
     query results.
     """
     for model in [IndexModel, CollectionModel]:
-        model.__repr__ = lambda self: json.dumps(self.to_dict(), indent=4, sort_keys=False)
+        install_json_repr_override(model)

{pinecone-5.2.0.dev0 → pinecone-5.2.0.dev2}/pinecone/data/features/bulk_import.py

@@ -1,13 +1,13 @@
 import warnings
 
-from typing import Optional, Union, Literal
+from typing import Optional, Union, Literal, Iterator, List
 
 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
 from pinecone.core_ea.openapi.shared import API_VERSION
 
-from pinecone.utils import parse_non_empty_args
+from pinecone.utils import parse_non_empty_args, install_json_repr_override, setup_openapi_client
 
 from pinecone.core_ea.openapi.db_data.models import (
     StartImportRequest,
@@ -17,7 +17,8 @@ from pinecone.core_ea.openapi.db_data.models import (
     ImportErrorMode as ImportErrorModeClass,
 )
 
-from ...utils import setup_openapi_client
+for m in [StartImportResponse, ImportListResponse, ImportModel]:
+    install_json_repr_override(m)
 
 
 def prerelease_feature(func):
@@ -69,9 +70,11 @@ class ImportFeatureMixin:
         Args:
             uri (str): The URI of the data to import. The URI must start with the scheme of a supported storage provider.
             integration (Optional[str], optional): Defaults to None.
+            error_mode: Defaults to "CONTINUE". If set to "CONTINUE", the import operation will continue even if some
+                records fail to import. Pass "ABORT" to stop the import operation if any records fail to import.
 
         Returns:
-            _type_: _description_
+            StartImportResponse: Contains the id of the import operation.
         """
         args_dict = parse_non_empty_args(
             [
@@ -83,7 +86,7 @@ class ImportFeatureMixin:
         return self.__import_operations_api.start_import(StartImportRequest(**args_dict))
 
     @prerelease_feature
-    def list_imports(self, **kwargs):
+    def list_imports(self, **kwargs) -> Iterator[List[ImportModel]]:
         """
         The list_imports operation accepts all of the same arguments as list_imports_paginated, and returns a generator that yields
         a list of operations in each page of results. It automatically handles pagination tokens on your
@@ -143,24 +146,24 @@ class ImportFeatureMixin:
         return self.__import_operations_api.list_imports(**args_dict)
 
     @prerelease_feature
-    def describe_import(self, operation_id: str) -> ImportModel:
+    def describe_import(self, id: str) -> ImportModel:
         """
         describe_import is used to get detailed information about a specific import operation.
 
         Args:
-            operation_id (str): The id of the import operation. This value is returned when
+            id (str): The id of the import operation. This value is returned when
             starting an import, and can be looked up using list_imports.
 
         Returns:
             ImportModel: An object containing operation id, status, and other details.
         """
-        return self.__import_operations_api.describe_import(operation_id=operation_id)
+        return self.__import_operations_api.describe_import(id=id)
 
     @prerelease_feature
-    def cancel_import(self, operation_id: str):
+    def cancel_import(self, id: str):
         """Cancel an import operation.
 
         Args:
-            operation_id (str): The id of the import operation to cancel.
+            id (str): The id of the import operation to cancel.
         """
-        return self.__import_operations_api.cancel_import(operation_id=operation_id)
+        return self.__import_operations_api.cancel_import(id=id)

{pinecone-5.2.0.dev0 → pinecone-5.2.0.dev2}/pinecone/utils/__init__.py

@@ -11,3 +11,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

pinecone-5.2.0.dev2/pinecone/utils/repr_overrides.py

@@ -0,0 +1,5 @@
+import json
+
+
+def install_json_repr_override(klass):
+    klass.__repr__ = lambda self: json.dumps(self.to_dict(), indent=4, sort_keys=False)

{pinecone-5.2.0.dev0 → pinecone-5.2.0.dev2}/pyproject.toml

@@ -16,7 +16,7 @@ exclude = '''
 
 [tool.poetry]
 name = "pinecone"
-version = "5.2.0.dev0"
+version = "5.2.0.dev2"
 packages = [
     { include="pinecone", from="." },
 ]

pinecone-5.2.0.dev0/pinecone/__version__

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