Add null check to vModelId

Signed-off-by: Rob Geada <[email protected]>
kserve · Oct 17, 2023 · bc1c6ec · bc1c6ec
1 parent b475614
commit bc1c6ec
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, vModelId, 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, vModelId, 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();
                     }