From de2825bb89d9fdd0586950e9c93eb7dae872b6a0 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Fri, 20 Dec 2024 23:09:40 -0800
Subject: [PATCH] refac

---
 backend/open_webui/utils/middleware.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index d876539de..9261c26b3 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -608,15 +608,16 @@ async def process_chat_response(request, response, user, events, metadata, tasks
                                 is None
                             ):
                                 webhook_url = Users.get_user_webhook_url_by_id(user.id)
+                                webui_url = f"{request.headers.get('x-forwarded-proto', request.url.scheme)}://{request.headers.get('x-forwarded-host', f'{request.client.host}:{request.url.port}')}"
                                 if webhook_url:
                                     post_webhook(
                                         webhook_url,
-                                        f"{title} - {WEBUI_URL}/c/{metadata['chat_id']}\n\n{content}",
+                                        f"{title} - {webui_url}/c/{metadata['chat_id']}\n\n{content}",
                                         {
                                             "action": "chat",
                                             "message": content,
                                             "title": title,
-                                            "url": f"{WEBUI_URL}/c/{metadata['chat_id']}",
+                                            "url": f"{webui_url}/c/{metadata['chat_id']}",
                                         },
                                     )