40 lines
1.5 KiB
Python
40 lines
1.5 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 pydantic import Field
|
|
|
|
from pretor.tool_plugin.base_tool import BaseToolData
|
|
from pretor.utils.ray_hook import ray_actor_hook
|
|
from typing import List, Literal, Dict
|
|
|
|
class ApprovalToolData(BaseToolData):
|
|
is_system: bool = True
|
|
action_scope: List[Literal["control_node", "consciousness_node", "supervisory_node", "growth_node", "", ""]] = [
|
|
"control_node", "consciousness_node"]
|
|
config_args: Dict[str, str] = {}
|
|
|
|
|
|
async def approval(message: str, event_id: str) -> str:
|
|
"""
|
|
当任务存在某些高风险操作或者计划需要让用户审批，发送请求给用户等待用户审批
|
|
Args:
|
|
message: 发送给用户的请求
|
|
event_id:
|
|
|
|
Returns:
|
|
用户的审批结果
|
|
"""
|
|
actor_list = ray_actor_hook("global_state_machine")
|
|
await actor_list.global_state_machine.put_pending.remote(event_id, message)
|
|
reply = await actor_list.global_state_machine.get_received.remote(event_id)
|
|
return reply |