Skip to content

Commit 9b49d40

Browse files
authored
feat: Enable ModelMetadata endpoint (#27)
MLServer and Triton both support the ModelMetadata endpoint, so we should surface it for end users. Here, the `ModelMetadata` endpoint is enabled in the adapter logic for MLServer and Triton. Ref: kserve/modelmesh-serving#136 Signed-off-by: Paul Van Eck <[email protected]>
1 parent 473fff4 commit 9b49d40

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

model-mesh-mlserver-adapter/server/server.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -526,6 +526,7 @@ func (s *MLServerAdapterServer) RuntimeStatus(ctx context.Context, req *mmesh.Ru
526526
mis := make(map[string]*mmesh.RuntimeStatusResponse_MethodInfo)
527527

528528
mis[mlserverServiceName+"/ModelInfer"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1}
529+
mis[mlserverServiceName+"/ModelMetadata"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1}
529530
runtimeStatus.MethodInfos = mis
530531

531532
log.Info("runtimeStatus", "Status", runtimeStatus)

model-mesh-triton-adapter/server/server.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,7 @@ func (s *TritonAdapterServer) RuntimeStatus(ctx context.Context, req *mmesh.Runt
264264

265265
// only support Transform for now
266266
mis[tritonServiceName+"/ModelInfer"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1}
267+
mis[tritonServiceName+"/ModelMetadata"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1}
267268
runtimeStatus.MethodInfos = mis
268269

269270
log.Info("runtimeStatus", "Status", runtimeStatus)

0 commit comments

Comments
 (0)