langchain 1.0.0a9__tar.gz → 1.0.0a10__tar.gz

{langchain-1.0.0a9 → langchain-1.0.0a10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain
-Version: 1.0.0a9
+Version: 1.0.0a10
 Summary: Building applications with LLMs through composability
 License: MIT
 Project-URL: Source Code, https://github.com/langchain-ai/langchain/tree/master/libs/langchain
@@ -9,7 +9,7 @@ Project-URL: repository, https://github.com/langchain-ai/langchain
 Requires-Python: <4.0.0,>=3.10.0
 Requires-Dist: langchain-core<2.0.0,>=0.3.75
 Requires-Dist: langchain-text-splitters<2.0.0,>=0.3.11
-Requires-Dist: langgraph<2.0.0,>=0.6.7
+Requires-Dist: langgraph==1.0.0a4
 Requires-Dist: pydantic<3.0.0,>=2.7.4
 Provides-Extra: community
 Requires-Dist: langchain-community; extra == "community"

{langchain-1.0.0a9 → langchain-1.0.0a10}/langchain/__init__.py

@@ -2,7 +2,7 @@
 
 from typing import Any
 
-__version__ = "1.0.0a8"
+__version__ = "1.0.0a10"
 
 
 def __getattr__(name: str) -> Any:  # noqa: ANN401

{langchain-1.0.0a9 → langchain-1.0.0a10}/langchain/agents/__init__.py

@@ -1,7 +1,7 @@
 """langgraph.prebuilt exposes a higher-level API for creating and executing agents and tools."""
 
 from langchain.agents.react_agent import AgentState, create_agent
-from langchain.agents.tool_node import ToolNode
+from langchain.tools import ToolNode
 
 __all__ = [
     "AgentState",

{langchain-1.0.0a9 → langchain-1.0.0a10}/langchain/agents/middleware/human_in_the_loop.py

@@ -169,7 +169,7 @@ class HumanInTheLoopMiddleware(AgentMiddleware):
             return None
 
         # Process all tool calls that require interrupts
-        approved_tool_calls: list[ToolCall] = auto_approved_tool_calls.copy()
+        revised_tool_calls: list[ToolCall] = auto_approved_tool_calls.copy()
         artificial_tool_messages: list[ToolMessage] = []
 
         # Create interrupt requests for all tools that need approval
@@ -210,10 +210,10 @@ class HumanInTheLoopMiddleware(AgentMiddleware):
             config = self.interrupt_on[tool_call["name"]]
 
             if response["type"] == "accept" and config.get("allow_accept"):
-                approved_tool_calls.append(tool_call)
+                revised_tool_calls.append(tool_call)
             elif response["type"] == "edit" and config.get("allow_edit"):
                 edited_action = response["args"]
-                approved_tool_calls.append(
+                revised_tool_calls.append(
                     ToolCall(
                         type="tool_call",
                         name=edited_action["action"],
@@ -233,6 +233,7 @@ class HumanInTheLoopMiddleware(AgentMiddleware):
                     tool_call_id=tool_call["id"],
                     status="error",
                 )
+                revised_tool_calls.append(tool_call)
                 artificial_tool_messages.append(tool_message)
             else:
                 allowed_actions = [
@@ -249,9 +250,6 @@ class HumanInTheLoopMiddleware(AgentMiddleware):
                 raise ValueError(msg)
 
         # Update the AI message to only include approved tool calls
-        last_ai_msg.tool_calls = approved_tool_calls
+        last_ai_msg.tool_calls = revised_tool_calls
 
-        if len(approved_tool_calls) > 0:
-            return {"messages": [last_ai_msg, *artificial_tool_messages]}
-
-        return {"jump_to": "model", "messages": artificial_tool_messages}
+        return {"messages": [last_ai_msg, *artificial_tool_messages]}

{langchain-1.0.0a9 → langchain-1.0.0a10}/langchain/agents/middleware_agent.py

@@ -33,8 +33,8 @@ from langchain.agents.structured_output import (
     StructuredOutputValidationError,
     ToolStrategy,
 )
-from langchain.agents.tool_node import ToolNode
 from langchain.chat_models import init_chat_model
+from langchain.tools import ToolNode
 
 STRUCTURED_OUTPUT_ERROR_TEMPLATE = "Error: {error}\n Please fix your mistakes."
 
@@ -532,29 +532,34 @@ def _make_model_to_tools_edge(
     first_node: str, structured_output_tools: dict[str, OutputToolBinding], tool_node: ToolNode
 ) -> Callable[[dict[str, Any]], str | list[Send] | None]:
     def model_to_tools(state: dict[str, Any]) -> str | list[Send] | None:
+        # 1. if there's an explicit jump_to in the state, use it
         if jump_to := state.get("jump_to"):
             return _resolve_jump(jump_to, first_node)
 
         last_ai_message, tool_messages = _fetch_last_ai_and_tool_messages(state["messages"])
         tool_message_ids = [m.tool_call_id for m in tool_messages]
 
+        # 2. if the model hasn't called any tools, jump to END
+        # this is the classic exit condition for an agent loop
+        if len(last_ai_message.tool_calls) == 0:
+            return END
+
         pending_tool_calls = [
             c
             for c in last_ai_message.tool_calls
             if c["id"] not in tool_message_ids and c["name"] not in structured_output_tools
         ]
 
+        # 3. if there are pending tool calls, jump to the tool node
         if pending_tool_calls:
-            # imo we should not be injecting state, store here,
-            # this should be done by the tool node itself ideally but this is a consequence
-            # of using Send w/ tool calls directly which allows more intuitive interrupt behavior
-            # largely internal so can be fixed later
             pending_tool_calls = [
                 tool_node.inject_tool_args(call, state, None) for call in pending_tool_calls
             ]
             return [Send("tools", [tool_call]) for tool_call in pending_tool_calls]
 
-        return END
+        # 4. AIMessage has tool calls, but there are no pending tool calls
+        # which suggests the injection of artificial tool messages. jump to the first node
+        return first_node
 
     return model_to_tools
 

{langchain-1.0.0a9 → langchain-1.0.0a10}/langchain/agents/react_agent.py

@@ -54,8 +54,8 @@ from langchain.agents.structured_output import (
     StructuredOutputValidationError,
     ToolStrategy,
 )
-from langchain.agents.tool_node import ToolNode
 from langchain.chat_models import init_chat_model
+from langchain.tools import ToolNode
 
 if TYPE_CHECKING:
     from langchain_core.tools import BaseTool
@@ -971,7 +971,7 @@ def create_agent(  # noqa: D417
                 return model.bind_tools(tools)
             ```
 
-            !!! note "Dynamic Model Requirements"
+            .. note::
                 Ensure returned models have appropriate tools bound via
                 `.bind_tools()` and support required functionality. Bound tools
                 must be a subset of those specified in the `tools` parameter.
@@ -985,10 +985,10 @@ def create_agent(  # noqa: D417
               of the list of messages in state["messages"].
             - SystemMessage: this is added to the beginning of the list of messages
               in state["messages"].
-            - Callable: This function should take in full graph state and the output is then passed
-              to the language model.
-            - Runnable: This runnable should take in full graph state and the output is then passed
-              to the language model.
+            - Callable: This function should take in full graph state and the output is
+              then passed to the language model.
+            - Runnable: This runnable should take in full graph state and the output is
+              then passed to the language model.
 
         response_format: An optional UsingToolStrategy configuration for structured responses.
 
@@ -1002,7 +1002,8 @@ def create_agent(  # noqa: D417
 
                 - schemas: A sequence of ResponseSchema objects that define
                   the structured output format
-                - tool_choice: Either "required" or "auto" to control when structured output is used
+                - tool_choice: Either "required" or "auto" to control when structured
+                  output is used
 
             Each ResponseSchema contains:
 
@@ -1011,12 +1012,12 @@ def create_agent(  # noqa: D417
                 - description: Optional custom description (defaults to model docstring)
                 - strict: Whether to enforce strict validation
 
-            !!! Important
+            .. important::
                 `response_format` requires the model to support tool calling
 
-            !!! Note
-                Structured responses are handled directly in the model call node via tool calls,
-                eliminating the need for separate structured response nodes.
+            .. note::
+                Structured responses are handled directly in the model call node via
+                tool calls, eliminating the need for separate structured response nodes.
 
         pre_model_hook: An optional node to add before the `agent` node
             (i.e., the node that calls the LLM).
@@ -1037,12 +1038,12 @@ def create_agent(  # noqa: D417
                 }
                 ```
 
-            !!! Important
+            .. important::
                 At least one of `messages` or `llm_input_messages` MUST be provided
                 and will be used as an input to the `agent` node.
                 The rest of the keys will be added to the graph state.
 
-            !!! Warning
+            .. warning::
                 If you are returning `messages` in the pre-model hook,
                 you should OVERWRITE the `messages` key by doing the following:
 
@@ -1059,7 +1060,7 @@ def create_agent(  # noqa: D417
             Post-model hook must be a callable or a runnable that takes in
             current graph state and returns a state update.
 
-            !!! Note
+            .. note::
                 Only available with `version="v2"`.
         state_schema: An optional state schema that defines graph state.
             Must have `messages` and `remaining_steps` keys.
@@ -1092,7 +1093,7 @@ def create_agent(  # noqa: D417
             another graph as a subgraph node -
             particularly useful for building multi-agent systems.
 
-    !!! warning "`config_schema` Deprecated"
+    .. warning::
         The `config_schema` parameter is deprecated in v0.6.0 and support will be removed in v2.0.0.
         Please use `context_schema` instead to specify the schema for run-scoped context.
 

{langchain-1.0.0a9 → langchain-1.0.0a10}/langchain/tools/__init__.py

@@ -8,10 +8,19 @@ from langchain_core.tools import (
     tool,
 )
 
+from langchain.tools.tool_node import (
+    InjectedState,
+    InjectedStore,
+    ToolNode,
+)
+
 __all__ = [
     "BaseTool",
+    "InjectedState",
+    "InjectedStore",
     "InjectedToolArg",
     "InjectedToolCallId",
     "ToolException",
+    "ToolNode",
     "tool",
 ]

{langchain-1.0.0a9/langchain/agents → langchain-1.0.0a10/langchain/tools}/tool_node.py

@@ -21,7 +21,7 @@ Key Components:
 Typical Usage:
     ```python
     from langchain_core.tools import tool
-    from langchain.agents import ToolNode
+    from langchain.tools import ToolNode
 
 
     @tool
@@ -344,7 +344,7 @@ class ToolNode(RunnableCallable):
         Basic usage:
 
         ```python
-        from langchain.agents import ToolNode
+        from langchain.tools import ToolNode
         from langchain_core.tools import tool
 
         @tool
@@ -359,7 +359,7 @@ class ToolNode(RunnableCallable):
 
         ```python
         from typing_extensions import Annotated
-        from langgraph.agents.tool_node import InjectedState
+        from langchain.tools import InjectedState
 
         @tool
         def context_tool(query: str, state: Annotated[dict, InjectedState]) -> str:
@@ -885,7 +885,8 @@ def tools_condition(
 
         ```python
         from langgraph.graph import StateGraph
-        from langgraph.agents.tool_node import ToolNode, tools_condition
+        from langchain.tools import ToolNode
+        from langchain.tools.tool_node import tools_condition
         from typing_extensions import TypedDict
 
 
@@ -950,9 +951,7 @@ class InjectedState(InjectedToolArg):
         from typing_extensions import Annotated, TypedDict
 
         from langchain_core.messages import BaseMessage, AIMessage
-        from langchain_core.tools import tool
-
-        from langgraph.agents.tool_node import InjectedState, ToolNode
+        from langchain.tools import InjectedState, ToolNode, tool
 
 
         class AgentState(TypedDict):
@@ -1020,15 +1019,14 @@ class InjectedStore(InjectedToolArg):
     for maintaining context, user preferences, or any other data that needs to
     persist beyond individual workflow executions.
 
-    !!! Warning
+    .. warning::
         `InjectedStore` annotation requires `langchain-core >= 0.3.8`
 
     Example:
         ```python
         from typing_extensions import Annotated
-        from langchain_core.tools import tool
         from langgraph.store.memory import InMemoryStore
-        from langgraph.agents.tool_node import InjectedStore, ToolNode
+        from langchain.tools import InjectedStore, ToolNode, tool
 
         @tool
         def save_preference(

{langchain-1.0.0a9 → langchain-1.0.0a10}/pyproject.toml

@@ -10,11 +10,11 @@ requires-python = ">=3.10.0,<4.0.0"
 dependencies = [
     "langchain-core>=0.3.75,<2.0.0",
     "langchain-text-splitters>=0.3.11,<2.0.0",
-    "langgraph>=0.6.7,<2.0.0",
+    "langgraph==1.0.0a4",
     "pydantic>=2.7.4,<3.0.0",
 ]
 name = "langchain"
-version = "1.0.0a9"
+version = "1.0.0a10"
 description = "Building applications with LLMs through composability"
 readme = "README.md"
 
@@ -111,6 +111,9 @@ test_integration = [
     "langchain-text-splitters",
 ]
 
+[tool.uv]
+prerelease = "allow"
+
 [tool.uv.sources.langchain-core]
 path = "../core"
 editable = true
@@ -153,15 +156,16 @@ unfixable = [
     "B028",
 ]
 
-[tool.ruff.lint.pydocstyle]
-convention = "google"
-
 [tool.ruff.lint.pyupgrade]
 keep-runtime-typing = true
 
 [tool.ruff.lint.flake8-annotations]
 allow-star-arg-any = true
 
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+ignore-var-parameters = true
+
 [tool.ruff.lint.per-file-ignores]
 "tests/*" = [
     "D1",
@@ -215,6 +219,15 @@ allow-star-arg-any = true
     "BLE001",
     "N801",
 ]
+"langchain/tools/tool_node.py" = [
+    "ANN401",
+    "A002",
+    "A001",
+    "B904",
+    "PLR2004",
+    "C901",
+    "TRY004",
+]
 
 [tool.mypy]
 strict = true

{langchain-1.0.0a9 → langchain-1.0.0a10}/tests/unit_tests/agents/test_middleware_agent.py

@@ -21,7 +21,7 @@ from langchain_core.messages import (
 from langchain_core.tools import tool, InjectedToolCallId
 
 from langchain.agents.middleware_agent import create_agent
-from langchain.agents.tool_node import InjectedState
+from langchain.tools import InjectedState
 from langchain.agents.middleware.human_in_the_loop import (
     HumanInTheLoopMiddleware,
     ActionRequest,
@@ -510,11 +510,12 @@ def test_human_in_the_loop_middleware_single_tool_response() -> None:
         result = middleware.after_model(state)
         assert result is not None
         assert "messages" in result
-        assert len(result["messages"]) == 1  # Only tool message when no approved tool calls
-        assert isinstance(result["messages"][0], ToolMessage)
-        assert result["messages"][0].content == "Custom response message"
-        assert result["messages"][0].name == "test_tool"
-        assert result["messages"][0].tool_call_id == "1"
+        assert len(result["messages"]) == 2
+        assert isinstance(result["messages"][0], AIMessage)
+        assert isinstance(result["messages"][1], ToolMessage)
+        assert result["messages"][1].content == "Custom response message"
+        assert result["messages"][1].name == "test_tool"
+        assert result["messages"][1].tool_call_id == "1"
 
 
 def test_human_in_the_loop_middleware_multiple_tools_mixed_responses() -> None:
@@ -552,10 +553,11 @@ def test_human_in_the_loop_middleware_multiple_tools_mixed_responses() -> None:
             len(result["messages"]) == 2
         )  # AI message with accepted tool call + tool message for rejected
 
-        # First message should be the AI message with updated tool calls
+        # First message should be the AI message with both tool calls
         updated_ai_message = result["messages"][0]
-        assert len(updated_ai_message.tool_calls) == 1  # Only accepted tool call
+        assert len(updated_ai_message.tool_calls) == 2  # Both tool calls remain
         assert updated_ai_message.tool_calls[0]["name"] == "get_forecast"  # Accepted
+        assert updated_ai_message.tool_calls[1]["name"] == "get_temperature"  # Got response
 
         # Second message should be the tool message for the rejected tool call
         tool_message = result["messages"][1]
@@ -1405,3 +1407,19 @@ def test_injected_state_in_middleware_agent() -> None:
     assert tool_message.name == "test_state"
     assert "success" in tool_message.content
     assert tool_message.tool_call_id == "test_call_1"
+
+
+def test_jump_to_is_ephemeral() -> None:
+    class MyMiddleware(AgentMiddleware):
+        def before_model(self, state: AgentState) -> dict[str, Any]:
+            assert "jump_to" not in state
+            return {"jump_to": "model"}
+
+        def after_model(self, state: AgentState) -> dict[str, Any]:
+            assert "jump_to" not in state
+            return {"jump_to": "model"}
+
+    agent = create_agent(model=FakeToolCallingModel(), middleware=[MyMiddleware()])
+    agent = agent.compile()
+    result = agent.invoke({"messages": [HumanMessage("Hello")]})
+    assert "jump_to" not in result

{langchain-1.0.0a9 → langchain-1.0.0a10}/tests/unit_tests/agents/test_react_agent.py

@@ -36,9 +36,11 @@ from langchain.agents import (
     create_agent,
 )
 from langchain.agents.react_agent import _validate_chat_history
-from langchain.agents.tool_node import (
+from langchain.tools import (
     InjectedState,
     InjectedStore,
+)
+from langchain.tools.tool_node import (
     _get_state_args,
     _infer_handled_types,
 )

{langchain-1.0.0a9 → langchain-1.0.0a10}/tests/unit_tests/agents/test_tool_node.py

@@ -35,13 +35,11 @@ from typing_extensions import TypedDict
 from langchain.agents import (
     ToolNode,
 )
-from langchain.agents.tool_node import (
-    TOOL_CALL_ERROR_TEMPLATE,
+from langchain.tools import (
     InjectedState,
     InjectedStore,
-    ToolInvocationError,
-    tools_condition,
 )
+from langchain.tools.tool_node import TOOL_CALL_ERROR_TEMPLATE, ToolInvocationError, tools_condition
 
 from .messages import _AnyIdHumanMessage, _AnyIdToolMessage
 from .model import FakeToolCallingModel

{langchain-1.0.0a9 → langchain-1.0.0a10}/tests/unit_tests/chat_models/test_chat_models.py

@@ -79,7 +79,7 @@ def test_configurable() -> None:
 
     Example:
 
-    .. python::
+    .. code-block:: python
 
         # This creates a configurable model without specifying which model
         model = init_chat_model()
@@ -88,10 +88,7 @@ def test_configurable() -> None:
         model.get_num_tokens("hello")  # AttributeError!
 
         # This works - provides model at runtime
-        response = model.invoke(
-            "Hello",
-            config={"configurable": {"model": "gpt-4o"}}
-        )
+        response = model.invoke("Hello", config={"configurable": {"model": "gpt-4o"}})
 
     """
     model = init_chat_model()
@@ -208,7 +205,7 @@ def test_configurable_with_default() -> None:
 
     Example:
 
-    .. python::
+    .. code-block:: python
 
         # This creates a configurable model with default parameters (model)
         model = init_chat_model("gpt-4o", configurable_fields="any", config_prefix="bar")
@@ -218,8 +215,7 @@ def test_configurable_with_default() -> None:
 
         # This also works - switches to Claude at runtime
         response = model.invoke(
-            "Hello",
-            config={"configurable": {"my_model_model": "claude-3-sonnet-20240229"}}
+            "Hello", config={"configurable": {"my_model_model": "claude-3-sonnet-20240229"}}
         )
 
     """

{langchain-1.0.0a9 → langchain-1.0.0a10}/tests/unit_tests/tools/test_imports.py

@@ -2,9 +2,12 @@ from langchain import tools
 
 EXPECTED_ALL = {
     "BaseTool",
+    "InjectedState",
+    "InjectedStore",
     "InjectedToolArg",
     "InjectedToolCallId",
     "ToolException",
+    "ToolNode",
     "tool",
 }