From 81f6d8b8bf6fe51998d5efbdb24296a70a3188e2 Mon Sep 17 00:00:00 2001 From: Leon Kiefer Date: Sun, 21 Jan 2024 01:38:42 +0100 Subject: [PATCH] fix: Add ModelStreamInfer to triton MethodInfos (#80) --- model-mesh-triton-adapter/server/server.go | 1 + 1 file changed, 1 insertion(+) diff --git a/model-mesh-triton-adapter/server/server.go b/model-mesh-triton-adapter/server/server.go index 6b831f14..f79fd3b2 100644 --- a/model-mesh-triton-adapter/server/server.go +++ b/model-mesh-triton-adapter/server/server.go @@ -266,6 +266,7 @@ func (s *TritonAdapterServer) RuntimeStatus(ctx context.Context, req *mmesh.Runt mis := make(map[string]*mmesh.RuntimeStatusResponse_MethodInfo) mis[tritonServiceName+"/ModelInfer"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1} + mis[tritonServiceName+"/ModelStreamInfer"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1} mis[tritonServiceName+"/ModelMetadata"] = &mmesh.RuntimeStatusResponse_MethodInfo{IdInjectionPath: path1} runtimeStatus.MethodInfos = mis