From ca397c31210b61c2af46f5380e832d0eba1c41c8 Mon Sep 17 00:00:00 2001 From: Weves Date: Mon, 30 Oct 2023 16:15:45 -0700 Subject: [PATCH] Add more logging --- backend/danswer/document_index/vespa/index.py | 2 ++ backend/danswer/search/search_runner.py | 1 + 2 files changed, 3 insertions(+) diff --git a/backend/danswer/document_index/vespa/index.py b/backend/danswer/document_index/vespa/index.py index ac1c655ea..ed9827745 100644 --- a/backend/danswer/document_index/vespa/index.py +++ b/backend/danswer/document_index/vespa/index.py @@ -58,6 +58,7 @@ from danswer.search.search_runner import query_processing from danswer.search.search_runner import remove_stop_words from danswer.utils.batching import batch_generator from danswer.utils.logger import setup_logger +from danswer.utils.timing import log_function_time logger = setup_logger() @@ -429,6 +430,7 @@ def _vespa_hit_to_inference_chunk(hit: dict[str, Any]) -> InferenceChunk: ) +@log_function_time() def _query_vespa(query_params: Mapping[str, str | int]) -> list[InferenceChunk]: if "query" in query_params and not cast(str, query_params["query"]).strip(): raise ValueError("No/empty query received") diff --git a/backend/danswer/search/search_runner.py b/backend/danswer/search/search_runner.py index 057e7752c..2a9eade69 100644 --- a/backend/danswer/search/search_runner.py +++ b/backend/danswer/search/search_runner.py @@ -61,6 +61,7 @@ def query_processing( return query +@log_function_time() def embed_query( query: str, embedding_model: SentenceTransformer | None = None,