[Feature] SwanLab Lark callback (#4830)

Author: dykderrick

docs/source/Instruction/命令行参数.md

@@ -515,6 +515,8 @@ soft overlong 奖励参数
 - swanlab_project: swanlab的project，需要在页面中预先创建好:[https://swanlab.cn/space/~](https://swanlab.cn/space/~)。
 - swanlab_workspace: 默认为None，会使用api-key对应的username。
 - swanlab_exp_name: 实验名，可以为空，为空时默认传入--output_dir的值。
+- swanlab_lark_webhook_url: 默认为None。swanlab的lark webhook url，用于推送实验结果到飞书。
+- swanlab_lark_secret: 默认为None。swanlab的lark secret，用于推送实验结果到飞书。
 - swanlab_mode: 可选cloud和local，云模式或者本地模式。
 
 ### 推理参数

docs/source_en/Instruction/Command-line-parameters.md

@@ -535,6 +535,8 @@ Soft overlong reward parameters:
 - **swanlab_project**: SwanLab's project, which needs to be created in advance on the page: [https://swanlab.cn/space/~](https://swanlab.cn/space/~)
 - **swanlab_workspace**: Defaults to `None`, will use the username associated with the API key
 - **swanlab_exp_name**: Experiment name, can be left empty. If empty, the value of `--output_dir` will be used by default
+- swanlab_lark_webhook_url: Defaults to None. SwanLab's Lark webhook URL, used for pushing experiment results to Lark.
+- swanlab_lark_secret: Defaults to None. SwanLab's Lark secret, used for pushing experiment results to Lark.
 - **swanlab_mode**: Optional values are `cloud` and `local`, representing cloud mode or local mode
 
 

swift/llm/argument/train_args.py

@@ -68,6 +68,8 @@ class SwanlabArguments:
     swanlab_project: Optional[str] = None
     swanlab_workspace: Optional[str] = None
     swanlab_exp_name: Optional[str] = None
+    swanlab_lark_webhook_url: Optional[str] = None
+    swanlab_lark_secret: Optional[str] = None
     swanlab_mode: Literal['cloud', 'local'] = 'cloud'
 
     def _init_swanlab(self):
@@ -80,6 +82,15 @@ def _init_swanlab(self):
         from swanlab.integration.transformers import SwanLabCallback
         if self.swanlab_token:
             swanlab.login(self.swanlab_token)
+
+        if self.swanlab_lark_webhook_url is not None:
+            from swanlab.plugin.notification import LarkCallback
+            lark_callback = LarkCallback(
+                webhook_url=self.swanlab_lark_webhook_url,
+                secret=self.swanlab_lark_secret,
+            )
+            swanlab.register_callbacks([lark_callback])
+
         INTEGRATION_TO_CALLBACK['swanlab'] = SwanLabCallback(
             project=self.swanlab_project,
             workspace=self.swanlab_workspace,

swift/ui/llm_train/report_to.py

@@ -47,6 +47,18 @@ class ReportTo(BaseUI):
                 'en': 'Experiment of SwanLab'
             },
         },
+        'swanlab_lark_webhook_url': {
+            'label': {
+                'zh': 'SwanLab飞书Webhook地址',
+                'en': 'Webhook URL of SwanLab Lark Callback'
+            },
+        },
+        'swanlab_lark_secret': {
+            'label': {
+                'zh': 'SwanLab飞书Secret',
+                'en': 'Secret of SwanLab Lark Callback'
+            },
+        },
         'swanlab_mode': {
             'label': {
                 'zh': 'SwanLab工作模式',
@@ -69,6 +81,9 @@ def do_build_ui(cls, base_tab: Type['BaseUI']):
                         scale=20)
                     gr.Textbox(elem_id='swanlab_token', lines=1, scale=20)
                     gr.Textbox(elem_id='swanlab_project', lines=1, scale=20)
+                with gr.Row():
+                    gr.Textbox(elem_id='swanlab_lark_webhook_url', lines=1, scale=20)
+                    gr.Textbox(elem_id='swanlab_lark_secret', lines=1, scale=20)
                 with gr.Row():
                     gr.Textbox(elem_id='swanlab_workspace', lines=1, scale=20)
                     gr.Textbox(elem_id='swanlab_exp_name', lines=1, scale=20)