enh: webhook notification

backend/open_webui/models/users.py

@@ -168,6 +168,18 @@ class UsersTable:
         except Exception:
             return None
 
+    def get_user_webhook_url_by_id(self, id: str) -> Optional[str]:
+        try:
+            with get_db() as db:
+                user = db.query(User).filter_by(id=id).first()
+                return (
+                    user.settings.get("ui", {})
+                    .get("notifications", {})
+                    .get("webhook_url", None)
+                )
+        except Exception:
+            return None
+
     def update_user_role_by_id(self, id: str, role: str) -> Optional[UserModel]:
         try:
             with get_db() as db:

backend/open_webui/socket/main.py

@@ -249,3 +249,9 @@ def get_event_call(request_info):
         return response
 
     return __event_call__
+
+
+def get_user_id_from_session_pool(sid):
+    print("get_user_id_from_session_pool", sid)
+    print(SESSION_POOL.get(sid))
+    return SESSION_POOL.get(sid)

backend/open_webui/utils/middleware.py

@@ -18,15 +18,18 @@ from starlette.responses import Response, StreamingResponse
 
 
 from open_webui.models.chats import Chats
+from open_webui.models.users import Users
 from open_webui.socket.main import (
     get_event_call,
     get_event_emitter,
+    get_user_id_from_session_pool,
 )
 from open_webui.routers.tasks import (
     generate_queries,
     generate_title,
     generate_chat_tags,
 )
+from open_webui.utils.webhook import post_webhook
 
 
 from open_webui.models.users import UserModel
@@ -55,7 +58,12 @@ from open_webui.utils.plugin import load_function_module_by_id
 from open_webui.tasks import create_task
 
 from open_webui.config import DEFAULT_TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE
-from open_webui.env import SRC_LOG_LEVELS, GLOBAL_LOG_LEVEL, BYPASS_MODEL_ACCESS_CONTROL
+from open_webui.env import (
+    SRC_LOG_LEVELS,
+    GLOBAL_LOG_LEVEL,
+    BYPASS_MODEL_ACCESS_CONTROL,
+    WEBUI_URL,
+)
 from open_webui.constants import TASKS
 
 
@@ -593,6 +601,25 @@ async def process_chat_response(request, response, user, events, metadata, tasks
 
                         if done:
                             data = {"done": True, "content": content, "title": title}
+
+                            if (
+                                get_user_id_from_session_pool(metadata["session_id"])
+                                is None
+                            ):
+                                webhook_url = Users.get_user_webhook_url_by_id(user.id)
+                                print(f"webhook_url: {webhook_url}")
+                                if webhook_url:
+                                    post_webhook(
+                                        webhook_url,
+                                        f"{title} - {WEBUI_URL}/{metadata['chat_id']}\n\n{content}",
+                                        {
+                                            "action": "chat",
+                                            "message": content,
+                                            "title": title,
+                                            "url": f"{WEBUI_URL}/{metadata['chat_id']}",
+                                        },
+                                    )
+
                         else:
                             continue
 

backend/open_webui/utils/webhook.py

@@ -11,6 +11,7 @@ log.setLevel(SRC_LOG_LEVELS["WEBHOOK"])
 
 def post_webhook(url: str, message: str, event_data: dict) -> bool:
     try:
+        log.debug(f"post_webhook: {url}, {message}, {event_data}")
         payload = {}
 
         # Slack and Google Chat Webhooks