浏览代码

refactor: replace print statements with logging

hwzhuhao 5 月之前
父节点
当前提交
fb2950201a
共有 2 个文件被更改,包括 6 次插入4 次删除
  1. 1 1
      backend/open_webui/utils/middleware.py
  2. 5 3
      backend/open_webui/utils/tools.py

+ 1 - 1
backend/open_webui/utils/middleware.py

@@ -651,7 +651,7 @@ def apply_params_to_form_data(form_data, model):
                     convert_logit_bias_input_to_json(params["logit_bias"])
                 )
             except Exception as e:
-                print(f"Error parsing logit_bias: {e}")
+                log.exception(f"Error parsing logit_bias: {e}")
 
     return form_data
 

+ 5 - 3
backend/open_webui/utils/tools.py

@@ -37,6 +37,7 @@ from open_webui.models.tools import Tools
 from open_webui.models.users import UserModel
 from open_webui.utils.plugin import load_tool_module_by_id
 from open_webui.env import (
+    SRC_LOG_LEVELS,
     AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA,
     AIOHTTP_CLIENT_SESSION_TOOL_SERVER_SSL,
 )
@@ -44,6 +45,7 @@ from open_webui.env import (
 import copy
 
 log = logging.getLogger(__name__)
+log.setLevel(SRC_LOG_LEVELS["MODELS"])
 
 
 def get_async_tool_function_and_apply_extra_params(
@@ -477,7 +479,7 @@ async def get_tool_server_data(token: str, url: str) -> Dict[str, Any]:
         "specs": convert_openapi_to_tool_payload(res),
     }
 
-    print("Fetched data:", data)
+    log.info("Fetched data:", data)
     return data
 
 
@@ -510,7 +512,7 @@ async def get_tool_servers_data(
     results = []
     for (idx, server, url, _), response in zip(server_entries, responses):
         if isinstance(response, Exception):
-            print(f"Failed to connect to {url} OpenAPI tool server")
+            log.error(f"Failed to connect to {url} OpenAPI tool server")
             continue
 
         results.append(
@@ -620,5 +622,5 @@ async def execute_tool_server(
 
     except Exception as err:
         error = str(err)
-        print("API Request Error:", error)
+        log.exception("API Request Error:", error)
         return {"error": error}