Fix typo

chroma-core · Dec 6, 2024 · 1b3cd98 · 1b3cd98
1 parent 99df1c6
commit 1b3cd98
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/chromadb/execution/executor/distributed.py b/chromadb/execution/executor/distributed.py
@@ -187,8 +187,8 @@ def _segment_scan(self, scan: Scan) -> SegmentScan:
         return SegmentScan(
             collection=collection_segments["collection"],
             knn_id=scope_to_segment[SegmentScope.VECTOR],
-            metadata_id=scope_to_segment[SegmentScope.VECTOR],
-            record_id=scope_to_segment[SegmentScope.VECTOR],
+            metadata_id=scope_to_segment[SegmentScope.METADATA],
+            record_id=scope_to_segment[SegmentScope.RECORD],
         )
 
     def _grpc_executuor_stub(self, scan: Scan) -> QueryExecutorStub:

diff --git a/chromadb/segment/impl/manager/distributed.py b/chromadb/segment/impl/manager/distributed.py
@@ -88,7 +88,7 @@ def delete_segments(self, collection_id: UUID) -> Sequence[UUID]:
     def get_collection_segments(self, collection_id: UUID) -> CollectionSegments:
         if collection_id not in self._collection_segment_cache:
             self._collection_segment_cache[collection_id] = self._sysdb.get_collection_with_segments(collection_id)
-        return self._segment_cache[collection_id]
+        return self._collection_segment_cache[collection_id]
 
     @trace_method(
         "DistributedSegmentManager.get_endpoint",