diff --git a/src/main/java/com/ibm/watson/modelmesh/ModelMeshApi.java b/src/main/java/com/ibm/watson/modelmesh/ModelMeshApi.java index 715c0efe..a58f5039 100644 --- a/src/main/java/com/ibm/watson/modelmesh/ModelMeshApi.java +++ b/src/main/java/com/ibm/watson/modelmesh/ModelMeshApi.java @@ -766,8 +766,9 @@ public void onHalfClose() { } } finally { if (payloadProcessor != null) { + String payloadModelId = vModelId == null ? resolvedModelId : vModelId; processPayload(reqMessage.readerIndex(reqReaderIndex), - requestId, resolvedModelId, methodName, headers, null, true); + requestId, payloadModelId, methodName, headers, null, true); } else { releaseReqMessage(); } @@ -803,7 +804,8 @@ public void onHalfClose() { data = response.data.readerIndex(respReaderIndex); metadata = response.metadata; } - processPayload(data, requestId, resolvedModelId, methodName, metadata, status, releaseResponse); + String payloadModelId = vModelId == null ? resolvedModelId : vModelId; + processPayload(data, requestId, payloadModelId, methodName, metadata, status, releaseResponse); } else if (releaseResponse && response != null) { response.release(); }