From 7fc8014e598901c5c4e067b413c8b01a178798c1 Mon Sep 17 00:00:00 2001 From: robgeada Date: Tue, 17 Oct 2023 15:43:42 +0100 Subject: [PATCH] Revert PayloadProcessor ModelID to original value --- src/main/java/com/ibm/watson/modelmesh/ModelMeshApi.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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(); }