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 f287132412
commit cbdffd27f9
3 changed files with 3 additions and 15 deletions

View file

@ -30,11 +30,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

@ -410,10 +410,7 @@ class TestIsExpectedClientError:
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",
@ -439,7 +436,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

@ -3716,10 +3716,7 @@ def test_image_edit_handler_keeps_the_sync_transform():
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():
@ -3751,7 +3748,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"