add black defaults in pyproject.toml, undo a bunch of the <100 line breaks

Author: austin-denoble

pinecone/__init__.py

@@ -2,9 +2,7 @@
 .. include:: ../pdoc/README.md
 """
 
-from .deprecated_plugins import (
-    check_for_deprecated_plugins as _check_for_deprecated_plugins,
-)
+from .deprecated_plugins import check_for_deprecated_plugins as _check_for_deprecated_plugins
 from .deprecation_warnings import *
 from .pinecone import Pinecone
 from .pinecone_asyncio import PineconeAsyncio
@@ -48,10 +46,7 @@
     "SearchRerank": ("pinecone.db_data.dataclasses", "SearchRerank"),
     "FetchResponse": ("pinecone.db_data.dataclasses", "FetchResponse"),
     "DeleteRequest": ("pinecone.db_data.models", "DeleteRequest"),
-    "DescribeIndexStatsRequest": (
-        "pinecone.db_data.models",
-        "DescribeIndexStatsRequest",
-    ),
+    "DescribeIndexStatsRequest": ("pinecone.db_data.models", "DescribeIndexStatsRequest"),
     "DescribeIndexStatsResponse": ("pinecone.db_data.models", "IndexDescription"),
     "RpcStatus": ("pinecone.db_data.models", "RpcStatus"),
     "ScoredVector": ("pinecone.db_data.models", "ScoredVector"),
@@ -60,14 +55,8 @@
     "QueryResponse": ("pinecone.db_data.models", "QueryResponse"),
     "UpsertResponse": ("pinecone.db_data.models", "UpsertResponse"),
     "UpdateRequest": ("pinecone.db_data.models", "UpdateRequest"),
-    "NamespaceDescription": (
-        "pinecone.core.openapi.db_data.models",
-        "NamespaceDescription",
-    ),
-    "ImportErrorMode": (
-        "pinecone.db_data.resources.sync.bulk_import",
-        "ImportErrorMode",
-    ),
+    "NamespaceDescription": ("pinecone.core.openapi.db_data.models", "NamespaceDescription"),
+    "ImportErrorMode": ("pinecone.db_data.resources.sync.bulk_import", "ImportErrorMode"),
     "VectorDictionaryMissingKeysError": (
         "pinecone.db_data.errors",
         "VectorDictionaryMissingKeysError",
@@ -78,10 +67,7 @@
     ),
     "VectorTupleLengthError": ("pinecone.db_data.errors", "VectorTupleLengthError"),
     "SparseValuesTypeError": ("pinecone.db_data.errors", "SparseValuesTypeError"),
-    "SparseValuesMissingKeysError": (
-        "pinecone.db_data.errors",
-        "SparseValuesMissingKeysError",
-    ),
+    "SparseValuesMissingKeysError": ("pinecone.db_data.errors", "SparseValuesMissingKeysError"),
     "SparseValuesDictionaryExpectedError": (
         "pinecone.db_data.errors",
         "SparseValuesDictionaryExpectedError",
@@ -104,10 +90,7 @@
     "IndexEmbed": ("pinecone.db_control.models", "IndexEmbed"),
     "ByocSpec": ("pinecone.db_control.models", "ByocSpec"),
     "ServerlessSpec": ("pinecone.db_control.models", "ServerlessSpec"),
-    "ServerlessSpecDefinition": (
-        "pinecone.db_control.models",
-        "ServerlessSpecDefinition",
-    ),
+    "ServerlessSpecDefinition": ("pinecone.db_control.models", "ServerlessSpecDefinition"),
     "PodSpec": ("pinecone.db_control.models", "PodSpec"),
     "PodSpecDefinition": ("pinecone.db_control.models", "PodSpecDefinition"),
     "PodType": ("pinecone.db_control.enums", "PodType"),

pinecone/db_control/request_factory.py

@@ -4,21 +4,15 @@
 
 from pinecone.utils import parse_non_empty_args, convert_enum_to_string
 
-from pinecone.core.openapi.db_control.model.create_collection_request import (
-    CreateCollectionRequest,
-)
+from pinecone.core.openapi.db_control.model.create_collection_request import CreateCollectionRequest
 from pinecone.core.openapi.db_control.model.create_index_for_model_request import (
     CreateIndexForModelRequest,
 )
 from pinecone.core.openapi.db_control.model.create_index_for_model_request_embed import (
     CreateIndexForModelRequestEmbed,
 )
-from pinecone.core.openapi.db_control.model.create_index_request import (
-    CreateIndexRequest,
-)
-from pinecone.core.openapi.db_control.model.configure_index_request import (
-    ConfigureIndexRequest,
-)
+from pinecone.core.openapi.db_control.model.create_index_request import CreateIndexRequest
+from pinecone.core.openapi.db_control.model.configure_index_request import ConfigureIndexRequest
 from pinecone.core.openapi.db_control.model.configure_index_request_spec import (
     ConfigureIndexRequestSpec,
 )
@@ -38,19 +32,11 @@
 )
 from pinecone.core.openapi.db_control.model.byoc_spec import ByocSpec as ByocSpecModel
 from pinecone.core.openapi.db_control.model.pod_spec import PodSpec as PodSpecModel
-from pinecone.core.openapi.db_control.model.pod_spec_metadata_config import (
-    PodSpecMetadataConfig,
-)
+from pinecone.core.openapi.db_control.model.pod_spec_metadata_config import PodSpecMetadataConfig
 from pinecone.core.openapi.db_control.model.create_index_from_backup_request import (
     CreateIndexFromBackupRequest,
 )
-from pinecone.db_control.models import (
-    ServerlessSpec,
-    PodSpec,
-    ByocSpec,
-    IndexModel,
-    IndexEmbed,
-)
+from pinecone.db_control.models import ServerlessSpec, PodSpec, ByocSpec, IndexModel, IndexEmbed
 
 from pinecone.db_control.enums import (
     Metric,
@@ -91,30 +77,18 @@ def __parse_deletion_protection(
         if deletion_protection in ["enabled", "disabled"]:
             return DeletionProtectionModel(deletion_protection)
         else:
-            raise ValueError(
-                "deletion_protection must be either 'enabled' or 'disabled'"
-            )
+            raise ValueError("deletion_protection must be either 'enabled' or 'disabled'")
 
     @staticmethod
-    def __parse_index_spec(
-        spec: Union[Dict, ServerlessSpec, PodSpec, ByocSpec],
-    ) -> IndexSpec:
+    def __parse_index_spec(spec: Union[Dict, ServerlessSpec, PodSpec, ByocSpec]) -> IndexSpec:
         if isinstance(spec, dict):
             if "serverless" in spec:
-                spec["serverless"]["cloud"] = convert_enum_to_string(
-                    spec["serverless"]["cloud"]
-                )
-                spec["serverless"]["region"] = convert_enum_to_string(
-                    spec["serverless"]["region"]
-                )
+                spec["serverless"]["cloud"] = convert_enum_to_string(spec["serverless"]["cloud"])
+                spec["serverless"]["region"] = convert_enum_to_string(spec["serverless"]["region"])
 
-                index_spec = IndexSpec(
-                    serverless=ServerlessSpecModel(**spec["serverless"])
-                )
+                index_spec = IndexSpec(serverless=ServerlessSpecModel(**spec["serverless"]))
             elif "pod" in spec:
-                spec["pod"]["environment"] = convert_enum_to_string(
-                    spec["pod"]["environment"]
-                )
+                spec["pod"]["environment"] = convert_enum_to_string(spec["pod"]["environment"])
                 args_dict = parse_non_empty_args(
                     [
                         ("environment", spec["pod"].get("environment")),
@@ -133,9 +107,7 @@ def __parse_index_spec(
             elif "byoc" in spec:
                 index_spec = IndexSpec(byoc=ByocSpecModel(**spec["byoc"]))
             else:
-                raise ValueError(
-                    "spec must contain either 'serverless', 'pod', or 'byoc' key"
-                )
+                raise ValueError("spec must contain either 'serverless', 'pod', or 'byoc' key")
         elif isinstance(spec, ServerlessSpec):
             index_spec = IndexSpec(
                 serverless=ServerlessSpecModel(cloud=spec.cloud, region=spec.region)
@@ -155,17 +127,13 @@ def __parse_index_spec(
                 )
 
             index_spec = IndexSpec(
-                pod=PodSpecModel(
-                    environment=spec.environment, pod_type=spec.pod_type, **args_dict
-                )
+                pod=PodSpecModel(environment=spec.environment, pod_type=spec.pod_type, **args_dict)
             )
         elif isinstance(spec, ByocSpec):
             args_dict = parse_non_empty_args([("environment", spec.environment)])
             index_spec = IndexSpec(byoc=ByocSpecModel(**args_dict))
         else:
-            raise TypeError(
-                "spec must be of type dict, ServerlessSpec, PodSpec, or ByocSpec"
-            )
+            raise TypeError("spec must be of type dict, ServerlessSpec, PodSpec, or ByocSpec")
 
         return index_spec
 
@@ -175,9 +143,7 @@ def create_index_request(
         spec: Union[Dict, ServerlessSpec, PodSpec, ByocSpec],
         dimension: Optional[int] = None,
         metric: Optional[Union[Metric, str]] = Metric.COSINE,
-        deletion_protection: Optional[
-            Union[DeletionProtection, str]
-        ] = DeletionProtection.DISABLED,
+        deletion_protection: Optional[Union[DeletionProtection, str]] = DeletionProtection.DISABLED,
         vector_type: Optional[Union[VectorType, str]] = VectorType.DENSE,
         tags: Optional[Dict[str, str]] = None,
     ) -> CreateIndexRequest:
@@ -186,9 +152,7 @@ def create_index_request(
         if vector_type is not None:
             vector_type = convert_enum_to_string(vector_type)
         if deletion_protection is not None:
-            dp = PineconeDBControlRequestFactory.__parse_deletion_protection(
-                deletion_protection
-            )
+            dp = PineconeDBControlRequestFactory.__parse_deletion_protection(deletion_protection)
         else:
             dp = None
 
@@ -219,16 +183,12 @@ def create_index_for_model_request(
         region: Union[AwsRegion, GcpRegion, AzureRegion, str],
         embed: Union[IndexEmbed, CreateIndexForModelEmbedTypedDict],
         tags: Optional[Dict[str, str]] = None,
-        deletion_protection: Optional[
-            Union[DeletionProtection, str]
-        ] = DeletionProtection.DISABLED,
+        deletion_protection: Optional[Union[DeletionProtection, str]] = DeletionProtection.DISABLED,
     ) -> CreateIndexForModelRequest:
         cloud = convert_enum_to_string(cloud)
         region = convert_enum_to_string(region)
         if deletion_protection is not None:
-            dp = PineconeDBControlRequestFactory.__parse_deletion_protection(
-                deletion_protection
-            )
+            dp = PineconeDBControlRequestFactory.__parse_deletion_protection(deletion_protection)
         else:
             dp = None
         tags_obj = PineconeDBControlRequestFactory.__parse_tags(tags)
@@ -265,23 +225,17 @@ def create_index_for_model_request(
     @staticmethod
     def create_index_from_backup_request(
         name: str,
-        deletion_protection: Optional[
-            Union[DeletionProtection, str]
-        ] = DeletionProtection.DISABLED,
+        deletion_protection: Optional[Union[DeletionProtection, str]] = DeletionProtection.DISABLED,
         tags: Optional[Dict[str, str]] = None,
     ) -> CreateIndexFromBackupRequest:
         if deletion_protection is not None:
-            dp = PineconeDBControlRequestFactory.__parse_deletion_protection(
-                deletion_protection
-            )
+            dp = PineconeDBControlRequestFactory.__parse_deletion_protection(deletion_protection)
         else:
             dp = None
 
         tags_obj = PineconeDBControlRequestFactory.__parse_tags(tags)
 
-        return CreateIndexFromBackupRequest(
-            name=name, deletion_protection=dp, tags=tags_obj
-        )
+        return CreateIndexFromBackupRequest(name=name, deletion_protection=dp, tags=tags_obj)
 
     @staticmethod
     def configure_index_request(
@@ -299,9 +253,7 @@ def configure_index_request(
         elif deletion_protection in ["enabled", "disabled"]:
             dp = DeletionProtectionModel(deletion_protection)
         else:
-            raise ValueError(
-                "deletion_protection must be either 'enabled' or 'disabled'"
-            )
+            raise ValueError("deletion_protection must be either 'enabled' or 'disabled'")
 
         fetched_tags = description.tags
         if fetched_tags is None:
@@ -329,9 +281,7 @@ def configure_index_request(
 
         spec = None
         if pod_config_args:
-            spec = ConfigureIndexRequestSpec(
-                pod=ConfigureIndexRequestSpecPod(**pod_config_args)
-            )
+            spec = ConfigureIndexRequestSpec(pod=ConfigureIndexRequestSpecPod(**pod_config_args))
 
         args_dict = parse_non_empty_args(
             [

pinecone/db_control/resources/asyncio/index.py

@@ -1,6 +1,6 @@
 import logging
 import asyncio
-from typing import Optional, Dict, Union
+from typing import Optional, Dict, Union, Any
 
 
 from pinecone.db_control.models import (
@@ -47,9 +47,7 @@ async def create(
         dimension: Optional[int] = None,
         metric: Optional[Union[Metric, str]] = Metric.COSINE,
         timeout: Optional[int] = None,
-        deletion_protection: Optional[
-            Union[DeletionProtection, str]
-        ] = DeletionProtection.DISABLED,
+        deletion_protection: Optional[Union[DeletionProtection, str]] = DeletionProtection.DISABLED,
         vector_type: Optional[Union[VectorType, str]] = VectorType.DENSE,
         tags: Optional[Dict[str, str]] = None,
     ) -> IndexModel:
@@ -77,9 +75,7 @@ async def create_for_model(
         region: Union[AwsRegion, GcpRegion, AzureRegion, str],
         embed: Union[IndexEmbed, CreateIndexForModelEmbedTypedDict],
         tags: Optional[Dict[str, str]] = None,
-        deletion_protection: Optional[
-            Union[DeletionProtection, str]
-        ] = DeletionProtection.DISABLED,
+        deletion_protection: Optional[Union[DeletionProtection, str]] = DeletionProtection.DISABLED,
         timeout: Optional[int] = None,
     ) -> IndexModel:
         req = PineconeDBControlRequestFactory.create_index_for_model_request(
@@ -102,9 +98,7 @@ async def create_from_backup(
         *,
         name: str,
         backup_id: str,
-        deletion_protection: Optional[
-            Union[DeletionProtection, str]
-        ] = DeletionProtection.DISABLED,
+        deletion_protection: Optional[Union[DeletionProtection, str]] = DeletionProtection.DISABLED,
         tags: Optional[Dict[str, str]] = None,
         timeout: Optional[int] = None,
     ) -> IndexModel:
@@ -116,9 +110,7 @@ async def create_from_backup(
         )
         return await self.__poll_describe_index_until_ready(name, timeout)
 
-    async def __poll_describe_index_until_ready(
-        self, name: str, timeout: Optional[int] = None
-    ):
+    async def __poll_describe_index_until_ready(self, name: str, timeout: Optional[int] = None):
         total_wait_time = 0
         while True:
             description = await self.describe(name=name)

pinecone/db_control/resources/sync/index.py

@@ -4,11 +4,7 @@
 
 from pinecone.db_control.index_host_store import IndexHostStore
 
-from pinecone.db_control.models import (
-    IndexModel,
-    IndexList,
-    IndexEmbed,
-)
+from pinecone.db_control.models import IndexModel, IndexList, IndexEmbed
 from pinecone.utils import docslinks, require_kwargs, PluginAware
 
 from pinecone.db_control.types import CreateIndexForModelEmbedTypedDict
@@ -149,9 +145,7 @@ def create_from_backup(
             return self.describe(name=name)
         return self.__poll_describe_index_until_ready(name, timeout)
 
-    def __poll_describe_index_until_ready(
-        self, name: str, timeout: Optional[int] = None
-    ):
+    def __poll_describe_index_until_ready(self, name: str, timeout: Optional[int] = None):
         total_wait_time = 0
         while True:
             description = self.describe(name=name)

pinecone/legacy_pinecone_interface.py

@@ -26,10 +26,7 @@
         GcpRegion,
         AzureRegion,
     )
-    from pinecone.db_control.types import (
-        CreateIndexForModelEmbedTypedDict,
-        ConfigureIndexEmbed,
-    )
+    from pinecone.db_control.types import CreateIndexForModelEmbedTypedDict, ConfigureIndexEmbed
 
 
 class LegacyPineconeDBControlInterface(ABC):

pinecone/pinecone.py

@@ -18,10 +18,7 @@
     from pinecone.db_data import _Index as Index, _IndexAsyncio as IndexAsyncio
     from pinecone.db_control.index_host_store import IndexHostStore
     from pinecone.core.openapi.db_control.api.manage_indexes_api import ManageIndexesApi
-    from pinecone.db_control.types import (
-        CreateIndexForModelEmbedTypedDict,
-        ConfigureIndexEmbed,
-    )
+    from pinecone.db_control.types import CreateIndexForModelEmbedTypedDict, ConfigureIndexEmbed
     from pinecone.db_control.enums import (
         Metric,
         VectorType,
@@ -228,9 +225,7 @@ def __init__(
         )
         """ :meta private: """
 
-        self._openapi_config = ConfigBuilder.build_openapi_config(
-            self._config, **kwargs
-        )
+        self._openapi_config = ConfigBuilder.build_openapi_config(self._config, **kwargs)
         """ :meta private: """
 
         if pool_threads is None: