Skip to content

Commit 76238c6

Browse files
Merge pull request #91 from FederatedAI/develop-2.0.3
fix bug, data security issues
2 parents ff78e7b + 6b168e5 commit 76238c6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/provider/GuestSingleInferenceProvider.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,11 @@ public ReturnResult doService(Context context, InboundPackage inboundPackage, Ou
5353
ModelProcessor modelProcessor = model.getModelProcessor();
5454
InferenceRequest inferenceRequest = (InferenceRequest) inboundPackage.getBody();
5555
Map<String, Future> futureMap = Maps.newHashMap();
56-
List<FederatedRpcInvoker.RpcDataWraper> rpcList = this.buildRpcDataWraper(context, Dict.FEDERATED_INFERENCE, inferenceRequest);
56+
57+
InferenceRequest remoteInferenceRequest = new InferenceRequest();
58+
remoteInferenceRequest.setSendToRemoteFeatureData(inferenceRequest.getSendToRemoteFeatureData());
59+
60+
List<FederatedRpcInvoker.RpcDataWraper> rpcList = this.buildRpcDataWraper(context, Dict.FEDERATED_INFERENCE, remoteInferenceRequest);
5761
rpcList.forEach((rpcDataWraper -> {
5862
ListenableFuture<ReturnResult> future = federatedRpcInvoker.singleInferenceRpcWithCache(context, rpcDataWraper, MetaInfo.PROPERTY_REMOTE_MODEL_INFERENCE_RESULT_CACHE_SWITCH);
5963
futureMap.put(rpcDataWraper.getHostModel().getPartId(), future);

0 commit comments

Comments
 (0)