scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(e2e): scope virtual keys to the deployment under test

Browse source
This commit is contained in:
mateo-berri 2026-07-14 17:12:15 -07:00
parent 04193649ee
commit a0c4e4684a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/e2e/llm_translation/test_messages_mid_conversation_system_e2e.py
View file

@ -156,7 +156,7 @@ class TestBedrockInvokeMidConversationSystem:
model = _register_invoke_deployment(
endpoints_client, resources, FLAGGED_INVOKE_MODEL
)
key = resources.key()
key = resources.key(models=[model])
system_block = _cacheable_system_block(unique_marker())
primed = _prime_prompt_cache(endpoints_client, key, model, system_block)
@ -196,7 +196,7 @@ class TestBedrockInvokeMidConversationSystem:
model = _register_invoke_deployment(
endpoints_client, resources, UNFLAGGED_INVOKE_MODEL
)
key = resources.key()
key = resources.key(models=[model])
body = RichMessagesRequest(
model=model,