This commit is contained in:
Timothy Jaeryang Baek 2025-08-25 01:12:14 +04:00
parent 3da6049a4a
commit 0ea421ea20

View file

@ -329,12 +329,13 @@ def merge_ollama_models_lists(model_lists):
for idx, model_list in enumerate(model_lists): for idx, model_list in enumerate(model_lists):
if model_list is not None: if model_list is not None:
for model in model_list: for model in model_list:
id = model["model"] id = model.get("model")
if id not in merged_models: if id is not None:
model["urls"] = [idx] if id not in merged_models:
merged_models[id] = model model["urls"] = [idx]
else: merged_models[id] = model
merged_models[id]["urls"].append(idx) else:
merged_models[id]["urls"].append(idx)
return list(merged_models.values()) return list(merged_models.values())