mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-13 23:11:40 +00:00
refactor(websearch): cut the helper and test docstrings to one line
This commit is contained in:
parent
f287132412
commit
cbdffd27f9
3 changed files with 3 additions and 15 deletions
|
|
@ -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}"
|
||||
|
|
|
|||
|
|
@ -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):
|
||||
|
|
|
|||
|
|
@ -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"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue