Skip to content

修复 Brak 自定义通知 url 结构 #1437

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Zeyi-Lin merged 2 commits into from
Feb 3, 2026
Merged

修复 Brak 自定义通知 url 结构 #1437

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ef3bfe7
Fix notification URL construction logic
h27539 Feb 1, 2026
cb264aa
update
Zeyi-Lin Feb 3, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 21 additions & 8 deletions swanlab/plugin/notification.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -463,7 +463,6 @@ def __str__(self) -> str:
}
def __init__(
self,
key: str,
url: str = 'https://api.day.app',
title: str = 'SwanLab',
bark_level: Literal['critical', 'active', 'timeSensitive', 'passive'] = 'active',
Expand All @@ -475,7 +474,6 @@ def __init__(
"""
初始化 Bark callback 配置
详细内容见 https://bark.day.app/#/tutorial?id=%e8%af%b7%e6%b1%82%e5%8f%82%e6%95%b0
:param key: bark中的device key
:param url: bark推送链接
:param title: 推送的通知标题，默认为SwanLab
:param bark_level: bark推送等级
Expand All @@ -484,7 +482,6 @@ def __init__(
:param click_jump: 是否点击链接跳转，默认为True
:param language: 推送语言
"""
self.key = key
self.url = url
self.title = title
if bark_level not in ['critical', 'active', 'timeSensitive', 'passive']:
Expand Down Expand Up @@ -532,19 +529,18 @@ def _create_notification_message(self, error: Optional[str] = None) -> Dict[str,
'level': self.bark_level,
'icon': self.icon,
'group': group,
'device_key': self.key,
'url': exp_link if self.click_jump else None,
}
data = {k: v for k, v in data.items() if v is not None}
return data


def send_notification(self, data: dict):
def _send_notification(self, data: dict):
"""发送通知，也可以直接构建数据字典进行消息发送"""
if self.url.endswith('/'):
url = self.url + 'push'
url = self.url
else:
url = self.url + '/push'
url = self.url + '/'

headers = {
'Content-Type': 'application/json; charset=utf-8',
Expand All @@ -562,6 +558,23 @@ def send_notification(self, data: dict):
return
print("✅ Bark sending successfully")

def send_msg(self, content: str, title: str="SwanLab Message"):
"""发送通知，也可以直接构建数据字典进行消息发送"""
url = self.url.rstrip('/') + '/'
if title:
url += f"{title}/{content}"
else:
url += content

resp = requests.post(
url=url,
)
resp.raise_for_status()
result: Dict[str, Any] = resp.json()
if result.get("errcode") and result["errcode"] != 0:
print(f"❌ Bark sending failed: {result.get('errmsg')}")
return
print("✅ Bark sending successfully")

def on_init(self, proj_name: str, workspace: str, public: Optional[bool] = None, logdir: Optional[str] = None, *args, **kwargs):
self.project = proj_name
Expand All @@ -582,7 +595,7 @@ def before_init_experiment(

def on_stop(self, error: Optional[str] = None, *args, **kwargs):
content = self._create_notification_message(error)
self.send_notification(content)
self._send_notification(content)


class TelegramBot:
Expand Down