langchain-dev-utils 1.3.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.16"
+__version__ = "1.3.1"

langchain_dev_utils/agents/factory.py

@@ -5,9 +5,9 @@ from langchain.agents.middleware.types import (
     AgentMiddleware,
     AgentState,
     ResponseT,
+    StateT_co,
     _InputAgentState,
     _OutputAgentState,
-    StateT_co,
 )
 from langchain.agents.structured_output import ResponseFormat
 from langchain_core.messages import SystemMessage

langchain_dev_utils/agents/middleware/__init__.py

@@ -1,4 +1,5 @@
 from .format_prompt import format_prompt
+from .handoffs import HandoffsAgentMiddleware, create_handoffs_tool
 from .model_fallback import ModelFallbackMiddleware
 from .model_router import ModelRouterMiddleware
 from .plan import (
@@ -24,4 +25,6 @@ __all__ = [
     "ModelRouterMiddleware",
     "ToolCallRepairMiddleware",
     "format_prompt",
+    "create_handoffs_tool",
+    "HandoffsAgentMiddleware",
 ]

langchain_dev_utils/agents/middleware/handoffs.py

@@ -0,0 +1,138 @@
+from typing import Any, Awaitable, Callable
+
+from langchain.agents import AgentState
+from langchain.agents.middleware import AgentMiddleware, ModelRequest, ModelResponse
+from langchain.agents.middleware.types import ModelCallResult
+from langchain.tools import BaseTool, ToolRuntime, tool
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_core.messages import SystemMessage, ToolMessage
+from langgraph.types import Command
+from typing_extensions import NotRequired, Optional, TypedDict
+
+from langchain_dev_utils.chat_models import load_chat_model
+
+
+class MultiAgentState(AgentState):
+    active_agent: NotRequired[str]
+
+
+class AgentConfig(TypedDict):
+    model: NotRequired[str | BaseChatModel]
+    prompt: str | SystemMessage
+    tools: list[BaseTool | dict[str, Any]]
+    default: NotRequired[bool]
+
+
+def create_handoffs_tool(
+    agent_name: str,
+    tool_name: Optional[str] = None,
+    tool_description: Optional[str] = None,
+):
+    """Create a tool for handoffs to a specified agent.
+
+    Args:
+        agent_name (str): The name of the agent to transfer to.
+        tool_name (Optional[str], optional): The name of the tool. Defaults to None.
+        tool_description (Optional[str], optional): The description of the tool. Defaults to None.
+
+    Returns:
+        BaseTool: A tool instance for handoffs to the specified agent.
+
+    Example:
+        Basic usage
+        >>> from langchain_dev_utils.agents.middleware import create_handoffs_tool
+        >>> handoffs_tool = create_handoffs_tool("time_agent")
+    """
+    if tool_name is None:
+        tool_name = f"transfer_to_{agent_name}"
+        if not tool_name.endswith("_agent"):
+            tool_name += "_agent"
+
+    if tool_description is None:
+        tool_description = f"Transfer to the {agent_name}"
+
+    @tool(name_or_callable=tool_name, description=tool_description)
+    def handoffs_tool(runtime: ToolRuntime) -> Command:
+        return Command(
+            update={
+                "messages": [
+                    ToolMessage(
+                        content=f"Transferred to {agent_name}",
+                        tool_call_id=runtime.tool_call_id,
+                    )
+                ],
+                "active_agent": agent_name,
+            }
+        )
+
+    return handoffs_tool
+
+
+def _get_default_active_agent(state: dict[str, AgentConfig]) -> Optional[str]:
+    for agent_name, config in state.items():
+        if config.get("default", False):
+            return agent_name
+    return None
+
+
+class HandoffsAgentMiddleware(AgentMiddleware):
+    """Agent middleware for switching between multiple agents.
+    This middleware dynamically replaces model call parameters based on the currently active agent configuration, enabling seamless switching between different agents.
+
+    Args:
+        agents_config (dict[str, AgentConfig]): A dictionary of agent configurations.
+
+    Examples:
+        ```python
+        from langchain_dev_utils.agents.middleware import HandoffsAgentMiddleware
+        middleware = HandoffsAgentMiddleware(agents_config)
+        ```
+    """
+
+    state_schema = MultiAgentState
+
+    def __init__(self, agents_config: dict[str, AgentConfig]):
+        default_agent_name = _get_default_active_agent(agents_config)
+        if default_agent_name is None:
+            raise ValueError(
+                "No default agent found, you must set one by set default=True"
+            )
+        self.default_agent_name = default_agent_name
+        self.agents_config = agents_config
+
+    def _get_active_agent_config(self, request: ModelRequest) -> dict[str, Any]:
+        active_agent_name = request.state.get("active_agent", self.default_agent_name)
+
+        _config = self.agents_config[active_agent_name]
+
+        params = {}
+        if _config.get("model"):
+            model = _config.get("model")
+            if isinstance(model, str):
+                model = load_chat_model(model)
+            params["model"] = model
+        if _config.get("prompt"):
+            params["system_prompt"] = _config.get("prompt")
+        if _config.get("tools"):
+            params["tools"] = _config.get("tools")
+        return params
+
+    def wrap_model_call(
+        self, request: ModelRequest, handler: Callable[[ModelRequest], ModelResponse]
+    ) -> ModelCallResult:
+        override_kwargs = self._get_active_agent_config(request)
+        if override_kwargs:
+            return handler(request.override(**override_kwargs))
+        else:
+            return handler(request)
+
+    async def awrap_model_call(
+        self,
+        request: ModelRequest,
+        handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
+    ) -> ModelCallResult:
+        override_kwargs = self._get_active_agent_config(request)
+        if override_kwargs:
+            return await handler(request.override(**override_kwargs))
+        else:
+            return await handler(request)

{langchain_dev_utils-1.3.0.dist-info → langchain_dev_utils-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.3.0
+Version: 1.3.1
 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
@@ -8,6 +8,7 @@ Project-URL: documentation, https://tbice123123.github.io/langchain-dev-utils
 Author-email: tiebingice <tiebingice123@outlook.com>
 License-File: LICENSE
 Requires-Python: >=3.11
+Requires-Dist: langchain-core>=1.2.5
 Requires-Dist: langchain>=1.2.0
 Requires-Dist: langgraph>=1.0.0
 Provides-Extra: standard

{langchain_dev_utils-1.3.0.dist-info → langchain_dev_utils-1.3.1.dist-info}/RECORD

@@ -1,13 +1,14 @@
-langchain_dev_utils/__init__.py,sha256=7RIJ-Nh9kHJf5O5NvahUeP8DNXq6oIzbYcIt_yKv0lQ,23
+langchain_dev_utils/__init__.py,sha256=-ypEJktJToAL9by62JJKWEzDo_KPCQtmE5kwFgX24z4,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/factory.py,sha256=mr-Q3fxxV_yEN3l8p-Rk0uxXvKjDUGEZDtrg9HRwr-8,3732
+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/middleware/__init__.py,sha256=EECbcYcHXQAMA-guJNRGwCVi9jG957d0nOaoIuyIKC0,832
+langchain_dev_utils/agents/middleware/__init__.py,sha256=n3PsbL8LSmariEe94rUMoT0GnUQMrZUNdSj-z4om2cY,962
 langchain_dev_utils/agents/middleware/format_prompt.py,sha256=LzYiQXCRvkpfDhGPxhZwdepeU3j-HUWPJqcx3FWsfT4,2357
+langchain_dev_utils/agents/middleware/handoffs.py,sha256=aCD90yTiAPyG8pC_DAr0eDi-iYW57W0wt6LzIoKLT_Y,5024
 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=0qDCmenxgY_zrwMfOyYlgLfhYNw-HszNLeeOkfj14NA,16002
@@ -32,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.0.dist-info/METADATA,sha256=_fA9O0n1NOwnh6Vg4ExNxmJ0Sa7yBJdhwvOyK4Cbycw,4515
-langchain_dev_utils-1.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.3.0.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.3.0.dist-info/RECORD,,
+langchain_dev_utils-1.3.1.dist-info/METADATA,sha256=6UJeWibp3qpwITK0THhTdFpM8tq7XyLyoUvbkMnRhLc,4552
+langchain_dev_utils-1.3.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.3.1.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.3.1.dist-info/RECORD,,