diff --git a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java b/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java index 25ff52fb..1e076a2c 100644 --- a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java +++ b/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java @@ -50,6 +50,9 @@ public void doPreProcess(Context context, InboundPackage inboundPackage, Outboun } context.setCaseId(batchInferenceRequest.getCaseid()); context.setServiceId(batchInferenceRequest.getServiceId()); + if (batchInferenceRequest.getApplyId() != null) { + context.setApplyId(batchInferenceRequest.getApplyId()); + } } catch (Exception e) { throw new GuestInvalidParamException(e.getMessage()); } diff --git a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java b/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java index 4de9babd..975697d6 100644 --- a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java +++ b/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java @@ -48,6 +48,9 @@ public void doPreProcess(Context context, InboundPackage inboundPackage, Outboun } context.setCaseId(inferenceRequest.getCaseid()); context.setServiceId(inferenceRequest.getServiceId()); + if (inferenceRequest.getApplyId() != null) { + context.setApplyId(inferenceRequest.getApplyId()); + } } catch (Exception e) { throw new GuestInvalidParamException(e.getMessage()); }