Log results in APM (#70)

This commit is contained in:
Tanay Soni 2020-04-22 11:28:23 +02:00 committed by GitHub
parent ce4adf3fd2
commit 3fe319ab20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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