Skip to content

Commit 378bfa8

Browse files
committed
fix: test with node agent
1 parent 1735bac commit 378bfa8

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

packages/uipath-openai-agents/src/uipath_openai_agents/runtime/schema.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def _add_agent_and_tools(current_agent: Agent) -> None:
171171
UiPathRuntimeNode(
172172
id=agent_name,
173173
name=agent_name,
174-
type="model",
174+
type="node",
175175
subgraph=None,
176176
metadata=None,
177177
)

packages/uipath-openai-agents/tests/test_multi_layer_agents.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -209,11 +209,11 @@ def test_multi_layer_agent_node_types():
209209
assert node_types["__start__"] == "__start__"
210210
assert node_types["__end__"] == "__end__"
211211

212-
# Verify agent nodes are of type "model"
213-
assert node_types["orchestrator_agent"] == "model"
214-
assert node_types["hr_agent"] == "model"
215-
assert node_types["procurement_agent"] == "model"
216-
assert node_types["policy_agent"] == "model"
212+
# Verify agent nodes are of type "node"
213+
assert node_types["orchestrator_agent"] == "node"
214+
assert node_types["hr_agent"] == "node"
215+
assert node_types["procurement_agent"] == "node"
216+
assert node_types["policy_agent"] == "node"
217217

218218
# Verify aggregated tools nodes are of type "tool"
219219
assert node_types["hr_agent_tools"] == "tool"

0 commit comments

Comments
 (0)