b23f49e5f4
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: allansimon-mistral <allan.simon@ext.mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
"""Session ID generation with a stable suffix preserved across compact/fork/rewind."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import secrets
|
|
|
|
|
|
def generate_session_id(*, suffix: str | None = None) -> str:
|
|
"""Generate a UUID-shaped session ID with an optional stable suffix.
|
|
|
|
The last segment (12 hex chars after the final hyphen) is either
|
|
the provided *suffix* or freshly random. The first four segments
|
|
(20 hex chars) are always random.
|
|
|
|
Format: ``xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx``
|
|
"""
|
|
head = secrets.token_hex(10) # 10 bytes = 20 hex chars
|
|
tail = suffix or secrets.token_hex(6) # 6 bytes = 12 hex chars
|
|
return f"{head[:8]}-{head[8:12]}-{head[12:16]}-{head[16:20]}-{tail}"
|
|
|
|
|
|
def extract_suffix(session_id: str) -> str:
|
|
"""Extract the stable suffix (last segment after the final hyphen)."""
|
|
return session_id.rsplit("-", 1)[-1]
|
|
|
|
|
|
_SHORT_LEN = 8
|
|
|
|
|
|
def shorten_session_id(session_id: str, *, from_end: bool = False) -> str:
|
|
"""Return a short human-readable slice of a session ID (8 chars)."""
|
|
if from_end:
|
|
return session_id[-_SHORT_LEN:]
|
|
return session_id[:_SHORT_LEN]
|