(
request: GetOnlineDocumentsRequest,
)
| 462 | response_model=OnlineFeaturesResponse, |
| 463 | ) |
| 464 | async def retrieve_online_documents( |
| 465 | request: GetOnlineDocumentsRequest, |
| 466 | ) -> Any: |
| 467 | with feast_metrics.track_request_latency("/retrieve-online-documents"): |
| 468 | logger.warning( |
| 469 | "This endpoint is in alpha and will be moved to /get-online-features when stable." |
| 470 | ) |
| 471 | features = await _get_features(request, store) |
| 472 | |
| 473 | read_params = dict( |
| 474 | features=features, |
| 475 | query=request.query, |
| 476 | top_k=request.top_k, |
| 477 | ) |
| 478 | if request.api_version == 2 and request.query_string is not None: |
| 479 | read_params["query_string"] = request.query_string |
| 480 | |
| 481 | if request.api_version == 2: |
| 482 | read_params["include_feature_view_version_metadata"] = ( |
| 483 | request.include_feature_view_version_metadata |
| 484 | ) |
| 485 | response = await run_in_threadpool( |
| 486 | lambda: store.retrieve_online_documents_v2(**read_params) # type: ignore |
| 487 | ) |
| 488 | else: |
| 489 | response = await run_in_threadpool( |
| 490 | lambda: store.retrieve_online_documents(**read_params) # type: ignore |
| 491 | ) |
| 492 | |
| 493 | response_dict = await run_in_threadpool( |
| 494 | convert_response_to_dict, response.proto |
| 495 | ) |
| 496 | return JSONResponse(content=response_dict) |
| 497 | |
| 498 | @app.post("/push", dependencies=[Depends(inject_user_details)]) |
| 499 | async def push(request: PushFeaturesRequest) -> Response: |
nothing calls this directly
no test coverage detected