diff --git a/chromadb/segment/impl/metadata/grpc_segment.py b/chromadb/segment/impl/metadata/grpc_segment.py index b08f8ba948bb..54718a0284f5 100644 --- a/chromadb/segment/impl/metadata/grpc_segment.py +++ b/chromadb/segment/impl/metadata/grpc_segment.py @@ -58,7 +58,6 @@ def count(self, request_version_context: RequestVersionContext) -> int: response: pb.CountRecordsResponse = self._metadata_reader_stub.CountRecords( request, timeout=self._request_timeout_seconds, - version_context=to_proto_request_version_context(request_version_context), ) return response.count @@ -108,12 +107,12 @@ def get_metadata( limit=limit, offset=offset, include_metadata=include_metadata, + version_context=to_proto_request_version_context(request_version_context), ) response: pb.QueryMetadataResponse = self._metadata_reader_stub.QueryMetadata( request, timeout=self._request_timeout_seconds, - version_context=to_proto_request_version_context(request_version_context), ) results: List[MetadataEmbeddingRecord] = [] for record in response.records: diff --git a/chromadb/segment/impl/vector/grpc_segment.py b/chromadb/segment/impl/vector/grpc_segment.py index f253dafc1ac1..576409b7bc11 100644 --- a/chromadb/segment/impl/vector/grpc_segment.py +++ b/chromadb/segment/impl/vector/grpc_segment.py @@ -65,11 +65,11 @@ def get_vectors( ids=ids, segment_id=self._segment["id"].hex, collection_id=self._segment["collection"].hex, + version_context=to_proto_request_version_context(request_version_context), ) response: GetVectorsResponse = self._vector_reader_stub.GetVectors( request, timeout=self._request_timeout_seconds, - version_context=to_proto_request_version_context(request_version_context), ) results: List[VectorEmbeddingRecord] = [] for vector in response.records: