From 56aaae3d2cec7a06fc4b932f7c8b3659269f7841 Mon Sep 17 00:00:00 2001 From: Sicheng Pan Date: Fri, 13 Dec 2024 10:30:49 -0800 Subject: [PATCH] Send both segment id and full info for backward compat --- chromadb/proto/convert.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/chromadb/proto/convert.py b/chromadb/proto/convert.py index 51d30bc608a..2a18475cf98 100644 --- a/chromadb/proto/convert.py +++ b/chromadb/proto/convert.py @@ -573,6 +573,9 @@ def to_proto_where_document(where_document: WhereDocument) -> chroma_pb.WhereDoc def to_proto_scan(scan: Scan) -> query_pb.ScanOperator: return query_pb.ScanOperator( collection=to_proto_collection(scan.collection), + knn_id=scan.knn["id"].hex, + metadata_id=scan.metadata["id"].hex, + record_id=scan.record["id"].hex, knn=to_proto_segment(scan.knn), metadata=to_proto_segment(scan.metadata), record=to_proto_segment(scan.record),