Added Config to Subagents and allowed streaming

src/deepagents/sub_agent.py

@@ -9,7 +9,7 @@
 from langchain.chat_models import init_chat_model
 from typing import Annotated, NotRequired, Any, Union, Optional, Callable
 from langgraph.types import Command
-from langchain_core.runnables import Runnable
+from langchain_core.runnables import Runnable, RunnableConfig
 
 from langgraph.prebuilt import InjectedState
 
@@ -106,14 +106,20 @@ def _create_task_tool(
     async def task(
         description: str,
         subagent_type: str,
+        special_config_param: RunnableConfig,
         state: Annotated[DeepAgentState, InjectedState],
         tool_call_id: Annotated[str, InjectedToolCallId],
     ):
         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]
         state["messages"] = [{"role": "user", "content": description}]
-        result = await sub_agent.ainvoke(state)
+        if special_config_param.get('configurable').get("stream_mode") == "stream":
+            result = None
+            async for chunk in sub_agent.astream(input=state, config=special_config_param):
+                result = chunk
+        else:
+            result = await sub_agent.ainvoke(input=state, config=special_config_param)
         return Command(
             update={
                 "files": result.get("files", {}),
@@ -148,14 +154,15 @@ def _create_sync_task_tool(
     def task(
         description: str,
         subagent_type: str,
+        special_config_param: RunnableConfig,
         state: Annotated[DeepAgentState, InjectedState],
         tool_call_id: Annotated[str, InjectedToolCallId],
     ):
         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]
         state["messages"] = [{"role": "user", "content": description}]
-        result = sub_agent.invoke(state)
+        result = sub_agent.invoke(input=state, config=special_config_param)
         return Command(
             update={
                 "files": result.get("files", {}),