用企业微信机器人获取企业微信群聊天记录并发送给gpt然后将gpt返回的内容发到企业微信群编写python代码
以下是基于企业微信API和GPT-2模型的Python代码,可以实现自动获取企业微信群聊天记录并发送给GPT-2,然后将GPT-2返回的内容发到企业微信群中。
首先,需要准备好企业微信的API密钥和机器人的ID,可以在企业微信管理后台中获取。
import requests
import json
# 企业微信API密钥和机器人ID
CORP_ID = 'your_corp_id'
AGENT_ID = 'your_agent_id'
SECRET = 'your_secret'
# 企业微信API基础地址
API_BASE = 'https://qyapi.weixin.qq.com/cgi-bin'
# GPT-2模型的API地址
GPT_API = 'http://localhost:5000/generate'
# 获取企业微信API访问令牌
def get_access_token():
url = f'{API_BASE}/gettoken?corpid={CORP_ID}&corpsecret={SECRET}'
res = requests.get(url).json()
if res.get('errcode') == 0:
return res.get('access_token')
else:
print(res.get('errmsg'))
return None
# 获取企业微信群聊天记录
def get_chat_history(chat_id, token):
url = f'{API_BASE}/appchat/getchatrecord?access_token={token}'
data = {
'chatid': chat_id,
'starttime': '0',
'endtime': '9999999999',
'limit': 1000,
'msgid': 1
}
res = requests.post(url, json=data).json()
if res.get('errcode') == 0:
return res.get('recordlist')
else:
print(res.get('errmsg'))
return None
# 发送企业微信消息
def send_message(chat_id, content, token):
url = f'{API_BASE}/message/send?access_token={token}'
data = {
'touser': '',
'toparty': chat_id,
'totag': '',
'msgtype': 'text',
'agentid': AGENT_ID,
'text': {
'content': content
},
'safe': 0,
'enable_id_trans': 0,
'enable_duplicate_check': 0,
'duplicate_check_interval': 1800
}
res = requests.post(url, json=data).json()
if res.get('errcode') == 0:
print('Message sent successfully.')
else:
print(res.get('errmsg'))
# 获取企业微信API访问令牌
token = get_access_token()
# 获取企业微信群聊天记录
chat_id = 'your_chat_id'
history = get_chat_history(chat_id, token)
# 提取聊天记录中的文本内容
chat_text = ''
for record in history:
if record.get('msgtype') == 'text':
chat_text += f"{record.get('sender')}: {record.get('text').get('content')}\n"
# 调用GPT-2模型生成文本
gpt_input = {'text': chat_text}
res = requests.post(GPT_API, json=gpt_input).json()
gpt_output = res.get('text')
# 发送GPT-2生成的文本到企业微信群
send_message(chat_id, gpt_output, token)
需要将代码中的your_corp_id、your_agent_id和your_secret替换为自己的企业微信API密钥和机器人ID,以及your_chat_id替换为要获取聊天记录的群的ID。
此外,还需要在本地启动一个GPT-2模型的API服务,代码中的GPT_API需要替换为实际的API地址。
以上代码可以放到一个定时任务中执行,可以实现自动化获取企业微信群聊天记录并发送给GPT-2,然后将GPT-2生成的文本自动发送到企业微信群中。这对于一些需要自动化处理大量聊天记录的场景非常有用
原文地址: https://www.cveoy.top/t/topic/deXi 著作权归作者所有。请勿转载和采集!