Timothy Jaeryang Baek před 1 měsícem
rodič
revize
0ea421ea20
1 změnil soubory, kde provedl 7 přidání a 6 odebrání
  1. 7 6
      backend/open_webui/routers/ollama.py

+ 7 - 6
backend/open_webui/routers/ollama.py

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