From df11a25a59956318c5b36ce704a1655ddd47f64f Mon Sep 17 00:00:00 2001 From: Xiaodong Ye Date: Sat, 6 May 2023 14:44:32 +0800 Subject: [PATCH] Add model card in response for ModelMetadata API Signed-off-by: Xiaodong Ye --- src/grpc/grpc_server.cc | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/grpc/grpc_server.cc b/src/grpc/grpc_server.cc index 9795e147f4..2414af499d 100644 --- a/src/grpc/grpc_server.cc +++ b/src/grpc/grpc_server.cc @@ -752,6 +752,15 @@ CommonHandler::RegisterModelMetadata() GOTO_IF_ERR(err, earlyexit); response->set_platform(std::string(platform, platformlen)); + if (model_metadata_json.Find("card")) { + const char* card; + size_t cardlen; + err = model_metadata_json.MemberAsString("card", &card, &cardlen); + GOTO_IF_ERR(err, earlyexit); + + response->set_card(std::string(card, cardlen)); + } + if (model_metadata_json.Find("inputs")) { triton::common::TritonJson::Value inputs_json; err = model_metadata_json.MemberAsArray("inputs", &inputs_json);