feat: add prompt opening behavior modes
This commit is contained in:
@@ -366,6 +366,7 @@ class PromptBrainTests(unittest.IsolatedAsyncioTestCase):
|
||||
tools=[tool],
|
||||
startup={
|
||||
"execution_mode": "sequential",
|
||||
"opening_mode": "confirmation",
|
||||
"entry_mode": "generate",
|
||||
"opening_message": {
|
||||
"title": "重要提示",
|
||||
@@ -485,6 +486,7 @@ class PromptBrainTests(unittest.IsolatedAsyncioTestCase):
|
||||
type="prompt",
|
||||
greeting="请先确认",
|
||||
startup={
|
||||
"opening_mode": "confirmation",
|
||||
"entry_mode": "wait_user",
|
||||
"opening_message": {
|
||||
"title": "重要提示",
|
||||
@@ -527,7 +529,10 @@ class PromptBrainTests(unittest.IsolatedAsyncioTestCase):
|
||||
cfg = AssistantConfig(
|
||||
type="prompt",
|
||||
greeting="固定开场白",
|
||||
startup={"entry_mode": "generate"},
|
||||
startup={
|
||||
"opening_mode": "playback",
|
||||
"entry_mode": "generate",
|
||||
},
|
||||
)
|
||||
brain = build_brain(cfg)
|
||||
queued = []
|
||||
@@ -561,45 +566,22 @@ class PromptBrainTests(unittest.IsolatedAsyncioTestCase):
|
||||
1,
|
||||
)
|
||||
|
||||
async def test_opening_confirmation_can_keep_speech_playing_before_entry(self):
|
||||
async def test_interruptible_opening_user_input_suppresses_entry_reply(self):
|
||||
cfg = AssistantConfig(
|
||||
type="prompt",
|
||||
greeting="必须完整播放的开场白",
|
||||
greeting="可以打断的开场白",
|
||||
startup={
|
||||
"opening_mode": "interruptible",
|
||||
"entry_mode": "generate",
|
||||
"opening_message": {
|
||||
"title": "重要提示",
|
||||
"message": "请确认已阅读。",
|
||||
"skip_speech_on_confirm": False,
|
||||
},
|
||||
},
|
||||
)
|
||||
brain = build_brain(cfg)
|
||||
queued = []
|
||||
input_states = []
|
||||
message_displayed = asyncio.Event()
|
||||
|
||||
class TrackedCallEnd(FakeCallEnd):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.playback_completion = None
|
||||
|
||||
def track_speech(self):
|
||||
self.tracked_speeches += 1
|
||||
self.playback_completion = asyncio.get_running_loop().create_future()
|
||||
return self.playback_completion
|
||||
|
||||
class FakeClientTools:
|
||||
async def call(self, *_args, **options):
|
||||
self.options = options
|
||||
message_displayed.set()
|
||||
return {"status": "ok", "data": {"action": "confirmed"}}
|
||||
|
||||
async def queue_frame(frame):
|
||||
queued.append(frame)
|
||||
|
||||
call_end = TrackedCallEnd()
|
||||
client_tools = FakeClientTools()
|
||||
await brain.setup(
|
||||
cfg,
|
||||
BrainRuntime(
|
||||
@@ -608,24 +590,52 @@ class PromptBrainTests(unittest.IsolatedAsyncioTestCase):
|
||||
queue_frame=queue_frame,
|
||||
set_system_prompt=lambda _prompt: None,
|
||||
set_tools=lambda _tools: None,
|
||||
call_end=call_end,
|
||||
client_tools=client_tools,
|
||||
call_end=FakeCallEnd(),
|
||||
set_input_enabled=input_states.append,
|
||||
),
|
||||
)
|
||||
|
||||
await brain.on_connected(greeting_pending=False)
|
||||
opening_task = asyncio.create_task(brain.on_client_ready())
|
||||
await message_displayed.wait()
|
||||
await asyncio.sleep(0)
|
||||
await brain.on_connected(greeting_pending=True)
|
||||
self.assertEqual(input_states, [])
|
||||
|
||||
await brain.on_interruption_processed()
|
||||
await brain.on_greeting_finished()
|
||||
|
||||
self.assertFalse(client_tools.options["interrupt_on_result"])
|
||||
self.assertFalse(opening_task.done())
|
||||
self.assertFalse(any(isinstance(frame, LLMRunFrame) for frame in queued))
|
||||
|
||||
call_end.playback_completion.set_result(None)
|
||||
await opening_task
|
||||
self.assertEqual(input_states, [False, True])
|
||||
async def test_interruptible_opening_natural_finish_runs_entry_behavior(self):
|
||||
cfg = AssistantConfig(
|
||||
type="prompt",
|
||||
greeting="可以打断的开场白",
|
||||
startup={
|
||||
"opening_mode": "interruptible",
|
||||
"entry_mode": "generate",
|
||||
},
|
||||
)
|
||||
brain = build_brain(cfg)
|
||||
queued = []
|
||||
input_states = []
|
||||
|
||||
async def queue_frame(frame):
|
||||
queued.append(frame)
|
||||
|
||||
await brain.setup(
|
||||
cfg,
|
||||
BrainRuntime(
|
||||
context=LLMContext(messages=[]),
|
||||
llm=FakeLLM(),
|
||||
queue_frame=queue_frame,
|
||||
set_system_prompt=lambda _prompt: None,
|
||||
set_tools=lambda _tools: None,
|
||||
call_end=FakeCallEnd(),
|
||||
set_input_enabled=input_states.append,
|
||||
),
|
||||
)
|
||||
|
||||
await brain.on_connected(greeting_pending=True)
|
||||
await brain.on_greeting_finished()
|
||||
|
||||
self.assertEqual(input_states, [])
|
||||
self.assertEqual(
|
||||
sum(isinstance(frame, LLMRunFrame) for frame in queued),
|
||||
1,
|
||||
|
||||
Reference in New Issue
Block a user