scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test(cost): assert the item-count fallback through the public cost entry point

Browse source 
Two fallback tests reached for _count_web_search_calls directly. Bill them
through get_cost_for_built_in_tools instead: same guarantee, and the assertion
is now the charge a caller would see rather than an internal count.
This commit is contained in:
longwind48 2026-08-25 13:45:10 +08:00
parent 19d043b268
commit 836283bdb7
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py
View file

@ -425,11 +425,11 @@ class TestBedrockMantleResponsesNativeWebSearch:
assert params["tools"] == [_function_tool()]
def test_web_search_is_not_advertised_unless_the_request_asks_for_it(self, local_model_cost_map):
"""The tool type widens only for requests that carry it, so a capable model's
other requests keep the provider-wide set and never reach the cost map."""
"""The tool type set widens only for requests that carry a Web Search tool, so a capable
model's other requests keep the provider-wide set and never reach the cost map."""
cfg = BedrockMantleResponsesAPIConfig()
supported = cfg._supported_response_tool_types(tools=[_function_tool()], model="openai.gpt-5.6-terra")
assert "web_search" not in supported
assert "web_search" not in supported, "a request carrying no web_search tool must not widen the set"
assert "function" in supported
def test_web_search_hoisted_out_of_codex_additional_tools(self, local_model_cost_map):