github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-03-18 02:53:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

simplified diff

Browse Source
This commit is contained in:
Sydney Runkle
2026-03-03 14:27:45 -08:00
parent 6aa74a76b7
commit c52c5d5bf0
4 changed files with 21 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
libs/langchain_v1/pyproject.toml
View File

@@ -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

9
libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py
View File

@@ -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"]

3
libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call_state_update.py
View File

@@ -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

42
libs/langchain_v1/tests/unit_tests/agents/test_agent_name.py
View File

@@ -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"])