f

Author: matiasedgeandnode

pyproject.toml

@@ -49,4 +49,9 @@ markers = [
 dev = [
     "ruff>=0.13.2",
     "mypy>=1.8.0",
+    "pytest>=8.0.0",
+    "pytest-asyncio>=0.24.0",
+    "pydantic-settings>=2.0.0",
+    "python-dotenv>=1.0.0",
+    "langchain-tests==1.1.2",
 ]

python/langchain-ampersend/pyproject.toml

@@ -18,4 +18,5 @@ build-backend = "uv_build"
 test = [
     "pytest>=8.0.0",
     "pytest-asyncio>=0.24.0",
+    "langchain-tests==1.1.2",  # Pin to avoid breaking CI
 ]

python/langchain-ampersend/tests/unit/a2a/test_standard.py

@@ -0,0 +1,60 @@
+"""Standard LangChain unit tests for tools."""
+
+from typing import Type
+from unittest.mock import MagicMock
+
+from a2a.types import AgentCapabilities, AgentCard
+from langchain_ampersend.a2a.toolkit import (
+    A2AToolkit,
+    GetAgentDetailsTool,
+    SendMessageTool,
+)
+from langchain_tests.unit_tests import ToolsUnitTests
+
+
+def _make_mock_toolkit() -> A2AToolkit:
+    """Create a mock toolkit with initialized card."""
+    mock_treasurer = MagicMock()
+    toolkit = A2AToolkit(
+        remote_agent_url="http://test-agent.com",
+        treasurer=mock_treasurer,
+    )
+    toolkit._card = AgentCard(
+        name="test_agent",
+        description="A test agent",
+        url="http://test-agent.com",
+        capabilities=AgentCapabilities(streaming=True),
+        default_input_modes=[],
+        default_output_modes=[],
+        skills=[],
+        version="1.0.0",
+    )
+    return toolkit
+
+
+class TestGetAgentDetailsToolStandard(ToolsUnitTests):
+    @property
+    def tool_constructor(self) -> Type[GetAgentDetailsTool]:
+        return GetAgentDetailsTool
+
+    @property
+    def tool_constructor_params(self) -> dict:
+        return {"toolkit": _make_mock_toolkit()}
+
+    @property
+    def tool_invoke_params_example(self) -> dict:
+        return {}  # GetAgentDetailsTool takes no parameters
+
+
+class TestSendMessageToolStandard(ToolsUnitTests):
+    @property
+    def tool_constructor(self) -> Type[SendMessageTool]:
+        return SendMessageTool
+
+    @property
+    def tool_constructor_params(self) -> dict:
+        return {"toolkit": _make_mock_toolkit()}
+
+    @property
+    def tool_invoke_params_example(self) -> dict:
+        return {"message": "Hello, agent!"}