langchain-dev-utils 1.2.16__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/agents/middleware/model_fallback.py

@@ -22,7 +22,7 @@ class ModelFallbackMiddleware(_ModelFallbackMiddleware):
 
         fallback = ModelFallbackMiddleware(
             "vllm:qwen3-8b", ## Try first on error
-            "openrouter:meta-llama/llama-3.3-8b-instruct:free" #Then this
+            "vllm:gpt-oss-20b", #Then this
         )
 
         agent = create_agent(
@@ -30,7 +30,7 @@ class ModelFallbackMiddleware(_ModelFallbackMiddleware):
             middleware=[fallback],
         )
 
-        # If primary fails: tries qwen3-8b, then llama-3.3-8b-instruct
+        # If primary fails: tries qwen3-8b, then gpt-oss-20b
         result = await agent.invoke({"messages": [HumanMessage("Hello")]})
         ```
     """

langchain_dev_utils/chat_models/adapters/openai_compatible.py

@@ -126,13 +126,12 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
     Base template class for OpenAI-compatible chat model implementations.
 
     This class provides a foundation for integrating various LLM providers that
-    offer OpenAI-compatible APIs (such as vLLM, OpenRouter, ZAI, Moonshot,
-    and many others). It enhances the base OpenAI functionality by:
+    offer OpenAI-compatible APIs. It enhances the base OpenAI functionality by:
 
     **1. Supports output of more types of reasoning content (reasoning_content)**
     ChatOpenAI can only output reasoning content natively supported by official
     OpenAI models, while OpenAICompatibleChatModel can output reasoning content
-    from other model providers (e.g., OpenRouter, vLLM).
+    from other model providers.
 
     **2. Dynamically adapts to choose the most suitable structured-output method**
     OpenAICompatibleChatModel adds method="auto" (default), which selects the best
@@ -593,7 +592,7 @@ def _create_openai_compatible_model(
     configuring environment variable mappings and default base URLs specific to each provider.
 
     Args:
-        provider: Provider identifier (e.g., `vllm`,`openrouter`)
+        provider: Provider identifier (e.g.`vllm`)
         base_url: Default API base URL for the provider
         compatibility_options: Optional configuration for the provider
 

{langchain_dev_utils-1.2.16.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.2.16
+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,7 +8,8 @@ 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>=1.1.0
+Requires-Dist: langchain-core>=1.2.5
+Requires-Dist: langchain>=1.2.0
 Requires-Dist: langgraph>=1.0.0
 Provides-Extra: standard
 Requires-Dist: json-repair>=0.53.1; extra == 'standard'

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

@@ -1,14 +1,15 @@
-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/model_fallback.py,sha256=nivtXXF4cwyOBv6p7RW12nXtNg87wjTWxO3BKIYiroI,1674
+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
 langchain_dev_utils/agents/middleware/summarization.py,sha256=IoZ2PM1OC3AXwf0DWpfreuPOAipeiYu0KPmAABWXuY0,3087
@@ -19,7 +20,7 @@ langchain_dev_utils/chat_models/__init__.py,sha256=YSLUyHrWEEj4y4DtGFCOnDW02VIYZ
 langchain_dev_utils/chat_models/base.py,sha256=BzaoCIv145eE8b5wNDsbZDHn4EAxe4vdlptp7qXPWKk,11625
 langchain_dev_utils/chat_models/types.py,sha256=MD3cv_ZIe9fCdgwisNfuxAOhy-j4YSs1ZOQYyCjlNKs,927
 langchain_dev_utils/chat_models/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=4iJgMGAReiJ668O9ZGAZbduxNY2PhEJJonDhQDBge44,24596
+langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=PkOoYNwp2kEhXalUHXCCAnL7LXfeiNoeiQ678suEse4,24492
 langchain_dev_utils/embeddings/__init__.py,sha256=zbEOaV86TUi9Zrg_dH9dpdgacWg31HMJTlTQknA9EKk,244
 langchain_dev_utils/embeddings/base.py,sha256=BGoWY0L7nG9iRV3d4sSagXhECXrwvS1xA-A_OVltn3k,9406
 langchain_dev_utils/message_convert/__init__.py,sha256=ZGrHGXPKMrZ_p9MqfIVZ4jgbEyb7aC4Q7X-muuThIYU,457
@@ -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.2.16.dist-info/METADATA,sha256=2lobuT6h5sJqxw25CaeDtdEQrk2EvROnPLhuF-ny4mI,4516
-langchain_dev_utils-1.2.16.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.16.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.16.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,,