github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-06-09 18:50:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(langchain): check agents integration tests with mypy (#34308)

Browse Source
This commit is contained in:
Christophe Bornet
2025-12-12 13:55:34 +01:00
committed by GitHub
parent d6b5f05f33
commit bbc1d46efe
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
libs/langchain_v1/pyproject.toml
View File

@@ -94,7 +94,7 @@ line-length = 100
strict = true strict = true
ignore_missing_imports = true ignore_missing_imports = true
enable_error_code = "deprecated" enable_error_code = "deprecated"
exclude = ["tests/unit_tests/agents/*", "tests/integration_tests/agents/*"] exclude = ["tests/unit_tests/agents/*"]
# TODO: activate for 'strict' checking # TODO: activate for 'strict' checking
disallow_any_generics = false disallow_any_generics = false

12
libs/langchain_v1/tests/integration_tests/agents/middleware/test_shell_tool_integration.py
View File

@@ -7,9 +7,13 @@ from typing import Any
import pytest import pytest
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from langgraph._internal._typing import StateLike
from langgraph.graph.state import CompiledStateGraph
from typing_extensions import TypedDict
from langchain.agents import create_agent from langchain.agents import create_agent
from langchain.agents.middleware.shell_tool import ShellToolMiddleware from langchain.agents.middleware.shell_tool import ShellToolMiddleware
from langchain.agents.middleware.types import _InputAgentState
def _get_model(provider: str) -> Any: def _get_model(provider: str) -> Any:
@@ -33,7 +37,7 @@ def test_shell_tool_basic_execution(tmp_path: Path, provider: str) -> None:
pytest.importorskip(f"langchain_{provider}") pytest.importorskip(f"langchain_{provider}")
workspace = tmp_path / "workspace" workspace = tmp_path / "workspace"
agent = create_agent( agent: CompiledStateGraph[Any, Any, _InputAgentState, Any] = create_agent(
model=_get_model(provider), model=_get_model(provider),
middleware=[ShellToolMiddleware(workspace_root=workspace)], middleware=[ShellToolMiddleware(workspace_root=workspace)],
) )
@@ -55,7 +59,7 @@ def test_shell_tool_basic_execution(tmp_path: Path, provider: str) -> None:
def test_shell_session_persistence(tmp_path: Path) -> None: def test_shell_session_persistence(tmp_path: Path) -> None:
"""Test shell session state persists across multiple tool calls.""" """Test shell session state persists across multiple tool calls."""
workspace = tmp_path / "workspace" workspace = tmp_path / "workspace"
agent = create_agent( agent: CompiledStateGraph[Any, Any, _InputAgentState, Any] = create_agent(
model=_get_model("anthropic"), model=_get_model("anthropic"),
middleware=[ShellToolMiddleware(workspace_root=workspace)], middleware=[ShellToolMiddleware(workspace_root=workspace)],
) )
@@ -82,7 +86,7 @@ def test_shell_session_persistence(tmp_path: Path) -> None:
def test_shell_tool_error_handling(tmp_path: Path) -> None: def test_shell_tool_error_handling(tmp_path: Path) -> None:
"""Test shell tool captures command errors.""" """Test shell tool captures command errors."""
workspace = tmp_path / "workspace" workspace = tmp_path / "workspace"
agent = create_agent( agent: CompiledStateGraph[Any, Any, _InputAgentState, Any] = create_agent(
model=_get_model("anthropic"), model=_get_model("anthropic"),
middleware=[ShellToolMiddleware(workspace_root=workspace)], middleware=[ShellToolMiddleware(workspace_root=workspace)],
) )
@@ -121,7 +125,7 @@ def test_shell_tool_with_custom_tools(tmp_path: Path) -> None:
"""Greet someone by name.""" """Greet someone by name."""
return f"Hello, {name}!" return f"Hello, {name}!"
agent = create_agent( agent: CompiledStateGraph[Any, Any, _InputAgentState, Any] = create_agent(
model=_get_model("anthropic"), model=_get_model("anthropic"),
tools=[custom_greeting], tools=[custom_greeting],
middleware=[ShellToolMiddleware(workspace_root=workspace)], middleware=[ShellToolMiddleware(workspace_root=workspace)],