Skip to content

Commit

Permalink
Fixing register agent tests (#595)
Browse files Browse the repository at this point in the history
* Fixing register agent tests

Signed-off-by: Joshua Palis <[email protected]>

* fixing register agent IT

Signed-off-by: Joshua Palis <[email protected]>

* Fixing agent type

Signed-off-by: Joshua Palis <[email protected]>

---------

Signed-off-by: Joshua Palis <[email protected]>
(cherry picked from commit ebcc8f7)
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
  • Loading branch information
github-actions[bot] committed Mar 20, 2024
1 parent d03ca83 commit ee3402e
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
import org.opensearch.flowframework.exception.FlowFrameworkException;
import org.opensearch.flowframework.indices.FlowFrameworkIndicesHandler;
import org.opensearch.ml.client.MachineLearningNodeClient;
import org.opensearch.ml.common.MLAgentType;
import org.opensearch.ml.common.agent.LLMSpec;
import org.opensearch.ml.common.agent.MLAgent;
import org.opensearch.ml.common.agent.MLMemorySpec;
Expand Down Expand Up @@ -69,7 +70,7 @@ public void setUp() throws Exception {
Map.ofEntries(
Map.entry("name", "test"),
Map.entry("description", "description"),
Map.entry("type", "type"),
Map.entry("type", MLAgentType.FLOW.name()),
Map.entry("llm.model_id", "xyz"),
Map.entry("llm.parameters", Collections.emptyMap()),
Map.entry("tools", tools),
Expand Down
4 changes: 2 additions & 2 deletions src/test/resources/template/agent-framework.json
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@
},
"user_inputs": {
"name": "Sub Agent",
"type": "cot",
"type": "CONVERSATIONAL",
"description": "this is a test agent",
"parameters": {
"hello": "world"
Expand Down Expand Up @@ -139,7 +139,7 @@
},
"user_inputs": {
"name": "DEMO-Test_Agent_For_CoT",
"type": "cot",
"type": "CONVERSATIONAL",
"description": "this is a test agent",
"parameters": {
"hello": "world"
Expand Down

0 comments on commit ee3402e

Please sign in to comment.