From 3a488d6b3f1c20dedf9234326801b9fced1a3e01 Mon Sep 17 00:00:00 2001
From: Apeksha_2022 <70309060+Apeksha9389@users.noreply.github.com>
Date: Thu, 8 Oct 2026 22:57:34 +0530
Subject: [PATCH 1/2] feat: add AGENT, GET_AGENT_CARD and CANCEL_AGENT task
 types

---
 src/conductor/client/workflow/task/task_type.py | 3 +++
 1 file changed, 3 insertions(+)

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"

From f6d6fa4c0ac3e2afd7bfc58e068945647b12a732 Mon Sep 17 00:00:00 2001
From: Apeksha_2022 <70309060+Apeksha9389@users.noreply.github.com>
Date: Thu, 8 Oct 2026 22:57:56 +0530
Subject: [PATCH 2/2] test: cover agent task types

---
 tests/unit/workflow/test_task_type.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 tests/unit/workflow/test_task_type.py

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()