diff --git a/pinecone/data/__init__.py b/pinecone/data/__init__.py index 02a67767..a6a791fc 100644 --- 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 index 90f7f827..9f52589c 100644 --- 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.