Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai>
Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai>
Co-Authored-By: Clément Drouin <clement.drouin@mistral.ai>
Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai>
Co-Authored-By: Clément Siriex <clement.sirieix@mistral.ai>
Co-Authored-By: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai>
Co-Authored-By: Thaddee Tyl <thaddee.tyl@gmail.com>
Co-Authored-By: David Brochart <david.brochart@gmail.com>
Co-Authored-By: Joseph Guhlin <joseph.guhlin@gmail.com>
Co-Authored-By: Thomas Kenbeek <thomaskenbeek@gmail.com>
Co-Authored-By: Remenby31 <baptiste.cruvellier31@gmail.com>
This commit is contained in:
Mathias Gesbert 2026-01-27 16:39:30 +01:00 committed by Mathias Gesbert
parent 79f215d91c
commit d33db9fff8
217 changed files with 16911 additions and 4305 deletions

View file

@ -9,7 +9,7 @@ import respx
from tests.mock.utils import mock_llm_chunk
from tests.stubs.fake_backend import FakeBackend
from vibe.core.agent import Agent
from vibe.core.agent_loop import AgentLoop
from vibe.core.config import (
ModelConfig,
ProviderConfig,
@ -294,7 +294,7 @@ class TestAPIToolFormatHandlerReasoningContent:
assert result.reasoning_content is None
class TestAgentStreamingReasoningEvents:
class TestAgentLoopStreamingReasoningEvents:
@pytest.mark.asyncio
async def test_streaming_accumulates_reasoning_in_message(self):
backend = FakeBackend([
@ -302,7 +302,7 @@ class TestAgentStreamingReasoningEvents:
mock_llm_chunk(content="", reasoning_content="Second thought."),
mock_llm_chunk(content="Final answer."),
])
agent = Agent(make_config(), backend=backend, enable_streaming=True)
agent = AgentLoop(make_config(), backend=backend, enable_streaming=True)
[_ async for _ in agent.act("Think and answer")]
@ -316,7 +316,7 @@ class TestAgentStreamingReasoningEvents:
mock_llm_chunk(content="Hello "),
mock_llm_chunk(content="world!"),
])
agent = Agent(make_config(), backend=backend, enable_streaming=True)
agent = AgentLoop(make_config(), backend=backend, enable_streaming=True)
events = [event async for event in agent.act("Say hello")]