MCPcopy
hub / github.com/feast-dev/feast / retrieve_online_documents

Function retrieve_online_documents

sdk/python/feast/feature_server.py:464–496  ·  view source on GitHub ↗
(
        request: GetOnlineDocumentsRequest,
    )

Source from the content-addressed store, hash-verified

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:

Callers

nothing calls this directly

Calls 3

_get_featuresFunction · 0.70

Tested by

no test coverage detected