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

v2.17.0 (#822)

Browse source 
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai>
Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com>
Co-authored-by: Hdandria <henri.dandria@mistral.ai>
Co-authored-by: Ivana Dunisijevic <ivana.dunisijevic@mistral.ai>
Co-authored-by: Jean Burellier <sheplu@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Mert Unsal <mert.unsal@mistral.ai>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com>
Co-authored-by: Pierre Rossinès <pierre.rossines@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: renovate-mistral[bot] <253709520+renovate-mistral[bot]@users.noreply.github.com>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Clément Drouin 2026-06-19 11:01:24 +02:00 • committed by GitHub
parent 564a14365e
commit 6bedf271ce
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
223 changed files with 10533 additions and 6947 deletions
Download patch file Download diff file Expand all files Collapse all files

30
tests/backend/test_image_encoding_cache.py
View file

@ -11,7 +11,7 @@ from vibe.core.llm.backend._image import (
to_base64,
to_data_uri,
)
from vibe.core.types import ImageAttachment
from vibe.core.types import FileImageSource, ImageAttachment, InlineImageSource
PNG_BYTES = b"\x89PNG\r\n\x1a\n" + b"\x00" * 16
@ -24,7 +24,9 @@ def _clear_cache() -> None:
def _att(tmp_path: Path, name: str = "shot.png") -> ImageAttachment:
p = tmp_path / name
p.write_bytes(PNG_BYTES)
return ImageAttachment(path=p, alias=name, mime_type="image/png")
return ImageAttachment(
source=FileImageSource(path=p), alias=name, mime_type="image/png"
)
def test_repeated_calls_hit_cache_and_skip_disk(tmp_path: Path) -> None:
@ -46,13 +48,14 @@ def test_cache_invalidates_when_file_mtime_changes(tmp_path: Path) -> None:
to_base64(att)
assert _encode_cached.cache_info().misses == 1
assert isinstance(att.source, FileImageSource)
new_bytes = PNG_BYTES + b"\x01"
att.path.write_bytes(new_bytes)
att.source.path.write_bytes(new_bytes)
# Force a distinct mtime even on coarse-resolution filesystems.
import os
stat = att.path.stat()
os.utime(att.path, ns=(stat.st_atime_ns, stat.st_mtime_ns + 1_000_000))
stat = att.source.path.stat()
os.utime(att.source.path, ns=(stat.st_atime_ns, stat.st_mtime_ns + 1_000_000))
refreshed = to_base64(att)
assert refreshed == base64.b64encode(new_bytes).decode("ascii")
@ -61,8 +64,23 @@ def test_cache_invalidates_when_file_mtime_changes(tmp_path: Path) -> None:
def test_missing_file_raises_image_read_error(tmp_path: Path) -> None:
att = ImageAttachment(
path=tmp_path / "nope.png", alias="nope.png", mime_type="image/png"
source=FileImageSource(path=tmp_path / "nope.png"),
alias="nope.png",
mime_type="image/png",
)
with pytest.raises(ImageReadError):
to_data_uri(att)
def test_inline_data_is_returned_without_touching_disk() -> None:
encoded = base64.b64encode(PNG_BYTES).decode("ascii")
att = ImageAttachment(
source=InlineImageSource(data=encoded),
alias="pasted.png",
mime_type="image/png",
)
assert to_base64(att) == encoded
assert to_data_uri(att) == f"data:image/png;base64,{encoded}"
assert _encode_cached.cache_info().misses == 0