Rename _Index class back to Index

pinecone-io · Dec 16, 2024 · 8e0de62 · 8e0de62
1 parent bb9d9c4
commit 8e0de62
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 29 deletions.
diff --git a/pinecone/data/__init__.py b/pinecone/data/__init__.py
@@ -1,4 +1,12 @@
-from .index import *
+from .index import (
+    Index as _Index,
+    FetchResponse,
+    QueryResponse,
+    DescribeIndexStatsResponse,
+    UpsertResponse,
+    SparseValues,
+    Vector,
+)
 from .import_error import Index, IndexClientInstantiationError
 from .errors import (
     VectorDictionaryMissingKeysError,

diff --git a/pinecone/data/index.py b/pinecone/data/index.py
@@ -11,18 +11,10 @@
 from pinecone.core.openapi.db_data import API_VERSION
 from pinecone.core.openapi.db_data.models import (
     FetchResponse,
-    QueryRequest,
     QueryResponse,
-    RpcStatus,
-    ScoredVector,
-    SingleQueryResults,
     IndexDescription as DescribeIndexStatsResponse,
-    UpsertRequest,
     UpsertResponse,
     Vector,
-    DeleteRequest,
-    UpdateRequest,
-    DescribeIndexStatsRequest,
     ListResponse,
     SparseValues,
 )
@@ -45,32 +37,13 @@
 
 logger = logging.getLogger(__name__)
 
-__all__ = [
-    "_Index",
-    "FetchResponse",
-    "QueryRequest",
-    "QueryResponse",
-    "RpcStatus",
-    "ScoredVector",
-    "SingleQueryResults",
-    "DescribeIndexStatsResponse",
-    "UpsertRequest",
-    "UpsertResponse",
-    "UpdateRequest",
-    "Vector",
-    "DeleteRequest",
-    "UpdateRequest",
-    "DescribeIndexStatsRequest",
-    "SparseValues",
-]
-
 
 def parse_query_response(response: QueryResponse):
     response._data_store.pop("results", None)
     return response
 
 
-class _Index(IndexInterface, ImportFeatureMixin):
+class Index(IndexInterface, ImportFeatureMixin):
     """
     A client for interacting with a Pinecone index via REST API.
     For improved performance, use the Pinecone GRPC index client.