UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-29 07:12:07 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge branch 'open-webui:dev' into dev

Browse Source
This commit is contained in:
Tiancong Li 2024-12-21 15:48:08 +08:00 committed by GitHub
commit 9d3ab2e40e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/utils/middleware.py
View File

@ -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']}",
},
)

6
backend/open_webui/utils/webhook.py
View File

@ -19,7 +19,11 @@ def post_webhook(url: str, message: str, event_data: dict) -> bool:
payload["text"] = message
# Discord Webhooks
elif "https://discord.com/api/webhooks" in url:
payload["content"] = message
payload["content"] = (
message
if len(message) > 2000
else f"{message[: 2000 - 14]}... (truncated)"
)
# Microsoft Teams Webhooks
elif "webhook.office.com" in url:
action = event_data.get("action", "undefined")