scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-24 00:52:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 91

test(e2e): assert the forwarded Converse body without requiring the model to accept safety_identifier

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-21 07:37:28 +00:00
parent 0317903a44
commit b38504b5a6
1 changed files with 8 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/e2e/llm_translation/test_responses_e2e.py
View file

@ -19,7 +19,6 @@ from e2e_config import PROVIDER_EDGE_ADVERTISE_HOST, PROVIDER_EDGE_BIND_HOST, un
from e2e_http import (
assert_client_error,
require_successful_call,
unwrap,
)
from endpoints_client import (
EndpointsClient,
@ -359,20 +358,15 @@ class TestResponses:
safety_identifier: Final = f"end-user-{unique_marker()}"
if endpoint == "/v1/responses":
responses_result: Final = endpoints_client.responses(
key, model, "reply with one word", safety_identifier=safety_identifier
)
require_successful_call(responses_result)
endpoints_client.responses(key, model, "reply with one word", safety_identifier=safety_identifier)
else:
unwrap(
endpoints_client.proxy.chat(
key,
ChatBody(
model=model,
messages=[ChatMessage(role="user", content="reply with one word")],
safety_identifier=safety_identifier,
),
)
endpoints_client.proxy.chat(
key,
ChatBody(
model=model,
messages=[ChatMessage(role="user", content="reply with one word")],
safety_identifier=safety_identifier,
),
)
forwarded: Final = tuple(body.additionalModelRequestFields for body in capture.bodies)