v2.3.0 (#429)
Co-authored-by: Carlo <carloantonio.patti@mistral.ai> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Clement Sirieix <clem.sirieix@gmail.com> 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: Thomas Kenbeek <thomas.kenbeek@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
parent
a560a47ce8
commit
5d2e01a6d7
139 changed files with 7152 additions and 1457 deletions
|
|
@ -531,7 +531,7 @@ class TestAutoCompactIntegration:
|
|||
|
||||
roles = [r for r, _ in observed]
|
||||
assert roles == [Role.system, Role.user, Role.assistant]
|
||||
assert observed[1][1] is not None and "<summary>" in observed[1][1]
|
||||
assert observed[1][1] == "Hello"
|
||||
|
||||
|
||||
class TestClearHistoryFullReset:
|
||||
|
|
@ -623,6 +623,37 @@ class TestClearHistoryFullReset:
|
|||
assert agent.session_id == agent.session_logger.session_id
|
||||
|
||||
|
||||
class TestClearHistoryObserverBugfix:
|
||||
@pytest.mark.asyncio
|
||||
async def test_clear_history_observer_sees_new_messages(
|
||||
self, observer_capture
|
||||
) -> None:
|
||||
"""Bug fix: clear_history previously left a stale index, so new messages
|
||||
appended after clearing were never observed.
|
||||
"""
|
||||
observed, observer = observer_capture
|
||||
backend = FakeBackend([
|
||||
[mock_llm_chunk(content="First")],
|
||||
[mock_llm_chunk(content="Second")],
|
||||
])
|
||||
agent = build_test_agent_loop(
|
||||
config=make_config(), message_observer=observer, backend=backend
|
||||
)
|
||||
|
||||
async for _ in agent.act("Hello"):
|
||||
pass
|
||||
|
||||
await agent.clear_history()
|
||||
observed.clear()
|
||||
|
||||
async for _ in agent.act("After clear"):
|
||||
pass
|
||||
|
||||
roles = [msg.role for msg in observed]
|
||||
assert Role.user in roles
|
||||
assert Role.assistant in roles
|
||||
|
||||
|
||||
class TestStatsEdgeCases:
|
||||
@pytest.mark.asyncio
|
||||
async def test_session_cost_approximation_on_model_change(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue