fix: Add ModelStreamInfer to triton MethodInfos (#80)

Legion2 · Legion2 · commit 81f6d8b8bf6f · 2024-01-21T16:21:39.000+01:00
diff --git 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
 
