NCBM/plyngent
1
0
Fork 0
mirror of https://github.com/NCBM/plyngent.git synced 2026-07-23 05:55:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

core/cli: update display and state for streaming, error, cancel events

Browse Source
This commit is contained in:
worldmozara
2026-07-14 22:11:30 +08:00
parent 42b50716f4
commit ed028dc00b
3 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_agent/test_loop.py
+4 -4
View File
@@ -96,10 +96,10 @@ async def test_run_chat_loop_text_only() -> None:
]
)
messages: list[AnyChatMessage] = [UserChatMessage(content="hi")]
events = [e async for e in run_chat_loop(client, messages, model="m")]
assert isinstance(events[0], AssistantMessageEvent)
assert isinstance(events[1], TextDeltaEvent)
assert events[1].content == "hello"
events = [e async for e in run_chat_loop(client, messages, model="m", stream=False)]
assert isinstance(events[0], TextDeltaEvent)
assert events[0].content == "hello"
assert isinstance(events[1], AssistantMessageEvent)
assert len(messages) == 2 # noqa: PLR2004
assert len(client.calls) == 1