deepagents 0.3.7a1__py3-none-any.whl → 0.3.9__py3-none-any.whl

deepagents/middleware/memory.py

@@ -74,7 +74,7 @@ logger = logging.getLogger(__name__)
 
 
 class MemoryState(AgentState):
-    """State schema for MemoryMiddleware.
+    """State schema for `MemoryMiddleware`.
 
     Attributes:
         memory_contents: Dict mapping source paths to their loaded content.
@@ -85,7 +85,7 @@ class MemoryState(AgentState):
 
 
 class MemoryStateUpdate(TypedDict):
-    """State update for MemoryMiddleware."""
+    """State update for `MemoryMiddleware`."""
 
     memory_contents: dict[str, str]
 
@@ -153,14 +153,15 @@ MEMORY_SYSTEM_PROMPT = """<agent_memory>
 
 
 class MemoryMiddleware(AgentMiddleware):
-    """Middleware for loading agent memory from AGENTS.md files.
+    """Middleware for loading agent memory from `AGENTS.md` files.
 
     Loads memory content from configured sources and injects into the system prompt.
+
     Supports multiple sources that are combined together.
 
     Args:
         backend: Backend instance or factory function for file operations.
-        sources: List of MemorySource configurations specifying paths and names.
+        sources: List of `MemorySource` configurations specifying paths and names.
     """
 
     state_schema = MemoryState
@@ -176,9 +177,12 @@ class MemoryMiddleware(AgentMiddleware):
         Args:
             backend: Backend instance or factory function that takes runtime
                      and returns a backend. Use a factory for StateBackend.
-            sources: List of memory file paths to load (e.g., ["~/.deepagents/AGENTS.md",
-                     "./.deepagents/AGENTS.md"]). Display names are automatically derived
-                     from the paths. Sources are loaded in order.
+            sources: List of memory file paths to load (e.g., `["~/.deepagents/AGENTS.md",
+                     "./.deepagents/AGENTS.md"]`).
+
+                     Display names are automatically derived from the paths.
+
+                     Sources are loaded in order.
         """
         self._backend = backend
         self.sources = sources

deepagents/middleware/skills.py

@@ -560,6 +560,8 @@ class SkillsMiddleware(AgentMiddleware):
         lines = []
         for skill in skills:
             lines.append(f"- **{skill['name']}**: {skill['description']}")
+            if skill["allowed_tools"]:
+                lines.append(f"  -> Allowed tools: {', '.join(skill['allowed_tools'])}")
             lines.append(f"  -> Read `{skill['path']}` for full instructions")
 
         return "\n".join(lines)
@@ -601,7 +603,7 @@ class SkillsMiddleware(AgentMiddleware):
             config: Runnable config.
 
         Returns:
-            State update with skills_metadata populated, or None if already present
+            State update with `skills_metadata` populated, or `None` if already present
         """
         # Skip if skills_metadata is already present in state (even if empty)
         if "skills_metadata" in state:
@@ -636,7 +638,7 @@ class SkillsMiddleware(AgentMiddleware):
             config: Runnable config.
 
         Returns:
-            State update with skills_metadata populated, or None if already present
+            State update with `skills_metadata` populated, or `None` if already present
         """
         # Skip if skills_metadata is already present in state (even if empty)
         if "skills_metadata" in state:

deepagents/middleware/subagents.py

@@ -1,7 +1,7 @@
 """Middleware for providing subagents to an agent via a `task` tool."""
 
 from collections.abc import Awaitable, Callable, Sequence
-from typing import Any, NotRequired, TypedDict, cast
+from typing import Annotated, Any, NotRequired, TypedDict, cast
 
 from langchain.agents import create_agent
 from langchain.agents.middleware import HumanInTheLoopMiddleware, InterruptOnConfig
@@ -73,10 +73,14 @@ class SubAgent(TypedDict):
 class CompiledSubAgent(TypedDict):
     """A pre-compiled agent spec.
 
