olivebot/agent/tools/Say.py
xszyou e48b8d8a2d 第4次更新
1、优化prompt;
2、解决日程删除错误;
3、iotm demo tool整体更新。
2023-12-18 16:52:53 +08:00

33 lines
826 B
Python

import os
from typing import Any
import requests
from langchain.tools import BaseTool
import fay_booter
from core.interact import Interact
class Say(BaseTool):
name = "Say"
description = """此工具用于沟通及提醒,使用时请传入说话内容作为参数,例如:“该下班了,请注意休息”"""
def __init__(self):
super().__init__()
async def _arun(self, *args: Any, **kwargs: Any) -> Any:
# 用例中没有用到 arun 不予具体实现
pass
def _run(self, para: str) -> str:
interact = Interact("agent", 1, {'user': '', 'msg': para})
fay_booter.feiFei.on_interact(interact)
return "语音输出了:" + para
if __name__ == "__main__":
tool = Say()
info = tool.run("该下班了,请注意休息")
print(info)