langchain-dev-utils 1.3.2__py3-none-any.whl → 1.3.3__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.3.2"
+__version__ = "1.3.3"

langchain_dev_utils/agents/__init__.py

@@ -1,4 +1,4 @@
 from .factory import create_agent
-from .wrap import wrap_agent_as_tool
+from .wrap import wrap_agent_as_tool, wrap_all_agents_as_tool
 
-__all__ = ["create_agent", "wrap_agent_as_tool"]
+__all__ = ["create_agent", "wrap_agent_as_tool", "wrap_all_agents_as_tool"]

langchain_dev_utils/agents/middleware/handoffs.py

@@ -78,7 +78,19 @@ def _transform_agent_config(
         dict[str, AgentConfig]: The transformed agent config.
     """
 
+    new_config = {}
     for agent_name, _cfg in config.items():
+        new_config[agent_name] = {}
+
+        if "model" in _cfg:
+            new_config[agent_name]["model"] = _cfg["model"]
+        if "prompt" in _cfg:
+            new_config[agent_name]["prompt"] = _cfg["prompt"]
+        if "default" in _cfg:
+            new_config[agent_name]["default"] = _cfg["default"]
+        if "tools" in _cfg:
+            new_config[agent_name]["tools"] = _cfg["tools"]
+
         handoffs = _cfg.get("handoffs", [])
         if handoffs == "all":
             handoff_tools = [
@@ -102,8 +114,11 @@ def _transform_agent_config(
                 ]
             ]
 
-        _cfg["tools"] = [*_cfg.get("tools", []), *handoff_tools]
-    return config
+        new_config[agent_name]["tools"] = [
+            *new_config[agent_name].get("tools", []),
+            *handoff_tools,
+        ]
+    return new_config
 
 
 class HandoffAgentMiddleware(AgentMiddleware):

langchain_dev_utils/agents/wrap.py

@@ -5,11 +5,8 @@ from langchain.tools import ToolRuntime
 from langchain_core.messages import AnyMessage, HumanMessage
 from langchain_core.tools import BaseTool, StructuredTool
 from langgraph.graph.state import CompiledStateGraph
-from pydantic import BaseModel, Field
 
-
-class AgentToolInput(BaseModel):
-    request: str = Field(description="The input to the agent")
+from langchain_dev_utils.message_convert import format_sequence
 
 
 def _process_input(request: str, runtime: ToolRuntime) -> str:
@@ -138,3 +135,140 @@ def wrap_agent_as_tool(
         name=tool_name,
         description=tool_description,
     )
+
+
+def wrap_all_agents_as_tool(
+    agents: list[CompiledStateGraph],
+    tool_name: Optional[str] = None,
+    tool_description: Optional[str] = None,
+    pre_input_hooks: Optional[
+        tuple[
+            Callable[[str, ToolRuntime], str],
+            Callable[[str, ToolRuntime], Awaitable[str]],
+        ]
+        | Callable[[str, ToolRuntime], str]
+    ] = None,
+    post_output_hooks: Optional[
+        tuple[
+            Callable[[str, list[AnyMessage], ToolRuntime], Any],
+            Callable[[str, list[AnyMessage], ToolRuntime], Awaitable[Any]],
+        ]
+        | Callable[[str, list[AnyMessage], ToolRuntime], Any]
+    ] = None,
+) -> BaseTool:
+    """Wraps all agents as single tool
+
+    Args:
+        agents: The agents to wrap
+        tool_name: The name of the tool, default to "task"
+        tool_description: The description of the tool
+        pre_input_hooks: Hooks to run before the input is processed
+        post_output_hooks: Hooks to run after the output is processed
+
+    Returns:
+        BaseTool: The wrapped agents as single tool
+
+    Example:
+        >>> from langchain_dev_utils.agents import wrap_all_agents_as_tool, create_agent
+        >>>
+        >>> call_time_agent_tool = wrap_all_agents_as_tool(
+        ...     [time_agent,weather_agent],
+        ...     tool_name="call_sub_agents",
+        ...     tool_description="Used to invoke the sub-agents to perform tasks"
+        ... )
+        >>>
+        >>> agent = create_agent("vllm:qwen3-4b", tools=[call_sub_agents_tool], name="agent")
+
+        >>> response = agent.invoke({"messages": [HumanMessage(content="What time is it now?")]})
+        >>> response
+    """
+    if len(agents) <= 1:
+        raise ValueError("At least more than one agent must be provided")
+
+    agents_map = {}
+
+    for agent in agents:
+        if agent.name is None:
+            raise ValueError("Agent name must not be provided")
+        if agent.name in agents_map:
+            raise ValueError("Agent name must be unique")
+        agents_map[agent.name] = agent
+
+    process_input = _process_input
+    process_input_async = _process_input
+    process_output = _process_output
+    process_output_async = _process_output
+
+    if pre_input_hooks:
+        if isinstance(pre_input_hooks, tuple):
+            process_input = pre_input_hooks[0]
+            process_input_async = pre_input_hooks[1]
+        else:
+            process_input = pre_input_hooks
+            process_input_async = pre_input_hooks
+
+    if post_output_hooks:
+        if isinstance(post_output_hooks, tuple):
+            process_output = post_output_hooks[0]
+            process_output_async = post_output_hooks[1]
+        else:
+            process_output = post_output_hooks
+            process_output_async = post_output_hooks
+
+    def call_agent(
+        agent_name: str,
+        description: str,
+        runtime: ToolRuntime,
+    ) -> str:
+        task_description = (
+            process_input(description, runtime) if process_input else description
+        )
+
+        if agent_name not in agents_map:
+            raise ValueError(f"Agent {agent_name} not found")
+
+        messages = [HumanMessage(content=task_description)]
+        response = agents_map[agent_name].invoke({"messages": messages})
+
+        response = process_output(task_description, response["messages"], runtime)
+        return response
+
+    async def acall_agent(
+        agent_name: str,
+        description: str,
+        runtime: ToolRuntime,
+    ) -> str:
+        if asyncio.iscoroutinefunction(process_input_async):
+            task_description = await process_input_async(description, runtime)
+        else:
+            task_description = cast(str, process_input_async(description, runtime))
+
+        if agent_name not in agents_map:
+            raise ValueError(f"Agent {agent_name} not found")
+
+        messages = [HumanMessage(content=task_description)]
+        response = await agents_map[agent_name].ainvoke({"messages": messages})
+
+        if asyncio.iscoroutinefunction(process_output_async):
+            response = await process_output_async(
+                task_description, response["messages"], runtime
+            )
+        else:
+            response = process_output(task_description, response["messages"], runtime)
+
+        return response
+
+    if tool_name is None:
+        tool_name = "task"
+
+    if tool_description is None:
+        tool_description = (
+            "Launch an ephemeral subagent for a task.\nAvailable agents:\n "
+            + format_sequence(list(agents_map.keys()), with_num=True)
+        )
+    return StructuredTool.from_function(
+        func=call_agent,
+        coroutine=acall_agent,
+        name=tool_name,
+        description=tool_description,
+    )

{langchain_dev_utils-1.3.2.dist-info → langchain_dev_utils-1.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.3.2
+Version: 1.3.3
 Summary: A practical utility library for LangChain and LangGraph development
 Project-URL: Source Code, https://github.com/TBice123123/langchain-dev-utils
 Project-URL: repository, https://github.com/TBice123123/langchain-dev-utils

{langchain_dev_utils-1.3.2.dist-info → langchain_dev_utils-1.3.3.dist-info}/RECORD

@@ -1,14 +1,14 @@
-langchain_dev_utils/__init__.py,sha256=HgKA3RqZvC7slo8MgLyffCGwJbQ3cY6I7oUMFvGLWps,22
+langchain_dev_utils/__init__.py,sha256=Vi6om3KImlKsS_Wg5CjUgYffoi2zx7T-SRPnnGL0G7M,22
 langchain_dev_utils/_utils.py,sha256=MFEzR1BjXMj6HEVwt2x2omttFuDJ_rYAEbNqe99r9pM,1338
 langchain_dev_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_dev_utils/agents/__init__.py,sha256=PJ-lSDZv_AXMYA3H4fx-HzJa14tPbkGmq1HX8LNfaPo,125
+langchain_dev_utils/agents/__init__.py,sha256=69_biZzyJvW9OBT1g8TX_77mp9-I_TvWo9QtlvHq83E,177
 langchain_dev_utils/agents/factory.py,sha256=8XB6y_ddf58vXlTLHBL6KCirFqkD2GjtzsuOt98sS7U,3732
 langchain_dev_utils/agents/file_system.py,sha256=Yk3eetREE26WNrnTWLoiDUpOyCJ-rhjlfFDk6foLa1E,8468
 langchain_dev_utils/agents/plan.py,sha256=WwhoiJBmVYVI9bT8HfjCzTJ_SIp9WFil0gOeznv2omQ,6497
-langchain_dev_utils/agents/wrap.py,sha256=RuchoH_VotPmKFuYEn2SXoSgNxZhSA9jKM0Iv_8oHLk,4718
+langchain_dev_utils/agents/wrap.py,sha256=O8LSieRhlqL4cwf_BPNIiJD3lhTyL6civBHcQPWCqSU,9490
 langchain_dev_utils/agents/middleware/__init__.py,sha256=QVQibaNHvHPyNTZ2UNFfYL153ZboaCHcoioTHK0FsiY,710
 langchain_dev_utils/agents/middleware/format_prompt.py,sha256=LzYiQXCRvkpfDhGPxhZwdepeU3j-HUWPJqcx3FWsfT4,2357
-langchain_dev_utils/agents/middleware/handoffs.py,sha256=hYtm6NKUvfJQ_fxtTBuvEay185_RCB0yXAEdgSL7-Ls,6721
+langchain_dev_utils/agents/middleware/handoffs.py,sha256=r196Xk0Jws1Tz6JQuvy5HEc3HAAQejCxFmJpB6KrvLU,7230
 langchain_dev_utils/agents/middleware/model_fallback.py,sha256=8xiNjTJ0yiRkPLCRfAGNnqY1TLstj1Anmiqyv5w2mA8,1633
 langchain_dev_utils/agents/middleware/model_router.py,sha256=qBspvj9ZoKfmC1pHWTO0EHHfxjgCUd-TuSbqvZl0kmg,7977
 langchain_dev_utils/agents/middleware/plan.py,sha256=-ZLkp85QQTSCX9thMblacJ1N86h0BYPoTwCfJlJ_jzQ,14981
@@ -33,7 +33,7 @@ langchain_dev_utils/pipeline/types.py,sha256=T3aROKKXeWvd0jcH5XkgMDQfEkLfPaiOhhV
 langchain_dev_utils/tool_calling/__init__.py,sha256=mu_WxKMcu6RoTf4vkTPbA1WSBSNc6YIqyBtOQ6iVQj4,322
 langchain_dev_utils/tool_calling/human_in_the_loop.py,sha256=7Z_QO5OZUR6K8nLoIcafc6osnvX2IYNorOJcbx6bVso,9672
 langchain_dev_utils/tool_calling/utils.py,sha256=S4-KXQ8jWmpGTXYZitovF8rxKpaSSUkFruM8LDwvcvE,2765
-langchain_dev_utils-1.3.2.dist-info/METADATA,sha256=PLQKHjjqwlL9lvaW3Et1q0gYbZTtRdmTN6w5A4gJiYM,4552
-langchain_dev_utils-1.3.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.3.2.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.3.2.dist-info/RECORD,,
+langchain_dev_utils-1.3.3.dist-info/METADATA,sha256=PEt8qHRYBK2l-A2djg8eJEuz6uZdxZq-fPa6U6yR8Cc,4552
+langchain_dev_utils-1.3.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.3.3.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.3.3.dist-info/RECORD,,