From 31079709975bea127433ac1c0a44a0687868a7f0 Mon Sep 17 00:00:00 2001 From: Georgios Schinas Date: Tue, 25 Nov 2025 18:37:25 +0000 Subject: [PATCH] fix: return embedding metadata if available --- .../matching_engine/matching_engine_index_endpoint.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/google/cloud/aiplatform/matching_engine/matching_engine_index_endpoint.py b/google/cloud/aiplatform/matching_engine/matching_engine_index_endpoint.py index b4737dd488..5e520feff8 100644 --- a/google/cloud/aiplatform/matching_engine/matching_engine_index_endpoint.py +++ b/google/cloud/aiplatform/matching_engine/matching_engine_index_endpoint.py @@ -276,6 +276,9 @@ def from_index_datapoint( self.sparse_embedding_dimensions = ( index_datapoint.sparse_embedding.dimensions ) + # retrieve embedding metadata + if (index_datapoint.embedding_metadata is not None): + self.embedding_metadata = index_datapoint.embedding_metadata return self def from_embedding(self, embedding: match_service_pb2.Embedding) -> "MatchNeighbor":