Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,15 @@
import com.volcengine.Pair;
import com.volcengine.hiagent.api.model.*;
import com.volcengine.hiagent.api.model.base.EvaTaskItem;
import com.volcengine.hiagent.api.model.base.EvaTaskRunConfig;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static com.volcengine.hiagent.api.model.base.EvaTaskRunConfigInitiator.EvaTaskRunConfigInitiatorSDK;

public class EvaClient {
private ApiClient apiClient;

Expand All @@ -36,6 +39,8 @@ public void setApiClient(ApiClient apiClient) {
* deserialize the response body
*/
public CreateEvaTaskResponse createEvaTask(CreateEvaTaskRequest body) throws ApiException {
body.setRunConfig(new EvaTaskRunConfig(EvaTaskRunConfigInitiatorSDK, null, false));

Object localVarPostBody = body;

// create path and map variables
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,8 @@ public class CreateEvaTaskRequest {
private DatasetTaskConfigForModify DatasetConfig;
public EvaTaskRulesConfig RulesConfig;
private EvaTaskSource Source;
@Nullable
private EvaTaskRunConfig RunConfig;


public CreateEvaTaskRequest() {
Expand Down Expand Up @@ -119,4 +121,13 @@ public EvaTaskSource getSource() {
public void setSource(EvaTaskSource source) {
Source = source;
}

@Nullable
public EvaTaskRunConfig getRunConfig() {
return RunConfig;
}

public void setRunConfig(@Nullable EvaTaskRunConfig runConfig) {
RunConfig = runConfig;
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
package com.volcengine.hiagent.api.model.base;

import org.jetbrains.annotations.Nullable;

public class EvaTaskRunConfig {
@Nullable
private EvaTaskRunConfigInitiator Initiator;
@Nullable
private String AgenticTaskID;
@Nullable
private Boolean EnableAttribAnalysis;

public EvaTaskRunConfig() {
}

public EvaTaskRunConfig(@Nullable EvaTaskRunConfigInitiator initiator, @Nullable String agenticTaskID, @Nullable Boolean enableAttribAnalysis) {
Initiator = initiator;
AgenticTaskID = agenticTaskID;
EnableAttribAnalysis = enableAttribAnalysis;
}

@Nullable
public EvaTaskRunConfigInitiator getInitiator() {
return Initiator;
}

public void setInitiator(@Nullable EvaTaskRunConfigInitiator initiator) {
Initiator = initiator;
}

@Nullable
public String getAgenticTaskID() {
return AgenticTaskID;
}

public void setAgenticTaskID(@Nullable String agenticTaskID) {
AgenticTaskID = agenticTaskID;
}

@Nullable
public Boolean getEnableAttribAnalysis() {
return EnableAttribAnalysis;
}

public void setEnableAttribAnalysis(@Nullable Boolean enableAttribAnalysis) {
EnableAttribAnalysis = enableAttribAnalysis;
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
package com.volcengine.hiagent.api.model.base;

import com.google.gson.annotations.SerializedName;

public enum EvaTaskRunConfigInitiator {
/**
* 平台发起
*/
@SerializedName("Platform")
EvaTaskRunConfigInitiatorPlatform("Platform"),

/**
* SDK 发起
*/
@SerializedName("SDK")
EvaTaskRunConfigInitiatorSDK("SDK");

private final String value;

EvaTaskRunConfigInitiator(String value) {
this.value = value;
}

public String getValue() {
return value;
}

@Override
public String toString() {
return this.getValue();
}
}