mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-04 02:31:27 +00:00
fix tests accordingly
This commit is contained in:
parent
a23f13a645
commit
29895e2c4e
1 changed files with 2 additions and 2 deletions
|
|
@ -843,13 +843,13 @@ class TestCacheControl:
|
|||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{"type": "text", "text": "Hello", "cache_control": {"type": "ephemeral", "ttl": "300"}},
|
||||
{"type": "text", "text": "Hello", "cache_control": {"type": "ephemeral", "ttl": "5m"}},
|
||||
],
|
||||
}
|
||||
]
|
||||
body = self._transform("anthropic--claude-3-5-sonnet", messages)
|
||||
cc = self._template(body)[0]["content"][0]["cache_control"]
|
||||
assert cc == {"type": "ephemeral", "ttl": "300"}
|
||||
assert cc == {"type": "ephemeral", "ttl": "5m"}
|
||||
|
||||
def test_message_level_cache_control_folded_on_assistant_message(self):
|
||||
"""cache_control on an assistant turn is folded into a content block."""
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue