langgraph-prebuilt 0.2.2__tar.gz → 0.5.0__tar.gz

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/.gitignore

@@ -181,3 +181,4 @@ Chinook.db
 .vercel
 .turbo
 .editorconfig
+.scratch

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: langgraph-prebuilt
-Version: 0.2.2
+Version: 0.5.0
 Summary: Library with high-level APIs for creating and executing LangGraph agents and tools.
 Project-URL: Repository, https://www.github.com/langchain-ai/langgraph
 License-Expression: MIT
 License-File: LICENSE
 Requires-Python: >=3.9
 Requires-Dist: langchain-core>=0.3.22
-Requires-Dist: langgraph-checkpoint>=2.0.10
+Requires-Dist: langgraph-checkpoint>=2.1.0
 Description-Content-Type: text/markdown
 
 # LangGraph Prebuilt

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/langgraph/prebuilt/chat_agent_executor.py

@@ -36,8 +36,8 @@ from typing_extensions import Annotated, TypedDict
 
 from langgraph.errors import ErrorCode, create_error_message
 from langgraph.graph import END, StateGraph
-from langgraph.graph.graph import CompiledGraph
 from langgraph.graph.message import add_messages
+from langgraph.graph.state import CompiledStateGraph
 from langgraph.managed import IsLastStep, RemainingSteps
 from langgraph.prebuilt.tool_node import ToolNode
 from langgraph.store.base import BaseStore
@@ -140,7 +140,9 @@ def _get_prompt_runnable(prompt: Optional[Prompt]) -> Runnable:
     return prompt_runnable
 
 
