瀏覽代碼

Merge pull request #11198 from xinhai-ai/refa_filters

chore: move the filter_type check earlier to avoid unnecessary valve reads.
Timothy Jaeryang Baek 1 月之前
父節點
當前提交
c9e2ba56c6
共有 1 個文件被更改,包括 5 次插入4 次删除
  1. 5 4
      backend/open_webui/utils/filter.py

+ 5 - 4
backend/open_webui/utils/filter.py

@@ -49,6 +49,11 @@ async def process_filter_functions(
             function_module, _, _ = load_function_module_by_id(filter_id)
             function_module, _, _ = load_function_module_by_id(filter_id)
             request.app.state.FUNCTIONS[filter_id] = function_module
             request.app.state.FUNCTIONS[filter_id] = function_module
 
 
+        # Prepare handler function
+        handler = getattr(function_module, filter_type, None)
+        if not handler:
+            continue
+
         # Check if the function has a file_handler variable
         # Check if the function has a file_handler variable
         if filter_type == "inlet" and hasattr(function_module, "file_handler"):
         if filter_type == "inlet" and hasattr(function_module, "file_handler"):
             skip_files = function_module.file_handler
             skip_files = function_module.file_handler
@@ -60,10 +65,6 @@ async def process_filter_functions(
                 **(valves if valves else {})
                 **(valves if valves else {})
             )
             )
 
 
-        # Prepare handler function
-        handler = getattr(function_module, filter_type, None)
-        if not handler:
-            continue
 
 
         try:
         try:
             # Prepare parameters
             # Prepare parameters