mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-20 08:58:15 +00:00
feat: 新增 task_query 工具(查询最近任务列表),为工作记忆链提供可视化入口
This commit is contained in:
@ -105,6 +105,11 @@ def execute_tool(graph: Any, tool_name: str, arguments: dict) -> str:
|
||||
result = execute_task_link_info(graph, arguments)
|
||||
return json.dumps(result, ensure_ascii=False, default=str)
|
||||
|
||||
elif tool_name == "task_query":
|
||||
recorder.record("query", tool_name, "")
|
||||
result = execute_task_query(graph, arguments)
|
||||
return json.dumps(result, ensure_ascii=False, default=str)
|
||||
|
||||
return f"未知工具: {tool_name}"
|
||||
|
||||
except Exception as e:
|
||||
@ -352,3 +357,17 @@ def execute_task_link_info(graph: Any, arguments: dict) -> dict:
|
||||
"linked_nodes": info_node_names,
|
||||
"details": result
|
||||
}
|
||||
|
||||
def execute_task_query(graph: Any, arguments: dict) -> dict:
|
||||
"""查询最近的任务列表"""
|
||||
limit = arguments.get("limit", 10)
|
||||
state_filter = arguments.get("state_filter")
|
||||
|
||||
result = graph.get_recent_tasks(limit=limit, state_filter=state_filter)
|
||||
|
||||
return {
|
||||
"status": "success",
|
||||
"tasks": result["tasks"],
|
||||
"total": result["total"],
|
||||
"message": f"找到 {result['total']} 个任务"
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user