diff --git a/src/conductor/client/workflow/task/task_type.py b/src/conductor/client/workflow/task/task_type.py index 36108f72..a0a24b38 100644 --- a/src/conductor/client/workflow/task/task_type.py +++ b/src/conductor/client/workflow/task/task_type.py @@ -40,3 +40,6 @@ class TaskType(str, Enum): LIST_MCP_TOOLS = "LIST_MCP_TOOLS" CALL_MCP_TOOL = "CALL_MCP_TOOL" PULL_WORKFLOW_MESSAGES = "PULL_WORKFLOW_MESSAGES" + AGENT = "AGENT" + GET_AGENT_CARD = "GET_AGENT_CARD" + CANCEL_AGENT = "CANCEL_AGENT" diff --git a/tests/unit/workflow/test_task_type.py b/tests/unit/workflow/test_task_type.py new file mode 100644 index 00000000..24400e83 --- /dev/null +++ b/tests/unit/workflow/test_task_type.py @@ -0,0 +1,14 @@ +import unittest + +from conductor.client.workflow.task.task_type import TaskType + + +class TestTaskType(unittest.TestCase): + def test_agent_task_types(self): + self.assertEqual(TaskType.AGENT.value, "AGENT") + self.assertEqual(TaskType.GET_AGENT_CARD.value, "GET_AGENT_CARD") + self.assertEqual(TaskType.CANCEL_AGENT.value, "CANCEL_AGENT") + + +if __name__ == "__main__": + unittest.main()