-def _should_bind_tools(model: LanguageModelLike, tools: Sequence[BaseTool]) -> bool:
+def _should_bind_tools(
+    model: LanguageModelLike, tools: Sequence[BaseTool], num_builtin: int = 0
+) -> bool:
     if isinstance(model, RunnableSequence):
         model = next(
             (
@@ -158,9 +160,10 @@ def _should_bind_tools(model: LanguageModelLike, tools: Sequence[BaseTool]) -> b
         return True
 
     bound_tools = model.kwargs["tools"]
-    if len(tools) != len(bound_tools):
+    if len(tools) != len(bound_tools) - num_builtin:
         raise ValueError(
             "Number of tools in the model.bind_tools() and tools passed to create_react_agent must match"
+            f" Got {len(tools)} tools, expected {len(bound_tools) - num_builtin}"
         )
 
     tool_names = set(tool.name for tool in tools)
@@ -257,7 +260,7 @@ def create_react_agent(
     debug: bool = False,
     version: Literal["v1", "v2"] = "v2",
     name: Optional[str] = None,
-) -> CompiledGraph:
+) -> CompiledStateGraph:
     """Creates an agent graph that calls tools in a loop until a stopping condition is met.
 
     For more details on using `create_react_agent`, visit [Agents](https://langchain-ai.github.io/langgraph/agents/overview/) documentation.
@@ -444,9 +447,8 @@ def create_react_agent(
     tool_calling_enabled = len(tool_classes) > 0
 
     if (
-        _should_bind_tools(model, tool_classes)
+        _should_bind_tools(model, tool_classes, num_builtin=len(llm_builtin_tools))
         and len(tool_classes) > 0
-        or (len(llm_builtin_tools) > 0)
     ):
         model = cast(BaseChatModel, model).bind_tools(tool_classes + llm_builtin_tools)  # type: ignore[operator]
 
@@ -592,10 +594,10 @@ def create_react_agent(
         workflow.add_node(
             "agent",
             RunnableCallable(call_model, acall_model),
-            input=input_schema,
+            input_schema=input_schema,
         )
         if pre_model_hook is not None:
-            workflow.add_node("pre_model_hook", pre_model_hook)
+            workflow.add_node("pre_model_hook", pre_model_hook)  # type: ignore[arg-type]
             workflow.add_edge("pre_model_hook", "agent")
             entrypoint = "pre_model_hook"
         else:
@@ -604,14 +606,15 @@ def create_react_agent(
         workflow.set_entry_point(entrypoint)
 
         if post_model_hook is not None:
-            workflow.add_node("post_model_hook", post_model_hook)
+            workflow.add_node("post_model_hook", post_model_hook)  # type: ignore[arg-type]
             workflow.add_edge("agent", "post_model_hook")
 
         if response_format is not None:
             workflow.add_node(
                 "generate_structured_response",
                 RunnableCallable(
-                    generate_structured_response, agenerate_structured_response
+                    generate_structured_response,
+                    agenerate_structured_response,
                 ),
             )
             if post_model_hook is not None:
@@ -658,14 +661,16 @@ def create_react_agent(
 
     # Define the two nodes we will cycle between
     workflow.add_node(
-        "agent", RunnableCallable(call_model, acall_model), input=input_schema
+        "agent",
+        RunnableCallable(call_model, acall_model),
+        input_schema=input_schema,
     )
     workflow.add_node("tools", tool_node)
 
     # Optionally add a pre-model hook node that will be called
     # every time before the "agent" (LLM-calling node)
     if pre_model_hook is not None:
-        workflow.add_node("pre_model_hook", pre_model_hook)
+        workflow.add_node("pre_model_hook", pre_model_hook)  # type: ignore[arg-type]
         workflow.add_edge("pre_model_hook", "agent")
         entrypoint = "pre_model_hook"
     else:
@@ -680,7 +685,7 @@ def create_react_agent(
 
     # Add a post model hook node if post_model_hook is provided
     if post_model_hook is not None:
-        workflow.add_node("post_model_hook", post_model_hook)
+        workflow.add_node("post_model_hook", post_model_hook)  # type: ignore[arg-type]
         agent_paths.append("post_model_hook")
         workflow.add_edge("agent", "post_model_hook")
     else:
@@ -691,7 +696,8 @@ def create_react_agent(
         workflow.add_node(
             "generate_structured_response",
             RunnableCallable(
-                generate_structured_response, agenerate_structured_response
+                generate_structured_response,
+                agenerate_structured_response,
             ),
         )
         if post_model_hook is not None:

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/langgraph/prebuilt/tool_node.py

@@ -850,7 +850,7 @@ def _get_store_arg(tool: BaseTool) -> Optional[str]:
             if _is_injection(type_arg, InjectedStore)
         ]
         if len(injections) > 1:
-            ValueError(
+            raise ValueError(
                 "A tool argument should not be annotated with InjectedStore more than "
                 f"once. Received arg {name} with annotations {injections}."
             )

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-prebuilt"
-version = "0.2.2"
+version = "0.5.0"
 description = "Library with high-level APIs for creating and executing LangGraph agents and tools."
 authors = []
 requires-python = ">=3.9"
@@ -12,7 +12,7 @@ readme = "README.md"
 license = "MIT"
 license-files = ['LICENSE']
 dependencies = [
-    "langgraph-checkpoint>=2.0.10",
+    "langgraph-checkpoint>=2.1.0",
     "langchain-core>=0.3.22",
 ]
 

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/tests/conftest.py

@@ -2,8 +2,6 @@ from collections.abc import AsyncIterator, Iterator
 from uuid import UUID
 
 import pytest
-from langchain_core import __version__ as core_version
-from packaging import version
 from pytest_mock import MockerFixture
 
 from langgraph.checkpoint.base import BaseCheckpointSaver
@@ -31,11 +29,6 @@ from tests.conftest_store import (
 
 pytest.register_assert_rewrite("tests.memory_assert")
 
-# TODO: fix this once core is released
-IS_LANGCHAIN_CORE_030_OR_GREATER = version.parse(core_version) >= version.parse(
-    "0.3.0.dev0"
-)
-
 
 @pytest.fixture
 def anyio_backend():

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/tests/memory_assert.py

@@ -13,9 +13,9 @@ from langgraph.checkpoint.base import (
     CheckpointMetadata,
     CheckpointTuple,
     SerializerProtocol,
-    copy_checkpoint,
 )
 from langgraph.checkpoint.memory import InMemorySaver, PersistentDict
+from langgraph.pregel.checkpoint import copy_checkpoint
 
 
 class NoopSerializer(SerializerProtocol):

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/tests/model.py

@@ -73,9 +73,12 @@ class FakeToolCallingModel(BaseChatModel):
 
         tool_dicts = []
         for tool in tools:
+            if isinstance(tool, dict):
+                tool_dicts.append(tool)
+                continue
             if not isinstance(tool, BaseTool):
                 raise TypeError(
-                    "Only BaseTool is supported by FakeToolCallingModel.bind_tools"
+                    "Only BaseTool and dict is supported by FakeToolCallingModel.bind_tools"
                 )
 
             # NOTE: this is a simplified tool spec for testing purposes only

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/tests/test_react_agent.py

@@ -55,7 +55,6 @@ from langgraph.store.memory import InMemoryStore
 from langgraph.types import Command, Interrupt, interrupt
 from langgraph.utils.config import get_stream_writer
 from tests.any_str import AnyStr
-from tests.conftest import IS_LANGCHAIN_CORE_030_OR_GREATER
 from tests.messages import _AnyIdHumanMessage, _AnyIdToolMessage
 from tests.model import FakeToolCallingModel
 
@@ -91,9 +90,7 @@ def test_no_prompt(sync_checkpointer: BaseCheckpointSaver, version: str) -> None
     assert saved.metadata == {
         "parents": {},
         "source": "loop",
-        "writes": {"agent": {"messages": [AIMessage(content="hi?", id="0")]}},
         "step": 1,
-        "thread_id": "123",
     }
     assert saved.pending_writes == []
 
@@ -119,9 +116,7 @@ async def test_no_prompt_async(async_checkpointer: BaseCheckpointSaver) -> None:
     assert saved.metadata == {
         "parents": {},
         "source": "loop",
-        "writes": {"agent": {"messages": [AIMessage(content="hi?", id="0")]}},
         "step": 1,
-        "thread_id": "123",
     }
     assert saved.pending_writes == []
 
@@ -275,7 +270,8 @@ async def test_prompt_with_store_async():
 
 @pytest.mark.parametrize("tool_style", ["openai", "anthropic"])
 @pytest.mark.parametrize("version", REACT_TOOL_CALL_VERSIONS)
-def test_model_with_tools(tool_style: str, version: str):
+@pytest.mark.parametrize("include_builtin", [True, False])
+def test_model_with_tools(tool_style: str, version: str, include_builtin: bool):
     model = FakeToolCallingModel(tool_style=tool_style)
 
     @dec_tool
@@ -288,10 +284,27 @@ def test_model_with_tools(tool_style: str, version: str):
         """Tool 2 docstring."""
         return f"Tool 2: {some_val}"
 
+    tools = [tool1, tool2]
+    if include_builtin:
+        tools.append(
+            {
+                "type": "mcp",
+                "server_label": "atest_sever",
+                "server_url": "https://some.mcp.somewhere.com/sse",
+                "headers": {"foo": "bar"},
+                "allowed_tools": [
+                    "mcp_tool_1",
+                    "set_active_account",
+                    "get_url_markdown",
+                    "get_url_screenshot",
+                ],
+                "require_approval": "never",
+            }
+        )
     # check valid agent constructor
     agent = create_react_agent(
-        model.bind_tools([tool1, tool2]),
-        [tool1, tool2],
+        model.bind_tools(tools),
+        tools,
         version=version,
     )
     result = agent.nodes["tools"].invoke(
@@ -456,10 +469,6 @@ def test__infer_handled_types() -> None:
         _infer_handled_types(handler)
 
 
-@pytest.mark.skipif(
-    not IS_LANGCHAIN_CORE_030_OR_GREATER,
-    reason="Pydantic v1 is required for this test to pass in langchain-core < 0.3",
-)
 @pytest.mark.parametrize("version", REACT_TOOL_CALL_VERSIONS)
 def test_react_agent_with_structured_response(version: str) -> None:
     class WeatherResponse(BaseModel):
@@ -496,10 +505,6 @@ class CustomStatePydantic(AgentStatePydantic):
     user_name: Optional[str] = None
 
 
-@pytest.mark.skipif(
-    not IS_LANGCHAIN_CORE_030_OR_GREATER,
-    reason="Langchain core 0.3.0 or greater is required",
-)
 @pytest.mark.parametrize("version", REACT_TOOL_CALL_VERSIONS)
 @pytest.mark.parametrize("state_schema", [CustomState, CustomStatePydantic])
 def test_react_agent_update_state(
@@ -565,10 +570,6 @@ def test_react_agent_update_state(
     assert tool_message.name == "get_user_name"
 
 
-@pytest.mark.skipif(
-    not IS_LANGCHAIN_CORE_030_OR_GREATER,
-    reason="Langchain core 0.3.0 or greater is required",
-)
 @pytest.mark.parametrize("version", REACT_TOOL_CALL_VERSIONS)
 def test_react_agent_parallel_tool_calls(
     sync_checkpointer: BaseCheckpointSaver, version: str
@@ -832,10 +833,6 @@ def test_create_react_agent_inject_vars(
     assert result["foo"] == 2
 
 
-@pytest.mark.skipif(
-    not IS_LANGCHAIN_CORE_030_OR_GREATER,
-    reason="Langchain core 0.3.0 or greater is required",
-)
 def test_tool_node_inject_store() -> None:
     store = InMemoryStore()
     namespace = ("test",)
@@ -1108,14 +1105,17 @@ def test_react_with_subgraph_tools(
         return {"result": state["a"] + state["b"]}
 
     add_subgraph = (
-        StateGraph(State, output=Output).add_node(add).add_edge(START, "add").compile()
+        StateGraph(State, output_schema=Output)
+        .add_node(add)
+        .add_edge(START, "add")
+        .compile()
     )
 
     def multiply(state):
         return {"result": state["a"] * state["b"]}
 
     multiply_subgraph = (
-        StateGraph(State, output=Output)
+        StateGraph(State, output_schema=Output)
         .add_node(multiply)
         .add_edge(START, "multiply")
         .compile()

{langgraph_prebuilt-0.2.2 → langgraph_prebuilt-0.5.0}/tests/test_tool_node.py

@@ -18,7 +18,6 @@ from langgraph.errors import NodeInterrupt
 from langgraph.prebuilt import ToolNode
 from langgraph.prebuilt.tool_node import TOOL_CALL_ERROR_TEMPLATE
 from langgraph.types import Command, Send
-from tests.conftest import IS_LANGCHAIN_CORE_030_OR_GREATER
 
 pytestmark = pytest.mark.anyio
 
@@ -492,10 +491,6 @@ def test_tool_node_node_interrupt():
             assert exc_info.value == "foo"
 
 
-@pytest.mark.skipif(
-    not IS_LANGCHAIN_CORE_030_OR_GREATER,
-    reason="Langchain core 0.3.0 or greater is required",
-)
 @pytest.mark.parametrize("input_type", ["dict", "tool_calls"])
 async def test_tool_node_command(input_type: str):
     from langchain_core.tools.base import InjectedToolCallId
@@ -797,10 +792,6 @@ async def test_tool_node_command(input_type: str):
     ) == [Command(update={"messages": []}, graph=Command.PARENT)]
 
 
-@pytest.mark.skipif(
-    not IS_LANGCHAIN_CORE_030_OR_GREATER,
-    reason="Langchain core 0.3.0 or greater is required",
-)
 async def test_tool_node_command_list_input():
     from langchain_core.tools.base import InjectedToolCallId