refactor(websearch): cut the helper and test docstrings to one line

This commit is contained in:
Priyansh Nandwana 2026-09-06 11:34:39 +05:30
parent 8c612c268e
commit de54d8a72c
3 changed files with 3 additions and 15 deletions

View file

@ -31,11 +31,7 @@ def is_codex_user_agent(user_agent: str) -> bool:
def qualify_provider_stripped_model(model: str, custom_llm_provider: str) -> str:
"""Put the provider prefix back on a provider-stripped model.
A sub-path provider (``bedrock/mantle/...``) leaves a slash in the remainder, so
treating any slash as "already qualified" would drop the prefix.
"""
"""Put the provider prefix back on a provider-stripped model."""
if not custom_llm_provider or model.startswith(f"{custom_llm_provider}/"):
return model
return f"{custom_llm_provider}/{model}"

View file

@ -561,10 +561,7 @@ class TestProviderResponseHeadersInHiddenParams:
class TestQualifyProviderStrippedModel:
"""#38829: an agentic follow-up re-dispatched the provider-stripped model. For providers
that route through a sub-path the remainder still holds a slash, and the old
"any slash means already qualified" test dropped the prefix, leaving a string
litellm.acompletion could not resolve a provider from."""
"""#38829: a model whose remainder still holds a slash must keep its provider prefix."""
@pytest.mark.parametrize(
"model,provider,expected",
@ -590,7 +587,6 @@ class TestQualifyProviderStrippedModel:
assert qualify_provider_stripped_model(model, provider) == model
def test_a_provider_that_only_shares_a_prefix_is_still_qualified(self):
"""`openai` must not be read as a prefix of `openai_like`."""
assert qualify_provider_stripped_model("openai_like/foo", "openai") == "openai/openai_like/foo"
def test_no_provider_leaves_the_model_untouched(self):

View file

@ -4305,10 +4305,7 @@ async def test_async_text_to_speech_handler_records_upstream_response_headers():
class TestAgenticFollowUpKeepsTheProviderPrefix:
"""#38829: the follow-up re-dispatched the provider-stripped model. For a provider that
routes through a sub-path the remainder still holds a slash, so the old
"a slash means already qualified" test dropped the prefix and the follow-up raised
"LLM Provider NOT provided"."""
"""#38829: the follow-up must re-dispatch a model litellm.acompletion can route."""
@staticmethod
def _plan():
@ -4340,7 +4337,6 @@ class TestAgenticFollowUpKeepsTheProviderPrefix:
@pytest.mark.asyncio
async def test_a_sub_path_model_still_resolves_a_provider(self):
"""bedrock/mantle/... reaches the handler as mantle/..., which alone is unroutable."""
response = await self._run_followup("mantle/anthropic.claude-sonnet-5", "bedrock")
assert response.choices[0].message.content == "ok from followup"