github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 23:29:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

standard-tests: add cache_control to Anthropic inputs test (#31516)

Browse Source
This commit is contained in:
ccurme 2025-06-06 10:00:43 -04:00 committed by GitHub
parent a1f068eb85
commit 9c639035c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libs/standard-tests/langchain_tests/integration_tests/chat_models.py
View File

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