Revert PayloadProcessor ModelID to original value

kserve · Oct 17, 2023 · 7fc8014 · 7fc8014
1 parent 6630fbb
commit 7fc8014
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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();
                     }