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

v2.9.4 (#669)

Browse source 
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Jean-Baptiste Muscat <jeanbaptiste.muscatdupuis@mistral.ai>
Co-authored-by: JeroenvdV <JeroenvdV@users.noreply.github.com>
Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@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: allansimon-mistral <allan.simon@ext.mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Clément Sirieix 2026-05-05 14:40:11 +02:00 • committed by GitHub
parent 71f373c60c
commit 4972dd5694
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
50 changed files with 2892 additions and 842 deletions
Download patch file Download diff file Expand all files Collapse all files

178
tests/session/test_saved_sessions.py Normal file
View file

@ -0,0 +1,178 @@
from __future__ import annotations
import json
from pathlib import Path
import pytest
from vibe.core.config import SessionLoggingConfig
from vibe.core.session.saved_sessions import update_saved_session_title
@pytest.fixture
def temp_session_dir(tmp_path: Path) -> Path:
session_dir = tmp_path / "sessions"
session_dir.mkdir()
return session_dir
@pytest.fixture
def session_config(temp_session_dir: Path) -> SessionLoggingConfig:
return SessionLoggingConfig(
save_dir=str(temp_session_dir), session_prefix="test", enabled=True
)
class TestUpdateSavedSessionTitle:
@pytest.mark.asyncio
async def test_updates_saved_session_title_without_losing_existing_metadata(
self, session_config: SessionLoggingConfig
) -> None:
session_dir = Path(session_config.save_dir)
saved_session_dir = session_dir / "test_20240101_120000_aaaaaaaa"
saved_session_dir.mkdir()
(saved_session_dir / "messages.jsonl").write_text(
'{"role": "user", "content": "Hello"}\n', encoding="utf-8"
)
original_metadata = {
"session_id": "aaaaaaaa-1111",
"start_time": "2024-01-01T12:00:00Z",
"end_time": "2024-01-01T12:05:00Z",
"git_commit": None,
"git_branch": None,
"username": "test-user",
"environment": {"working_directory": "/home/user/project"},
"title": "Old title",
"stats": {"steps": 2},
"total_messages": 1,
"tools_available": [
{
"type": "function",
"function": {"name": "bash", "description": "Run shell commands"},
}
],
"config": {"active_model": "test-model"},
"system_prompt": {"role": "system", "content": "You are helpful"},
}
metadata_file = saved_session_dir / "meta.json"
metadata_file.write_text(json.dumps(original_metadata), encoding="utf-8")
updated_metadata = await update_saved_session_title(
"aaaaaaaa-1111", "Renamed session", session_config
)
assert updated_metadata == {
**original_metadata,
"title": "Renamed session",
"title_source": "manual",
}
assert json.loads(metadata_file.read_text(encoding="utf-8")) == updated_metadata
@pytest.mark.asyncio
async def test_rejects_empty_title(
self, session_config: SessionLoggingConfig
) -> None:
session_dir = Path(session_config.save_dir)
saved_session_dir = session_dir / "test_20240101_120000_bbbbbbbb"
saved_session_dir.mkdir()
(saved_session_dir / "messages.jsonl").write_text(
'{"role": "user", "content": "Hello"}\n', encoding="utf-8"
)
original_metadata = {
"session_id": "bbbbbbbb-2222",
"start_time": "2024-01-01T12:00:00Z",
"end_time": "2024-01-01T12:05:00Z",
"git_commit": None,
"git_branch": None,
"username": "test-user",
"environment": {"working_directory": "/home/user/project"},
"title": "Manual title",
"title_source": "manual",
"stats": {"steps": 2},
}
metadata_file = saved_session_dir / "meta.json"
metadata_file.write_text(json.dumps(original_metadata), encoding="utf-8")
with pytest.raises(ValueError, match="Session title cannot be empty."):
await update_saved_session_title("bbbbbbbb-2222", " ", session_config)
@pytest.mark.asyncio
async def test_preserves_saved_session_end_time_when_updating_title(
self, session_config: SessionLoggingConfig
) -> None:
session_dir = Path(session_config.save_dir)
saved_session_dir = session_dir / "test_20240101_120000_cccccccc"
saved_session_dir.mkdir()
(saved_session_dir / "messages.jsonl").write_text(
'{"role": "user", "content": "Hello"}\n', encoding="utf-8"
)
original_metadata = {
"session_id": "cccccccc-3333",
"start_time": "2024-01-01T12:00:00Z",
"end_time": "2024-01-01T12:05:00Z",
"git_commit": None,
"git_branch": None,
"username": "test-user",
"environment": {"working_directory": "/home/user/project"},
"title": "Old title",
}
metadata_file = saved_session_dir / "meta.json"
metadata_file.write_text(json.dumps(original_metadata), encoding="utf-8")
updated_metadata = await update_saved_session_title(
"cccccccc-3333", "Renamed session", session_config
)
assert updated_metadata == {
**original_metadata,
"title": "Renamed session",
"title_source": "manual",
}
assert json.loads(metadata_file.read_text(encoding="utf-8")) == updated_metadata
@pytest.mark.asyncio
async def test_raises_for_missing_saved_session(
self, session_config: SessionLoggingConfig
) -> None:
with pytest.raises(ValueError, match="Session not found: missing-session"):
await update_saved_session_title(
"missing-session", "Renamed", session_config
)
@pytest.mark.asyncio
async def test_requires_exact_saved_session_id(
self, session_config: SessionLoggingConfig
) -> None:
session_dir = Path(session_config.save_dir)
saved_session_dir = session_dir / "test_20240101_120000_dddddddd"
saved_session_dir.mkdir()
(saved_session_dir / "messages.jsonl").write_text(
'{"role": "user", "content": "Hello"}\n', encoding="utf-8"
)
original_metadata = {
"session_id": "dddddddd-4444",
"start_time": "2024-01-01T12:00:00Z",
"end_time": "2024-01-01T12:05:00Z",
"git_commit": None,
"git_branch": None,
"username": "test-user",
"environment": {"working_directory": "/home/user/project"},
"title": "Old title",
}
metadata_file = saved_session_dir / "meta.json"
metadata_file.write_text(json.dumps(original_metadata), encoding="utf-8")
with pytest.raises(ValueError, match="Session not found: dddddddd"):
await update_saved_session_title("dddddddd", "Renamed", session_config)
assert (
json.loads(metadata_file.read_text(encoding="utf-8")) == original_metadata
)