mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
bug fix de depluciate model list (#9775)
This commit is contained in:
parent
34bdf36eab
commit
3a7061a05c
3 changed files with 91 additions and 4 deletions
|
@ -13,7 +13,7 @@ model/{model_id}/update - PATCH endpoint for model update.
|
|||
import asyncio
|
||||
import json
|
||||
import uuid
|
||||
from typing import Literal, Optional, Union, cast
|
||||
from typing import Dict, List, Literal, Optional, Union, cast
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException, Request, status
|
||||
from pydantic import BaseModel
|
||||
|
@ -846,3 +846,24 @@ async def update_model(
|
|||
param=getattr(e, "param", "None"),
|
||||
code=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
|
||||
|
||||
def _deduplicate_litellm_router_models(models: List[Dict]) -> List[Dict]:
|
||||
"""
|
||||
Deduplicate models based on their model_info.id field.
|
||||
Returns a list of unique models keeping only the first occurrence of each model ID.
|
||||
|
||||
Args:
|
||||
models: List of model dictionaries containing model_info
|
||||
|
||||
Returns:
|
||||
List of deduplicated model dictionaries
|
||||
"""
|
||||
seen_ids = set()
|
||||
unique_models = []
|
||||
for model in models:
|
||||
model_id = model.get("model_info", {}).get("id", None)
|
||||
if model_id is not None and model_id not in seen_ids:
|
||||
unique_models.append(model)
|
||||
seen_ids.add(model_id)
|
||||
return unique_models
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue