diff --git a/docs/index.rst b/docs/index.rst index a32202b9..d3e8d6e5 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -11,7 +11,7 @@ This standard client (Client) specializes in performing database operations with Moreover, the standard client supports Hierarchical Navigable Small World (HNSW) vector searches, allowing users to find vectors similar to a given query vector within an index. -This admin client (AdminClient) is designed to conduct Proximus administrative operation such +This admin client (AdminClient) is designed to conduct AVS administrative operation such as creating indexes, querying index information, and dropping indexes. Please explore the modules below for more information on API usage and details. diff --git a/pyproject.toml b/pyproject.toml index c5683e9d..49fcad2a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "aerospike-vector-search" -description = "Aerospike Proximus Client Library for Python" +description = "Aerospike Vector Search Client Library for Python" authors = [ { name = "Aerospike, Inc.", email = "info@aerospike.com" } ] @@ -22,7 +22,7 @@ classifiers = [ "Programming Language :: Python :: Implementation :: CPython", "Topic :: Database" ] -version = "0.6.0" +version = "0.6.1" requires-python = ">3.8" dependencies = [ "grpcio", diff --git a/src/aerospike_vector_search/aio/client.py b/src/aerospike_vector_search/aio/client.py index 124ad307..7c248c6b 100644 --- a/src/aerospike_vector_search/aio/client.py +++ b/src/aerospike_vector_search/aio/client.py @@ -338,15 +338,10 @@ async def vector_search( ) try: - results_stream = transact_stub.VectorSearch(vector_search_request) + return [self._respond_neighbor(result) async for result in transact_stub.VectorSearch(vector_search_request)] except grpc.RpcError as e: logger.error("Failed with error: %s", e) raise types.AVSServerError(rpc_error=e) - async_results = [] - async for result in results_stream: - async_results.append(self._respond_neighbor(result)) - - return async_results async def wait_for_index_completion( self, diff --git a/src/aerospike_vector_search/client.py b/src/aerospike_vector_search/client.py index 5ce843f4..f9b7d783 100644 --- a/src/aerospike_vector_search/client.py +++ b/src/aerospike_vector_search/client.py @@ -313,15 +313,10 @@ def vector_search( ) try: - results_stream = transact_stub.VectorSearch(vector_search_request) + return [self._respond_neighbor(result) for result in transact_stub.VectorSearch(vector_search_request)] except grpc.RpcError as e: logger.error("Failed with error: %s", e) raise types.AVSServerError(rpc_error=e) - results = [] - for result in results_stream: - results.append(self._respond_neighbor(result)) - - return results def wait_for_index_completion( self, diff --git a/src/aerospike_vector_search/shared/base_channel_provider.py b/src/aerospike_vector_search/shared/base_channel_provider.py index 0b923be3..39de3682 100644 --- a/src/aerospike_vector_search/shared/base_channel_provider.py +++ b/src/aerospike_vector_search/shared/base_channel_provider.py @@ -22,7 +22,7 @@ def __init__( class BaseChannelProvider(object): - """Proximus Channel Provider""" + """AVS Channel Provider""" def __init__( self,