mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
test(e2e): tolerate provider retries in safety_identifier capture assertion
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
83223885e6
commit
ba93c7402a
1 changed files with 5 additions and 3 deletions
|
|
@ -374,9 +374,11 @@ class TestResponses:
|
|||
)
|
||||
)
|
||||
|
||||
assert [body.additionalModelRequestFields for body in capture.bodies] == [
|
||||
{"safety_identifier": safety_identifier}
|
||||
], f"{endpoint} did not forward safety_identifier to Bedrock Converse: {capture.bodies}"
|
||||
forwarded: Final = tuple(body.additionalModelRequestFields for body in capture.bodies)
|
||||
assert forwarded, f"{endpoint} produced no Bedrock Converse request"
|
||||
assert forwarded == ({"safety_identifier": safety_identifier},) * len(forwarded), (
|
||||
f"{endpoint} did not forward safety_identifier to Bedrock Converse on every attempt: {capture.bodies}"
|
||||
)
|
||||
|
||||
@pytest.mark.skip(reason="stage red: product gap, /v1/responses 500s (aresponses TypeError) on missing input instead of 400")
|
||||
@pytest.mark.covers("llm.responses.openai.input_validation.nonstream.works")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue