2.0.0
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:
parent
79f215d91c
commit
d33db9fff8
217 changed files with 16911 additions and 4305 deletions
|
|
@ -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")]
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue