01c19c13e8
*实现agent ReAct与LLM chain自动切换逻辑✓ 聊天窗区分任务消息✓ 修复删除日程bug✓ 优化远程音频逻辑✓ 等待处理引入加载中效果✓ 优化prompt以解决日程任务递归调用问题✓ 修复一次性日程清除的bug✓
30 lines
716 B
Python
30 lines
716 B
Python
import os
|
|
from typing import Any
|
|
|
|
from langchain.tools import BaseTool
|
|
from agent import agent_service
|
|
|
|
|
|
class SetChatStatus(BaseTool):
|
|
name = "SetChatStatus"
|
|
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
|
|
return "设置聊天状态成功"
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tool = SetChatStatus()
|
|
info = tool.run("该下班了，请注意休息")
|
|
print(info)
|