diff --git a/libs/langchain_v1/pyproject.toml b/libs/langchain_v1/pyproject.toml index 1390381b5b5..c5fe5bf7318 100644 --- a/libs/langchain_v1/pyproject.toml +++ b/libs/langchain_v1/pyproject.toml @@ -100,7 +100,7 @@ langchain-tests = { path = "../standard-tests", editable = true } langchain-text-splitters = { path = "../text-splitters", editable = true } langchain-openai = { path = "../partners/openai", editable = true } langchain-anthropic = { path = "../partners/anthropic", editable = true } -langgraph = { git = "https://github.com/langchain-ai/langgraph.git", branch = "1.1", subdirectory = "libs/langgraph" } +langgraph = { git = "https://github.com/langchain-ai/langgraph.git", branch = "sr/1.1-v2-default", subdirectory = "libs/langgraph" } [tool.ruff] line-length = 100 diff --git a/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py b/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py index 3c2af86a29e..20c93af4316 100644 --- a/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py +++ b/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py @@ -97,7 +97,8 @@ def test_create_agent_invoke( thread1 = {"configurable": {"thread_id": "1"}} result = agent_one.invoke({"messages": ["hello"]}, thread1) # v2 stream format returns GraphOutput; unwrap to get the dict - result_dict = result.value if isinstance(result, GraphOutput) else result + assert isinstance(result, GraphOutput) + result_dict = result.value assert result_dict == { "messages": [ _AnyIdHumanMessage(content="hello"), @@ -200,7 +201,8 @@ def test_create_agent_jump( thread1 = {"configurable": {"thread_id": "1"}} result = agent_one.invoke({"messages": []}, thread1) - result_dict = result.value if isinstance(result, GraphOutput) else result + assert isinstance(result, GraphOutput) + result_dict = result.value assert result_dict == {"messages": []} assert calls == ["NoopSeven.before_model", "NoopEight.before_model"] @@ -699,7 +701,8 @@ async def test_create_agent_async_jump() -> None: result = await agent.ainvoke({"messages": []}) - result_dict = result.value if isinstance(result, GraphOutput) else result + assert isinstance(result, GraphOutput) + result_dict = result.value assert result_dict == {"messages": []} assert calls == ["AsyncMiddlewareOne.abefore_model", "AsyncMiddlewareTwo.abefore_model"] diff --git a/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call_state_update.py b/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call_state_update.py index e898732d5c6..cbc7b7de0a7 100644 --- a/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call_state_update.py +++ b/libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call_state_update.py @@ -630,7 +630,8 @@ class TestComposition: result = agent.invoke({"messages": [HumanMessage("Hi")]}) - result_dict = result.value if isinstance(result, GraphOutput) else result + assert isinstance(result, GraphOutput) + result_dict = result.value assert result_dict.get("structured_response") == {"key": "value"} messages = result_dict["messages"] assert len(messages) == 2 diff --git a/libs/langchain_v1/tests/unit_tests/agents/test_agent_name.py b/libs/langchain_v1/tests/unit_tests/agents/test_agent_name.py index d20fafb14c1..9463706e3f9 100644 --- a/libs/langchain_v1/tests/unit_tests/agents/test_agent_name.py +++ b/libs/langchain_v1/tests/unit_tests/agents/test_agent_name.py @@ -118,11 +118,8 @@ def test_lc_agent_name_in_stream_metadata() -> None: {"messages": [HumanMessage("Hello")]}, stream_mode="messages", ): - if isinstance(event, dict): - assert event["type"] == "messages" - _chunk, metadata = event["data"] - else: - _chunk, metadata = event + assert event["type"] == "messages" + _chunk, metadata = event["data"] if "lc_agent_name" in metadata: metadata_with_agent_name.append(metadata["lc_agent_name"]) @@ -141,11 +138,8 @@ def test_lc_agent_name_not_in_stream_metadata_when_name_not_provided() -> None: {"messages": [HumanMessage("Hello")]}, stream_mode="messages", ): - if isinstance(event, dict): - assert event["type"] == "messages" - _chunk, metadata = event["data"] - else: - _chunk, metadata = event + assert event["type"] == "messages" + _chunk, metadata = event["data"] assert "lc_agent_name" not in metadata @@ -164,11 +158,8 @@ def test_lc_agent_name_in_stream_metadata_multiple_iterations() -> None: {"messages": [HumanMessage("Call a tool")]}, stream_mode="messages", ): - if isinstance(event, dict): - assert event["type"] == "messages" - _chunk, metadata = event["data"] - else: - _chunk, metadata = event + assert event["type"] == "messages" + _chunk, metadata = event["data"] if "lc_agent_name" in metadata: metadata_with_agent_name.append(metadata["lc_agent_name"]) @@ -190,11 +181,8 @@ async def test_lc_agent_name_in_astream_metadata() -> None: {"messages": [HumanMessage("Hello async")]}, stream_mode="messages", ): - if isinstance(event, dict): - assert event["type"] == "messages" - _chunk, metadata = event["data"] - else: - _chunk, metadata = event + assert event["type"] == "messages" + _chunk, metadata = event["data"] if "lc_agent_name" in metadata: metadata_with_agent_name.append(metadata["lc_agent_name"]) @@ -213,11 +201,8 @@ async def test_lc_agent_name_not_in_astream_metadata_when_name_not_provided() -> {"messages": [HumanMessage("Hello async")]}, stream_mode="messages", ): - if isinstance(event, dict): - assert event["type"] == "messages" - _chunk, metadata = event["data"] - else: - _chunk, metadata = event + assert event["type"] == "messages" + _chunk, metadata = event["data"] assert "lc_agent_name" not in metadata @@ -236,11 +221,8 @@ async def test_lc_agent_name_in_astream_metadata_multiple_iterations() -> None: {"messages": [HumanMessage("Call tool async")]}, stream_mode="messages", ): - if isinstance(event, dict): - assert event["type"] == "messages" - _chunk, metadata = event["data"] - else: - _chunk, metadata = event + assert event["type"] == "messages" + _chunk, metadata = event["data"] if "lc_agent_name" in metadata: metadata_with_agent_name.append(metadata["lc_agent_name"])