fix incorrect parameter

chroma-core · Sep 25, 2024 · fe32510 · fe32510
1 parent c6b27fc
commit fe32510
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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: