From 284d322a9f29e9ce24f77332d1eef9240eaa59d2 Mon Sep 17 00:00:00 2001 From: xiongyao Date: Mon, 28 Apr 2025 21:52:42 +0800 Subject: [PATCH] optimize something --- api/app_factory.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/api/app_factory.py b/api/app_factory.py index c467b8b0eb..e5e5eed054 100644 --- a/api/app_factory.py +++ b/api/app_factory.py @@ -29,19 +29,25 @@ def create_flask_app_with_configs() -> DifyApp: # log request data info request_id = get_request_id() - if "application/json" in request.headers.get("content-type", "").lower(): - logging.info( - f"[before request]|request_id: {request_id}," - f" method: {request.method}, url: {request.url}, request_data: {request.get_json()}" - ) - else: - logging.info(f"[before request]|request_id: {request_id}, method: {request.method}, url: {request.url}") + + try: + if request.method.lower() == "post" and "application/json" in request.headers.get("content-type", + "").lower(): + logging.info( + f"[before request]|request_id: {request_id}," + f" method: {request.method}, url: {request.url}, request_data: {request.get_json()}" + ) + + else: + logging.info(f"[before request]|request_id: {request_id}, method: {request.method}, url: {request.url}") + except Exception as e: + logging.info(f"before_request handler err {e}") # add extra `request_id` field for every response data @dify_app.after_request def add_extra_info(resp): obj = resp.get_json() - if isinstance(obj, dict): + if obj is not None and isinstance(obj, dict): request_id = get_request_id() obj["request_id"] = request_id resp.set_data(json.dumps(obj))