From ff7e35581b7f22ae6a2932a06d70b35be6f5bc09 Mon Sep 17 00:00:00 2001 From: Tanay Soni Date: Tue, 7 Jul 2020 12:28:41 +0200 Subject: [PATCH] Add response time in logs (#201) --- rest_api/controller/search.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/rest_api/controller/search.py b/rest_api/controller/search.py index d6f1e13cc..628d8da0b 100644 --- a/rest_api/controller/search.py +++ b/rest_api/controller/search.py @@ -1,4 +1,5 @@ import logging +import time from datetime import datetime from typing import List, Dict, Optional @@ -111,6 +112,7 @@ doc_qa_limiter = RequestLimiter(CONCURRENT_REQUEST_PER_WORKER) @router.post("/models/{model_id}/doc-qa", response_model=Answers, response_model_exclude_unset=True) def doc_qa(model_id: int, request: Question): with doc_qa_limiter.run(): + start_time = time.time() finder = FINDERS.get(model_id, None) if not finder: raise HTTPException( @@ -135,7 +137,8 @@ def doc_qa(model_id: int, request: Question): results.append(result) elasticapm.set_custom_context({"results": results}) - logger.info({"request": request.json(), "results": results}) + end_time = time.time() + logger.info({"request": request.json(), "results": results, "time": f"{(end_time - start_time):.2f}"}) return {"results": results}