scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 213

test(sap): cover content validator shapes for cache_control passthrough

Browse source
This commit is contained in:
Devin AI 2026-07-28 16:01:43 +00:00
parent 53308754bb
commit c99355d441
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
tests/test_litellm/llms/sap/chat/test_sap_transformation.py
View file

@ -776,3 +776,23 @@ class TestSAPPromptCaching:
assert template[0]["content"] == "part one\npart two"
assert template[1]["content"] == "answer"
assert template[2]["content"] == "tool result"
def test_content_shapes_accepted_by_the_content_validator(self):
from litellm.llms.sap.chat.models import SAPMessage
assert SAPMessage(role="system", content=[]).content == ""
assert SAPMessage(role="system", content={"type": "text", "text": "hi"}).content == "hi"
cached = SAPMessage(
role="system",
content={"type": "text", "text": "hi", "cache_control": {"type": "ephemeral"}},
)
assert [block.model_dump(by_alias=True, exclude_unset=True) for block in cached.content] == [
{"type": "text", "text": "hi", "cache_control": {"type": "ephemeral"}}
]
assert SAPMessage(
role="system",
content=["plain", {"type": "text", "text": ""}, {"type": "text", "text": "block"}],
).content == "plain\nblock"
with pytest.raises(ValidationError):
SAPMessage(role="system", content=7)