Преглед изворни кода

refactor: standardize filter valve retrieval logic

landerrosette пре 1 месец
родитељ
комит
2b7dd6e399
1 измењених фајлова са 3 додато и 3 уклоњено
  1. 3 3
      backend/open_webui/utils/filter.py

+ 3 - 3
backend/open_webui/utils/filter.py

@@ -12,9 +12,9 @@ log.setLevel(SRC_LOG_LEVELS["MAIN"])
 def get_sorted_filter_ids(model: dict):
     def get_priority(function_id):
         function = Functions.get_function_by_id(function_id)
-        if function is not None and hasattr(function, "valves"):
-            # TODO: Fix FunctionModel to include vavles
-            return (function.valves if function.valves else {}).get("priority", 0)
+        if function is not None:
+            valves = Functions.get_function_valves_by_id(function_id)
+            return valves.get("priority", 0) if valves else 0
         return 0
 
     filter_ids = [function.id for function in Functions.get_global_filter_functions()]