51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
# Copyright 2026 zhaoxi826
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from kilostar.plugin.tool_plugin.base_tool import BaseToolData
|
|
from kilostar.utils.ray_hook import ray_actor_hook
|
|
from typing import List, Literal, Dict
|
|
|
|
|
|
class ApprovalToolData(BaseToolData):
|
|
"""``approval`` 工具的元数据：默认面向 control/consciousness 两类节点开放。"""
|
|
|
|
is_system: bool = True
|
|
action_scope: List[
|
|
Literal[
|
|
"control_node",
|
|
"consciousness_node",
|
|
"regulatory_node",
|
|
"growth_node",
|
|
"",
|
|
]
|
|
] = ["control_node", "consciousness_node"]
|
|
config_args: Dict[str, str] = {}
|
|
category: str = "system"
|
|
|
|
|
|
async def approval(message: str, trace_id: str) -> str:
|
|
"""
|
|
当任务存在某些高风险操作或者计划需要让用户审批，发送请求给用户等待用户审批
|
|
Args:
|
|
message: 发送给用户的请求
|
|
trace_id:
|
|
|
|
Returns:
|
|
用户的审批结果
|
|
"""
|
|
actor_list = ray_actor_hook("global_state_machine")
|
|
await actor_list.global_state_machine.put_pending.remote(trace_id, message)
|
|
reply = await actor_list.global_state_machine.get_received.remote(trace_id)
|
|
return reply
|