From 5398e60329ec759f50ef3861749fd0b2ddac1531 Mon Sep 17 00:00:00 2001 From: Owen Halpert Date: Mon, 21 Sep 2026 08:30:19 -0700 Subject: [PATCH] Read turbopuffer query result ids from the raw response --- vectordb_bench/backend/clients/turbopuffer/turbopuffer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vectordb_bench/backend/clients/turbopuffer/turbopuffer.py b/vectordb_bench/backend/clients/turbopuffer/turbopuffer.py index 0b87462d4..ce261acdc 100644 --- a/vectordb_bench/backend/clients/turbopuffer/turbopuffer.py +++ b/vectordb_bench/backend/clients/turbopuffer/turbopuffer.py @@ -440,8 +440,9 @@ def search_embedding( query_kwargs["include_attributes"] = [self._vector_field] elif payload_profile == PayloadProfile.SCALAR_LABEL: query_kwargs["include_attributes"] = [self._scalar_payload_label_field] - res = self._namespace_for_tenant(tenant).query(**query_kwargs) - return [int(row.id) for row in res.rows] if res.rows is not None else [] + raw = self._namespace_for_tenant(tenant).with_raw_response.query(**query_kwargs) + rows = loads(raw.http_response.content).get("rows") + return [int(row["id"]) for row in rows] if rows else [] def search_documents( self,