22d1e4ce10
2024.01.01: openai token计算✓ 优化ReAct Agent 与 LLM Chain自动切换逻辑✓ *添加双记忆机制：长时记忆流及短时聊天记忆✓ 修复record.py asr bug✓ 提高远程音频（android 连接器）的稳定性✓ 修复执行时间计算bug✓ 优化语音输出逻辑✓
37 lines
1013 B
Python
37 lines
1013 B
Python
import os
|
|
from typing import Any
|
|
|
|
import requests
|
|
from langchain.tools import BaseTool
|
|
import fay_booter
|
|
from core.interact import Interact
|
|
from agent import agent_service
|
|
|
|
|
|
class Say(BaseTool):
|
|
name = "语音输出工具"
|
|
description = """此工具用于语音输出，使用时请传入说话内容作为参数，例如：“该下班了，请注意休息”"""
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
async def _arun(self, *args: Any, **kwargs: Any) -> Any:
|
|
# 用例中没有用到 arun 不予具体实现
|
|
pass
|
|
|
|
|
|
def _run(self, para: str) -> str:
|
|
agent_service.agent.is_chat = True
|
|
agent_service.agent.is_use_say_tool = True
|
|
agent_service.agent.say_tool_text = para
|
|
interact = Interact("audio", 1, {'user': '', 'msg': para})
|
|
fay_booter.feiFei.on_interact(interact)
|
|
return "语音输出了：" + para
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tool = Say()
|
|
info = tool.run("该下班了，请注意休息")
|
|
print(info)
|