diff --git a/ci/test/.test.yml b/ci/test/.test.yml
index 7f4d74e6..a533335e 100644
--- a/ci/test/.test.yml
+++ b/ci/test/.test.yml
@@ -110,6 +110,12 @@
   variables:
     TEST_PATH: tests/agents/test_max_turns.py
 
+
+🤖 agents test_agent_inference:
+  <<: *run_test
+  variables:
+    TEST_PATH: tests/agents/test_agent_inference.py
+
 # ⚙️ core test_openai_chatcompletions:
 #   <<: *run_test
 #   variables:
diff --git a/src/cai/agents/patterns/red_team.py b/src/cai/agents/patterns/red_team.py
index 406934d2..01d56b52 100644
--- a/src/cai/agents/patterns/red_team.py
+++ b/src/cai/agents/patterns/red_team.py
@@ -37,10 +37,10 @@ def thought_agent_handoff(ctf=None):  # pylint: disable=unused-argument
     return thought_agent
 
 
-# Register handoff tools to enable inter-agent communication pathways
-redteam_agent.tools.append(transfer_to_dns_agent)
-dns_smtp_agent.tools.append(redteam_agent_handoff)
-thought_agent.tools.append(redteam_agent_handoff)
+# Register handoff functions to enable inter-agent communication pathways
+redteam_agent.functions.append(transfer_to_dns_agent)
+dns_smtp_agent.functions.append(redteam_agent_handoff)
+thought_agent.functions.append(redteam_agent_handoff)
 
 # Initialize the swarm pattern with the thought agent as the entry point
 redteam_swarm_pattern = thought_agent