From 340f703ff58703dc22df44a0c19ff4ba82bc430d Mon Sep 17 00:00:00 2001 From: Weves Date: Wed, 1 Nov 2023 23:20:55 -0700 Subject: [PATCH] test --- backend/danswer/document_index/vespa/index.py | 1 + backend/scripts/benchmark_search_isolated.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/danswer/document_index/vespa/index.py b/backend/danswer/document_index/vespa/index.py index b18f7909ec..b2ac3c8a08 100644 --- a/backend/danswer/document_index/vespa/index.py +++ b/backend/danswer/document_index/vespa/index.py @@ -451,6 +451,7 @@ def _query_vespa(query_params: Mapping[str, str | int]) -> list[InferenceChunk]: response_json = response.json() logger.debug("Response: %s", response_json) + logger.info("timing info: %s", response_json.get("timing")) hits = response_json["root"].get("children", []) for hit in hits: diff --git a/backend/scripts/benchmark_search_isolated.py b/backend/scripts/benchmark_search_isolated.py index 73fb447156..af08c1683b 100644 --- a/backend/scripts/benchmark_search_isolated.py +++ b/backend/scripts/benchmark_search_isolated.py @@ -141,7 +141,7 @@ def _measure_vespa_latency(filters: dict = {}): ) query = generate_random_sentence() query_embedding = embed_query(query) - num_to_retrieve = 50 + num_to_retrieve = 1 params: dict[str, str | int] = { "yql": yql, "query": query,