mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-08-21 23:22:27 +02:00
Eugene YurtsevandGitHub
b58a7fb2fe
* Add support for ToolOutput response format. * I don't love the name -- it's confusing unless you know that it's parameterizing a strategy. We should determine if we want to support our old strategy for doing things -- it has a higher latency (one extra LLM call), but it's a reasonable built-in strategy as it doesn't do anything awkward with conversation history. (Wouldn't surprising if it has overall better performance than tool choice for longer conversations)
87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
from typing import Callable, Union
|
|
|
|
import pytest
|
|
from pydantic import BaseModel
|
|
from syrupy import SnapshotAssertion
|
|
|
|
from langgraph.prebuilt import create_react_agent
|
|
from tests.model import FakeToolCallingModel
|
|
|
|
model = FakeToolCallingModel()
|
|
|
|
|
|
def tool() -> None:
|
|
"""Testing tool."""
|
|
...
|
|
|
|
|
|
def tool2() -> None:
|
|
"""Another testing tool."""
|
|
...
|
|
|
|
|
|
def pre_model_hook() -> None:
|
|
"""Pre-model hook."""
|
|
...
|
|
|
|
|
|
def post_model_hook() -> None:
|
|
"""Post-model hook."""
|
|
...
|
|
|
|
|
|
class ResponseFormat(BaseModel):
|
|
"""Response format for the agent."""
|
|
|
|
result: str
|
|
|
|
|
|
@pytest.mark.parametrize("tools", [[], [tool]])
|
|
@pytest.mark.parametrize("pre_model_hook", [None, pre_model_hook])
|
|
@pytest.mark.parametrize("post_model_hook", [None, post_model_hook])
|
|
def test_react_agent_graph_structure(
|
|
snapshot: SnapshotAssertion,
|
|
tools: list[Callable],
|
|
pre_model_hook: Union[Callable, None],
|
|
post_model_hook: Union[Callable, None],
|
|
) -> None:
|
|
agent = create_react_agent(
|
|
model,
|
|
tools=tools,
|
|
pre_model_hook=pre_model_hook,
|
|
post_model_hook=post_model_hook,
|
|
)
|
|
try:
|
|
assert agent.get_graph().draw_mermaid(with_styles=False) == snapshot
|
|
except Exception as e:
|
|
raise ValueError(
|
|
"The graph structure has changed. Please update the snapshot."
|
|
"Configuration used:\n"
|
|
f"tools: {tools}, "
|
|
f"pre_model_hook: {pre_model_hook}, "
|
|
f"post_model_hook: {post_model_hook}, "
|
|
) from e
|
|
|
|
|
|
@pytest.mark.parametrize("tools", [[], [tool, tool2]], ids=["no_tools", "two_tools"])
|
|
@pytest.mark.parametrize(
|
|
"pre_model_hook", [None, pre_model_hook], ids=["no_pre_hook", "with_pre_hook"]
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"post_model_hook", [None, post_model_hook], ids=["no_post_hook", "with_post_hook"]
|
|
)
|
|
def test_react_agent_graph_structure_with_individual_nodes(
|
|
snapshot: SnapshotAssertion,
|
|
tools: list[Callable],
|
|
pre_model_hook: Union[Callable, None],
|
|
post_model_hook: Union[Callable, None],
|
|
) -> None:
|
|
agent = create_react_agent(
|
|
model,
|
|
tools=tools,
|
|
pre_model_hook=pre_model_hook,
|
|
post_model_hook=post_model_hook,
|
|
use_individual_tool_nodes=True,
|
|
)
|
|
assert agent.get_graph().draw_mermaid(with_styles=False) == snapshot
|