34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
import abc
|
||
|
|
from typing import Any
|
||
|
|
from langchain.tools import BaseTool
|
||
|
|
import re
|
||
|
|
import random
|
||
|
|
class ToRemind(BaseTool, abc.ABC):
|
||
|
|
name: str = "ToRemind"
|
||
|
|
description: str = ("用于实时发送信息提醒主人做某事项（不能带时间），传入事项内容作为参数。")
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
async def _arun(self, *args: Any, **kwargs: Any) -> Any:
|
||
|
|
# 用例中没有用到 arun 不予具体实现
|
||
|
|
pass
|
||
|
|
|
||
|
|
def _run(self, para: str) -> str:
|
||
|
|
|
||
|
|
para = para.replace("提醒", "回复")
|
||
|
|
demo = [
|
||
|
|
"主人！是时候（事项内容）了喔~",
|
||
|
|
"亲爱的主人，现在是（事项内容）的时候啦！",
|
||
|
|
"嘿，主人，该（事项内容）了哦~",
|
||
|
|
"温馨提醒：（事项内容）的时间到啦，主人！",
|
||
|
|
"小提醒：主人，现在可以（事项内容）了~"
|
||
|
|
]
|
||
|
|
|
||
|
|
return f"直接以中文友善{para}，如"+ random.choice(demo)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
my_timer = ToRemind()
|
||
|
|
result = my_timer._run("提醒主人叫咖啡")
|
||
|
|
print(result)
|