langchain-dev-utils 1.3.6__tar.gz → 1.3.7__tar.gz

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.3.6
+Version: 1.3.7
 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.6 → langchain_dev_utils-1.3.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langchain-dev-utils"
-version = "1.3.6"
+version = "1.3.7"
 description = "A practical utility library for LangChain and LangGraph development"
 readme = "README.md"
 authors = [{ name = "tiebingice", email = "tiebingice123@outlook.com" }]

langchain_dev_utils-1.3.7/src/langchain_dev_utils/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.3.7"

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.3.7}/src/langchain_dev_utils/agents/middleware/handoffs.py

@@ -1,4 +1,4 @@
-from typing import Any, Awaitable, Callable, Literal
+from typing import Any, Awaitable, Callable, Literal, cast
 
 from langchain.agents import AgentState
 from langchain.agents.middleware import AgentMiddleware, ModelRequest, ModelResponse
@@ -128,6 +128,7 @@ class HandoffAgentMiddleware(AgentMiddleware):
     Args:
         agents_config (dict[str, AgentConfig]): A dictionary of agent configurations.
         custom_handoffs_tool_descriptions (Optional[dict[str, str]]): A dictionary of custom tool descriptions for handoffs tools. Defaults to None.
+        handoffs_tool_overrides (Optional[dict[str, BaseTool]]): A dictionary of handoffs tools to override. Defaults to None.
 
     Examples:
         ```python
@@ -142,6 +143,7 @@ class HandoffAgentMiddleware(AgentMiddleware):
         self,
         agents_config: dict[str, AgentConfig],
         custom_handoffs_tool_descriptions: Optional[dict[str, str]] = None,
+        handoffs_tool_overrides: Optional[dict[str, BaseTool]] = None,
     ) -> None:
         default_agent_name = _get_default_active_agent(agents_config)
         if default_agent_name is None:
@@ -152,13 +154,23 @@ class HandoffAgentMiddleware(AgentMiddleware):
         if custom_handoffs_tool_descriptions is None:
             custom_handoffs_tool_descriptions = {}
 
-        handoffs_tools = [
-            _create_handoffs_tool(
-                agent_name,
-                custom_handoffs_tool_descriptions.get(agent_name),
-            )
-            for agent_name in agents_config.keys()
-        ]
+        if handoffs_tool_overrides is None:
+            handoffs_tool_overrides = {}
+
+        handoffs_tools = []
+        for agent_name in agents_config.keys():
+            if not handoffs_tool_overrides.get(agent_name):
+                handoffs_tools.append(
+                    _create_handoffs_tool(
+                        agent_name,
+                        custom_handoffs_tool_descriptions.get(agent_name),
+                    )
+                )
+            else:
+                handoffs_tools.append(
+                    cast(BaseTool, handoffs_tool_overrides.get(agent_name))
+                )
+
         self.default_agent_name = default_agent_name
         self.agents_config = _transform_agent_config(
             agents_config,
@@ -166,7 +178,7 @@ class HandoffAgentMiddleware(AgentMiddleware):
         )
         self.tools = handoffs_tools
 
-    def _get_active_agent_config(self, request: ModelRequest) -> dict[str, Any]:
+    def _get_override_request(self, request: ModelRequest) -> ModelRequest:
         active_agent_name = request.state.get("active_agent", self.default_agent_name)
 
         _config = self.agents_config[active_agent_name]
@@ -181,24 +193,22 @@ class HandoffAgentMiddleware(AgentMiddleware):
             params["system_prompt"] = _config.get("prompt")
         if _config.get("tools"):
             params["tools"] = _config.get("tools")
-        return params
+
+        if params:
+            return request.override(**params)
+        else:
+            return request
 
     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)
+        override_request = self._get_override_request(request)
+        return handler(override_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)
+        override_request = self._get_override_request(request)
+        return await handler(override_request)

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.3.7}/src/langchain_dev_utils/agents/middleware/model_router.py

@@ -150,7 +150,7 @@ class ModelRouterMiddleware(AgentMiddleware):
         model_name = await self._aselect_model(state["messages"])
         return {"router_model_selection": model_name}
 
-    def _get_override_kwargs(self, request: ModelRequest) -> dict[str, Any]:
+    def _get_override_request(self, request: ModelRequest) -> ModelRequest:
         model_dict = {
             item["model_name"]: {
                 "tools": item.get("tools", None),
@@ -180,24 +180,21 @@ class ModelRouterMiddleware(AgentMiddleware):
                     content=model_values["system_prompt"]
                 )
 
-        return override_kwargs
+        if override_kwargs:
+            return request.override(**override_kwargs)
+        else:
+            return request
 
     def wrap_model_call(
         self, request: ModelRequest, handler: Callable[[ModelRequest], ModelResponse]
     ) -> ModelCallResult:
-        override_kwargs = self._get_override_kwargs(request)
-        if override_kwargs:
-            return handler(request.override(**override_kwargs))
-        else:
-            return handler(request)
+        override_request = self._get_override_request(request)
+        return handler(override_request)
 
     async def awrap_model_call(
         self,
         request: ModelRequest,
         handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelCallResult:
-        override_kwargs = self._get_override_kwargs(request)
-        if override_kwargs:
-            return await handler(request.override(**override_kwargs))
-        else:
-            return await handler(request)
+        override_request = self._get_override_request(request)
+        return await handler(override_request)

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.3.7}/src/langchain_dev_utils/agents/middleware/plan.py

@@ -335,12 +335,8 @@ class PlanMiddleware(AgentMiddleware):
         self.system_prompt = system_prompt
         self.tools = tools
 
-    def wrap_model_call(
-        self,
-        request: ModelRequest,
-        handler: Callable[[ModelRequest], ModelResponse],
-    ) -> ModelCallResult:
-        """Update the system message to include the plan system prompt."""
+    def _get_override_request(self, request: ModelRequest) -> ModelRequest:
+        """Add the plan system prompt to the system message."""
         if request.system_message is not None:
             new_system_content = [
                 *request.system_message.content_blocks,
@@ -351,7 +347,15 @@ class PlanMiddleware(AgentMiddleware):
         new_system_message = SystemMessage(
             content=cast("list[str | dict[str, str]]", new_system_content)
         )
-        return handler(request.override(system_message=new_system_message))
+        return request.override(system_message=new_system_message)
+
+    def wrap_model_call(
+        self,
+        request: ModelRequest,
+        handler: Callable[[ModelRequest], ModelResponse],
+    ) -> ModelCallResult:
+        override_request = self._get_override_request(request)
+        return handler(override_request)
 
     async def awrap_model_call(
         self,
@@ -359,14 +363,5 @@ class PlanMiddleware(AgentMiddleware):
         handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelCallResult:
         """Update the system message to include the plan system prompt."""
-        if request.system_message is not None:
-            new_system_content = [
-                *request.system_message.content_blocks,
-                {"type": "text", "text": f"\n\n{self.system_prompt}"},
-            ]
-        else:
-            new_system_content = [{"type": "text", "text": self.system_prompt}]
-        new_system_message = SystemMessage(
-            content=cast("list[str | dict[str, str]]", new_system_content)
-        )
-        return await handler(request.override(system_message=new_system_message))
+        override_request = self._get_override_request(request)
+        return await handler(override_request)

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.3.7}/tests/test_handoffs_middleware.py

@@ -1,8 +1,9 @@
 from datetime import datetime
 
-from langchain.tools import tool
+from langchain.tools import BaseTool, ToolRuntime, tool
 from langchain_core.messages import HumanMessage, ToolMessage
 from langgraph.checkpoint.memory import InMemorySaver
+from langgraph.types import Command
 
 from langchain_dev_utils.agents import create_agent
 from langchain_dev_utils.agents.middleware import (
@@ -24,6 +25,22 @@ def run_code(code: str) -> str:
     return "Running code successfully"
 
 
+@tool
+def transfer_to_coding_agent(runtime: ToolRuntime) -> Command:
+    """This tool help you transfer to the coding agent."""
+    return Command(
+        update={
+            "messages": [
+                ToolMessage(
+                    content="Successfully transferred to coding agent",
+                    tool_call_id=runtime.tool_call_id,
+                )
+            ],
+            "active_agent": "code_agent",
+        }
+    )
+
+
 agents_config: dict[str, AgentConfig] = {
     "time_agent": {
         "model": "zai:glm-4.5",
@@ -56,14 +73,24 @@ agents_config: dict[str, AgentConfig] = {
 custom_tool_descriptions: dict[str, str] = {
     "time_agent": "transfer to the time agent to answer time-related questions",
     "talk_agent": "transfer to the talk agent to answer user questions",
-    "code_agent": "transfer to the code agent to answer code-related questions",
+}
+
+
+handoffs_tool_map: dict[str, BaseTool] = {
+    "code_agent": transfer_to_coding_agent,
 }
 
 
 def test_handoffs_middleware():
     agent = create_agent(
         model="dashscope:qwen3-max",
-        middleware=[HandoffAgentMiddleware(agents_config, custom_tool_descriptions)],
+        middleware=[
+            HandoffAgentMiddleware(
+                agents_config=agents_config,
+                custom_handoffs_tool_descriptions=custom_tool_descriptions,
+                handoffs_tool_overrides=handoffs_tool_map,
+            )
+        ],
         tools=[
             get_current_time,
             run_code,
@@ -95,13 +122,26 @@ def test_handoffs_middleware():
         == "qwen3-coder-plus"
     )
     assert isinstance(response["messages"][-2], ToolMessage)
+    assert any(
+        message
+        for message in response["messages"]
+        if isinstance(message, ToolMessage)
+        and "Successfully transferred to coding agent" in message.content
+        and message.name == "transfer_to_coding_agent"
+    )
     assert "active_agent" in response and response["active_agent"] == "code_agent"
 
 
 async def test_handoffs_middleware_async():
     agent = create_agent(
         model="dashscope:qwen3-max",
-        middleware=[HandoffAgentMiddleware(agents_config, custom_tool_descriptions)],
+        middleware=[
+            HandoffAgentMiddleware(
+                agents_config=agents_config,
+                custom_handoffs_tool_descriptions=custom_tool_descriptions,
+                handoffs_tool_overrides=handoffs_tool_map,
+            )
+        ],
         tools=[
             get_current_time,
             run_code,
@@ -133,4 +173,11 @@ async def test_handoffs_middleware_async():
         == "qwen3-coder-plus"
     )
     assert isinstance(response["messages"][-2], ToolMessage)
+    assert any(
+        message
+        for message in response["messages"]
+        if isinstance(message, ToolMessage)
+        and "Successfully transferred to coding agent" in message.content
+        and message.name == "transfer_to_coding_agent"
+    )
     assert "active_agent" in response and response["active_agent"] == "code_agent"

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.3.7}/tests/test_load_model.py

@@ -76,17 +76,6 @@ async def test_model_tool_calling_async(
     assert hasattr(response, "tool_calls") and len(response.tool_calls) == 1
 
 
-def test_model_with_reasoning(reasoning_model: BaseChatModel):
-    response = reasoning_model.invoke("hello?")
-    assert response.additional_kwargs.get("reasoning_content")
-
-
-@pytest.mark.asyncio
-async def test_model_with_reasoning_async(reasoning_model: BaseChatModel):
-    response = await reasoning_model.ainvoke("hello？")
-    assert response.additional_kwargs.get("reasoning_content")
-
-
 def test_model_profile():
     model = load_chat_model("dashscope:qwen-flash")
     assert model.profile == ALI_PROFILES["qwen-flash"]

langchain_dev_utils-1.3.6/src/langchain_dev_utils/__init__.py

@@ -1 +0,0 @@
-__version__ = "1.3.6"