From 83bf62193364ef90282ef9bf1af43018b01c8873 Mon Sep 17 00:00:00 2001
From: Tushar Gupta <114675969+ummtushar@users.noreply.github.com>
Date: Sun, 21 Sep 2025 23:00:15 +0200
Subject: [PATCH] Move sub-agent creation from _create_task_tool to inside task
 function

This change moves the creation of the agents dictionary from the tool creation time to inside the task function, making it dynamic per invocation. This allows for potential future flexibility but may impact performance.
---
 src/deepagents/sub_agent.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/deepagents/sub_agent.py b/src/deepagents/sub_agent.py
index 2bf3ef2..be2c3f9 100644
--- a/src/deepagents/sub_agent.py
+++ b/src/deepagents/sub_agent.py
@@ -95,9 +95,6 @@ def _create_task_tool(
     state_schema,
     post_model_hook: Optional[Callable] = None,
 ):
-    agents = _get_agents(
-        tools, instructions, subagents, model, state_schema, post_model_hook
-    )
     other_agents_string = _get_subagent_description(subagents)
 
     @tool(
@@ -109,6 +106,10 @@ async def task(
         state: Annotated[DeepAgentState, InjectedState],
         tool_call_id: Annotated[str, InjectedToolCallId],
     ):
+        agents = _get_agents(
+            tools, instructions, subagents, model, state_schema, post_model_hook
+        )
+        
         if subagent_type not in agents:
             return f"Error: invoked agent of type {subagent_type}, the only allowed types are {[f'`{k}`' for k in agents]}"
         sub_agent = agents[subagent_type]
@@ -136,9 +137,6 @@ def _create_sync_task_tool(
     state_schema,
     post_model_hook: Optional[Callable] = None,
 ):
-    agents = _get_agents(
-        tools, instructions, subagents, model, state_schema, post_model_hook
-    )
     other_agents_string = _get_subagent_description(subagents)
 
     @tool(
@@ -150,6 +148,10 @@ def task(
         state: Annotated[DeepAgentState, InjectedState],
         tool_call_id: Annotated[str, InjectedToolCallId],
     ):
+        agents = _get_agents(
+            tools, instructions, subagents, model, state_schema, post_model_hook
+        )
+        
         if subagent_type not in agents:
             return f"Error: invoked agent of type {subagent_type}, the only allowed types are {[f'`{k}`' for k in agents]}"
         sub_agent = agents[subagent_type]