v2.14.0 (#743)
Co-authored-by: Alexis Tacnet <alexis@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@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: p.vezia <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Hiba Chaabnia <Hiba-Chaabnia@users.noreply.github.com> Co-authored-by: Nikhil Bhima <nikhilbhima@users.noreply.github.com> Co-authored-by: Nkipohcs <Nkipohcs@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
parent
ad0d5c9520
commit
3f8487f761
197 changed files with 10819 additions and 2830 deletions
51
tests/core/test_llm_message_merge.py
Normal file
51
tests/core/test_llm_message_merge.py
Normal file
|
|
@ -0,0 +1,51 @@
|
|||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from vibe.core.types import ImageAttachment, LLMMessage, Role
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def image_a(tmp_path: Path) -> ImageAttachment:
|
||||
p = tmp_path / "a.png"
|
||||
p.write_bytes(b"\x89PNG")
|
||||
return ImageAttachment(path=p, alias="a.png", mime_type="image/png")
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def image_b(tmp_path: Path) -> ImageAttachment:
|
||||
p = tmp_path / "b.png"
|
||||
p.write_bytes(b"\x89PNG")
|
||||
return ImageAttachment(path=p, alias="b.png", mime_type="image/png")
|
||||
|
||||
|
||||
def _msg(content: str, images: list[ImageAttachment] | None = None) -> LLMMessage:
|
||||
return LLMMessage(role=Role.assistant, content=content, images=images)
|
||||
|
||||
|
||||
def test_merge_prefers_self_images_when_present(
|
||||
image_a: ImageAttachment, image_b: ImageAttachment
|
||||
) -> None:
|
||||
merged = _msg("hi", images=[image_a]) + _msg(" there", images=[image_b])
|
||||
assert merged.images == [image_a]
|
||||
|
||||
|
||||
def test_merge_falls_back_to_other_when_self_is_none(image_b: ImageAttachment) -> None:
|
||||
merged = _msg("hi") + _msg(" there", images=[image_b])
|
||||
assert merged.images == [image_b]
|
||||
|
||||
|
||||
def test_merge_preserves_explicit_empty_self_images_over_other(
|
||||
image_b: ImageAttachment,
|
||||
) -> None:
|
||||
# An explicit `[]` (intentional clearing) must NOT silently inherit
|
||||
# `other.images`. Truthy/falsy-based merging would have flipped this.
|
||||
merged = _msg("hi", images=[]) + _msg(" there", images=[image_b])
|
||||
assert merged.images == []
|
||||
|
||||
|
||||
def test_merge_yields_none_when_both_sides_are_none() -> None:
|
||||
merged = _msg("hi") + _msg(" there")
|
||||
assert merged.images is None
|
||||
Loading…
Add table
Add a link
Reference in a new issue