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

v1.2.0

Browse source 
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai>
Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai>
Co-Authored-By: Kracekumar <kracethekingmaker@gmail.com>
This commit is contained in:
Quentin 2025-12-14 00:54:42 +01:00 • committed by Mathias Gesbert
parent 661588de0c
commit d8dbeeb31e
91 changed files with 4521 additions and 873 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_agent_observer_streaming.py
View file

@ -18,6 +18,7 @@ from vibe.core.middleware import (
MiddlewareResult,
ResetReason,
)
from vibe.core.modes import AgentMode
from vibe.core.tools.base import BaseToolConfig, ToolPermission
from vibe.core.tools.builtins.todo import TodoArgs
from vibe.core.types import (
@ -187,7 +188,7 @@ async def test_act_handles_streaming_with_tool_call_events_in_sequence() -> None
tools={"todo": BaseToolConfig(permission=ToolPermission.ALWAYS)},
),
backend=backend,
auto_approve=True,
mode=AgentMode.AUTO_APPROVE,
enable_streaming=True,
)
@ -229,7 +230,7 @@ async def test_act_handles_tool_call_chunk_with_content() -> None:
tools={"todo": BaseToolConfig(permission=ToolPermission.ALWAYS)},
),
backend=backend,
auto_approve=True,
mode=AgentMode.AUTO_APPROVE,
enable_streaming=True,
)
@ -274,7 +275,7 @@ async def test_act_merges_streamed_tool_call_arguments() -> None:
tools={"todo": BaseToolConfig(permission=ToolPermission.ALWAYS)},
),
backend=backend,
auto_approve=True,
mode=AgentMode.AUTO_APPROVE,
enable_streaming=True,
)
@ -366,7 +367,7 @@ async def test_act_handles_user_cancellation_during_streaming() -> None:
tools={"todo": BaseToolConfig(permission=ToolPermission.ASK)},
),
backend=backend,
auto_approve=False,
mode=AgentMode.DEFAULT,
enable_streaming=True,
)
middleware = CountingMiddleware()
@ -392,7 +393,7 @@ async def test_act_handles_user_cancellation_during_streaming() -> None:
assert events[-1].skipped is True
assert events[-1].skip_reason is not None
assert "<user_cancellation>" in events[-1].skip_reason
assert agent.interaction_logger.save_interaction.await_count == 2
assert agent.interaction_logger.save_interaction.await_count == 1
@pytest.mark.asyncio