This commit is contained in:
Timothy J. Baek 2024-10-23 12:41:41 -07:00
parent 68a47f6a9f
commit b16c4bc1df

View File

@ -1103,7 +1103,7 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
if model["owned_by"] == "arena": if model["owned_by"] == "arena":
model_ids = model.get("info", {}).get("meta", {}).get("model_ids") model_ids = model.get("info", {}).get("meta", {}).get("model_ids")
filter_mode = model.get("info", {}).get("meta", {}).get("filter_mode") filter_mode = model.get("info", {}).get("meta", {}).get("filter_mode")
if filter_mode == 'exclude': if model_ids and filter_mode == 'exclude':
model_ids = [ model_ids = [
model["id"] model["id"]
for model in await get_all_models() for model in await get_all_models()