uvos/vibe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

v2.9.4 (#669)

Browse source 
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Jean-Baptiste Muscat <jeanbaptiste.muscatdupuis@mistral.ai>
Co-authored-by: JeroenvdV <JeroenvdV@users.noreply.github.com>
Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai>
Co-authored-by: Val <102326092+vdeva@users.noreply.github.com>
Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com>
Co-authored-by: allansimon-mistral <allan.simon@ext.mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Clément Sirieix 2026-05-05 14:40:11 +02:00 • committed by GitHub
parent 71f373c60c
commit 4972dd5694
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
50 changed files with 2892 additions and 842 deletions
Download patch file Download diff file Expand all files Collapse all files

45
tests/test_agent_observer_streaming.py
View file

@ -536,6 +536,51 @@ async def test_context_too_long_non_streaming(observer_capture) -> None:
assert agent.session_logger.save_interaction.await_count == 1
class _NonRetryableError(Exception):
# Mimics Temporal's ``ApplicationError(non_retryable=True)`` without
# pulling temporalio into vibe's test deps. The wrap-site check relies
# on the truthy ``non_retryable`` attribute, not the concrete type.
non_retryable = True
@pytest.mark.asyncio
async def test_non_retryable_passes_through_streaming(observer_capture) -> None:
observed, observer = observer_capture
backend = FakeBackend(exception_to_raise=_NonRetryableError("auth failed"))
agent = build_test_agent_loop(
config=make_config(),
backend=backend,
message_observer=observer,
enable_streaming=True,
)
agent.session_logger.save_interaction = AsyncMock(return_value=None)
with pytest.raises(_NonRetryableError, match="auth failed"):
[_ async for _ in agent.act("Trigger non-retryable failure while streaming")]
assert [role for role, _ in observed] == [Role.system, Role.user]
assert agent.session_logger.save_interaction.await_count == 1
@pytest.mark.asyncio
async def test_non_retryable_passes_through_non_streaming(observer_capture) -> None:
observed, observer = observer_capture
backend = FakeBackend(exception_to_raise=_NonRetryableError("auth failed"))
agent = build_test_agent_loop(
config=make_config(),
backend=backend,
message_observer=observer,
enable_streaming=False,
)
agent.session_logger.save_interaction = AsyncMock(return_value=None)
with pytest.raises(_NonRetryableError, match="auth failed"):
[_ async for _ in agent.act("Trigger non-retryable failure without streaming")]
assert [role for role, _ in observed] == [Role.system, Role.user]
assert agent.session_logger.save_interaction.await_count == 1
def _snapshot_events(events: list) -> list[tuple[str, str]]:
return [
(type(e).__name__, e.content)