Skip to content

Commit 70ad223

Browse files
committed
remove useless log
Signed-off-by: xinyual <xinyual@amazon.com>
1 parent 20aa5bb commit 70ad223

File tree

3 files changed

+1
-15
lines changed

3 files changed

+1
-15
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java

+1-9
Original file line numberDiff line numberDiff line change
@@ -629,15 +629,7 @@ private void runReAct(
629629
listener.onResponse(ModelTensorOutput.builder().mlModelOutputs(finalModelTensors).build());
630630
}
631631
} else {
632-
ActionRequest request2 = new MLPredictionTaskRequest(
633-
llm.getModelId(),
634-
RemoteInferenceMLInput
635-
.builder()
636-
.algorithm(FunctionName.REMOTE)
637-
.inputDataset(RemoteInferenceInputDataSet.builder().parameters(tmpParameters).build())
638-
.build()
639-
);
640-
client.execute(MLPredictionTaskAction.INSTANCE, request2, (ActionListener<MLTaskResponse>) nextStepListener);
632+
client.execute(MLPredictionTaskAction.INSTANCE, request, (ActionListener<MLTaskResponse>) nextStepListener);
641633
}
642634
}
643635
}, e -> {

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/AgentTool.java

-4
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,7 @@ public AgentTool(Client client, String agentId) {
5353

5454
@Override
5555
public <T> void run(Map<String, String> parameters, ActionListener<T> listener) {
56-
log.info("Agent tool before");
57-
log.info(parameters);
5856
parameters = extractFromChatParameters(parameters);
59-
log.info("Agent tool after");
60-
log.info(parameters);
6157
AgentMLInput agentMLInput = AgentMLInput
6258
.AgentMLInputBuilder()
6359
.agentId(agentId)

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/MLModelTool.java

-2
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,6 @@ public MLModelTool(Client client, String modelId) {
6565

6666
@Override
6767
public <T> void run(Map<String, String> parameters, ActionListener<T> listener) {
68-
log.info("ML input");
69-
log.info(parameters);
7068
RemoteInferenceInputDataSet inputDataSet = RemoteInferenceInputDataSet.builder().parameters(parameters).build();
7169
ActionRequest request = new MLPredictionTaskRequest(
7270
modelId,

0 commit comments

Comments
 (0)