scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 175

test(cost): recapture vertex claude and mantle opus usage fixtures for the live speed key

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-11-09 17:08:08 +00:00
parent cfddaafe18
commit 3f70e58859
2 changed files with 21 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
tests/integration/cost_calculation/cases/bedrock_mantle.py
View file

@ -1012,20 +1012,22 @@ BEDROCK_MANTLE_CLAUDE_OPUS_5_5_STREAM: Final = CLAUDE_OPUS_5_5_STREAM.model_copy
}
)
BEDROCK_MANTLE_CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT: Final = with_usage(
CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT,
"bedrock_mantle-anthropic.claude-opus-5-5-messages_input_text",
_ADAPTIVE_USAGE,
_ADAPTIVE_EXPECTED,
).model_copy(
update={
"model": _MANTLE_CLAUDE_OPUS_MODEL,
"request": {
**CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT.request,
"thinking": {"type": "adaptive"},
"output_config": {"effort": "medium"},
},
}
BEDROCK_MANTLE_CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT: Final = _mantle_without_usage_fields(
with_usage(
CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT,
"bedrock_mantle-anthropic.claude-opus-5-5-messages_input_text",
_ADAPTIVE_USAGE,
_ADAPTIVE_EXPECTED,
).model_copy(
update={
"model": _MANTLE_CLAUDE_OPUS_MODEL,
"request": {
**CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT.request,
"thinking": {"type": "adaptive"},
"output_config": {"effort": "medium"},
},
}
)
)
BEDROCK_MANTLE_CLAUDE_OPUS_5_5_MESSAGES_STREAM: Final = CLAUDE_OPUS_5_5_MESSAGES_STREAM.model_copy(

5
tests/integration/cost_calculation/cases/vertex_ai.py
View file

@ -142,6 +142,7 @@ def _vertex_claude_usage(
},
"output_tokens": output_tokens,
"output_tokens_details": {"thinking_tokens": thinking_tokens},
"speed": "standard",
}
@ -230,6 +231,7 @@ def _vertex_claude_stream_response(model: str, stop_reason: str, text_deltas: tu
"cache_read_input_tokens": 0,
"cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0},
"output_tokens": 0,
"speed": "standard",
},
},
},
@ -2916,6 +2918,7 @@ VERTEX_AI_CLAUDE_SONNET_5_5_INPUT_TEXT: Final = CLAUDE_SONNET_5_5_INPUT_TEXT.mod
"cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0},
"output_tokens": 412,
"output_tokens_details": {"thinking_tokens": 0},
"speed": "standard",
},
},
),
@ -2958,6 +2961,7 @@ VERTEX_AI_CLAUDE_OPUS_5_5_INPUT_TEXT: Final = CLAUDE_OPUS_5_5_INPUT_TEXT.model_c
"cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0},
"output_tokens": 412,
"output_tokens_details": {"thinking_tokens": 0},
"speed": "standard",
},
},
),
@ -3001,6 +3005,7 @@ VERTEX_AI_CLAUDE_FABLE_5_1_INPUT_TEXT: Final = CLAUDE_FABLE_5_1_INPUT_TEXT.model
"cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0},
"output_tokens": 412,
"output_tokens_details": {"thinking_tokens": 0},
"speed": "standard",
},
},
),