-    Important: The runnable's state schema must include a 'messages' key.
-    This is required for the subagent to communicate results back to the main agent.
+    !!! note
+
+        The runnable's state schema must include a 'messages' key.
+
+        This is required for the subagent to communicate results back to the main agent.
+
     When the subagent completes, the final message in the 'messages' list will be
-    extracted and returned as a ToolMessage to the parent agent.
+    extracted and returned as a `ToolMessage` to the parent agent.
     """
 
     name: str
@@ -90,8 +94,8 @@ class CompiledSubAgent(TypedDict):
 
     Create a custom agent using either:
 
-    1. LangChain's `create_agent()`: https://docs.langchain.com/oss/python/langchain/quickstart
-    2. A custom graph using langgraph: https://docs.langchain.com/oss/python/langgraph/quickstart
+    1. LangChain's [`create_agent()`](https://docs.langchain.com/oss/python/langchain/quickstart)
+    2. A custom graph using [`langgraph`](https://docs.langchain.com/oss/python/langgraph/quickstart)
 
     If you're creating a custom graph, make sure the state schema includes a 'messages' key.
     This is required for the subagent to communicate results back to the main agent.
@@ -395,8 +399,11 @@ def _create_task_tool(
         task_description = task_description.format(available_agents=subagent_description_str)
 
     def task(
-        description: str,
-        subagent_type: str,
+        description: Annotated[
+            str,
+            "A detailed description of the task for the subagent to perform autonomously. Include all necessary context and specify the expected output format.",  # noqa: E501
+        ],
+        subagent_type: Annotated[str, "The type of subagent to use. Must be one of the available agent types listed in the tool description."],
         runtime: ToolRuntime,
     ) -> str | Command:
         if subagent_type not in subagent_graphs:
@@ -410,8 +417,11 @@ def _create_task_tool(
         return _return_command_with_state_update(result, runtime.tool_call_id)
 
     async def atask(
-        description: str,
-        subagent_type: str,
+        description: Annotated[
+            str,
+            "A detailed description of the task for the subagent to perform autonomously. Include all necessary context and specify the expected output format.",  # noqa: E501
+        ],
+        subagent_type: Annotated[str, "The type of subagent to use. Must be one of the available agent types listed in the tool description."],
         runtime: ToolRuntime,
     ) -> str | Command:
         if subagent_type not in subagent_graphs:
@@ -450,18 +460,24 @@ class SubAgentMiddleware(AgentMiddleware):
 
     Args:
         default_model: The model to use for subagents.
-            Can be a LanguageModelLike or a dict for init_chat_model.
+
+            Can be a `LanguageModelLike` or a dict for `init_chat_model`.
         default_tools: The tools to use for the default general-purpose subagent.
-        default_middleware: Default middleware to apply to all subagents. If `None` (default),
-            no default middleware is applied. Pass a list to specify custom middleware.
-        default_interrupt_on: The tool configs to use for the default general-purpose subagent. These
-            are also the fallback for any subagents that don't specify their own tool configs.
+        default_middleware: Default middleware to apply to all subagents.
+
+            If `None`, no default middleware is applied.
+
+            Pass a list to specify custom middleware.
+        default_interrupt_on: The tool configs to use for the default general-purpose subagent.
+
+            These are also the fallback for any subagents that don't specify their own tool configs.
         subagents: A list of additional subagents to provide to the agent.
         system_prompt: Full system prompt override. When provided, completely replaces
             the agent's system prompt.
-        general_purpose_agent: Whether to include the general-purpose agent. Defaults to `True`.
-        task_description: Custom description for the task tool. If `None`, uses the
-            default description template.
+        general_purpose_agent: Whether to include the general-purpose agent.
+        task_description: Custom description for the task tool.
+
+            If `None`, uses the default description template.
 
     Example:
         ```python
@@ -505,7 +521,7 @@ class SubAgentMiddleware(AgentMiddleware):
         general_purpose_agent: bool = True,
         task_description: str | None = None,
     ) -> None:
-        """Initialize the SubAgentMiddleware."""
+        """Initialize the `SubAgentMiddleware`."""
         super().__init__()
         self.system_prompt = system_prompt
         task_tool = _create_task_tool(