2024-11-20 23:44:47 +08:00
|
|
|
import abc
|
|
|
|
|
from typing import Any
|
|
|
|
|
from langchain.tools import BaseTool
|
|
|
|
|
import requests
|
|
|
|
|
import json
|
|
|
|
|
|
2024-12-11 15:29:38 +08:00
|
|
|
url = "http://127.0.0.1:4008/send"
|
2024-11-20 23:44:47 +08:00
|
|
|
headers = {'Content-Type': 'application/json'}
|
|
|
|
|
data = {
|
|
|
|
|
"message": "你好",
|
|
|
|
|
"receiver": "@2efc4e10cf2eafd0b0125930e4b96ed0cebffa75b2fd272590e38763225a282b"
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SendWX(BaseTool, abc.ABC):
|
|
|
|
|
name = "SendWX"
|
|
|
|
|
description = "给主人微信发送消息，传入参数是:('消息内容')"
|
|
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
super().__init__()
|
|
|
|
|
|
|
|
|
|
async def _arun(self, *args: Any, **kwargs: Any) -> Any:
|
|
|
|
|
# 用例中没有用到 arun 不予具体实现
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
def _run(self, para) -> str:
|
|
|
|
|
global data
|
|
|
|
|
data['message'] = para
|
|
|
|
|
response = requests.post(url, headers=headers, data=json.dumps(data))
|
|
|
|
|
return "成功给主人,发送微信消息：{}".format(para)
|
|
|
|
|
|
