Enhance: tools wecom bot support markdown message (#5791)

This commit is contained in:
quicksand 2024-07-01 18:19:47 +08:00 committed by GitHub
parent 1d3e96ffa6
commit 2a27568537
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 39 additions and 7 deletions

View File

@ -22,7 +22,21 @@ class WecomGroupBotTool(BuiltinTool):
return self.create_text_message(
f'Invalid parameter hook_key ${hook_key}, not a valid UUID')
msgtype = 'text'
message_type = tool_parameters.get('message_type', 'text')
if message_type == 'markdown':
payload = {
"msgtype": 'markdown',
"markdown": {
"content": content,
}
}
else:
payload = {
"msgtype": 'text',
"text": {
"content": content,
}
}
api_url = 'https://qyapi.weixin.qq.com/cgi-bin/webhook/send'
headers = {
'Content-Type': 'application/json',
@ -30,12 +44,6 @@ class WecomGroupBotTool(BuiltinTool):
params = {
'key': hook_key,
}
payload = {
"msgtype": msgtype,
"text": {
"content": content,
}
}
try:
res = httpx.post(api_url, headers=headers, params=params, json=payload)

View File

@ -38,3 +38,27 @@ parameters:
pt_BR: Content to sent to the group.
llm_description: Content of the message
form: llm
- name: message_type
type: select
default: text
required: true
label:
en_US: Wecom Group bot message type
zh_Hans: 群机器人webhook的消息类型
pt_BR: Wecom Group bot message type
human_description:
en_US: Wecom Group bot message type
zh_Hans: 群机器人webhook的消息类型
pt_BR: Wecom Group bot message type
options:
- value: text
label:
en_US: Text
zh_Hans: 文本
pt_BR: Text
- value: markdown
label:
en_US: Markdown
zh_Hans: Markdown
pt_BR: Markdown
form: form