mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-23 00:42:28 +00:00
af4a44fcbd
* Add rest api endpoint to delete documents by filter. * Remove parametrization of rest api tests * Make the paths in rest_api/config.py absolute * Fix path to pipelines.yaml * Restructuring test_rest_api.py to be able to test only my endpoint (and to make the suite more structured) * Convert DELETE /documents into POST /documents/delete_by_filters Co-authored by: sarthakj2109 <54064348+sarthakj2109@users.noreply.github.com>
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import logging
|
|
import time
|
|
|
|
from fastapi import APIRouter
|
|
|
|
from rest_api.application import PIPELINE
|
|
from rest_api.config import LOG_LEVEL, CONCURRENT_REQUEST_PER_WORKER
|
|
from rest_api.schema import QueryRequest, QueryResponse
|
|
from rest_api.controller.utils import RequestLimiter
|
|
|
|
|
|
logging.getLogger("haystack").setLevel(LOG_LEVEL)
|
|
logger = logging.getLogger("haystack")
|
|
|
|
|
|
router = APIRouter()
|
|
concurrency_limiter = RequestLimiter(CONCURRENT_REQUEST_PER_WORKER)
|
|
|
|
|
|
@router.get("/initialized")
|
|
def initialized():
|
|
"""
|
|
This endpoint can be used during startup to understand if the
|
|
server is ready to take any requests, or is still loading.
|
|
|
|
The recommended approach is to call this endpoint with a short timeout,
|
|
like 500ms, and in case of no reply, consider the server busy.
|
|
"""
|
|
return True
|
|
|
|
|
|
@router.post("/query", response_model=QueryResponse)
|
|
def query(request: QueryRequest):
|
|
with concurrency_limiter.run():
|
|
result = _process_request(PIPELINE, request)
|
|
return result
|
|
|
|
|
|
def _process_request(pipeline, request) -> QueryResponse:
|
|
start_time = time.time()
|
|
|
|
params = request.params or {}
|
|
params["filters"] = params.get("filters") or {}
|
|
filters = {}
|
|
if "filters" in params: # put filter values into a list and remove filters with null value
|
|
for key, values in params["filters"].items():
|
|
if values is None:
|
|
continue
|
|
if not isinstance(values, list):
|
|
values = [values]
|
|
filters[key] = values
|
|
params["filters"] = filters
|
|
result = pipeline.run(query=request.query, params=params)
|
|
end_time = time.time()
|
|
logger.info({"request": request.dict(), "response": result, "time": f"{(end_time - start_time):.2f}"})
|
|
|
|
return result
|