yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 13:06:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add response time in logs (#201)

Browse Source
This commit is contained in:
Tanay Soni 2020-07-07 12:28:41 +02:00 committed by GitHub
parent 68d604d82b
commit ff7e35581b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rest_api/controller/search.py
View File

@ -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}