From 3fe319ab20ab4609669c3123ba821f19f8a747c2 Mon Sep 17 00:00:00 2001 From: Tanay Soni Date: Wed, 22 Apr 2020 11:28:23 +0200 Subject: [PATCH] Log results in APM (#70) --- haystack/api/controller/model.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/haystack/api/controller/model.py b/haystack/api/controller/model.py index 0e08f89d4..3ccb65eea 100644 --- a/haystack/api/controller/model.py +++ b/haystack/api/controller/model.py @@ -2,6 +2,7 @@ import logging from datetime import datetime from typing import List, Dict, Optional +import elasticapm from fastapi import APIRouter from fastapi import HTTPException from pydantic import BaseModel @@ -119,7 +120,8 @@ def doc_qa(model_id: int, request: Question): ) results.append(result) - logger.info({"request": request.json(), "results": results}) + elasticapm.set_custom_context({"results": results}) + logger.info({"request": request.json(), "results": results}) return {"results": results} @@ -144,6 +146,7 @@ def faq_qa(model_id: int, request: Question): ) results.append(result) - logger.info({"request": request.json(), "results": results}) + elasticapm.set_custom_context({"results": results}) + logger.info({"request": request.json(), "results": results}) - return {"results": results} + return {"results": results}