diff --git a/libs/standard-tests/langchain_tests/integration_tests/chat_models.py b/libs/standard-tests/langchain_tests/integration_tests/chat_models.py index 854b98f9abb..0a699e1c828 100644 --- a/libs/standard-tests/langchain_tests/integration_tests/chat_models.py +++ b/libs/standard-tests/langchain_tests/integration_tests/chat_models.py @@ -2582,15 +2582,25 @@ class ChatModelIntegrationTests(ChatModelTests): if not self.supports_anthropic_inputs: pytest.skip("Model does not explicitly support Anthropic inputs.") - class color_picker(BaseModelV1): - """Input your fav color and get a random fact about it.""" - - fav_color: str + # Anthropic-format tool + color_picker = { + "name": "color_picker", + "input_schema": { + "type": "object", + "properties": { + "fav_color": {"type": "string"}, + }, + "required": ["fav_color"], + }, + "description": "Input your fav color and get a random fact about it.", + "cache_control": {"type": "ephemeral"}, + } human_content: List[dict] = [ { "type": "text", "text": "what's your favorite color in this image", + "cache_control": {"type": "ephemeral"}, }, ] if self.supports_image_inputs: