scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-13 03:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 66

fix(ui): update RouterSettingsAccordion FallbackSelectionForm mock and apply ruff/prettier formatting

Browse source 
Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-10-10 02:37:40 +00:00
parent 0bd1fcd940
commit e4c094da71
4 changed files with 23 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
litellm/router_utils/fallback_event_handlers.py
View file

@ -554,21 +554,27 @@ def get_fallback_model_group(fallbacks: list[Any], model_group: str) -> tuple[li
"""
generic_fallback_idx: int | None = None
stripped_model_fallback: list[str] | None = None
pattern_model_fallback: list[str] | None = None # mutable-ok: resolved inside the loop like stripped_model_fallback # rebind-ok: one assignment inside the loop
pattern_model_fallback: list[str] | None = (
None # mutable-ok: resolved inside the loop like stripped_model_fallback # rebind-ok: one assignment inside the loop
)
fallback_model_group: list[str] | None = None
fallback_keys: Final = tuple(next(iter(item)) for item in fallbacks if isinstance(item, dict) and item)
prefixed_model_group: Final = _provider_prefixed_model_group(model_group, fallback_keys)
## check for specific model group-specific fallbacks
for idx, item in enumerate(fallbacks):
if isinstance(item, dict):
fallback_key = cast(str, next(iter(item))) # cast-ok: fallback entries are single-key dicts; the key is always a str
fallback_key = cast(
str, next(iter(item))
) # cast-ok: fallback entries are single-key dicts; the key is always a str
if fallback_key == model_group: # check exact match
fallback_model_group = item[model_group]
break
elif fallback_key == prefixed_model_group or _check_stripped_model_group(
model_group=model_group, fallback_key=fallback_key
): # check generic fallback
stripped_model_fallback = cast(list[str], item[fallback_key]) # cast-ok: fallback chains are lists of model names
stripped_model_fallback = cast(
list[str], item[fallback_key]
) # cast-ok: fallback chains are lists of model names
elif fallback_key == "*": # check generic fallback
generic_fallback_idx = idx
elif (
@ -576,7 +582,9 @@ def get_fallback_model_group(fallbacks: list[Any], model_group: str) -> tuple[li
and "*" in fallback_key
and fnmatch.fnmatchcase(model_group, fallback_key)
):
pattern_model_fallback = cast(list[str], item[fallback_key]) # cast-ok: fallback chains are lists of model names
pattern_model_fallback = cast(
list[str], item[fallback_key]
) # cast-ok: fallback chains are lists of model names
elif isinstance(item, str):
fallback_model_group = [item]
## if none, check for generic fallback
@ -584,7 +592,9 @@ def get_fallback_model_group(fallbacks: list[Any], model_group: str) -> tuple[li
if stripped_model_fallback is not None:
fallback_model_group = stripped_model_fallback
elif pattern_model_fallback is not None:
fallback_model_group = pattern_model_fallback # rebind-ok: one assignment per priority branch, same as stripped and generic
fallback_model_group = (
pattern_model_fallback # rebind-ok: one assignment per priority branch, same as stripped and generic
)
elif generic_fallback_idx is not None:
fallback_model_group = fallbacks[generic_fallback_idx]["*"]

1
ui/litellm-dashboard/src/components/Settings/RouterSettings/Fallbacks/AddFallbacks.tsx
View file

@ -63,7 +63,6 @@ export default function AddFallbacks({ accessToken, value = [], onChange }: AddF
}
}, [accessToken, isModalVisible]);
const handleCancel = () => {
setIsModalVisible(false);
// Reset to initial state

4
ui/litellm-dashboard/src/components/Settings/RouterSettings/Fallbacks/fallbackOptions.test.ts
View file

@ -63,9 +63,7 @@ describe("fallbackChainOptions", () => {
describe("resolveFallbackProvider", () => {
it("prefers the cost map litellm_provider", () => {
expect(
resolveFallbackProvider("gpt-4", { "gpt-4": { litellm_provider: "azure" } }, modelGroups),
).toBe("azure");
expect(resolveFallbackProvider("gpt-4", { "gpt-4": { litellm_provider: "azure" } }, modelGroups)).toBe("azure");
});
it("falls back to the model group providers when the cost map misses", () => {

9
ui/litellm-dashboard/src/components/common_components/RouterSettingsAccordion.test.tsx
View file

@ -16,8 +16,13 @@ vi.mock("@/components/llm_calls/fetch_models", () => ({
}));
vi.mock("../Settings/RouterSettings/Fallbacks/FallbackSelectionForm", () => ({
FallbackSelectionForm: ({ availableModels }: { availableModels: string[] }) => (
<div data-testid="available-models">{availableModels.join(",")}</div>
FallbackSelectionForm: ({ modelGroups }: { modelGroups: { model_group: string }[] }) => (
<div data-testid="available-models">
{modelGroups
.map((modelGroup) => modelGroup.model_group)
.sort()
.join(",")}
</div>
),
}));