mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
90763daf81
Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Literal
|
|
|
|
from vibe.core.config import VibeConfig
|
|
from vibe.core.logger import logger
|
|
from vibe.core.nuage.client import WorkflowsClient
|
|
from vibe.core.nuage.workflow import WorkflowExecutionStatus
|
|
from vibe.core.session.session_loader import SessionLoader
|
|
|
|
ResumeSessionSource = Literal["local", "remote"]
|
|
|
|
SHORT_SESSION_ID_LEN = 8
|
|
|
|
|
|
def short_session_id(session_id: str, source: ResumeSessionSource = "local") -> str:
|
|
if source == "remote":
|
|
return session_id[-SHORT_SESSION_ID_LEN:]
|
|
return session_id[:SHORT_SESSION_ID_LEN]
|
|
|
|
|
|
_ACTIVE_STATUSES = {
|
|
WorkflowExecutionStatus.RUNNING,
|
|
WorkflowExecutionStatus.RETRYING_AFTER_ERROR,
|
|
}
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ResumeSessionInfo:
|
|
session_id: str
|
|
source: ResumeSessionSource
|
|
cwd: str
|
|
title: str | None
|
|
end_time: str | None
|
|
status: str | None = None
|
|
|
|
@property
|
|
def option_id(self) -> str:
|
|
return f"{self.source}:{self.session_id}"
|
|
|
|
|
|
def list_local_resume_sessions(
|
|
config: VibeConfig, cwd: str | None
|
|
) -> list[ResumeSessionInfo]:
|
|
return [
|
|
ResumeSessionInfo(
|
|
session_id=session["session_id"],
|
|
source="local",
|
|
cwd=session["cwd"],
|
|
title=session.get("title"),
|
|
end_time=session.get("end_time"),
|
|
)
|
|
for session in SessionLoader.list_sessions(config.session_logging, cwd=cwd)
|
|
]
|
|
|
|
|
|
async def list_remote_resume_sessions(config: VibeConfig) -> list[ResumeSessionInfo]:
|
|
if not config.nuage_enabled or not config.nuage_api_key:
|
|
logger.debug("Remote resume listing skipped: missing Nuage configuration")
|
|
return []
|
|
|
|
async with WorkflowsClient(
|
|
base_url=config.nuage_base_url,
|
|
api_key=config.nuage_api_key,
|
|
timeout=config.api_timeout,
|
|
) as client:
|
|
response = await client.get_workflow_runs(
|
|
workflow_identifier=config.nuage_workflow_id, page_size=50
|
|
)
|
|
|
|
sessions: list[ResumeSessionInfo] = []
|
|
for execution in response.executions:
|
|
if execution.status not in _ACTIVE_STATUSES:
|
|
continue
|
|
|
|
sessions.append(
|
|
ResumeSessionInfo(
|
|
session_id=execution.execution_id,
|
|
source="remote",
|
|
cwd="",
|
|
title="Vibe Nuage",
|
|
end_time=(
|
|
execution.end_time.isoformat()
|
|
if execution.end_time
|
|
else execution.start_time.isoformat()
|
|
),
|
|
status=execution.status,
|
|
)
|
|
)
|
|
|
|
logger.debug("Remote resume listing filtered sessions: %d", len(sessions))
|
|
return sessions
|