Dummy impl for local SysDB

chroma-core · Dec 6, 2024 · 99df1c6 · 99df1c6
1 parent e8e9a6c
commit 99df1c6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/chromadb/db/mixins/sysdb.py b/chromadb/db/mixins/sysdb.py
@@ -24,6 +24,7 @@
 )
 from chromadb.ingest import Producer
 from chromadb.types import (
+    CollectionSegments,
     Database,
     OptionalArgument,
     Segment,
@@ -488,6 +489,13 @@ def get_collections(
 
             return collections
 
+    @override
+    def get_collection_with_segments(self, collection_id: UUID) -> CollectionSegments:
+        return CollectionSegments(
+            collection=self.get_collections(id=collection_id)[0],
+            segments=self.get_segments(collection=collection_id),
+        )
+
     @trace_method("SqlSysDB.delete_segment", OpenTelemetryGranularity.ALL)
     @override
     def delete_segment(self, collection: UUID, id: UUID) -> None: